Function parseTablesFromQuery

src/utils/sqlAnalysis.ts:35–62

(sql: string)

// Returns alias → ParsedTableRef. Handles quoted identifiers, schema.table, and comma-separated FROM.
export const parseTablesFromQuery = (sql: string): Map<string, ParsedTableRef> | null => {
  if (!sql || sql.length === 0) return null;

  const fromSection = extractFromSection(sql);
  if (!fromSection) return null;

  const tableMap = new Map<string, ParsedTableRef>();
  const fromPattern =
    /(?:from|join|,)\s+("(?:[^"]|"")*"|`[^`]+`|[a-zA-Z_][a-zA-Z0-9_]*)(?:\.("(?:[^"]|"")*"|`[^`]+`|[a-zA-Z_][a-zA-Z0-9_]*))?(?:\s+(?:as\s+)?("(?:[^"]|"")*"|`[^`]+`|(?!(?:join|left|right|inner|outer|cross|natural|full|on|using|where|group|order|having|limit|offset|union|intersect|except|for|fetch|window|lateral|tablesample|qualify|straight_join)\b)[a-zA-Z_][a-zA-Z0-9_]*))?/gi;

  let match;
  let matchCount = 0;
  const MAX_MATCHES = 10;

  while ((match = fromPattern.exec(fromSection)) !== null && matchCount++ < MAX_MATCHES) {
    const schemaToken = match[2] ? match[1] : undefined;
    const tableToken = match[2] ?? match[1];
    if (!tableToken) continue;

    const tableName = stripIdentifierQuotes(tableToken);
    const schema = schemaToken ? stripIdentifierQuotes(schemaToken) : undefined;
    const aliasToken = match[3];
    const alias = aliasToken ? stripIdentifierQuotes(aliasToken) : tableName;
    tableMap.set(alias, { name: tableName, schema });
  }

  return tableMap.size > 0 ? tableMap : null;
};

// Optimized statement extractor - avoid full text scan when possible
export const getCurrentStatement = (model: { getValue: () => string; getOffsetAt: (position: { lineNumber: number; column: number }) => number }, position: { lineNumber: number; column: number }): string => {