fix: fixed codelens placement for main and test blocks

Author: soorya-u

src/zigMainCodeLens.ts

@@ -16,16 +16,19 @@ export default class ZigMainCodeLensProvider implements vscode.CodeLensProvider
         const codeLenses: vscode.CodeLens[] = [];
         const text = document.getText();
 
-        const mainRegex = /^(?!\s*\/\/\/?\s*)\s*pub\s+fn\s+main\s*\(/gm;
+        const mainRegex = /^(?![ \t]*\/\/\/?\s*)[ \t]*pub\s+fn\s+main\s*\(/gm;
+
         let match;
         while ((match = mainRegex.exec(text))) {
             const position = document.positionAt(match.index);
             const line = document.lineAt(position.line);
-            const nextLine = Math.min(line.lineNumber + 1, document.lineCount - 1);
-            const range = new vscode.Range(
-                document.lineAt(nextLine).range.start,
-                document.lineAt(nextLine).range.start,
-            );
+
+            let targetLine = line.lineNumber + 1;
+            const prevLineText = document.lineAt(targetLine - 1).text.trim();
+            if (prevLineText) targetLine--;
+
+            const range = document.lineAt(targetLine).range;
+
             codeLenses.push(
                 new vscode.CodeLens(range, { title: "Run", command: "zig.run", arguments: [document.uri.fsPath] }),
             );

src/zigTestRunnerProvider.ts

@@ -70,7 +70,8 @@ export default class ZigTestRunnerProvider {
     private _updateTestItems(textDocument: vscode.TextDocument) {
         if (textDocument.languageId !== "zig") return;
 
-        const regex = /^(?!\s*\/\/\/?\s*)\s*\btest\s+(?:"([^"]+)"|([A-Za-z0-9_][\w]*)|@"([^"]+)")\s*\{/gm;
+        const regex = /^(?![ \t]*\/\/\/?\s*)[ \t]*\btest\s+(?:"([^"]+)"|([A-Za-z0-9_]\w*)|@"([^"]+)")\s*\{/gm;
+
         const matches = Array.from(textDocument.getText().matchAll(regex));
         this.deleteTestForAFile(textDocument.uri);
 
@@ -79,11 +80,7 @@ export default class ZigTestRunnerProvider {
             const isDocTest = !match[1];
             const position = textDocument.positionAt(match.index);
             const line = textDocument.lineAt(position.line);
-            const nextLine = Math.min(line.lineNumber + 1, textDocument.lineCount - 1);
-            const range = new vscode.Range(
-                textDocument.lineAt(nextLine).range.start,
-                textDocument.lineAt(nextLine).range.start,
-            );
+            const range = textDocument.lineAt(line.lineNumber).range;
 
             const fileName = path.basename(textDocument.uri.fsPath);