feat: Add VSCode code lens to show the unit file type and provide a w…

…ay to change it

src/index.ts

@@ -8,6 +8,7 @@ import { vscodeConfigNS } from "./config/vscode-config";
 import { HintDataManagers } from "./hint-data/manager/multiple";
 import { SystemdLint } from "./vscode-lint";
 import { SystemdCommands } from "./vscode-commands";
+import { SystemdCodeLens } from "./vscode-codelens";
 import { SystemdDocumentManager } from "./vscode-documents";
 
 export function activate(context: ExtensionContext) {
@@ -23,6 +24,7 @@ export function activate(context: ExtensionContext) {
     const completion = new SystemdCompletionProvider(config, hintDataManager);
     const signature = new SystemdSignatureProvider(hintDataManager);
     const lint = new SystemdLint(config, hintDataManager);
+    const codeLens = new SystemdCodeLens(config, hintDataManager);
     const commands = new SystemdCommands();
 
     subs.push(
@@ -38,6 +40,7 @@ export function activate(context: ExtensionContext) {
     subs.push(signature.register());
     subs.push(languages.registerHoverProvider(selector, signature));
     subs.push(languages.registerCodeActionsProvider(selector, lint));
+    subs.push(languages.registerCodeLensProvider(selector, codeLens));
     subs.push(
         workspace.onDidOpenTextDocument((doc) => {
             if (docs.onDidOpenTextDocument(doc)) lint.onDidOpenTextDocument(doc);

src/vscode-codelens.ts

@@ -0,0 +1,22 @@
+import { TextDocument, CancellationToken, CodeLensProvider, CodeLens } from "vscode";
+import { ExtensionConfig } from "./config/vscode-config-loader";
+import { HintDataManagers } from "./hint-data/manager/multiple";
+import { systemdFileTypeNames } from "./parser/file-info";
+import { SystemdCommands } from "./vscode-commands";
+import { SystemdDocumentManager } from "./vscode-documents";
+
+export class SystemdCodeLens implements CodeLensProvider {
+    constructor(private config: ExtensionConfig, private readonly managers: HintDataManagers) {}
+
+    provideCodeLenses(document: TextDocument, token: CancellationToken) {
+        const manager = SystemdDocumentManager.instance;
+        const fileType = manager.getType(document);
+        const codeLens = new CodeLens(
+            document.lineAt(0).range,
+            SystemdCommands.get("changeUnitFileType", "Systemd Unit Type: " + systemdFileTypeNames[fileType], [
+                document,
+            ])
+        );
+        return [codeLens];
+    }
+}