improve line numbers for linking

continuedev · Jun 11, 2024 · d6a3b48 · d6a3b48
1 parent 0bbd9dc
commit d6a3b48
Showing 1 changed file with 16 additions and 5 deletions.
diff --git a/gui/src/components/markdown/LinkableCode.tsx b/gui/src/components/markdown/LinkableCode.tsx
@@ -51,17 +51,28 @@ function LinkableCode(props: any) {
     );
     if (contextItemFileMatch) {
       setIsLink(true);
-      setFilepath(contextItemFileMatch.description);
-      setLine(0);
+      if (contextItemFileMatch.description.includes(" (")) {
+        // Get line number from filename
+        const [startLine, _] = contextItemFileMatch.description
+          .split(" (")[1]
+          .split(")")[0]
+          .split("-");
+        setFilepath(contextItemFileMatch.description.split(" (")[0]);
+        setLine(parseInt(startLine));
+      } else {
+        setFilepath(contextItemFileMatch.description.split(" (")[0]);
+        setLine(0);
+      }
     }
 
     if (content.length > 6) {
-      const contextItemContentMatch = contextItems.find((item) =>
-        item.content.includes(content),
+      const contextItemContentMatch = contextItems.find(
+        (item) =>
+          item.id.providerTitle === "file" && item.content.includes(content),
       );
       if (contextItemContentMatch) {
         setIsLink(true);
-        setFilepath(contextItemContentMatch.description);
+        setFilepath(contextItemContentMatch.description.split(" (")[0]);
         setLine(
           contextItemContentMatch.content
             .split("\n")