import sqlite3
import os

# 确保数据库文件存放在正确的目录
DB_PATH = 'database.db'

def init_db():
    # 连接到数据库（如果不存在则创建）
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()

    # 创建外链表
    cursor.execute('''
    CREATE TABLE IF NOT EXISTS exlinks (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        exlink_id TEXT NOT NULL,
        qr_limit INTEGER DEFAULT 3,
        drivers TEXT,  -- 存储为JSON字符串，例如 '["quark","wechat"]'
        use_limit INTEGER DEFAULT 0,
        created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    )
    ''')

    # 创建用户表（如果需要的话）
    cursor.execute('''
    CREATE TABLE IF NOT EXISTS users (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        username TEXT NOT NULL UNIQUE,
        token TEXT,
        created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
    )
    ''')

    # 提交更改
    conn.commit()
    conn.close()

if __name__ == '__main__':
    init_db()
    print("Database initialized successfully!")