Enhanced AST Code Generation with Runtime Schema Support

[devin-ai-integration]

Enhanced AST Code Generation with Runtime Schema Support

Overview

This PR enhances the generateTsAstCodeFromPgAst() function by adding a new generateTsAstCodeFromPgAstWithSchema() function that uses runtime schema information to determine whether nodes should be wrapped or not, generating code with correct builder paths.

Changes Made

New Function: generateTsAstCodeFromPgAstWithSchema

• Takes both a PostgreSQL AST and runtime schema as parameters
• Uses the runtime schema's isNode property to determine node wrapping
• Generates t.ast.*() calls for unwrapped nodes and t.nodes.*() calls for wrapped nodes
• Includes sophisticated field-based node type detection for complex AST structures

Key Features

• Schema-aware code generation: Dynamically determines correct builder paths based on runtime schema
• Field matching algorithm: Identifies node types by comparing object fields against schema specifications
• Array handling: Properly preserves array structures in generated code
• Backward compatibility: Original generateTsAstCodeFromPgAst function remains unchanged

Implementation Details

• Added findNodeTypeByFields() helper function for dynamic node type identification
• Enhanced getValueNode() function with proper array handling
• Created schema lookup map for efficient runtime schema access
• Supports both required and optional field matching

Tests Added

• Enhanced AST generation test: Verifies correct wrapped vs unwrapped node generation
• Complex AST test: Tests mixed wrapping patterns with WithClause and nested structures
• Snapshot testing: Validates generated code structure and correctness

Example Usage

import { generateTsAstCodeFromPgAstWithSchema } from '@pgsql/proto-parser';
import { runtimeSchema } from './runtime-schema';

const ast = /* your PostgreSQL AST */;
const enhancedCode = generateTsAstCodeFromPgAstWithSchema(ast, runtimeSchema);

Generated Code Examples

The enhanced function now correctly generates:

• t.nodes.selectStmt({...}) for wrapped SelectStmt nodes
• t.nodes.withClause({...}) for wrapped WithClause nodes
• t.ast.limitOption(...) for unwrapped enum values

Testing

• ✅ All existing tests pass (42/42)
• ✅ New tests added with comprehensive coverage
• ✅ Snapshot tests validate correct code generation
• ✅ No breaking changes to existing functionality

Link to Devin run

https://app.devin.ai/sessions/3fc9c4d281d84299922c41d82c9c0477

Requested by

Dan Lynch ([email protected])

[devin-ai-integration]

🤖 Devin AI Engineer

I'll be helping with this pull request! Here's what you should know:

✅ I will automatically:

• Address comments on this PR. Add '(aside)' to your comment to have me ignore it.
• Look at CI failures and help fix them

Note: I can only respond to comments from users who have write access to this repository.

⚙️ Control Options:

• Disable automatic comment and CI monitoring