"""
更新用户Cookie Cloud参数
"""
import json
import sqlite3
import os

DB_PATH = os.path.join(os.path.dirname(__file__), 'backend', 'instance', 'cookie_monitor.db')

def update_user_cookie_cloud():
    """更新用户Cookie Cloud参数"""
    with open('config.json', 'r', encoding='utf-8') as f:
        config = json.load(f)

    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()

    # 添加新列（如果不存在）
    columns_to_add = [
        ('cookie_cloud_uuid', 'VARCHAR(255)'),
        ('cookie_cloud_password', 'VARCHAR(255)'),
        ('cookie_cloud_api_url', 'VARCHAR(255)')
    ]
    
    for col_name, col_type in columns_to_add:
        try:
            cursor.execute(f"ALTER TABLE users ADD COLUMN {col_name} {col_type}")
            print(f"添加列: {col_name}")
        except sqlite3.OperationalError as e:
            if 'duplicate column name' in str(e).lower():
                print(f"列已存在: {col_name}")
            else:
                raise

    for user_config in config['users']:
        user_name = user_config['name']
        cc = user_config.get('cookie_cloud', {})
        
        cursor.execute(
            "UPDATE users SET cookie_cloud_uuid = ?, cookie_cloud_password = ?, cookie_cloud_api_url = ? WHERE name = ?",
            (
                cc.get('uuid', ''),
                cc.get('password', ''),
                cc.get('api_url', ''),
                user_name
            )
        )
        print(f"更新用户 {user_name}: UUID={cc.get('uuid', '')}, API={cc.get('api_url', '')}")

    conn.commit()
    conn.close()
    print("\n更新完成!")

if __name__ == '__main__':
    update_user_cookie_cloud()