Function parseSQLStatement

packages/smart-cells/src/parsers/sql-parser.ts:173–323

* Parse a SQL statement from a Python code string. * * @param code - The Python code string to parse * @returns The parsed SQL statement or null if parsing fails

(code: string)

 * @returns The parsed SQL statement or null if parsing fails
 */
function parseSQLStatement(code: string): SQLParseInfo | null {
  try {
    const tree = parsePythonAST(code);
    const cursor = tree.cursor();

    // Trees start with a Script node
    if (cursor.name === "Script") {
      cursor.next();
    }
    const assignStmt = findAssignment(cursor);
    if (!assignStmt) {
      return null;
    }

    // Code after the assignment statement is not allowed
    if (code.slice(assignStmt.to).trim().length > 0) {
      return null;
    }

    let dfName: string | null = null;
    let sqlString: string | null = null;
    let engine: string | undefined;
    let output: boolean | undefined;
    let startPosition = 0;

    // Parse the assignment
    const assignCursor = assignStmt.cursor();
    assignCursor.firstChild(); // Move to first child of assignment

    // First child should be the variable name
    if (assignCursor.name === "VariableName") {
      dfName = code.slice(assignCursor.from, assignCursor.to);
    }

    if (!dfName) {
      return null;
    }

    // Move to the expression part (after the =)
    // We need to find the value being assigned (after AssignOp)
    let foundAssignOp = false;
    let rightHandSide: SyntaxNode | null = null;

    while (assignCursor.nextSibling()) {
      if (assignCursor.name === "AssignOp") {
        foundAssignOp = true;
      } else if (foundAssignOp && !rightHandSide) {
        // This is the right-hand side of the assignment
        rightHandSide = assignCursor.node;
        break;
      }
    }

    if (!rightHandSide) {
      return null;
    }

    // Check if the right-hand side is wrapped in a conditional or binary expression