GameTime/lib/infrastructure/local/app_database.dart

import 'package:drift/drift.dart';
import 'package:drift_flutter/drift_flutter.dart';

import 'tables.dart';

part 'app_database.g.dart';

@DriftDatabase(
  tables: [
    ActiveRestStates,
    ActiveScoreStopwatchStates,
    ActiveSetResults,
    ActiveWorkoutSessions,
    ChangeLogEntries,
    Exercises,
    MediaAssets,
    ProgramExercises,
    Programs,
    WorkoutHistories,
    WorkoutHistorySetResults,
    WorkoutTemplateExerciseOverrides,
    WorkoutTemplatePrograms,
    WorkoutTemplates,
  ],
)
final class AppDatabase extends _$AppDatabase {
  AppDatabase(super.executor);

  factory AppDatabase.open() {
    return AppDatabase(
      driftDatabase(
        name: 'gametime',
        native: const DriftNativeOptions(shareAcrossIsolates: true),
      ),
    );
  }

  @override
  int get schemaVersion => 4;

  @override
  MigrationStrategy get migration {
    return MigrationStrategy(
      onCreate: (migrator) async {
        await migrator.createAll();
        await _createIndexes();
      },
      onUpgrade: (migrator, from, to) async {
        if (from < 2) {
          await customStatement(
            'ALTER TABLE active_set_results ADD COLUMN status TEXT NOT NULL '
            "DEFAULT 'completed' CHECK (status IN ('completed', 'skipped'))",
          );
          await customStatement(
            'ALTER TABLE workout_history_set_results ADD COLUMN status TEXT '
            "NOT NULL DEFAULT 'completed' CHECK (status IN ('completed', "
            "'skipped'))",
          );
        }
        if (from < 3) {
          await _migrateToSchema3(migrator);
        }
        if (from < 4) {
          await _migrateToSchema4();
        }
        await _createIndexes();
      },
      beforeOpen: (details) async {
        await customStatement('PRAGMA foreign_keys = ON');
      },
    );
  }

  Future<void> _createIndexes() async {
    for (final tableName in _syncableTableNames) {
      await customStatement(
        'CREATE INDEX IF NOT EXISTS idx_${tableName}_deleted_at '
        'ON $tableName (deleted_at)',
      );
      await customStatement(
        'CREATE INDEX IF NOT EXISTS idx_${tableName}_updated_at '
        'ON $tableName (updated_at)',
      );
      await customStatement(
        'CREATE INDEX IF NOT EXISTS idx_${tableName}_sync_state '
        'ON $tableName (sync_state)',
      );
      await customStatement(
        'CREATE INDEX IF NOT EXISTS idx_${tableName}_local_revision '
        'ON $tableName (local_revision)',
      );
    }

    await customStatement(
      'CREATE INDEX IF NOT EXISTS idx_program_exercises_program_id '
      'ON program_exercises (program_id)',
    );
    await customStatement(
      'CREATE INDEX IF NOT EXISTS idx_workout_template_programs_template_id '
      'ON workout_template_programs (workout_template_id)',
    );
    await customStatement(
      'CREATE INDEX IF NOT EXISTS idx_active_set_results_session_id '
      'ON active_set_results (active_workout_session_id)',
    );
    await customStatement(
      'CREATE INDEX IF NOT EXISTS idx_active_score_stopwatch_states_session_id '
      'ON active_score_stopwatch_states (active_workout_session_id)',
    );
    await customStatement(
      'CREATE INDEX IF NOT EXISTS idx_active_rest_states_session_id '
      'ON active_rest_states (active_workout_session_id)',
    );
    await customStatement(
      'CREATE UNIQUE INDEX IF NOT EXISTS '
      'idx_active_workout_sessions_single_open '
      'ON active_workout_sessions ((1)) '
      'WHERE deleted_at IS NULL '
      "AND status IN ('running', 'paused', 'savedExit')",
    );
    await customStatement(
      'CREATE INDEX IF NOT EXISTS idx_workout_history_started_at '
      'ON workout_history (started_at)',
    );
    await customStatement(
      'CREATE INDEX IF NOT EXISTS idx_workout_history_set_results_history_id '
      'ON workout_history_set_results (workout_history_id)',
    );
    await customStatement(
      'CREATE INDEX IF NOT EXISTS idx_change_log_entity '
      'ON change_log (entity_type, entity_id)',
    );
    await customStatement(
      'CREATE INDEX IF NOT EXISTS idx_change_log_local_revision '
      'ON change_log (local_revision)',
    );
    await customStatement(
      'CREATE INDEX IF NOT EXISTS idx_change_log_synced_at '
      'ON change_log (synced_at)',
    );
  }
}

const _syncableTableNames = [
  'active_rest_states',
  'active_score_stopwatch_states',
  'active_set_results',
  'active_workout_sessions',
  'exercises',
  'media_assets',
  'program_exercises',
  'programs',
  'workout_history',
  'workout_history_set_results',
  'workout_template_exercise_overrides',
  'workout_template_programs',
  'workout_templates',
];

extension on AppDatabase {
  Future<void> _migrateToSchema3(Migrator migrator) async {
    await customStatement(
      'ALTER TABLE exercises ADD COLUMN score_input_mode TEXT NOT NULL '
      "DEFAULT 'manual' CHECK (score_input_mode IN ('manual', 'stopwatch'))",
    );
    await customStatement(
      'ALTER TABLE program_exercises ADD COLUMN score_input_mode_snapshot TEXT '
      "NOT NULL DEFAULT 'manual' CHECK (score_input_mode_snapshot IN "
      "('manual', 'stopwatch'))",
    );
    await customStatement(
      'ALTER TABLE program_exercises ADD COLUMN target_score_time_ms INTEGER '
      'CHECK (target_score_time_ms IS NULL OR target_score_time_ms > 0)',
    );
    await customStatement(
      'ALTER TABLE workout_template_exercise_overrides ADD COLUMN '
      'target_score_time_ms_override INTEGER CHECK '
      '(target_score_time_ms_override IS NULL OR '
      'target_score_time_ms_override > 0)',
    );
    await customStatement(
      'ALTER TABLE active_set_results ADD COLUMN actual_score_time_ms INTEGER '
      'CHECK (actual_score_time_ms IS NULL OR actual_score_time_ms >= 0)',
    );
    await customStatement(
      'ALTER TABLE active_set_results ADD COLUMN score_input_mode_snapshot '
      "TEXT NOT NULL DEFAULT 'manual' CHECK (score_input_mode_snapshot IN "
      "('manual', 'stopwatch'))",
    );
    await customStatement(
      'ALTER TABLE workout_history_set_results ADD COLUMN '
      'score_input_mode_snapshot TEXT NOT NULL DEFAULT '
      "'manual' CHECK (score_input_mode_snapshot IN ('manual', 'stopwatch'))",
    );
    await customStatement(
      'ALTER TABLE workout_history_set_results ADD COLUMN '
      'target_score_time_ms_snapshot INTEGER CHECK '
      '(target_score_time_ms_snapshot IS NULL OR '
      'target_score_time_ms_snapshot > 0)',
    );
    await customStatement(
      'ALTER TABLE workout_history_set_results ADD COLUMN '
      'actual_score_time_ms INTEGER CHECK (actual_score_time_ms IS NULL OR '
      'actual_score_time_ms >= 0)',
    );
    await migrator.createTable(activeScoreStopwatchStates);
  }

  Future<void> _migrateToSchema4() async {
    await customStatement(
      'ALTER TABLE exercises ADD COLUMN default_target_time_seconds INTEGER '
      'CHECK (default_target_time_seconds IS NULL OR '
      'default_target_time_seconds > 0)',
    );
    await customStatement(
      'ALTER TABLE exercises ADD COLUMN default_target_reps INTEGER '
      'CHECK (default_target_reps IS NULL OR default_target_reps > 0)',
    );
    await customStatement(
      'ALTER TABLE exercises ADD COLUMN default_target_score REAL '
      'CHECK (default_target_score IS NULL OR default_target_score > 0)',
    );
    await customStatement(
      'ALTER TABLE exercises ADD COLUMN default_target_score_time_ms INTEGER '
      'CHECK (default_target_score_time_ms IS NULL OR '
      'default_target_score_time_ms > 0)',
    );
  }
}