import type { WireMigration, WireMigrationRecord } from './migration';

interface V1_1ApprovalResult {
  readonly decision: 'approved' | 'rejected' | 'cancelled';
  readonly scope?: 'session';
  readonly feedback?: string;
  readonly selectedLabel?: string;
}

interface V1_1ApprovalResultRecord extends WireMigrationRecord {
  readonly type: 'permission.record_approval_result';
  readonly turnId: number;
  readonly toolCallId: string;
  readonly toolName: string;
  readonly action: string;
  readonly sessionApprovalRule?: string;
  readonly result: V1_1ApprovalResult;
}

const LEGACY_SESSION_APPROVAL_ACTION_TO_PATTERN: Readonly<Record<string, string>> = {
  'run command': 'Bash',
  'stop background task': 'TaskStop',
  'edit file': 'Write',
  'edit file outside of working directory': 'Write',
  'write file': 'Write',
};

const LEGACY_SESSION_APPROVAL_UNRESTORABLE_ACTIONS = new Set<string>([
  'run command in plan mode',
  'run background command',
]);

export const migrateV1_1ToV1_2: WireMigration = {
  sourceVersion: '1.1',
  targetVersion: '1.2',
  migrateRecord(record: WireMigrationRecord): WireMigrationRecord {
    if (record.type !== 'permission.record_approval_result') return record;
    const approvalRecord = record as V1_1ApprovalResultRecord;
    if (
      approvalRecord.result.decision !== 'approved' ||
      approvalRecord.result.scope !== 'session'
    ) {
      return record;
    }
    if (approvalRecord.sessionApprovalRule !== undefined) return record;

    const pattern = LEGACY_SESSION_APPROVAL_UNRESTORABLE_ACTIONS.has(approvalRecord.action)
      ? undefined
      : LEGACY_SESSION_APPROVAL_ACTION_TO_PATTERN[approvalRecord.action] ??
        approvalRecord.toolName;
    if (pattern === undefined) return record;

    return {
      ...record,
      sessionApprovalRule: pattern,
    };
  },
};