refactor(ts-plugin): split filter and contextSpan assignment into separate passes

Keep the filter callback side-effect-free; move `contextSpan` mutation
into a dedicated for-loop pass.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>

Author: mizdra

packages/ts-plugin/src/language-service/feature/definition-and-bound-span.ts

@@ -10,13 +10,28 @@ export function getDefinitionAndBoundSpan(
     const result = languageService.getDefinitionAndBoundSpan(...args);
     if (!result) return;
     if (!result.definitions) return result;
+
+    // Exclude re-exported tokens from `@value ... from '...'`; the real definition lives in the target file.
     result.definitions = result.definitions.filter((def) => {
       const script = language.scripts.get(def.fileName);
       if (!isCSSModuleScript(script)) return true;
-
       const cssModule = script.generated.root[CMK_DATA_KEY];
+      const defName = unquote(def.name);
+      const importedValue = cssModule.tokenImporters
+        .flatMap((i) => (i.type === 'value' ? i.values : []))
+        .find((v) => {
+          const localName = v.localName ?? v.name;
+          const localLoc = v.localLoc ?? v.loc;
+          return localName === defName && localLoc.start.offset === def.textSpan.start;
+        });
+      return !importedValue;
+    });
 
-      // Search tokens and set `contextSpan`. `contextSpan` is used for Definition Preview in editors.
+    // Set `contextSpan` for local tokens. `contextSpan` is used for Definition Preview in editors.
+    for (const def of result.definitions) {
+      const script = language.scripts.get(def.fileName);
+      if (!isCSSModuleScript(script)) continue;
+      const cssModule = script.generated.root[CMK_DATA_KEY];
       const defName = unquote(def.name);
       const localToken = cssModule.localTokens.find(
         (t) => t.name === defName && t.loc.start.offset === def.textSpan.start,
@@ -26,20 +41,8 @@ export function getDefinitionAndBoundSpan(
           start: localToken.declarationLoc.start.offset,
           length: localToken.declarationLoc.end.offset - localToken.declarationLoc.start.offset,
         };
-        return true;
       }
-      // Exclude re-exported tokens from `@value ... from '...'`; the real definition lives in the target file.
-      const importedValue = cssModule.tokenImporters
-        .flatMap((i) => (i.type === 'value' ? i.values : []))
-        .find((v) => {
-          const localName = v.localName ?? v.name;
-          const localLoc = v.localLoc ?? v.loc;
-          return localName === defName && localLoc.start.offset === def.textSpan.start;
-        });
-      if (importedValue) return false;
-
-      return true;
-    });
+    }
     return result;
   };
 }