Merge branch 'main' into fix-wrong-root

Author: savannahostrowski

.github/labeler.yml

@@ -0,0 +1,26 @@
+docs:
+  - all:
+    - changed-files:
+      - any-glob-to-any-file:
+        - README.md
+        - SUPPORT.md
+      - all-globs-to-all-files:
+        - '!src/**'
+
+internal:
+  - all:
+    - changed-files:
+      - any-glob-to-any-file:
+        - .github/**
+        - .vscode/**
+        - .husky/**
+        - .gitignore
+        - .vscode-test.mjs
+        - .vscodeignore
+        - biome.json
+        - bun.lock
+        - esbuild.js
+        - tsconfig.json
+
+      - all-globs-to-all-files:
+        - '!src/**'

.github/workflows/labeler.yml

@@ -0,0 +1,33 @@
+name: Labels
+on:
+  pull_request_target:
+    types:
+      - opened
+      - synchronize
+      - reopened
+      # For label-checker
+      - labeled
+      - unlabeled
+
+jobs:
+  labeler:
+    permissions:
+      contents: read
+      pull-requests: write
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/labeler@v6
+      if: ${{ github.event.action != 'labeled' && github.event.action != 'unlabeled' }}
+    - run: echo "Done adding labels"
+  # Run this after labeler applied labels
+  check-labels:
+    needs:
+      - labeler
+    permissions:
+      pull-requests: read
+    runs-on: ubuntu-latest
+    steps:
+      - uses: docker://agilepathway/pull-request-label-checker:latest
+        with:
+          one_of: breaking,security,feature,bug,refactor,upgrade,docs,internal
+          repo_token: ${{ secrets.GITHUB_TOKEN }}

.github/workflows/latest-changes.yml

@@ -0,0 +1,34 @@
+name: Latest Changes
+
+on:
+  pull_request_target:
+    branches:
+      - main
+    types:
+      - closed
+  workflow_dispatch:
+    inputs:
+      number:
+        description: PR number
+        required: true
+
+jobs:
+  latest-changes:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Dump GitHub context
+        env:
+          GITHUB_CONTEXT: ${{ toJson(github) }}
+        run: echo "$GITHUB_CONTEXT"
+      - uses: actions/checkout@v6
+        with:
+          # To allow latest-changes to commit to the main branch
+          token: ${{ secrets.FASTAPI_VSCODE_LATEST_CHANGES }}
+      - uses: tiangolo/latest-changes@0.4.1
+        with:
+          token: ${{ secrets.GITHUB_TOKEN }}
+          latest_changes_file: release-notes.md
+          latest_changes_header: '# Latest Changes'
+          end_regex: '^## '
+          debug_logs: true
+          label_header_prefix: '## '

package.json

@@ -113,6 +113,11 @@
           "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')) to navigate to the corresponding route definition."
         }
       }
     }

release-notes.md

@@ -0,0 +1,3 @@
+# Release Notes
+
+## Latest Changes

src/core/pathUtils.ts

@@ -77,3 +77,35 @@ export function getPathSegments(path: string, count: number): string {
 export function countSegments(path: string): number {
   return path.split("/").filter(Boolean).length
 }
+
+/**
+ * Checks if a test path matches an endpoint path pattern.
+ * Endpoint paths may contain path parameters like {item_id} which match any segment.
+ *
+ * Examples:
+ *   pathMatchesEndpoint("/items/123", "/items/{item_id}") -> true
+ *   pathMatchesEndpoint("/items/123/details", "/items/{item_id}") -> false
+ *   pathMatchesEndpoint("/users/abc/posts/456", "/users/{user_id}/posts/{post_id}") -> true
+ *   pathMatchesEndpoint("/items/", "/items/{item_id}") -> false
+ */
+export function pathMatchesEndpoint(
+  testPath: string,
+  endpointPath: string,
+): boolean {
+  const testSegments = testPath.split("/").filter(Boolean)
+  const endpointSegments = endpointPath.split("/").filter(Boolean)
+
+  // Segment counts must match
+  if (testSegments.length !== endpointSegments.length) {
+    return false
+  }
+
+  return endpointSegments.every((seg, index) => {
+    // Path parameter (e.g., {item_id}) matches any segment
+    if (seg.startsWith("{") && seg.endsWith("}")) {
+      return true
+    }
+    // Literal segments must match exactly
+    return seg === testSegments[index]
+  })
+}

src/extension.ts

@@ -15,6 +15,7 @@ import {
   type EndpointTreeItem,
   EndpointTreeProvider,
 } from "./providers/EndpointTreeProvider"
+import { TestCodeLensProvider } from "./providers/TestCodeLensProvider"
 
 async function discoverFastAPIApps(parser: Parser): Promise<AppDefinition[]> {
   const apps: AppDefinition[] = []
@@ -102,11 +103,48 @@ export async function activate(context: vscode.ExtensionContext) {
   // Discover FastAPI endpoints from workspace
   const apps = await discoverFastAPIApps(parserService)
   const endpointProvider = new EndpointTreeProvider(apps)
+  const codeLensProvider = new TestCodeLensProvider(parserService, apps)
+
+  let refreshTimeout: NodeJS.Timeout | null = null
+
+  const triggerRefresh = () => {
+    if (refreshTimeout) {
+      clearTimeout(refreshTimeout)
+    }
+    refreshTimeout = setTimeout(async () => {
+      if (!parserService) {
+        return
+      }
+      const newApps = await discoverFastAPIApps(parserService)
+      endpointProvider.setApps(newApps)
+      codeLensProvider.setApps(newApps)
+    }, 500)
+  }
+
+  // Watch for changes in Python files to refresh endpoints
+  const watcher = vscode.workspace.createFileSystemWatcher("**/*.py")
+  watcher.onDidChange(triggerRefresh)
+  watcher.onDidCreate(triggerRefresh)
+  watcher.onDidDelete(triggerRefresh)
+  context.subscriptions.push(watcher)
 
   const treeView = vscode.window.createTreeView("endpoint-explorer", {
     treeDataProvider: endpointProvider,
   })
 
+  // Register CodeLens provider for test files
+  const config = vscode.workspace.getConfiguration("fastapi")
+  if (config.get<boolean>("showTestCodeLenses", true)) {
+    context.subscriptions.push(
+      vscode.languages.registerCodeLensProvider(
+        // Covers common test file patterns
+        // e.g., test_*.py, *_test.py, tests/*.py
+        { language: "python", pattern: "**/*test*.py" },
+        codeLensProvider,
+      ),
+    )
+  }
+
   context.subscriptions.push(
     treeView,
 
@@ -119,6 +157,7 @@ export async function activate(context: vscode.ExtensionContext) {
         clearImportCache()
         const newApps = await discoverFastAPIApps(parserService)
         endpointProvider.setApps(newApps)
+        codeLensProvider.setApps(newApps)
       },
     ),
 
@@ -162,6 +201,24 @@ export async function activate(context: vscode.ExtensionContext) {
     vscode.commands.registerCommand("fastapi-vscode.toggleRouters", () => {
       endpointProvider.toggleRouters()
     }),
+
+    vscode.commands.registerCommand(
+      "fastapi-vscode.goToDefinition",
+      (
+        locations: vscode.Location[],
+        fromUri: vscode.Uri,
+        fromPosition: vscode.Position,
+      ) => {
+        vscode.commands.executeCommand(
+          "editor.action.goToLocations",
+          fromUri,
+          fromPosition,
+          locations,
+          locations.length === 1 ? "goto" : "peek",
+          "No matching route found",
+        )
+      },
+    ),
   )
 }