User inferno prompt instead of handrolled one (#1867)

[lemmy-ui.git] / generate_translations.js

diff --git a/generate_translations.js b/generate_translations.js
index 594f574c3433778551159c28cc8ae5d32369ee6c..ac3e987521bdc7a5da613184f23be5275eccb444 100644 (file)
--- a/generate_translations.js
+++ b/generate_translations.js
@@ -30,30 +30,70 @@ fs.readdir(translationDir, (_err, files) => {
 const baseLanguage = "en";
 
 fs.readFile(`${translationDir}${baseLanguage}.json`, "utf8", (_, fileStr) => {
-  const keys = Object.keys(JSON.parse(fileStr));
+  const noOptionKeys = [];
+  const optionKeys = [];
+  const optionRegex = /\{\{(.+?)\}\}/g;
+  const optionMap = new Map();
+
+  for (const [key, val] of Object.entries(JSON.parse(fileStr))) {
+    const options = [];
+    for (
+      let match = optionRegex.exec(val);
+      match;
+      match = optionRegex.exec(val)
+    ) {
+      options.push(match[1]);
+    }
+
+    if (options.length > 0) {
+      optionMap.set(key, options);
+      optionKeys.push(key);
+    } else {
+      noOptionKeys.push(key);
+    }
+  }
+
+  const indent = "    ";
 
   const data = `import { i18n } from "i18next";
 
 declare module "i18next" {
-  export type I18nKeys = 
-${keys.map(key => `    | "${key}"`).join("\n")};
-  
+  export type NoOptionI18nKeys = 
+${noOptionKeys.map(key => `${indent}| "${key}"`).join("\n")};
+
+  export type OptionI18nKeys = 
+${optionKeys.map(key => `${indent}| "${key}"`).join("\n")};
+
+  export type I18nKeys = NoOptionI18nKeys | OptionI18nKeys;
+
+  export type TTypedOptions<TKey extends OptionI18nKeys> =${Array.from(
+    optionMap.entries()
+  ).reduce(
+    (acc, [key, options]) =>
+      `${acc} TKey extends \"${key}\" ? ${
+        options.reduce((acc, cur) => acc + `${cur}: string | number; `, "{ ") +
+        "}"
+      } :\n${indent}`,
+    ""
+  )} (Record<string, unknown> | string);
+
   export interface TFunctionTyped {
-    // basic usage
+    // Translation requires options
     <
+      TKey extends OptionI18nKeys | OptionI18nKeys[],
       TResult extends TFunctionResult = string,
-      TInterpolationMap extends Record<string, unknown> = StringMap
-    >(
-      key: I18nKeys | I18nKeys[],
-      options?: TOptions<TInterpolationMap> | string
+      TInterpolationMap extends TTypedOptions<TKey> = StringMap
+    > (
+      key: TKey,
+      options: TOptions<TInterpolationMap> | string
     ): TResult;
-    // overloaded usage
+
+    // Translation does not require options
     <
       TResult extends TFunctionResult = string,
       TInterpolationMap extends Record<string, unknown> = StringMap
-    >(
-      key: I18nKeys | I18nKeys[],
-      defaultValue?: string,
+    > (
+      key: NoOptionI18nKeys | NoOptionI18nKeys[],
       options?: TOptions<TInterpolationMap> | string
     ): TResult;
   }