⬅️ Back to Table of Contents

📄 optimizeAST

📊 Analysis Summary

Metric	Count
🔧 Functions	2
📦 Imports	2

📚 Table of Contents

• Imports
• Functions

🛠️ File Location:

📂 packages/rule-schema-to-typescript-types/src/optimizeAST.ts

📦 Imports

Name	Source
SchemaAST	./types.js
UnionAST	./types.js

---

Functions

optimizeAST(ast: SchemaAST | null): void

Parameters:

• ast SchemaAST | null

Returns: void

Calls:

• optimizeAST
• unwrapUnions
• uniqueElementsMap.set
• JSON.stringify
• uniqueElementsMap.values

Internal Comments:

// hacky way to deduplicate union members (x2)
// @ts-expect-error -- purposely overwriting the property with a flattened list (x4)

Code

export function optimizeAST(ast: SchemaAST | null): void {
  if (ast == null) {
    return;
  }

  switch (ast.type) {
    case 'array': {
      optimizeAST(ast.elementType);
      return;
    }

    case 'literal':
      return;

    case 'object': {
      for (const property of ast.properties) {
        optimizeAST(property.type);
      }
      optimizeAST(ast.indexSignature);
      return;
    }

    case 'tuple': {
      for (const element of ast.elements) {
        optimizeAST(element);
      }
      optimizeAST(ast.spreadType);
      return;
    }

    case 'type-reference':
      return;

    case 'union': {
      const elements = unwrapUnions(ast);
      for (const element of elements) {
        optimizeAST(element);
      }

      // hacky way to deduplicate union members
      const uniqueElementsMap = new Map<string, SchemaAST>();
      for (const element of elements) {
        uniqueElementsMap.set(JSON.stringify(element), element);
      }
      const uniqueElements = [...uniqueElementsMap.values()];

      // @ts-expect-error -- purposely overwriting the property with a flattened list
      ast.elements = uniqueElements;
      return;
    }
  }
}

unwrapUnions(union: UnionAST): SchemaAST[]

Parameters:

• union UnionAST

Returns: SchemaAST[]

Calls:

• elements.push
• unwrapUnions
• elements[0].commentLines.unshift

Internal Comments:

// preserve the union's comment lines by prepending them to the first element's lines (x6)

Code

function unwrapUnions(union: UnionAST): SchemaAST[] {
  const elements: SchemaAST[] = [];
  for (const element of union.elements) {
    if (element.type === 'union') {
      elements.push(...unwrapUnions(element));
    } else {
      elements.push(element);
    }
  }

  if (elements.length > 0) {
    // preserve the union's comment lines by prepending them to the first element's lines
    elements[0].commentLines.unshift(...union.commentLines);
  }

  return elements;
}

---

Generated by Syntax Scribe