import sys
import subprocess


def main():
    # 检查是否提供了参数
    if len(sys.argv) != 2:
        print("Usage: python up.py <number_of_containers>")
        sys.exit(1)

    # 读取用户输入的容器数量
    try:
        num_containers = int(sys.argv[1])
    except ValueError:
        print("Please provide an integer for the number of containers.")
        sys.exit(1)

    # 以下是启动容器的代码
    for i in range(1, num_containers + 1):
        container_name = f"clash-{i:04d}"  # 生成格式化的容器名
        print(f'{container_name}')
        config_file = f"config_{i:04d}.yaml"  # 生成格式化的配置文件名
        cmd = f"docker run -d --restart=always --log-opt max-size=1m -v $PWD/split_config/{config_file}:/root/.config/clash/config.yaml -p {17888 + i - 1}:8080 -p {27890 + i - 1}:7890 --name {container_name} laoyutang/clash-and-dashboard:latest"
        subprocess.run(cmd, shell=True)

    print(f"Started {num_containers} containers.")


if __name__ == "__main__":
    main()