import sqlite3
import os

DB_PATH = os.path.join(os.path.dirname(__file__), "onever_drive.db")

def migrate():
    print(f"Connecting to {DB_PATH}...")
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()
    
    columns_to_add = [
        ("exclude_patterns", "VARCHAR(255) DEFAULT '' NOT NULL"),
        ("skip_hidden", "BOOLEAN DEFAULT 0 NOT NULL"),
        ("skip_system", "BOOLEAN DEFAULT 0 NOT NULL"),
        ("skip_readonly", "BOOLEAN DEFAULT 0 NOT NULL")
    ]
    
    for col_name, col_def in columns_to_add:
        try:
            cursor.execute(f"ALTER TABLE backup_jobs ADD COLUMN {col_name} {col_def}")
            print(f"Column '{col_name}' added successfully to backup_jobs.")
        except sqlite3.OperationalError as e:
            print(f"Column '{col_name}' could not be added (maybe it already exists?): {e}")
            
    conn.commit()
    conn.close()

if __name__ == "__main__":
    migrate()