import random
import os

class Config:
    # --- 基础配置 ---
    _ENV = os.getenv("ROBOGO_ENV", "PROD").upper()
    if _ENV in ["FAT"]:
        BASE_URL = "https://robogo-fat.d-robotics.cc"
    else:
        BASE_URL = "https://robogo.d-robotics.cc"

    LOGIN_URL = f"{BASE_URL}/cloud-desktop/login"
    
    # --- 文件管理配置 ---
    # 定义为静态字符串，避免导入时触发 input
    FOLDER_NAME = f"UI_TEST_{random.randint(1, 1000000)}"

    # --- 开发机申请配置 ---
    DEV_MACHINE_NAME = f"UI_TEST_{random.randint(1, 1000000)}"
    DEV_MACHINE_SKU = "sku-e154fd3a-4719-db77-3911-21c6355349ec"
    SSH_PUBLIC_KEY = "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIBGp6Be0pt0Xy3Ipxm+AQTz6JQq8DAzIU6XHqD+/gzH6"
    SYSTEM_DISK = "100"

    # --- 登录凭证 (如果环境变量没有，则逻辑中会提示输入) ---
    AUTH_ACCOUNT = os.getenv("AUTH_ACCOUNT") or os.getenv("ROBOGO_USER", "")
    AUTH_PASSWORD = os.getenv("AUTH_PASSWORD") or os.getenv("ROBOGO_PWD", "")

    # --- 其他框架配置 ---
    TIMEOUT = 30000
    INSPECTION_INTERVAL = 300

    # --- 框架状态等待时长 (秒) ---
    WAIT = {
        'create': 60,
        'stop': 30,
        'delete': 30
    }

    # --- 云桌面 ---
    CLOUD_DESKTOP_SKU = "desktop-5880gpu12g-16c32g"
    CLOUD_DESKTOP_NAME = f"UI_TEST_{random.randint(1, 1000000)}"

    # --- 路径配置 ---
    BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
    TEST_DATA_DIR = os.path.join(BASE_DIR, "test_data")
    TEST_FILE = os.path.join(TEST_DATA_DIR, "Fruits-15.zip")

    # --- 镜像资产 ---
    MIRROR_NAME = f"UI_TEST_{random.randint(1, 1000000)}"
    MIRROR_SKU = "desktop-5880gpu12g-16c32g"