import {
	codeDiffSuggestionResponse,
	applyCodeDiffResponse,
	nodeExecutionSucceededResponse,
	aiEnabledWorkflowBaseRequirements,
	aiEnabledWithCodeDiffRequirements,
} from '../../../config/ai-assistant-fixtures';
import { test, expect } from '../../../fixtures/base';

test.describe('AI Assistant::enabled', {
	annotation: [
		{ type: 'owner', description: 'AI' },
	],
}, () => {
	test.describe('Code Node Error Help', () => {
		test('should apply code diff to code node', async ({ n8n, setupRequirements }) => {
			await setupRequirements(aiEnabledWithCodeDiffRequirements);

			let applySuggestionCalls = 0;
			await n8n.page.route('**/rest/ai/chat/apply-suggestion', async (route) => {
				applySuggestionCalls += 1;
				await route.fulfill({
					status: 200,
					contentType: 'application/json',
					body: JSON.stringify(applyCodeDiffResponse),
				});
			});

			await n8n.canvas.openNode('Code');

			await n8n.ndv.execute();

			await n8n.aiAssistant.getNodeErrorViewAssistantButton().click();

			await expect(n8n.aiAssistant.getChatMessagesAll()).toHaveCount(2);
			await expect(n8n.aiAssistant.getCodeDiffs()).toHaveCount(1);
			await expect(n8n.aiAssistant.getApplyCodeDiffButtons()).toHaveCount(1);

			await n8n.aiAssistant.getApplyCodeDiffButtons().first().click();

			await expect(n8n.aiAssistant.getApplyCodeDiffButtons()).toHaveCount(0);
			await expect(n8n.aiAssistant.getUndoReplaceCodeButtons()).toHaveCount(1);
			await expect(n8n.aiAssistant.getCodeReplacedMessage()).toBeVisible();
			await expect(n8n.ndv.getCodeEditor()).toContainText('item.json.myNewField = 1');

			await n8n.aiAssistant.getUndoReplaceCodeButtons().first().click();

			await expect(n8n.aiAssistant.getApplyCodeDiffButtons()).toHaveCount(1);
			await expect(n8n.aiAssistant.getCodeReplacedMessage()).toHaveCount(0);
			expect(applySuggestionCalls).toBe(1);
			await expect(n8n.ndv.getCodeEditor()).toContainText('item.json.myNewField = 1aaa');

			await n8n.aiAssistant.getApplyCodeDiffButtons().first().click();

			await expect(n8n.ndv.getCodeEditor()).toContainText('item.json.myNewField = 1');
		});

		test('should ignore node execution success and error messages after the node run successfully once', async ({
			n8n,
			setupRequirements,
		}) => {
			await setupRequirements(aiEnabledWorkflowBaseRequirements);

			let chatRequestCount = 0;
			await n8n.page.route('**/rest/ai/chat', async (route) => {
				chatRequestCount += 1;
				const response =
					chatRequestCount === 1 ? codeDiffSuggestionResponse : nodeExecutionSucceededResponse;
				await route.fulfill({
					status: 200,
					contentType: 'application/json',
					body: JSON.stringify(response),
				});
			});

			await n8n.canvas.openNode('Code');
			await n8n.ndv.execute();
			await n8n.aiAssistant.getNodeErrorViewAssistantButton().click();

			await n8n.ndv
				.getCodeEditor()
				.fill(
					"// Loop over input items and add a new field called 'myNewField' to the JSON of each one\nfor (const item of $input.all()) {\n  item.json.myNewField = 1;\n}\n\nreturn $input.all();",
				);

			await n8n.ndv.execute();

			await n8n.ndv
				.getCodeEditor()
				.fill(
					"// Loop over input items and add a new field called 'myNewField' to the JSON of each one\nfor (const item of $input.all()) {\n  item.json.myNewField = 1aaaa!;\n}\n\nreturn $input.all();",
				);

			await n8n.ndv.execute();

			await expect(n8n.aiAssistant.getChatMessagesAssistant().nth(2)).toContainText(
				'Code node ran successfully, did my solution help resolve your issue?',
			);
		});
	});
});