neodlp/src-tauri/src/migrations.rs

use tauri_plugin_sql::{Migration, MigrationKind};

pub fn get_migrations() -> Vec<Migration> {
    vec![Migration {
        version: 1,
        description: "create_initial_tables",
        sql: "
            CREATE TABLE IF NOT EXISTS video_info (
                id INTEGER PRIMARY KEY NOT NULL,
                video_id TEXT UNIQUE NOT NULL,
                title TEXT NOT NULL,
                url TEXT NOT NULL,
                host TEXT NOT NULL,
                thumbnail TEXT,
                channel TEXT,
                duration_string TEXT,
                release_date TEXT,
                view_count INTEGER,
                like_count INTEGER
            );
            CREATE TABLE IF NOT EXISTS playlist_info (
                id INTEGER PRIMARY KEY NOT NULL,
                playlist_id TEXT UNIQUE NOT NULL,
                playlist_title TEXT NOT NULL,
                playlist_url TEXT NOT NULL,
                playlist_n_entries INTEGER NOT NULL,
                playlist_channel TEXT
            );
            CREATE TABLE IF NOT EXISTS downloads (
                id INTEGER PRIMARY KEY NOT NULL,
                download_id TEXT UNIQUE NOT NULL,
                download_status TEXT NOT NULL,
                video_id TEXT NOT NULL,
                format_id TEXT NOT NULL,
                subtitle_id TEXT,
                queue_index INTEGER,
                playlist_id TEXT,
                playlist_index INTEGER,
                resolution TEXT,
                ext TEXT,
                abr REAL,
                vbr REAL,
                acodec TEXT,
                vcodec TEXT,
                dynamic_range TEXT,
                process_id INTEGER,
                status TEXT,
                progress REAL,
                total INTEGER,
                downloaded INTEGER,
                speed REAL,
                eta INTEGER,
                filepath TEXT,
                filetype TEXT,
                filesize INTEGER,
                FOREIGN KEY (video_id) REFERENCES video_info (video_id),
                FOREIGN KEY (playlist_id) REFERENCES playlist_info (playlist_id)
            );
            CREATE TABLE IF NOT EXISTS settings (
                id INTEGER PRIMARY KEY NOT NULL,
                key TEXT UNIQUE NOT NULL,
                value TEXT NOT NULL
            );
            CREATE TABLE IF NOT EXISTS kv_store (
                id INTEGER PRIMARY KEY NOT NULL,
                key TEXT UNIQUE NOT NULL,
                value TEXT NOT NULL
            );
            ",
        kind: MigrationKind::Up,
    },
    Migration {
        version: 2,
        description: "add_columns_to_downloads",
        sql: "
            -- Create temporary table with all new columns
            CREATE TABLE downloads_temp (
                id INTEGER PRIMARY KEY NOT NULL,
                download_id TEXT UNIQUE NOT NULL,
                download_status TEXT NOT NULL,
                video_id TEXT NOT NULL,
                format_id TEXT NOT NULL,
                subtitle_id TEXT,
                queue_index INTEGER,
                playlist_id TEXT,
                playlist_index INTEGER,
                resolution TEXT,
                ext TEXT,
                abr REAL,
                vbr REAL,
                acodec TEXT,
                vcodec TEXT,
                dynamic_range TEXT,
                process_id INTEGER,
                status TEXT,
                progress REAL,
                total INTEGER,
                downloaded INTEGER,
                speed REAL,
                eta INTEGER,
                filepath TEXT,
                filetype TEXT,
                filesize INTEGER,
                output_format TEXT,
                embed_metadata INTEGER NOT NULL DEFAULT 0,
                embed_thumbnail INTEGER NOT NULL DEFAULT 0,
                sponsorblock_remove TEXT,
                sponsorblock_mark TEXT,
                use_aria2 INTEGER NOT NULL DEFAULT 0,
                custom_command TEXT,
                queue_config TEXT,
                created_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
                updated_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
                FOREIGN KEY (video_id) REFERENCES video_info (video_id),
                FOREIGN KEY (playlist_id) REFERENCES playlist_info (playlist_id)
            );

            -- Copy all data from original table to temporary table with default values for new columns
            INSERT INTO downloads_temp SELECT
                id, download_id, download_status, video_id, format_id, subtitle_id,
                queue_index, playlist_id, playlist_index, resolution, ext, abr, vbr,
                acodec, vcodec, dynamic_range, process_id, status, progress, total,
                downloaded, speed, eta, filepath, filetype, filesize,
                NULL, -- output_format
                0,    -- embed_metadata
                0,    -- embed_thumbnail
                NULL, -- sponsorblock_remove
                NULL, -- sponsorblock_mark
                0,    -- use_aria2
                NULL, -- custom_command
                NULL, -- queue_config
                CURRENT_TIMESTAMP, -- created_at
                CURRENT_TIMESTAMP  -- updated_at
            FROM downloads;

            -- Drop the original table
            DROP TABLE downloads;

            -- Rename temporary table to original name
            ALTER TABLE downloads_temp RENAME TO downloads;

            -- Create trigger for updating updated_at timestamp
            CREATE TRIGGER IF NOT EXISTS update_downloads_updated_at
                AFTER UPDATE ON downloads
                FOR EACH ROW
            BEGIN
                UPDATE downloads SET updated_at = CURRENT_TIMESTAMP WHERE id = NEW.id;
            END;
            ",
        kind: MigrationKind::Up,
    },
    Migration {
        version: 3,
        description: "add_more_columns_and_indices_to_downloads",
        sql: "
            -- Create temporary table with all new columns
            CREATE TABLE downloads_temp (
                id INTEGER PRIMARY KEY NOT NULL,
                download_id TEXT UNIQUE NOT NULL,
                download_status TEXT NOT NULL,
                video_id TEXT NOT NULL,
                format_id TEXT NOT NULL,
                subtitle_id TEXT,
                queue_index INTEGER,
                playlist_id TEXT,
                playlist_indices TEXT,
                resolution TEXT,
                ext TEXT,
                abr REAL,
                vbr REAL,
                acodec TEXT,
                vcodec TEXT,
                dynamic_range TEXT,
                process_id INTEGER,
                status TEXT,
                item TEXT,
                progress REAL,
                total INTEGER,
                downloaded INTEGER,
                speed REAL,
                eta INTEGER,
                filepath TEXT,
                filetype TEXT,
                filesize INTEGER,
                output_format TEXT,
                embed_metadata INTEGER NOT NULL DEFAULT 0,
                embed_thumbnail INTEGER NOT NULL DEFAULT 0,
                square_crop_thumbnail INTEGER NOT NULL DEFAULT 0,
                sponsorblock_remove TEXT,
                sponsorblock_mark TEXT,
                use_aria2 INTEGER NOT NULL DEFAULT 0,
                custom_command TEXT,
                queue_config TEXT,
                created_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
                updated_at TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
                FOREIGN KEY (video_id) REFERENCES video_info (video_id),
                FOREIGN KEY (playlist_id) REFERENCES playlist_info (playlist_id)
            );

            -- Copy all data from original table to temporary table with default values for new columns
            INSERT INTO downloads_temp SELECT
                id, download_id, download_status, video_id, format_id, subtitle_id,
                queue_index, playlist_id,
                CAST(playlist_index AS TEXT),                                 -- Convert INTEGER playlist_index to TEXT playlist_indices
                resolution, ext, abr, vbr,
                acodec, vcodec, dynamic_range, process_id, status,
                CASE WHEN playlist_id IS NOT NULL THEN '1/1' ELSE NULL END,   -- item
                progress, total, downloaded, speed, eta,
                filepath, filetype, filesize,
                output_format, embed_metadata, embed_thumbnail,
                0,                                                            -- square_crop_thumbnail
                sponsorblock_remove, sponsorblock_mark, use_aria2,
                custom_command, queue_config, created_at, updated_at
            FROM downloads;

            -- Remove existing triggers
            DROP TRIGGER IF EXISTS update_downloads_updated_at;

            -- Drop the original table
            DROP TABLE downloads;

            -- Rename temporary table to original name
            ALTER TABLE downloads_temp RENAME TO downloads;

            -- Create trigger for updating updated_at timestamp
            CREATE TRIGGER IF NOT EXISTS update_downloads_updated_at
                AFTER UPDATE ON downloads
                FOR EACH ROW
            BEGIN
                UPDATE downloads SET updated_at = CURRENT_TIMESTAMP WHERE id = NEW.id;
            END;

            -- Add indexes to improve query performance
            CREATE INDEX IF NOT EXISTS idx_downloads_video_id ON downloads(video_id);
            CREATE INDEX IF NOT EXISTS idx_downloads_playlist_id ON downloads(playlist_id);
            CREATE INDEX IF NOT EXISTS idx_downloads_status_updated ON downloads(download_status, updated_at DESC);
            CREATE INDEX IF NOT EXISTS idx_downloads_id_desc ON downloads(id DESC);
        ",
        kind: MigrationKind::Up,
    }]
}