✨ Add endpoint discovery service for Endpoint Explorer

.vscodeignore

@@ -1,9 +1,20 @@
+# Source files (bundled into dist/)
 src/**
+esbuild.js
+wasm/**
+
+# Dependencies
 node_modules/**
+!node_modules/web-tree-sitter/package.json
+!node_modules/web-tree-sitter/web-tree-sitter.cjs
+
+# Build artifacts
 dist/test/**
+dist/*.map
+
+# Config/meta files
 .*/
 .*
 *.lock*
 *.json
 !package.json
-esbuild.js

README.md

@@ -1,10 +1,21 @@
 # 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
+- Supports `include_router` chains with prefix resolution
+
+## 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) |
+
+**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,59 @@
+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 })
+
+  // web-tree-sitter.wasm from node_modules
+  const coreSrc = path.join(
+    import.meta.dirname,
+    "node_modules",
+    "web-tree-sitter",
+    "web-tree-sitter.wasm",
+  )
+  copyFileSync(coreSrc, path.join(wasmDestDir, "web-tree-sitter.wasm"))
+  console.log("Copied web-tree-sitter.wasm -> dist/wasm/")
+
+  // tree-sitter-python.wasm from wasm/ directory (checked into repo)
+  const pythonSrc = path.join(
+    import.meta.dirname,
+    "wasm",
+    "tree-sitter-python.wasm",
+  )
+  copyFileSync(pythonSrc, path.join(wasmDestDir, "tree-sitter-python.wasm"))
+  console.log("Copied tree-sitter-python.wasm -> 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

@@ -2,40 +2,50 @@
   "name": "fastapi-vscode",
   "displayName": "FastAPI Extension",
   "description": "VS Code extension for FastAPI development",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "publisher": "FastAPILabs",
+  "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",
   "engines": {
     "vscode": "^1.85.0"
   },
-  "categories": [
-    "Other"
-  ],
+  "main": "./dist/extension.js",
   "activationEvents": [
     "workspaceContains:**/*.py"
   ],
-  "main": "./dist/extension.js",
+  "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"
       },
       {
         "command": "fastapi-vscode.reportIssue",
@@ -44,19 +54,29 @@
       }
     ],
     "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",
@@ -66,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"
         }
       ]
     },
@@ -90,6 +105,16 @@
           "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."
+        }
+      }
     }
   },
   "scripts": {
@@ -115,15 +140,9 @@
     "lint-staged": "^16.2.7",
     "typescript": "^5.0.0"
   },
-  "license": "MIT",
-  "repository": {
-    "type": "git",
-    "url": "https://github.com/fastapi/fastapi-vscode"
-  },
-  "bugs": {
-    "url": "https://github.com/fastapi/fastapi-vscode/issues"
+  "dependencies": {
+    "web-tree-sitter": "^0.26.3"
   },
-  "homepage": "https://github.com/fastapi/fastapi-vscode#readme",
   "lint-staged": {
     "*.{ts,js,json}": [
       "biome check --write"

src/core/analyzer.ts

@@ -0,0 +1,64 @@
+/**
+ * Analyzer module to extract FastAPI-related information from syntax trees.
+ */
+
+import { readFileSync } from "node:fs"
+import type { Tree } from "web-tree-sitter"
+import {
+  decoratorExtractor,
+  findNodesByType,
+  importExtractor,
+  includeRouterExtractor,
+  mountExtractor,
+  routerExtractor,
+} from "./extractors.js"
+import type { FileAnalysis } from "./internal"
+import type { Parser } from "./parser.js"
+
+function notNull<T>(value: T | null): value is T {
+  return value !== null
+}
+
+/** Analyze a syntax tree and extract FastAPI-related information */
+export function analyzeTree(tree: Tree, filePath: string): FileAnalysis {
+  const rootNode = tree.rootNode
+
+  // Get all decorated definitions (functions and classes with decorators)
+  const decoratedDefs = findNodesByType(rootNode, "decorated_definition")
+  const routes = decoratedDefs.map(decoratorExtractor).filter(notNull)
+
+  // Get all router assignments
+  const assignments = findNodesByType(rootNode, "assignment")
+  const routers = assignments.map(routerExtractor).filter(notNull)
+
+  // Get all include_router and mount calls
+  const callNodes = findNodesByType(rootNode, "call")
+  const includeRouters = callNodes.map(includeRouterExtractor).filter(notNull)
+  const mounts = callNodes.map(mountExtractor).filter(notNull)
+
+  // Get all import statements
+  const importNodes = findNodesByType(rootNode, "import_statement")
+  const importFromNodes = findNodesByType(rootNode, "import_from_statement")
+  const imports = [...importNodes, ...importFromNodes]
+    .map(importExtractor)
+    .filter(notNull)
+
+  return { filePath, routes, routers, includeRouters, mounts, imports }
+}
+
+/** Analyze a file given its path and a parser instance */
+export function analyzeFile(
+  filePath: string,
+  parser: Parser,
+): FileAnalysis | null {
+  try {
+    const code = readFileSync(filePath, "utf-8")
+    const tree = parser.parse(code)
+    if (!tree) {
+      return null
+    }
+    return analyzeTree(tree, filePath)
+  } catch {
+    return null
+  }
+}