/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the MIT License. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

import { commands } from 'vscode';
import { CodeReference } from '.';
import { IAuthenticationService } from '../../../../../../platform/authentication/common/authentication';
import { Disposable } from '../../../../../../util/vs/base/common/lifecycle';
import { IInstantiationService } from '../../../../../../util/vs/platform/instantiation/common/instantiation';
import { onCopilotToken } from '../../../lib/src/auth/copilotTokenNotifier';
import { ICompletionsCitationManager, IPDocumentCitation } from '../../../lib/src/citationManager';
import { OutputPaneShowCommand } from '../../../lib/src/snippy/constants';
import { copilotOutputLogTelemetry } from '../../../lib/src/snippy/telemetryHandlers';
import { notify } from './matchNotifier';
import { GitHubCopilotLogger } from './outputChannel';

/**
 * Citation manager that logs citations to the VS Code log. On the first citation encountered,
 * the user gets a notification.
 */
export class LoggingCitationManager extends Disposable implements ICompletionsCitationManager {
	declare _serviceBrand: undefined;

	private logger?: GitHubCopilotLogger;
	private readonly codeReference: CodeReference;

	constructor(
		@IInstantiationService private readonly instantiationService: IInstantiationService,
		@IAuthenticationService authenticationService: IAuthenticationService,
	) {
		super();
		this.codeReference = this._register(this.instantiationService.createInstance(CodeReference));
		const disposable = onCopilotToken(authenticationService, _ => {
			if (this.logger) {
				return;
			}
			this.logger = instantiationService.createInstance(GitHubCopilotLogger);
			const initialNotificationCommand = commands.registerCommand(OutputPaneShowCommand, () =>
				this.logger?.forceShow()
			);
			this.codeReference.addDisposable(initialNotificationCommand);
		});
		this.codeReference.addDisposable(disposable);
	}

	register() {
		return this.codeReference.register();
	}

	async handleIPCodeCitation(citation: IPDocumentCitation): Promise<void> {
		if (!this.codeReference.enabled || !this.logger || citation.details.length === 0) {
			return;
		}

		const start = citation.location?.start;
		const matchLocation = start ? `[Ln ${start.line + 1}, Col ${start.character + 1}]` : 'Location not available';
		const shortenedMatchText = `${citation.matchingText
			?.slice(0, 100)
			.replace(/[\r\n\t]+|^[ \t]+/gm, ' ')
			.trim()}...`;

		this.logger.info(citation.inDocumentUri, `Similar code at `, matchLocation, shortenedMatchText);
		for (const detail of citation.details) {
			const { license, url } = detail;
			this.logger.info(`License: ${license.replace('NOASSERTION', 'unknown')}, URL: ${url}`);
		}
		copilotOutputLogTelemetry.handleWrite({ instantiationService: this.instantiationService });
		await this.instantiationService.invokeFunction(notify);
	}
}