import type { MigrationContext, ReversibleMigration } from '../migration-types';

/**
 * Agent runs no longer share `execution_entity` with workflow executions.
 * They live in two dedicated tables:
 *   - `agent_execution_threads` — per-session aggregate (token usage,
 *     session number, title, emoji). Renamed from `execution_threads`.
 *   - `agent_execution` — per-message run record with typed
 *     columns. Replaces the agent rows that previously lived in
 *     `execution_entity` + free-form key/value rows in
 *     `execution_metadata`.
 */
export class CreateAgentExecutionTables1783000000001 implements ReversibleMigration {
	async up({ schemaBuilder: { createTable, column } }: MigrationContext) {
		// ── Create new agent execution threads and execution recording tables ──

		await createTable('agent_execution_threads')
			.withColumns(
				column('id').varchar(36).primary,
				column('agentId').varchar(36).notNull,
				column('agentName').varchar(255).notNull,
				column('projectId').varchar(255).notNull,
				column('sessionNumber').int.notNull.default(0),
				column('totalPromptTokens').int.notNull.default(0),
				column('totalCompletionTokens').int.notNull.default(0),
				column('totalCost').double.notNull.default(0),
				column('totalDuration').int.notNull.default(0),
				column('title').varchar(255),
				column('emoji').varchar(8),
			)
			.withIndexOn('agentId')
			.withIndexOn('projectId')
			.withForeignKey('agentId', {
				tableName: 'agents',
				columnName: 'id',
				onDelete: 'CASCADE',
			})
			.withForeignKey('projectId', {
				tableName: 'project',
				columnName: 'id',
				onDelete: 'CASCADE',
			}).withTimestamps;

		await createTable('agent_execution')
			.withColumns(
				column('id').varchar(36).primary,
				column('threadId').varchar(36).notNull,
				column('status').varchar(16).notNull.withEnumCheck(['success', 'error']),
				column('startedAt').timestampTimezone(3),
				column('stoppedAt').timestampTimezone(3),
				column('duration').int.notNull.default(0),
				column('userMessage').text.notNull,
				column('assistantResponse').text.notNull,
				column('model').varchar(255),
				column('promptTokens').int,
				column('completionTokens').int,
				column('totalTokens').int,
				column('cost').double,
				column('toolCalls').json,
				column('timeline').json,
				column('error').text,
				column('hitlStatus').varchar(16).withEnumCheck(['suspended', 'resumed']),
				column('workingMemory').text,
				column('source').varchar(32),
			)
			.withIndexOn(['threadId', 'createdAt'])
			.withForeignKey('threadId', {
				tableName: 'agent_execution_threads',
				columnName: 'id',
				onDelete: 'CASCADE',
			}).withTimestamps;
	}

	async down({ schemaBuilder: { dropTable } }: MigrationContext) {
		await dropTable('agent_execution');
		await dropTable('agent_execution_threads');
	}
}