#!/usr/bin/env node

/**
 * 트리거 상태 확인 및 수정 스크립트
 * memory_embedding_vec_insert와 memory_embedding_vec_update 트리거의
 * TF-IDF 차원 조건을 확인하고 필요시 수정합니다.
 * 
 * 사용법:
 *   npx tsx scripts/check-and-fix-trigger.ts [--fix]
 */

import { initializeDatabase } from '../src/infrastructure/database/database/init.js';
import Database from 'better-sqlite3';
import { mementoConfig } from '../src/shared/config/index.js';

interface TriggerInfo {
  name: string;
  sql: string;
  hasCorrectTfidfDimension: boolean;
}

/**
 * 트리거 정보 조회
 */
function getTriggerInfo(db: Database.Database, triggerName: string): TriggerInfo | null {
  const result = db
    .prepare(`
      SELECT name, sql FROM sqlite_master 
      WHERE type='trigger' AND name=?
    `)
    .get(triggerName) as { name: string; sql: string } | undefined;

  if (!result) {
    return null;
  }

  // TF-IDF 차원 조건 확인: dimensions = 512가 있어야 함
  const hasCorrectTfidfDimension = 
    result.sql.includes("embedding_provider = 'tfidf'") &&
    result.sql.includes('dimensions = 512');

  return {
    name: result.name,
    sql: result.sql,
    hasCorrectTfidfDimension
  };
}

/**
 * 트리거 수정
 */
function fixTriggers(db: Database.Database): void {
  console.log('🔧 트리거 수정 중...\n');

  // 기존 트리거 삭제
  db.exec('DROP TRIGGER IF EXISTS memory_embedding_vec_insert');
  db.exec('DROP TRIGGER IF EXISTS memory_embedding_vec_update');

  // memory_embedding_vec_insert 트리거 재생성 (TF-IDF 차원: 512)
  db.exec(`
    CREATE TRIGGER IF NOT EXISTS memory_embedding_vec_insert AFTER INSERT ON memory_embedding BEGIN
      INSERT INTO memory_item_vec(rowid, embedding) 
      SELECT NEW.id, json_extract(NEW.embedding, '$')
      WHERE NEW.dimensions = 384;
      
      INSERT INTO memory_item_vec_tfidf(rowid, embedding) 
      SELECT NEW.id, json_extract(NEW.embedding, '$')
      WHERE NEW.embedding_provider = 'tfidf' AND NEW.dimensions = 512 AND NEW.projection_type = 'native';
      
      INSERT INTO memory_item_vec_minilm(rowid, embedding) 
      SELECT NEW.id, json_extract(NEW.embedding, '$')
      WHERE NEW.embedding_provider = 'minilm' AND NEW.dimensions = 384 AND NEW.projection_type = 'native';
      
      INSERT INTO memory_item_vec_openai(rowid, embedding) 
      SELECT NEW.id, json_extract(NEW.embedding, '$')
      WHERE NEW.embedding_provider = 'openai' AND NEW.dimensions = 1536 AND NEW.projection_type = 'native';
      
      INSERT INTO memory_item_vec_gemini(rowid, embedding) 
      SELECT NEW.id, json_extract(NEW.embedding, '$')
      WHERE NEW.embedding_provider = 'gemini' AND NEW.dimensions = 768 AND NEW.projection_type = 'native';
    END
  `);

  // memory_embedding_vec_update 트리거 재생성 (TF-IDF 차원: 512)
  db.exec(`
    CREATE TRIGGER IF NOT EXISTS memory_embedding_vec_update AFTER UPDATE ON memory_embedding BEGIN
      DELETE FROM memory_item_vec WHERE rowid = NEW.id;
      DELETE FROM memory_item_vec_tfidf WHERE rowid = NEW.id;
      DELETE FROM memory_item_vec_minilm WHERE rowid = NEW.id;
      DELETE FROM memory_item_vec_openai WHERE rowid = NEW.id;
      DELETE FROM memory_item_vec_gemini WHERE rowid = NEW.id;

      INSERT INTO memory_item_vec(rowid, embedding) 
      SELECT NEW.id, json_extract(NEW.embedding, '$')
      WHERE NEW.dimensions = 384;
      
      INSERT INTO memory_item_vec_tfidf(rowid, embedding) 
      SELECT NEW.id, json_extract(NEW.embedding, '$')
      WHERE NEW.embedding_provider = 'tfidf' AND NEW.dimensions = 512 AND NEW.projection_type = 'native';
      
      INSERT INTO memory_item_vec_minilm(rowid, embedding) 
      SELECT NEW.id, json_extract(NEW.embedding, '$')
      WHERE NEW.embedding_provider = 'minilm' AND NEW.dimensions = 384 AND NEW.projection_type = 'native';
      
      INSERT INTO memory_item_vec_openai(rowid, embedding) 
      SELECT NEW.id, json_extract(NEW.embedding, '$')
      WHERE NEW.embedding_provider = 'openai' AND NEW.dimensions = 1536 AND NEW.projection_type = 'native';
      
      INSERT INTO memory_item_vec_gemini(rowid, embedding) 
      SELECT NEW.id, json_extract(NEW.embedding, '$')
      WHERE NEW.embedding_provider = 'gemini' AND NEW.dimensions = 768 AND NEW.projection_type = 'native';
    END
  `);

  console.log('✅ 트리거 수정 완료\n');
}

/**
 * 마이그레이션 버전 확인
 */
function checkMigrationVersion(db: Database.Database): void {
  const tableExists = db
    .prepare(`
      SELECT name FROM sqlite_master 
      WHERE type='table' AND name='memento_schema_version'
    `)
    .get() as { name: string } | undefined;

  if (!tableExists) {
    console.log('⚠️  memento_schema_version 테이블이 없습니다. 마이그레이션 시스템이 초기화되지 않았을 수 있습니다.\n');
    return;
  }

  const version = db
    .prepare(`
      SELECT version, migration_name, applied_at 
      FROM memento_schema_version 
      ORDER BY applied_at DESC 
      LIMIT 1
    `)
    .get() as { version: string; migration_name: string; applied_at: string } | undefined;

  if (version) {
    console.log(`📋 현재 스키마 버전: ${version.version}`);
    console.log(`   마이그레이션: ${version.migration_name}`);
    console.log(`   적용 일시: ${version.applied_at}\n`);
  } else {
    console.log('⚠️  적용된 마이그레이션이 없습니다.\n');
  }

  // 12.0 마이그레이션 확인
  const migration12 = db
    .prepare(`
      SELECT version, migration_name, applied_at 
      FROM memento_schema_version 
      WHERE version = '12.0'
    `)
    .get() as { version: string; migration_name: string; applied_at: string } | undefined;

  if (migration12) {
    console.log(`✅ 마이그레이션 12.0 (fix-tfidf-dimension-trigger)이 적용되었습니다.`);
    console.log(`   적용 일시: ${migration12.applied_at}\n`);
  } else {
    console.log('⚠️  마이그레이션 12.0 (fix-tfidf-dimension-trigger)이 아직 적용되지 않았습니다.\n');
  }
}

async function main() {
  const shouldFix = process.argv.includes('--fix');
  
  console.log('🔍 트리거 상태 확인 중...\n');
  console.log(`데이터베이스 경로: ${mementoConfig.dbPath}\n`);

  let db: Database.Database | null = null;

  try {
    db = await initializeDatabase();

    // 마이그레이션 버전 확인
    checkMigrationVersion(db);

    // 트리거 확인
    const insertTrigger = getTriggerInfo(db, 'memory_embedding_vec_insert');
    const updateTrigger = getTriggerInfo(db, 'memory_embedding_vec_update');

    if (!insertTrigger) {
      console.log('❌ memory_embedding_vec_insert 트리거가 존재하지 않습니다.');
      process.exit(1);
    }

    if (!updateTrigger) {
      console.log('❌ memory_embedding_vec_update 트리거가 존재하지 않습니다.');
      process.exit(1);
    }

    console.log('📊 트리거 상태:');
    console.log(`   memory_embedding_vec_insert: ${insertTrigger.hasCorrectTfidfDimension ? '✅ 올바름 (dimensions = 512)' : '❌ 잘못됨 (dimensions = 384)'}`);
    console.log(`   memory_embedding_vec_update: ${updateTrigger.hasCorrectTfidfDimension ? '✅ 올바름 (dimensions = 512)' : '❌ 잘못됨 (dimensions = 384)'}\n`);

    if (!insertTrigger.hasCorrectTfidfDimension || !updateTrigger.hasCorrectTfidfDimension) {
      console.log('⚠️  트리거에 문제가 있습니다. TF-IDF 차원 조건이 384로 설정되어 있습니다.');
      console.log('   올바른 값은 512입니다.\n');

      if (shouldFix) {
        fixTriggers(db);
        
        // 수정 후 재확인
        const fixedInsertTrigger = getTriggerInfo(db, 'memory_embedding_vec_insert');
        const fixedUpdateTrigger = getTriggerInfo(db, 'memory_embedding_vec_update');

        console.log('📊 수정 후 트리거 상태:');
        console.log(`   memory_embedding_vec_insert: ${fixedInsertTrigger?.hasCorrectTfidfDimension ? '✅ 올바름' : '❌ 여전히 문제 있음'}`);
        console.log(`   memory_embedding_vec_update: ${fixedUpdateTrigger?.hasCorrectTfidfDimension ? '✅ 올바름' : '❌ 여전히 문제 있음'}\n`);

        if (fixedInsertTrigger?.hasCorrectTfidfDimension && fixedUpdateTrigger?.hasCorrectTfidfDimension) {
          console.log('✅ 트리거가 성공적으로 수정되었습니다!');
          process.exit(0);
        } else {
          console.log('❌ 트리거 수정에 실패했습니다.');
          process.exit(1);
        }
      } else {
        console.log('💡 트리거를 수정하려면 --fix 플래그를 사용하세요:');
        console.log('   npx tsx scripts/check-and-fix-trigger.ts --fix\n');
        process.exit(1);
      }
    } else {
      console.log('✅ 모든 트리거가 올바르게 설정되어 있습니다!');
      process.exit(0);
    }
  } catch (error) {
    console.error('❌ 오류 발생:', error instanceof Error ? error.message : String(error));
    process.exit(1);
  } finally {
    if (db) {
      db.close();
    }
  }
}

main();