import * as vscode from "vscode";

export class AutoDevQuickFixProvider implements vscode.CodeActionProvider {
	public static readonly providedCodeActionKinds = [
		vscode.CodeActionKind.QuickFix,
	];

	provideCodeActions(
		document: vscode.TextDocument,
		range: vscode.Range | vscode.Selection,
		context: vscode.CodeActionContext,
		token: vscode.CancellationToken,
	): vscode.ProviderResult<(vscode.Command | vscode.CodeAction)[]> {
		if (context.diagnostics.length === 0) {
			return [];
		}

		const createQuickFix = (edit: boolean) => {
			const diagnostic = context.diagnostics[0];
			const quickFix = new vscode.CodeAction(
				edit ? "Fix with AutoDev" : "Ask AutoDev",
				vscode.CodeActionKind.QuickFix,
			);
			quickFix.isPreferred = false;
			const surroundingRange = new vscode.Range(
				Math.max(0, range.start.line - 3),
				0,
				Math.min(document.lineCount, range.end.line + 3),
				0,
			);
			quickFix.command = {
				command: "autodev.quickFix",
				title: "AutoDev Quick Fix",
				arguments: [
					diagnostic.message,
					document.getText(surroundingRange),
					edit,
				],
			};
			return quickFix;
		};

		return [
			createQuickFix(false),
		];
	}
}