fix(site): fix language detection for Dockerfile (#12188)

BrunoQuaresma · web-flow · commit be1edc39953a · 2024-02-16T12:50:36.000Z
diff --git a/site/src/modules/templates/TemplateFiles/TemplateFiles.tsx b/site/src/modules/templates/TemplateFiles/TemplateFiles.tsx
@@ -9,17 +9,6 @@ import { TemplateFileTree } from "./TemplateFileTree";
 import { Link } from "react-router-dom";
 import EditOutlined from "@mui/icons-material/EditOutlined";
 
-const languageByExtension: Record<string, string> = {
-  tf: "hcl",
-  hcl: "hcl",
-  md: "markdown",
-  mkd: "markdown",
-  Dockerfile: "dockerfile",
-  sh: "shell",
-  tpl: "tpl",
-  protobuf: "protobuf",
-  nix: "dockerfile",
-};
 interface TemplateFilesProps {
   currentFiles: TemplateVersionFiles;
   /**
@@ -135,9 +124,7 @@ export const TemplateFiles: FC<TemplateFilesProps> = ({
                     </div>
                   </header>
                   <SyntaxHighlighter
-                    language={
-                      languageByExtension[filename.split(".").pop() ?? ""]
-                    }
+                    language={getLanguage(filename)}
                     value={info.value}
                     compareWith={info.previousValue}
                     editorProps={{
@@ -159,6 +146,27 @@ export const TemplateFiles: FC<TemplateFilesProps> = ({
   );
 };
 
+const languageByExtension: Record<string, string> = {
+  tf: "hcl",
+  hcl: "hcl",
+  md: "markdown",
+  mkd: "markdown",
+  sh: "shell",
+  tpl: "tpl",
+  protobuf: "protobuf",
+  nix: "dockerfile",
+  json: "json",
+};
+
+const getLanguage = (filename: string) => {
+  // Dockerfile can be like images/Dockerfile or Dockerfile.java
+  if (filename.includes("Dockerfile")) {
+    return "dockerfile";
+  }
+  const extension = filename.split(".").pop();
+  return languageByExtension[extension ?? ""];
+};
+
 const numberOfLines = (content: string) => {
   return content.split("\n").length;
 };
