fix(generator): Output Parser.parseItem() calls with one valid type, …

…without the array (#551)
LuanRT · Dec 4, 2023 · bd487f8 · bd487f8
1 parent 48a5d4e
commit bd487f8
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/parser/generator.ts b/src/parser/generator.ts
@@ -448,12 +448,12 @@ export function toParser(key: string, inference_type: InferenceType, key_path: s
   switch (inference_type.type) {
     case 'renderer':
       {
-        parser = `Parser.parseItem(${key_path.join('.')}.${key}, [ ${inference_type.renderers.map((type) => `YTNodes.${type}`).join(', ')} ])`;
+        parser = `Parser.parseItem(${key_path.join('.')}.${key}, ${toParserValidTypes(inference_type.renderers)})`;
       }
       break;
     case 'renderer_list':
       {
-        parser = `Parser.parse(${key_path.join('.')}.${key}, true, [ ${inference_type.renderers.map((type) => `YTNodes.${type}`).join(', ')} ])`;
+        parser = `Parser.parse(${key_path.join('.')}.${key}, true, ${toParserValidTypes(inference_type.renderers)})`;
       }
       break;
     case 'object':
@@ -491,6 +491,14 @@ export function toParser(key: string, inference_type: InferenceType, key_path: s
   return parser;
 }
 
+function toParserValidTypes(types: string[]) {
+  if (types.length === 1) {
+    return `YTNodes.${types[0]}`;
+  }
+
+  return `[ ${types.map((type) => `YTNodes.${type}`).join(', ')} ]`;
+}
+
 function accessDataFromKeyPath(root: any, key_path: string[]) {
   let data = root;
   for (const key of key_path)