First commit.

Signed-off-by: Chen Xiao <abigwc@gmail.com>

client/node_modules/@chevrotain/cst-dts-gen/src/generate.ts

@@ -0,0 +1,97 @@
+import { GenerateDtsOptions } from "@chevrotain/types";
+import {
+  CstNodeTypeDefinition,
+  PropertyArrayType,
+  PropertyTypeDefinition,
+  RuleArrayType,
+  TokenArrayType,
+} from "./model.js";
+
+export function genDts(
+  model: CstNodeTypeDefinition[],
+  options: Required<GenerateDtsOptions>,
+): string {
+  let contentParts: string[] = [];
+
+  contentParts = contentParts.concat(
+    `import type { CstNode, ICstVisitor, IToken } from "chevrotain";`,
+  );
+
+  contentParts = contentParts.concat(
+    model.flatMap((node) => genCstNodeTypes(node)),
+  );
+
+  if (options.includeVisitorInterface) {
+    contentParts = contentParts.concat(
+      genVisitor(options.visitorInterfaceName, model),
+    );
+  }
+
+  return contentParts.join("\n\n") + "\n";
+}
+
+function genCstNodeTypes(node: CstNodeTypeDefinition) {
+  const nodeCstInterface = genNodeInterface(node);
+  const nodeChildrenInterface = genNodeChildrenType(node);
+
+  return [nodeCstInterface, nodeChildrenInterface];
+}
+
+function genNodeInterface(node: CstNodeTypeDefinition) {
+  const nodeInterfaceName = getNodeInterfaceName(node.name);
+  const childrenTypeName = getNodeChildrenTypeName(node.name);
+
+  return `export interface ${nodeInterfaceName} extends CstNode {
+  name: "${node.name}";
+  children: ${childrenTypeName};
+}`;
+}
+
+function genNodeChildrenType(node: CstNodeTypeDefinition) {
+  const typeName = getNodeChildrenTypeName(node.name);
+
+  return `export type ${typeName} = {
+  ${node.properties.map((property) => genChildProperty(property)).join("\n  ")}
+};`;
+}
+
+function genChildProperty(prop: PropertyTypeDefinition) {
+  const typeName = buildTypeString(prop.type);
+  return `${prop.name}${prop.optional ? "?" : ""}: ${typeName}[];`;
+}
+
+function genVisitor(name: string, nodes: CstNodeTypeDefinition[]) {
+  return `export interface ${name}<IN, OUT> extends ICstVisitor<IN, OUT> {
+  ${nodes.map((node) => genVisitorFunction(node)).join("\n  ")}
+}`;
+}
+
+function genVisitorFunction(node: CstNodeTypeDefinition) {
+  const childrenTypeName = getNodeChildrenTypeName(node.name);
+  return `${node.name}(children: ${childrenTypeName}, param?: IN): OUT;`;
+}
+
+function buildTypeString(type: PropertyArrayType) {
+  if (Array.isArray(type)) {
+    const typeNames = [...new Set(type.map((t) => getTypeString(t)))];
+    const typeString = typeNames.join(" | ");
+    return "(" + typeString + ")";
+  } else {
+    return getTypeString(type);
+  }
+}
+
+function getTypeString(type: TokenArrayType | RuleArrayType) {
+  if (type.kind === "token") {
+    return "IToken";
+  }
+  return getNodeInterfaceName(type.name);
+}
+
+function getNodeInterfaceName(ruleName: string) {
+  return ruleName.charAt(0).toUpperCase() + ruleName.slice(1) + "CstNode";
+}
+
+function getNodeChildrenTypeName(ruleName: string) {
+  return ruleName.charAt(0).toUpperCase() + ruleName.slice(1) + "CstChildren";
+}