import * as SQLite from 'expo-sqlite';

const db = SQLite.openDatabaseSync('upload_status.db');

export type UploadTask = {
    uri: string;
    filename: string;
    status: 'pending' | 'uploading' | 'success' | 'failed' | 'skipped';
    progress: number; // 0-100
    file_id?: string; // 后端返回的文件ID
};

// 初始化表
export function initUploadTable() {
    console.log('Initializing upload tasks table...');
    db.execSync(`
        CREATE TABLE IF NOT EXISTS upload_tasks (
            uri TEXT PRIMARY KEY NOT NULL,
            filename TEXT NOT NULL,
            status TEXT NOT NULL DEFAULT 'pending',
            progress INTEGER NOT NULL DEFAULT 0,
            file_id TEXT
        );
    `);
    console.log('Upload tasks table initialized');
}

// 插入新的上传任务
export async function insertUploadTask(uri: string, filename: string): Promise<void> {
    console.log('Inserting upload task:', uri, filename);
    db.runSync(
        'INSERT OR IGNORE INTO upload_tasks (uri, filename, status, progress) VALUES (?, ?, ?, ?);',
        uri,
        filename,
        'pending',
        0
    );
}

// 检查文件是否已上传或正在上传
export async function getUploadTaskStatus(uri: string): Promise<UploadTask | null> {
    console.log('Checking upload task status for:', uri);
    const result = db.getFirstSync<UploadTask>(
        'SELECT uri, filename, status, progress, file_id FROM upload_tasks WHERE uri = ?;',
        uri
    );
    return result || null;
}

// 更新上传任务的状态
export async function updateUploadTaskStatus(uri: string, status: UploadTask['status'], file_id?: string): Promise<void> {
    console.log('Updating upload task status:', uri, status, file_id);
    if (file_id) {
        db.runSync(
            'UPDATE upload_tasks SET status = ?, file_id = ? WHERE uri = ?;',
            status,
            file_id,
            uri
        );
    } else {
        db.runSync(
            'UPDATE upload_tasks SET status = ? WHERE uri = ?;',
            status,
            uri
        );
    }
}

// 更新上传任务的进度
export async function updateUploadTaskProgress(uri: string, progress: number): Promise<void> {
    console.log('Updating upload task progress:', uri, progress);
    db.runSync(
        'UPDATE upload_tasks SET progress = ? WHERE uri = ?;',
        progress,
        uri
    );
}

// 获取所有上传任务
export async function getUploadTasks(): Promise<UploadTask[]> {
    console.log('Fetching all upload tasks...');
    const results = db.getAllSync<UploadTask>(
        'SELECT uri, filename, status, progress, file_id FROM upload_tasks;'
    );
    return results;
}

// 清理已完成或失败的任务 (可选，根据需求添加)
export async function cleanUpUploadTasks(): Promise<void> {
    console.log('Cleaning up completed/failed upload tasks...');
    db.runSync(
        "DELETE FROM upload_tasks WHERE status = 'success' OR status = 'failed' OR status = 'skipped';"
    );
}