import fs from 'fs/promises';
import os from 'os';
import path from 'path';

import { test, expect, chatHubTestConfig } from './fixtures';
import { ChatHubChatPage } from '../../../pages/ChatHubChatPage';

test.use(chatHubTestConfig);

test.describe('File attachment @capability:proxy', {
	annotation: [
		{ type: 'owner', description: 'Chat' },
	],
}, () => {
	let tmpDir: string;
	let testImagePath: string;
	let testTextPath: string;

	test.beforeEach(async () => {
		tmpDir = await fs.mkdtemp(path.join(os.tmpdir(), 'chat-hub-test-'));
		testImagePath = path.join(tmpDir, 'test-image.png');
		testTextPath = path.join(tmpDir, 'test-file.txt');

		// 100x100 solid red square PNG
		const pngBuffer = Buffer.from(
			'iVBORw0KGgoAAAANSUhEUgAAAGQAAABkCAIAAAD/gAIDAAAAkElEQVR42u3QMQ0AAAjAsPk3DRb4eJpUQZviSIEsWbJkyUKBLFmyZMlCgSxZsmTJQoEsWbJkyUKBLFmyZMlCgSxZsmTJQoEsWbJkyUKBLFmyZMlCgSxZsmTJQoEsWbJkyUKBLFmyZMlCgSxZsmTJQoEsWbJkyUKBLFmyZMlCgSxZsmTJQoEsWbJkyUKBLFnvFp4t6yugc3LNAAAAAElFTkSuQmCC',
			'base64',
		);
		await fs.writeFile(testImagePath, pngBuffer);

		// Simple text file
		await fs.writeFile(testTextPath, 'I am a file');
	});

	test.afterEach(async () => {
		if (tmpDir) {
			await fs.rm(tmpDir, { recursive: true, force: true });
		}
	});

	test('image attachment', async ({ n8n, anthropicCredential: _ }) => {
		const page = new ChatHubChatPage(n8n.page);

		await n8n.navigate.toChatHub();
		await page.dismissWelcomeScreen();
		await expect(page.getModelSelectorButton()).toContainText(/claude/i); // auto-select a model

		await page.getFileInput().setInputFiles(testImagePath);
		await page.getChatInput().fill('What color is this image? Reply with just the color name.');
		await page.getSendButton().click();

		await expect(page.getChatMessages().nth(0)).toContainText('What color is this image?');
		await expect(page.getAttachmentsAt(0)).toHaveCount(1);
		await expect(page.getAttachmentsAt(0).nth(0)).toBeVisible();
		await expect(page.getChatMessages().nth(1)).toContainText(/red/i);

		// Verify image is persisted and can be opened in new tab
		await n8n.page.reload();
		const newPage = await page.openAttachmentAt(0, 0);

		await expect(newPage.locator('img')).toBeVisible();
		await expect(newPage.locator('img')).toHaveJSProperty('naturalWidth', 100);
		await expect(newPage.locator('img')).toHaveJSProperty('naturalHeight', 100);
		await newPage.close();
	});

	test('text file attachment', async ({ n8n, anthropicCredential: _ }) => {
		const page = new ChatHubChatPage(n8n.page);

		await n8n.navigate.toChatHub();
		await page.dismissWelcomeScreen();
		await expect(page.getModelSelectorButton()).toContainText(/claude/i);

		await page.getFileInput().setInputFiles(testTextPath);
		await page.getChatInput().fill('What is the exact content of this file?');
		await page.getSendButton().click();

		await expect(page.getChatMessages().nth(0)).toContainText('What is the exact content');
		await expect(page.getAttachmentsAt(0)).toHaveCount(1);
		await expect(page.getChatMessages().nth(1)).toContainText('I am a file');
	});

	test('reference attachment in subsequent message', async ({ n8n, anthropicCredential: _ }) => {
		const page = new ChatHubChatPage(n8n.page);

		await n8n.navigate.toChatHub();
		await page.dismissWelcomeScreen();
		await expect(page.getModelSelectorButton()).toContainText(/claude/i);

		// Send initial message with attachment
		await page.getFileInput().setInputFiles(testImagePath);
		await page.getChatInput().fill('What color is this image? Reply with just the color name.');
		await page.getSendButton().click();

		await expect(page.getChatMessages().nth(0)).toContainText('What color is this image?');
		await expect(page.getChatMessages().nth(1)).toContainText(/red/i);

		// Send follow-up question referencing the image
		await page.getChatInput().fill('Does the image include text? Reply just yes or no.');
		await page.getSendButton().click();

		await expect(page.getChatMessages().nth(2)).toContainText('Does the image include text?');
		await expect(page.getChatMessages().nth(3)).toContainText(/no/i);
	});
});