simplify extensions

Author: hate

apps/web/src/shared/utils/fileExtensions.ts

@@ -1,16 +1,34 @@
-const extensions: Record<string, string> = {
-  rust: "rs",
-  typescript: "ts",
-  javascript: "js",
-  python: "py",
+const extensionToLanguage: Record<string, string> = {
+  rs: "rust",
+  ts: "typescript",
+  tsx: "typescript",
+  js: "javascript",
+  jsx: "javascript",
+  py: "python",
   go: "go",
   java: "java",
   cpp: "cpp",
-  ruby: "rb",
+  cc: "cpp",
+  cxx: "cpp",
+  hpp: "cpp",
+  h: "c",
+  c: "c",
+  rb: "ruby",
   swift: "swift",
-  kotlin: "kt",
+  kt: "kotlin",
+  txt: "plaintext",
+  md: "plaintext",
+  text: "plaintext",
 };
 
 export function getFileExtension(language: string): string {
-  return extensions[language.toLowerCase()] ?? "txt";
+  const lang = language.toLowerCase();
+  const ext = Object.keys(extensionToLanguage).find(key => extensionToLanguage[key] === lang);
+  return ext ?? "txt";
+}
+
+export function getLanguageFromFilename(filename: string): string {
+  const match = filename.toLowerCase().match(/\.([a-z0-9]+)$/);
+  if (!match) return "plaintext";
+  return extensionToLanguage[match[1]] ?? "plaintext";
 }