from librouteros import connect, Api
import csv
import ssl
import threading

from steps import *

"""
自动配置 ros 脚本，需要安装 librouteros 库：

```
pip install librouteros
```

配置文件格式为 CSV，包含以下字段：

- index: 配置索引，【重要】这个 index 用来生成对应 l2tp-out 用户名中的序号，每个节点都是唯一且固定的
- name: 城市名
- code: 城市代码
- gateway: 网关地址
- public: 公网 IP
- mask: 公网子网掩码
- private: 内网 IP

如果需要添加或修改配置项，从 config.*.bak.csv 文件复制配置项到 config.csv 文件中

如果需要修改配置内容，在下面 “配置执行步骤” 部分添加或修改函数调用
"""

threads = []
failed = []

# 配置执行步骤
steps = [
    configNet,
    configJgAuth,
    configJgMixOuts,
    configJgScripts,
    configJgLogs,
]


def main():
    # 加载配置文件
    config_data = []
    with open("config.csv", "r", encoding="utf-8-sig") as file:
        reader = csv.DictReader(file)
        for row in reader:
            config_data.append(row)

    # 多线程执行 ros 配置
    for _, config in enumerate(config_data):
        thread = threading.Thread(target=start, args=(config,))
        thread.start()
        threads.append(thread)

    # 等待所有线程完成
    for thread in threads:
        thread.join()

    # 输出结果
    print("=" * 20)
    print(
        "配置完成统计({}/{}), 失败项：{}".format(
            len(config_data) - len(failed), len(config_data), len(failed)
        )
    )
    for _, (item, err) in enumerate(failed):
        print("{},{},{},{}".format(item["public"], item["name"], item["code"], err))


def start(config):
    print(
        "{}: 配置{}({})".format(
            str(config["index"]).zfill(3), config["name"], config["code"]
        )
    )

    # 连接到 ros
    conn: Api
    try:
        ctx = ssl.create_default_context()
        ctx.check_hostname = False
        ctx.load_verify_locations(f'certs/{config["cert"]}')
        conn = connect(
            username="admin",
            password="wyongk9815",
            host=config["public"],
            port=8729,
            ssl_wrapper=ctx.wrap_socket,
        )
    except Exception as err:
        failed.append((config, f"连接失败: {err}"))
        return

    # 配置 ros
    try:
        for step in steps:
            step(conn, config)
    except Exception as err:
        failed.append((config, f"配置失败: {err}"))

    # 关闭连接
    conn.close()


main()