Sep 13, 2023 · Sep 13, 2023 · Sep 13, 2023 · Sep 13, 2023 · Sep 13, 2023 · Sep 13, 2023
diff --git a/site/src/components/TemplateFiles/TemplateFiles.tsx b/site/src/components/TemplateFiles/TemplateFiles.tsx
 import { UseTabResult } from "hooks/useTab";
 import { FC } from "react";
 import { combineClasses } from "utils/combineClasses";
 import { TemplateVersionFiles } from "utils/templateVersion";
 import { AllowedExtension, TemplateVersionFiles } from "utils/templateVersion";
 import InsertDriveFileOutlined from "@mui/icons-material/InsertDriveFileOutlined";

 const iconByExtension: Record<string, JSX.Element> = {
 const iconByExtension: Record<AllowedExtension, JSX.Element> = {
  tf: <TerraformIcon />,
  md: <MarkdownIcon />,
  mkd: <MarkdownIcon />,
  Dockerfile: <DockerIcon />,
  protobuf: <InsertDriveFileOutlined />,
  sh: <InsertDriveFileOutlined />,
  tpl: <InsertDriveFileOutlined />,
 };

 const getExtension = (filename: string) => {
  return filename;
 };

 const languageByExtension: Record<string, string> = {
 const languageByExtension: Record<AllowedExtension, string> = {
  tf: "hcl",
  md: "markdown",
  mkd: "markdown",
  Dockerfile: "dockerfile",
  sh: "bash",
  tpl: "tpl",
  protobuf: "protobuf",
 };

 export const TemplateFiles: FC<{
      <div className={styles.tabs}>
        {filenames.map((filename, index) => {
          const tabValue = index.toString();
          const extension = getExtension(filename);
          const extension = getExtension(filename) as AllowedExtension;
          const icon = iconByExtension[extension];
          const hasDiff =
            previousFiles &&
      <SyntaxHighlighter
        value={currentFile}
        compareWith={previousFile}
        language={languageByExtension[getExtension(selectedFilename)]}
        language={
          languageByExtension[
            getExtension(selectedFilename) as AllowedExtension
          ]
        }
      />
    </div>
  );
diff --git a/site/src/pages/TemplateVersionPage/TemplateVersionPageView.stories.tsx b/site/src/pages/TemplateVersionPage/TemplateVersionPageView.stories.tsx
    currentFiles: {
      "README.md": readmeContent,
      "main.tf": `{}`,
      "some.tpl": `{{.Name}}`,
      "some.sh": `echo "Hello world"`,
    },
  },
 };
diff --git a/site/src/utils/templateVersion.ts b/site/src/utils/templateVersion.ts
  return files;
 };

 export const allowedExtensions = ["tf", "md", "Dockerfile", "protobuf"];
 export const allowedExtensions = [
  "tf",
  "md",
  "mkd",
  "Dockerfile",
  "protobuf",
  "sh",
  "tpl",
 ] as const;

 export type AllowedExtension = (typeof allowedExtensions)[number];

 export const isAllowedFile = (name: string) => {
  return allowedExtensions.some((ext) => name.endsWith(ext));
Original file line number	Diff line number	Diff line change
Expand Up		@@ -40,6 +40,8 @@ const defaultArgs: TemplateVersionPageViewProps = {
		currentFiles: {
		"README.md": readmeContent,
		"main.tf": `{}`,
		"some.tpl": `{{.Name}}`,
		"some.sh": `echo "Hello world"`,
		},
		},
		};
Expand Down