import {
	NodeOperationError,
	NodeConnectionTypes,
	type IExecuteFunctions,
	type INodeType,
	type INodeTypeDescription,
	type INode,
	AI_TRANSFORM_CODE_GENERATED_FOR_PROMPT,
	AI_TRANSFORM_JS_CODE,
	ensureError,
} from 'n8n-workflow';

import { JsTaskRunnerSandbox } from '../Code/JsTaskRunnerSandbox';

export class AiTransform implements INodeType {
	description: INodeTypeDescription = {
		displayName: 'AI Transform',
		name: 'aiTransform',
		icon: 'node:ai-transform',
		iconColor: 'violet',
		group: ['transform'],
		version: 1,
		description: 'Modify data based on instructions written in plain english',
		defaults: {
			name: 'AI Transform',
		},
		inputs: [NodeConnectionTypes.Main],
		outputs: [NodeConnectionTypes.Main],
		parameterPane: 'wide',
		hints: [
			{
				message:
					"This node doesn't have access to the contents of binary files. To use those contents here, use the 'Extract from File' node first.",
				displayCondition: '={{ $input.all().some(i => i.binary) }}',
				location: 'outputPane',
			},
		],
		properties: [
			{
				displayName: 'Instructions',
				name: 'instructions',
				type: 'button',
				default: '',
				description:
					"Provide instructions on how you want to transform the data, then click 'Generate code'. Use dot notation to refer to nested fields (e.g. address.street).",
				placeholder:
					"Example: Merge 'firstname' and 'lastname' into a field 'details.name' and sort by 'email'",
				typeOptions: {
					buttonConfig: {
						label: 'Generate code',
						hasInputField: true,
						inputFieldMaxLength: 500,
						action: {
							type: 'askAiCodeGeneration',
							target: AI_TRANSFORM_JS_CODE,
						},
					},
				},
			},
			{
				displayName: 'Code Generated For Prompt',
				name: AI_TRANSFORM_CODE_GENERATED_FOR_PROMPT,
				type: 'hidden',
				default: '',
			},
			{
				displayName: 'Generated JavaScript',
				name: AI_TRANSFORM_JS_CODE,
				type: 'string',
				typeOptions: {
					editor: 'jsEditor',
					editorIsReadOnly: true,
				},
				default: '',
				hint: 'Read-only. To edit this code, adjust the instructions or copy and paste it into a Code node.',
				noDataExpression: true,
			},
		],
	};

	async execute(this: IExecuteFunctions) {
		const workflowMode = this.getMode();
		const node = this.getNode();
		const code = getValidatedCode(this, node);

		const sandbox = new JsTaskRunnerSandbox(workflowMode, this);
		return [await sandbox.runCodeAllItems(code)];
	}
}

function getValidatedCode(executeFunctions: IExecuteFunctions, node: INode): string {
	try {
		const code = executeFunctions.getNodeParameter(AI_TRANSFORM_JS_CODE, 0) as string;
		if (code) {
			return code;
		}

		const instructions = executeFunctions.getNodeParameter('instructions', 0) as string;
		if (!instructions) {
			throw new NodeOperationError(node, 'Missing instructions to generate code', {
				description: "Enter your prompt in the 'Instructions' parameter and click 'Generate code'",
			});
		}
		throw new NodeOperationError(node, 'Missing code for data transformation', {
			description: "Click the 'Generate code' button to create the code",
		});
	} catch (error: unknown) {
		if (error instanceof NodeOperationError) throw error;

		throw new NodeOperationError(node, ensureError(error));
	}
}