update extension

koka-lang · Aug 28, 2023 · 3c29bf0 · 3c29bf0
1 parent 3bfe022
commit 3c29bf0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/support/vscode/koka.language-koka/src/extension.ts b/support/vscode/koka.language-koka/src/extension.ts
@@ -310,9 +310,12 @@ class MainCodeLensProvider implements vscode.CodeLensProvider {
 		const doc = document.getText()
     const main = doc.indexOf('\nfun main')
     if (main < 0){
+      if (doc.startsWith('fun main')){
+        return [this.createCodeLens(document, 0)]
+      }
       return []
     }
-		return [this.createCodeLens(document, main)]
+		return [this.createCodeLens(document, main+1)]
 	}
 
 	private createCodeLens(document: TextDocument, offset: number): CodeLens {