modify LanguageNormalizer.ts

kenkoooo · Oct 15, 2023 · 767ff59 · 767ff59
1 parent a9e6a2e
commit 767ff59
Showing 1 changed file with 18 additions and 4 deletions.
diff --git a/atcoder-problems-frontend/src/utils/LanguageNormalizer.ts b/atcoder-problems-frontend/src/utils/LanguageNormalizer.ts
@@ -1,7 +1,21 @@
+const mapping: [beginning: string, normalized: string][] = [
+  ["PyPy", "Python"],
+  ["Python (Cython", "Cython"],
+  ["Assembly x64", "Assembly x64"],
+  ["Awk", "AWK"],
+  ["IOI-Style", "C++"],
+  ["LuaJIT", "Lua"],
+  ["Seed7", "Seed7"],
+  ["Perl6", "Raku"],
+  ["Objective-C", "Objective-C"],
+];
+
 export const normalizeLanguage = (language: string): string => {
-  if (language.startsWith("Perl6")) {
-    return "Raku";
-  } else {
-    return language.replace(/\d*\s*\(.*\)$/, "");
+  for (const [beginning, normalized] of mapping) {
+    if (language.startsWith(beginning)) {
+      return normalized;
+    }
   }
+
+  return language.replace(/\s*[\d(\-].*/, "") || "Unknown";
 };