feat(fift/navigation): add lens to go to Tolk sources for this definition (#124)

Fixes #123

Author: i582

editors/code/src/extension.ts

@@ -196,6 +196,45 @@ async function startServer(context: vscode.ExtensionContext): Promise<vscode.Dis
     })
 }
 
+async function resolveFile(filePath: string): Promise<vscode.Uri> {
+    if (path.isAbsolute(filePath)) {
+        return vscode.Uri.file(filePath)
+    }
+
+    const workspaceFolders = vscode.workspace.workspaceFolders
+    if (!workspaceFolders || workspaceFolders.length === 0) {
+        throw new Error("No workspace folder found")
+    }
+
+    const workspaceRoot = workspaceFolders[0].uri.fsPath
+    const fullPath = path.join(workspaceRoot, "contracts", filePath)
+
+    try {
+        const uri = vscode.Uri.file(fullPath)
+        await vscode.workspace.fs.stat(uri)
+        return uri
+    } catch {
+        const foundFiles = await vscode.workspace.findFiles(
+            `**/${path.basename(filePath)}`,
+            "**/node_modules/**", // no need to search in node_modules
+            10,
+        )
+
+        if (foundFiles.length === 0) {
+            throw new Error(`File not found: ${filePath}`)
+        }
+
+        if (foundFiles.length === 1) {
+            return foundFiles[0]
+        } else {
+            const exactMatch = foundFiles.find(
+                file => file.fsPath.endsWith(filePath) || file.fsPath.includes(filePath),
+            )
+            return exactMatch ?? foundFiles[0]
+        }
+    }
+}
+
 function registerCommands(disposables: vscode.Disposable[]): void {
     disposables.push(
         vscode.commands.registerCommand("tolk.showToolchainInfo", async () => {
@@ -613,6 +652,23 @@ Node.js: ${info.environment.nodeVersion ?? "Unknown"}`
             void vscode.env.clipboard.writeText(str)
             void vscode.window.showInformationMessage(`Copied ${str} to clipboard`)
         }),
+        vscode.commands.registerCommand("ton.openFile", async (filePath: string, line?: number) => {
+            try {
+                const uri = await resolveFile(filePath)
+
+                if (line !== undefined && line > 0) {
+                    const uriWithFragment = uri.with({
+                        fragment: `L${line}`,
+                    })
+                    await vscode.commands.executeCommand("vscode.open", uriWithFragment)
+                } else {
+                    await vscode.commands.executeCommand("vscode.open", uri)
+                }
+            } catch (error) {
+                vscode.window.showErrorMessage(`Failed to open file: ${filePath}`)
+                console.error("Error opening file:", error)
+            }
+        }),
     )
 }
 

server/src/languages/fift/lens/index.ts

@@ -0,0 +1,40 @@
+//  SPDX-License-Identifier: MIT
+//  Copyright © 2025 TON Core
+import * as lsp from "vscode-languageserver"
+import {RecursiveVisitor} from "@server/visitor/visitor"
+import {asLspRange} from "@server/utils/position"
+import {FiftFile} from "@server/languages/fift/psi/FiftFile"
+import {trimPrefix} from "@server/utils/strings"
+
+export function collectFiftCodeLenses(file: FiftFile): lsp.CodeLens[] {
+    const result: lsp.CodeLens[] = []
+
+    RecursiveVisitor.visit(file.rootNode, (n): boolean => {
+        if (n.type === "definition") {
+            const prevComment = n.previousSibling
+            if (!prevComment || prevComment.type !== "comment") {
+                return true
+            }
+
+            const text = trimPrefix(prevComment.text, "// ")
+            const parts = text.split(/ /)
+            const pathAndLine = parts[0]
+            if (!pathAndLine) return true
+
+            const [path, line] = pathAndLine.split(":")
+
+            result.push({
+                range: asLspRange(n),
+                command: {
+                    title: `Go to Tolk sources (${pathAndLine})`,
+                    command: "ton.openFile",
+                    arguments: [path, Number(line)],
+                },
+            })
+        }
+
+        return true
+    })
+
+    return result
+}

server/src/server.ts

@@ -120,6 +120,7 @@ import {IndexingRootKind} from "@server/indexing/indexing"
 import {FuncIndexingRoot} from "@server/languages/func/indexing-root"
 import {formatTolkFile} from "@server/languages/tolk/format/format"
 import {collectFuncCodeLenses} from "@server/languages/func/lens"
+import {collectFiftCodeLenses} from "@server/languages/fift/lens"
 
 /**
  * Whenever LS is initialized.
@@ -980,6 +981,11 @@ connection.onInitialize(async (initParams: lsp.InitializeParams): Promise<lsp.In
         async (params: lsp.CodeLensParams): Promise<lsp.CodeLens[] | null> => {
             const uri = params.textDocument.uri
 
+            if (isFiftFile(uri)) {
+                const file = await findFiftFile(uri)
+                return collectFiftCodeLenses(file)
+            }
+
             if (isFuncFile(uri)) {
                 const file = await findFuncFile(uri)
                 return collectFuncCodeLenses(file)