✨ Add Test CodeLens navigation

.vscodeignore

@@ -7,5 +7,6 @@ src/**
 **/*.ts
 **/tsconfig.json
 esbuild.js
-node_modules/**
 bun.lockb
+node_modules/**
+!node_modules/web-tree-sitter/**

README.md

@@ -1,10 +1,23 @@
 # FastAPI VS Code Extension
 
-A VS Code extension for FastAPI development.
+A VS Code extension for FastAPI development that discovers and displays your API endpoints in a tree view.
 
 ## Features
 
-- Hello World command (example)
+- Automatic discovery of FastAPI routes and routers
+- Tree view showing all endpoints organized by router
+- Click to navigate to route definitions from the endpoint explorer
+- Supports `include_router` chains with prefix resolution
+- CodeLens in test files to jump from test client calls to routes
+
+## Settings
+
+| Setting | Description | Default |
+|---------|-------------|---------|
+| `fastapi.entryPoint` | Path to the main FastAPI application file (e.g., `src/main.py`). If not set, the extension searches common locations: `main.py`, `app/main.py`, `api/main.py`, `src/main.py`, `backend/app/main.py`. | `""` (auto-detect) |
+| `fastapi.showTestCodeLenses` | Show CodeLens links above test client calls (e.g., `client.get('/items')`) to navigate to the corresponding route definition. | `true` |
+
+**Note:** Currently the extension discovers one FastAPI app per workspace folder. If you have multiple apps, use separate workspace folders or configure `fastapi.entryPoint` to point to your primary app.
 
 ## Development
 

esbuild.js

@@ -1,27 +1,51 @@
+import { copyFileSync, globSync, mkdirSync } from "node:fs"
+import path from "node:path"
 import esbuild from "esbuild"
 
 const production = process.argv.includes("--production")
 const watch = process.argv.includes("--watch")
 
+function copyWasmFiles() {
+  const wasmDestDir = path.join(import.meta.dirname, "dist", "wasm")
+  mkdirSync(wasmDestDir, { recursive: true })
+
+  const wasmFiles = [
+    ["web-tree-sitter", "web-tree-sitter.wasm"],
+    ["tree-sitter-python", "tree-sitter-python.wasm"],
+  ]
+
+  for (const [pkg, file] of wasmFiles) {
+    const src = path.join(import.meta.dirname, "node_modules", pkg, file)
+    copyFileSync(src, path.join(wasmDestDir, file))
+    console.log(`Copied ${file} -> dist/wasm/`)
+  }
+}
+
 async function main() {
+  copyWasmFiles()
+
+  const entryPoints = ["src/extension.ts"]
+  if (!production) {
+    entryPoints.push(...globSync("src/test/**/*.test.ts"))
+  }
+
   const ctx = await esbuild.context({
-    entryPoints: [
-      "src/extension.ts",
-      "src/test/extension.test.ts",
-      "src/test/EndpointTreeProvider.test.ts",
-    ],
+    entryPoints,
     bundle: true,
     format: "cjs",
     minify: production,
     sourcemap: !production,
     sourcesContent: false,
     platform: "node",
+    target: "node20",
+    treeShaking: true,
     outdir: "dist",
     outbase: "src",
-    external: ["vscode"],
+    external: ["vscode", "web-tree-sitter"],
     logLevel: "info",
     define: {
       "process.env.NODE_ENV": production ? '"production"' : '"development"',
+      __DIST_ROOT__: JSON.stringify(path.join(import.meta.dirname, "dist")),
     },
   })
 

package.json

@@ -1,57 +1,82 @@
 {
   "name": "fastapi-vscode",
-  "displayName": "FastAPI Extension",
-  "description": "VS Code extension for FastAPI development",
-  "version": "0.0.1",
-  "publisher": "FastAPI Labs",
-  "engines": {
-    "vscode": "^1.85.0"
+  "version": "0.0.2",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/fastapi/fastapi-vscode"
+  },
+  "main": "./dist/extension.js",
+  "devDependencies": {
+    "@biomejs/biome": "^2.3.11",
+    "@types/bun": "latest",
+    "@types/mocha": "^10.0.10",
+    "@types/vscode": "^1.85.0",
+    "@vscode/test-cli": "^0.0.12",
+    "@vscode/test-electron": "^2.5.2",
+    "@vscode/vsce": "^3.7.1",
+    "esbuild": "^0.27.2",
+    "husky": "^9.1.7",
+    "lint-staged": "^16.2.7",
+    "typescript": "^5.0.0"
   },
-  "categories": [
-    "Other"
-  ],
   "activationEvents": [
     "workspaceContains:**/*.py"
   ],
-  "main": "./dist/extension.js",
+  "bugs": {
+    "url": "https://github.com/fastapi/fastapi-vscode/issues"
+  },
+  "categories": [
+    "Other"
+  ],
   "contributes": {
     "commands": [
       {
         "command": "fastapi-vscode.refreshEndpoints",
         "title": "Refresh Endpoints",
+        "category": "FastAPI",
         "icon": "$(refresh)"
       },
       {
-        "command": "fastapi-vscode.goToEndpoint",
-        "title": "Go to Definition"
+        "command": "fastapi-vscode.toggleRouters",
+        "title": "Toggle Expand/Collapse Routers",
+        "category": "FastAPI",
+        "icon": "$(fold)"
       },
       {
         "command": "fastapi-vscode.copyEndpointPath",
-        "title": "Copy Path"
+        "title": "Copy Endpoint Path",
+        "category": "FastAPI"
       },
       {
         "command": "fastapi-vscode.goToRouter",
-        "title": "Go to Router Definition"
-      },
-      {
-        "command": "fastapi-vscode.copyRouterPrefix",
-        "title": "Copy Prefix"
+        "title": "Go to Router Definition",
+        "category": "FastAPI"
       }
     ],
     "menus": {
+      "commandPalette": [
+        {
+          "command": "fastapi-vscode.copyEndpointPath",
+          "when": "false"
+        },
+        {
+          "command": "fastapi-vscode.goToRouter",
+          "when": "false"
+        }
+      ],
       "view/title": [
         {
           "command": "fastapi-vscode.refreshEndpoints",
           "when": "view == endpoint-explorer",
-          "group": "navigation"
+          "group": "navigation@1"
+        },
+        {
+          "command": "fastapi-vscode.toggleRouters",
+          "when": "view == endpoint-explorer",
+          "group": "navigation@2"
         }
       ],
       "view/item/context": [
-        {
-          "command": "fastapi-vscode.goToEndpoint",
-          "when": "view == endpoint-explorer && viewItem == route",
-          "group": "navigation"
-        },
         {
           "command": "fastapi-vscode.copyEndpointPath",
           "when": "view == endpoint-explorer && viewItem == route",
@@ -61,11 +86,6 @@
           "command": "fastapi-vscode.goToRouter",
           "when": "view == endpoint-explorer && viewItem == router",
           "group": "navigation"
-        },
-        {
-          "command": "fastapi-vscode.copyRouterPrefix",
-          "when": "view == endpoint-explorer && viewItem == router",
-          "group": "navigation"
         }
       ]
     },
@@ -85,45 +105,49 @@
           "name": "Endpoint Explorer"
         }
       ]
+    },
+    "configuration": {
+      "title": "FastAPI",
+      "properties": {
+        "fastapi.entryPoint": {
+          "type": "string",
+          "default": "",
+          "description": "Path to the main FastAPI application file (e.g., 'src/main.py'). If not set, the extension will search common locations."
+        },
+        "fastapi.showTestCodeLenses": {
+          "type": "boolean",
+          "default": true,
+          "description": "Show CodeLens links above test client calls (e.g., client.get('/items'), client.websocket('/ws')) to navigate to the corresponding route definition."
+        }
+      }
     }
   },
+  "description": "VS Code extension for FastAPI development",
+  "displayName": "FastAPI Extension",
+  "engines": {
+    "vscode": "^1.85.0"
+  },
+  "homepage": "https://github.com/fastapi/fastapi-vscode#readme",
+  "license": "MIT",
+  "lint-staged": {
+    "*.{ts,js,json}": [
+      "biome check --write"
+    ]
+  },
+  "publisher": "FastAPILabs",
   "scripts": {
     "vscode:prepublish": "bun run esbuild.js --production",
     "compile": "bun run esbuild.js",
     "watch": "bun run esbuild.js --watch",
     "package": "vsce package",
     "publish:marketplace": "vsce publish",
     "lint": "biome check --write --unsafe --no-errors-on-unmatched --files-ignore-unknown=true src/",
+    "pretest": "bun run compile",
     "test": "vscode-test",
     "prepare": "husky"
   },
-  "devDependencies": {
-    "@biomejs/biome": "^2.3.11",
-    "@types/bun": "latest",
-    "@types/mocha": "^10.0.10",
-    "@types/vscode": "^1.85.0",
-    "@vscode/test-cli": "^0.0.12",
-    "@vscode/test-electron": "^2.5.2",
-    "@vscode/vsce": "^3.7.1",
-    "esbuild": "^0.27.2",
-    "husky": "^9.1.7",
-    "lint-staged": "^16.2.7"
-  },
-  "peerDependencies": {
-    "typescript": "^5"
-  },
-  "license": "MIT",
-  "repository": {
-    "type": "git",
-    "url": "https://github.com/fastapi/fastapi-vscode"
-  },
-  "bugs": {
-    "url": "https://github.com/fastapi/fastapi-vscode/issues"
-  },
-  "homepage": "https://github.com/fastapi/fastapi-vscode#readme",
-  "lint-staged": {
-    "*.{ts,js,json}": [
-      "biome check --write"
-    ]
+  "dependencies": {
+    "tree-sitter-python": "^0.25.0",
+    "web-tree-sitter": "^0.26.3"
   }
 }