Method create

sqlx/lexer.ts:195–226

(code: string)

export class SyntaxTreeNode {
  public static create(code: string) {
    const parentNode = new SyntaxTreeNode(SyntaxTreeNodeType.SQL);
    let currentNode = parentNode;
    const nodeStack = [currentNode];
    lexer.reset(code);
    for (const token of lexer) {
      if (!token.type) {
        throw new Error("Undefined token type encountered.");
      }
      if (START_TOKEN_NODE_MAPPINGS.has(token.type)) {
        const childType = START_TOKEN_NODE_MAPPINGS.get(token.type)!;
        if (childType === SyntaxTreeNodeType.SQL && currentNode.type !== SyntaxTreeNodeType.SQL) {
          throw new Error("SQL syntax tree nodes may only be children of other SQL nodes.");
        }
        const newCurrentNode = new SyntaxTreeNode(childType, [token.value]);
        nodeStack.push(newCurrentNode);
        currentNode.push(newCurrentNode);
        currentNode = newCurrentNode;
      } else if (CLOSE_TOKEN_TYPES.has(token.type)) {
        currentNode.push(token.value);
        nodeStack.pop();
        currentNode = nodeStack[nodeStack.length - 1];
      } else if (WHOLE_TOKEN_NODE_MAPPINGS.has(token.type)) {
        currentNode.push(
          new SyntaxTreeNode(WHOLE_TOKEN_NODE_MAPPINGS.get(token.type)!).push(token.value)
        );
      } else {
        currentNode.push(token.value);
      }
    }
    return parentNode;
  }

  public static isSyntaxTreeNode(node: string | SyntaxTreeNode): node is SyntaxTreeNode {
    return typeof node !== "string";