import { describe, it, expect } from 'vitest';
import Database from 'better-sqlite3';
import { runMigrations } from '@main/db/migrations/index.js';

describe('migrations', () => {
  it('creates schema with intent + edited columns', () => {
    const db = new Database(':memory:');
    runMigrations(db);
    const cols = db.prepare(`PRAGMA table_info(notes)`).all().map((r: any) => r.name);
    expect(cols).toEqual(
      expect.arrayContaining([
        'id', 'raw_text', 'ai_title', 'ai_summary', 'ai_status', 'ai_error',
        'ai_provider', 'ai_generated_at',
        'title_edited_by_user', 'summary_edited_by_user',
        'user_intent', 'intent_prompted_at',
        'created_at', 'updated_at'
      ])
    );
    db.close();
  });

  it('is idempotent', () => {
    const db = new Database(':memory:');
    runMigrations(db);
    const before = (db.prepare('PRAGMA user_version').get() as any).user_version;
    runMigrations(db);
    const after = (db.prepare('PRAGMA user_version').get() as any).user_version;
    expect(after).toBe(before);
    db.close();
  });
});

describe('migration v3 — soft delete columns', () => {
  it('adds deleted_at, last_recalled_at, recall_dismissed_at to notes', () => {
    const db = new Database(':memory:');
    runMigrations(db);
    const cols = db.prepare(`PRAGMA table_info(notes)`).all().map((r: any) => r.name);
    expect(cols).toEqual(
      expect.arrayContaining(['deleted_at', 'last_recalled_at', 'recall_dismissed_at'])
    );
    db.close();
  });

  it('creates idx_notes_deleted_at index', () => {
    const db = new Database(':memory:');
    runMigrations(db);
    const indexes = db
      .prepare(`SELECT name FROM sqlite_master WHERE type='index' AND tbl_name='notes'`)
      .all() as Array<{ name: string }>;
    expect(indexes.map((i) => i.name)).toContain('idx_notes_deleted_at');
    db.close();
  });

  it('user_version reaches 3', () => {
    const db = new Database(':memory:');
    runMigrations(db);
    const row = db.prepare('PRAGMA user_version').get() as { user_version: number };
    expect(row.user_version).toBe(3);
    db.close();
  });

  it('all 3 new columns default to NULL', () => {
    const db = new Database(':memory:');
    runMigrations(db);
    db.prepare(
      `INSERT INTO notes (id, raw_text, ai_status, created_at, updated_at)
       VALUES ('n1', 't', 'pending', '2026-05-01T00:00:00Z', '2026-05-01T00:00:00Z')`
    ).run();
    const row = db.prepare('SELECT deleted_at, last_recalled_at, recall_dismissed_at FROM notes WHERE id=?').get('n1') as any;
    expect(row.deleted_at).toBeNull();
    expect(row.last_recalled_at).toBeNull();
    expect(row.recall_dismissed_at).toBeNull();
    db.close();
  });
});