Handle shell languages slightly better (#1043)

* Handle shell languages slightly better

Signed-off-by: nigel brown <[email protected]>

* linter love

Signed-off-by: nigel brown <[email protected]>

---------

Signed-off-by: nigel brown <[email protected]>

Author: therealnb

src/codegate/extract_snippets/message_extractor.py

@@ -138,7 +138,16 @@ def __init__(self):
             "rs": "rust",
             "java": "java",
         }
-        self._available_languages = ["python", "javascript", "typescript", "go", "rust", "java"]
+        self._available_languages = [
+            "sh",
+            "bash",
+            "python",
+            "javascript",
+            "typescript",
+            "go",
+            "rust",
+            "java",
+        ]  # noqa: E501
 
     @property
     @abstractmethod
@@ -194,7 +203,7 @@ def _ecosystem_from_message(self, message: str):
         Returns:
             Determined language based on message content
         """
-        return self._language_mapping.get(message, None)
+        return self._language_mapping.get(message, message)
 
     def _get_snippet_for_match(self, match: re.Match) -> CodeSnippet:
         matched_snippet = self._get_match_pattern_snippet(match)

src/codegate/pipeline/comment/output.py

@@ -60,7 +60,15 @@ async def _snippet_comment(self, snippet: CodeSnippet, context: PipelineContext)
                 liklihood = "likely"
             if snippet.language is not None:
                 language = snippet.language
-            comment = f"{comment}\n\n🛡️ CodeGate: The {language} supplied is {liklihood} unsafe. Please check carefully!\n\n"  # noqa: E501
+            if language not in [
+                "python",
+                "javascript",
+                "typescript",
+                "go",
+                "rust",
+                "java",
+            ]:  # noqa: E501
+                comment = f"{comment}\n\n🛡️ CodeGate: The {language} supplied is {liklihood} unsafe. Please check carefully!\n\n"  # noqa: E501
 
         snippet.libraries = PackageExtractor.extract_packages(snippet.code, snippet.language)