<?php

declare(strict_types=1);
/**
 * SPDX-FileCopyrightText: 2023 F7cloud GmbH and F7cloud contributors
 * SPDX-License-Identifier: AGPL-3.0-or-later
 */

namespace OCA\Assistant\Migration;

use Closure;
use OCP\DB\ISchemaWrapper;
use OCP\DB\Types;
use OCP\Migration\IOutput;
use OCP\Migration\SimpleMigrationStep;

class Version010003Date20231208083112 extends SimpleMigrationStep {
	/**
	 * @param IOutput $output
	 * @param Closure $schemaClosure The `\Closure` returns a `ISchemaWrapper`
	 * @param array $options
	 * @return void
	 */
	public function preSchemaChange(IOutput $output, Closure $schemaClosure, array $options): void {
	}

	/**
	 * @param IOutput $output
	 * @param Closure $schemaClosure The `\Closure` returns a `ISchemaWrapper`
	 * @param array $options
	 * @return null|ISchemaWrapper
	 */
	public function changeSchema(IOutput $output, Closure $schemaClosure, array $options) {
		/** @var ISchemaWrapper $schema */
		$schema = $schemaClosure();
		$schemaChanged = false;

		if (!$schema->hasTable('assistant_i_prompts')) {
			$schemaChanged = true;
			$table = $schema->createTable('assistant_i_prompts');
			$table->addColumn('id', Types::BIGINT, [
				'autoincrement' => true,
				'notnull' => true,
			]);
			$table->addColumn('user_id', Types::STRING, [
				'notnull' => true,
				'length' => 64,
			]);
			$table->addColumn('value', Types::STRING, [
				'notnull' => true,
				'length' => 1000,
			]);
			$table->addColumn('timestamp', Types::BIGINT, [
				'notnull' => true,
				'unsigned' => true,
			]);
			$table->setPrimaryKey(['id']);
			$table->addIndex(['user_id'], 'assistant_i_prompt_uid');
		}

		if (!$schema->hasTable('assistant_i_gens')) {
			$schemaChanged = true;
			$table = $schema->createTable('assistant_i_gens');
			$table->addColumn('id', Types::BIGINT, [
				'autoincrement' => true,
				'notnull' => true,
			]);
			$table->addColumn('image_gen_id', Types::STRING, [
				'notnull' => true,
			]);
			$table->addColumn('is_generated', Types::BOOLEAN, [
				'notnull' => false, 'default' => false,
			]);
			$table->addColumn('failed', Types::BOOLEAN, [
				'notnull' => false, 'default' => false,
			]);
			$table->addColumn('notify_ready', Types::BOOLEAN, [
				'notnull' => false, 'default' => false,
			]);
			$table->addColumn('prompt', Types::STRING, [
				'notnull' => true,
			]);
			$table->addColumn('user_id', Types::STRING, [
				'notnull' => true,
			]);
			$table->addColumn('timestamp', Types::BIGINT, [
				'notnull' => true,
				'unsigned' => true,
			]);
			$table->addColumn('exp_gen_time', Types::BIGINT, [
				'notnull' => true,
				'unsigned' => true,
			]);
			$table->setPrimaryKey(['id']);
			$table->addIndex(['image_gen_id'], 'assistant_i_gens_id');
		}

		if (!$schema->hasTable('assistant_i_files')) {
			$schemaChanged = true;
			$table = $schema->createTable('assistant_i_files');
			$table->addColumn('id', Types::BIGINT, [
				'autoincrement' => true,
				'notnull' => true,
			]);
			$table->addColumn('generation_id', Types::INTEGER, [
				'notnull' => true,
			]);
			$table->addColumn('file_name', Types::STRING, [
				'notnull' => true,
			]);
			$table->addColumn('hidden', Types::BOOLEAN, [
				'notnull' => false, 'default' => false,
			]);
			$table->setPrimaryKey(['id']);
			$table->addIndex(['generation_id'], 'assistant_gen_id');
		}

		if (!$schema->hasTable('assistant_stale_gens')) {
			$schemaChanged = true;
			$table = $schema->createTable('assistant_stale_gens');
			$table->addColumn('id', Types::BIGINT, [
				'autoincrement' => true,
				'notnull' => true,
			]);
			$table->addColumn('image_gen_id', Types::STRING, [
				'notnull' => true,
			]);
			$table->setPrimaryKey(['id']);
			$table->addIndex(['image_gen_id'], 'assistant_i_stale_gens_id');
		}

		if (!$schema->hasTable('assistant_t_prompts')) {
			$schemaChanged = true;
			$table = $schema->createTable('assistant_t_prompts');
			$table->addColumn('id', Types::BIGINT, [
				'autoincrement' => true,
				'notnull' => true,
			]);
			$table->addColumn('user_id', Types::STRING, [
				'notnull' => true,
				'length' => 64,
			]);
			$table->addColumn('value', Types::STRING, [
				'notnull' => true,
				'length' => 1000,
			]);
			$table->addColumn('timestamp', Types::BIGINT, [
				'notnull' => true,
				'unsigned' => true,
			]);
			$table->setPrimaryKey(['id']);
			$table->addIndex(['user_id'], 'assistant_t_prompts_uid');
			$table->addIndex(['user_id', 'timestamp'], 'assistant_t_prompts_uid_ts');
		}

		if (!$schema->hasTable('assistant_stt_transcripts')) {
			$schemaChanged = true;
			$table = $schema->createTable('assistant_stt_transcripts');

			$table->addColumn('id', Types::BIGINT, [
				'autoincrement' => true,
				'notnull' => true,
				'length' => 64,
			]);
			$table->addColumn('user_id', Types::STRING, [
				'notnull' => false,
				'length' => 64,
			]);
			$table->addColumn('transcript', Types::TEXT, [
				'notnull' => true,
			]);
			$table->addColumn('last_accessed', Types::DATETIME, [
				'notnull' => false,
			]);

			$table->setPrimaryKey(['id'], 'assistant_stt_transcript_id');
			$table->addIndex(['user_id'], 'assistant_stt_transcript_user');
			$table->addIndex(['last_accessed'], 'assistant_stt_transcript_la');
		}

		return $schemaChanged ? $schema : null;
	}

	/**
	 * @param IOutput $output
	 * @param Closure $schemaClosure The `\Closure` returns a `ISchemaWrapper`
	 * @param array $options
	 * @return void
	 */
	public function postSchemaChange(IOutput $output, Closure $schemaClosure, array $options): void {
	}
}