import {
	ConfigurationChangeEvent,
	Disposable,
	languages,
	workspace,
} from "vscode";
import {
	customCodeLensProvider,
	CustomCodeLensProvider,
} from "./CustomCodeLensProvider";

class CodeLensController implements Disposable {
	private internalProvider: CustomCodeLensProvider;
	private registeredProvider: Disposable | undefined;
	private configurationChangeListener: Disposable;

	constructor() {
		this.internalProvider = customCodeLensProvider;

		this.configurationChangeListener = workspace.onDidChangeConfiguration(
			(event: ConfigurationChangeEvent) => {
				if (event.affectsConfiguration("panfeng.editor.shortcuts")) {
					this.internalProvider.refresh();
				}
			},
			this
		);

		this.registeredProvider = languages.registerCodeLensProvider(
			{ scheme: "file" },
			this.internalProvider
		);
	}

	public dispose(): void {
		if (this.registeredProvider) {
			this.registeredProvider.dispose();
		}
		this.configurationChangeListener.dispose();
	}
}

export const codeLensController: CodeLensController = new CodeLensController();