Merge pull request #3111 from continuedev/tomasz/comment-diff-snippet

Comment out diff snippet
continuedev · Nov 28, 2024 · 2138122 · 2138122
2 parents 8f2edef + e9933a1
commit 2138122
Showing 1 changed file with 4 additions and 19 deletions.
diff --git a/core/autocomplete/templating/formatting.ts b/core/autocomplete/templating/formatting.ts
@@ -13,12 +13,14 @@ const getCommentMark = (helper: HelperVars) => {
 };
 
 const addCommentMarks = (text: string, helper: HelperVars) => {
+  const commentMark = getCommentMark(helper);
   const lines = [
     ...text
       .trim()
       .split("\n")
-      .map((line) => `${getCommentMark(helper)} ${line}`),
+      .map((line) => `${commentMark} ${line}`),
   ];
+
   return lines.join("\n");
 };
 
@@ -61,11 +63,6 @@ const commentifySnippet = (
   };
 };
 
-const SNIPPET_TYPES_TO_COMMENT = [
-  AutocompleteSnippetType.Code,
-  AutocompleteSnippetType.Clipboard,
-];
-
 export const formatSnippets = (
   helper: HelperVars,
   snippets: AutocompleteSnippet[],
@@ -88,19 +85,7 @@ export const formatSnippets = (
         }
       })
       .map((item) => {
-        if (SNIPPET_TYPES_TO_COMMENT.includes(item.type)) {
-          return commentifySnippet(helper, item);
-        }
-
-        return item;
-      })
-      .map((item) => {
-        if (SNIPPET_TYPES_TO_COMMENT.includes(item.type)) {
-          const commentMark = getCommentMark(helper);
-          return item.content + `\n${commentMark}\n${commentMark}`;
-        }
-
-        return item.content;
+        return commentifySnippet(helper, item).content;
       })
       .join("\n") + `\n${currentFilepathComment}`
   );