extension.ts cleanup

Author: savannahostrowski

package.json

@@ -46,13 +46,22 @@
         "command": "fastapi-vscode.copyEndpointPath",
         "title": "Copy Endpoint Path",
         "category": "FastAPI"
+      },
+      {
+        "command": "fastapi-vscode.goToRouter",
+        "title": "Go to Router Definition",
+        "category": "FastAPI"
       }
     ],
     "menus": {
       "commandPalette": [
         {
           "command": "fastapi-vscode.copyEndpointPath",
           "when": "false"
+        },
+        {
+          "command": "fastapi-vscode.goToRouter",
+          "when": "false"
         }
       ],
       "view/title": [
@@ -72,6 +81,11 @@
           "command": "fastapi-vscode.copyEndpointPath",
           "when": "view == endpoint-explorer && viewItem == route",
           "group": "navigation"
+        },
+        {
+          "command": "fastapi-vscode.goToRouter",
+          "when": "view == endpoint-explorer && viewItem == router",
+          "group": "navigation"
         }
       ]
     },

src/extension.ts

@@ -2,6 +2,7 @@
  * VSCode extension entry point for FastAPI endpoint discovery.
  */
 
+import { existsSync } from "node:fs"
 import { sep } from "node:path"
 import * as vscode from "vscode"
 import { clearImportCache } from "./core/importResolver"
@@ -34,18 +35,22 @@ async function discoverFastAPIApps(parser: Parser): Promise<AppDefinition[]> {
       const entryPath = customEntryPoint.startsWith("/")
         ? customEntryPoint
         : vscode.Uri.joinPath(folder.uri, customEntryPoint).fsPath
+
+      if (!existsSync(entryPath)) {
+        vscode.window.showWarningMessage(
+          `FastAPI entry point not found: ${customEntryPoint}`,
+        )
+        continue
+      }
+
       candidates = [entryPath]
     } else {
       // Scan for main.py and __init__.py files (likely FastAPI entry points)
       const mainFiles = await vscode.workspace.findFiles(
         new vscode.RelativePattern(folder, "**/main.py"),
-        undefined,
-        20,
       )
       const initFiles = await vscode.workspace.findFiles(
         new vscode.RelativePattern(folder, "**/__init__.py"),
-        undefined,
-        20,
       )
       // Prefer main.py, then __init__.py, sorted by path depth (shallower first)
       candidates = [...mainFiles, ...initFiles]
@@ -68,10 +73,6 @@ async function discoverFastAPIApps(parser: Parser): Promise<AppDefinition[]> {
 }
 
 function navigateToLocation(location: SourceLocation): void {
-  if (!location.filePath) {
-    vscode.window.showErrorMessage("File path is missing for the endpoint.")
-    return
-  }
   const uri = vscode.Uri.file(location.filePath)
   const position = new vscode.Position(location.line - 1, location.column)
   vscode.window.showTextDocument(uri, {

src/providers/EndpointTreeProvider.ts

@@ -39,9 +39,6 @@ const METHOD_ICONS: Record<RouteMethod, string> = {
   WEBSOCKET: "broadcast",
 }
 
-/** Parent directories that don't provide useful context for labels */
-const GENERIC_PARENT_DIRS = new Set(["routes", "api", "routers", "endpoints"])
-
 export class EndpointTreeProvider
   implements TreeDataProvider<EndpointTreeItem>
 {
@@ -346,17 +343,11 @@ export class EndpointTreeProvider
             // Add / prefix to tag-based labels for consistency
             routerLabel = `/${element.router.tags[0]}`
           } else {
-            // Use parent directory + filename for context (e.g., "integrations/router")
+            // Use filename as label (e.g., "api_routes" from "routes/api_routes.py")
             const filePath = element.router.location.filePath
-            const parts = filePath.split("/")
-            const fileName = parts.pop()?.replace(/\.py$/, "") ?? ""
-            const parentDir = parts.pop() ?? ""
-            // Skip generic parent dirs like "routes" or "api"
-            if (parentDir && !GENERIC_PARENT_DIRS.has(parentDir)) {
-              routerLabel = `${parentDir}/${fileName}`
-            } else {
-              routerLabel = fileName
-            }
+            const fileName =
+              filePath.split("/").pop()?.replace(/\.py$/, "") ?? ""
+            routerLabel = fileName
           }
         }
         const routerItem = new TreeItem(