Nearly working TS interface generator for new corpus types

latest_corpus2.rb

@@ -0,0 +1,86 @@
+HASH = Sequence::Corpus.as_json({})
+
+VALUES = HASH.fetch(:values)
+VALUES_TPL = "export type Celery%{name} = %{type};\n"
+VALUES_OVERRIDE = { float: "number", integer: "number" }
+
+def emit_values
+  output = VALUES.map do |val|
+    name = val.fetch("name")
+    type = VALUES_OVERRIDE.fetch(name, name)
+    VALUES_TPL % { name: name.capitalize, type: type }
+  end
+    .uniq
+    .sort
+  puts(output)
+end
+
+ENUMS = HASH.fetch(:enums)
+ENUM_TPL = "export type ALLOWED_%{name} = %{type};\n"
+
+def emit_enums
+  output = ENUMS.map do |enum|
+    name = enum.fetch("name").upcase
+    type = enum.fetch("allowed_values").sort.map(&:inspect).uniq.join(" | ")
+    ENUM_TPL % { name: name, type: type }
+  end
+    .uniq
+    .sort
+
+  puts output
+end
+
+ARGS = HASH
+  .fetch(:args)
+  .reduce(HashWithIndifferentAccess.new) do |acc, arg|
+    acc[arg.fetch("name")] = arg
+    acc
+  end
+
+NODES = HASH.fetch(:nodes)
+
+NODE_START    = [ "/** %{docs}%{tag_docs} */",
+                  "export interface %{camel_case} {",
+                  '  kind: "%{snake_case}";',
+                  "  args: {", ].join("\n")
+MIDDLE_CENTER =   "    %{arg_name}: %{arg_values};"
+BOTTOM_END    = [ "  }",
+                  "  body: (%{body_types})[] | undefined;",
+                  "}\n", ].join("\n")
+
+def emit_nodes()
+  nodes = NODES.map do |node|
+      tag_list    = node.fetch("tags").sort.uniq.join(", ")
+      name        = node.fetch("name").to_s
+      body_types  = node.fetch("allowed_body_types")
+      tpl_binding = {
+        body_types: body_types.sort.uniq.map(&:to_s).map(&:camelize).join(" | "),
+        camel_case: name.camelize,
+        docs:       node.fetch("docs"),
+        snake_case: name,
+        tag_docs:   "Tagged with properties: #{tag_list}."
+      }
+
+      one   = NODE_START % tpl_binding
+      two   = node.fetch("allowed_args").sort.map do |arg|
+                MIDDLE_CENTER % {
+                  arg_name: arg.to_s,
+                  arg_values: ARGS.fetch(arg)
+                                .fetch("allowed_values")
+                                .map(&:name)
+                                .map(&:camelize)
+                                .join(" | ")
+                }
+              end
+      three = BOTTOM_END % tpl_binding
+      [one, two, three].flatten.join("\n")
+    end
+    .compact
+    .uniq
+    .join("\n\n")
+  puts nodes
+end
+
+emit_values()
+emit_enums()
+emit_nodes()