import { workflowReviewsPolicySchema } from '@n8n/api-types';
import { SettingsRepository } from '@n8n/db';
import { Service } from '@n8n/di';
import { jsonParse, UserError } from 'n8n-workflow';

import { WORKFLOW_REVIEW_POLICY_SETTINGS_KEY } from '@/constants/workflow-reviews';

const DEFAULT_POLICY = { enabled: false };

@Service()
export class WorkflowReviewPolicyService {
	constructor(private readonly settingsRepository: SettingsRepository) {}

	async get(): Promise<{ enabled: boolean }> {
		const row = await this.settingsRepository.findByKey(WORKFLOW_REVIEW_POLICY_SETTINGS_KEY);
		if (!row) {
			return DEFAULT_POLICY;
		}

		const parsed = workflowReviewsPolicySchema.safeParse(
			jsonParse(row.value, { fallbackValue: DEFAULT_POLICY }),
		);
		if (!parsed.success) {
			return DEFAULT_POLICY;
		}

		return parsed.data;
	}

	async set(enabled: boolean): Promise<{ enabled: boolean }> {
		const parsed = workflowReviewsPolicySchema.safeParse({ enabled });
		if (!parsed.success) {
			throw new UserError('Invalid workflow reviews settings');
		}

		await this.settingsRepository.upsert(
			{
				key: WORKFLOW_REVIEW_POLICY_SETTINGS_KEY,
				value: JSON.stringify(parsed.data),
				loadOnStartup: true,
			},
			['key'],
		);

		return parsed.data;
	}
}