@databricks/appkit 0.33.0 → 0.34.1

package/dist/shared/src/sql/helpers.js.map

@@ -1 +1 @@
-{"version":3,"file":"helpers.js","names":[],"sources":["../../../../../shared/src/sql/helpers.ts"],"sourcesContent":["import type {\n  SQLBinaryMarker,\n  SQLBooleanMarker,\n  SQLDateMarker,\n  SQLNumberMarker,\n  SQLStringMarker,\n  SQLTimestampMarker,\n  SQLTypeMarker,\n} from \"./types\";\n\n/**\n * SQL helper namespace\n */\nexport const sql = {\n  /**\n   * Creates a DATE type parameter\n   * Accepts Date objects or ISO date strings (YYYY-MM-DD format)\n   * @param value - Date object or ISO date string\n   * @returns Marker object for DATE type parameter\n   * @example\n   * ```typescript\n   * const params = { startDate: sql.date(new Date(\"2024-01-01\")) };\n   * params = { startDate: \"2024-01-01\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { startDate: sql.date(\"2024-01-01\") };\n   * params = { startDate: \"2024-01-01\" }\n   * ```\n   */\n  date(value: Date | string): SQLDateMarker {\n    let dateValue: string = \"\";\n\n    // check if value is a Date object\n    if (value instanceof Date) {\n      dateValue = value.toISOString().split(\"T\")[0];\n    }\n    // check if value is a string\n    else if (typeof value === \"string\") {\n      // validate format\n      if (!/^\\d{4}-\\d{2}-\\d{2}$/.test(value)) {\n        throw new Error(\n          `sql.date() expects Date or ISO date string (YYYY-MM-DD format), got: ${value}`,\n        );\n      }\n      dateValue = value;\n    }\n    // if value is not a Date object or string, throw an error\n    else {\n      throw new Error(\n        `sql.date() expects Date or ISO date string (YYYY-MM-DD format), got: ${typeof value}`,\n      );\n    }\n\n    return {\n      __sql_type: \"DATE\",\n      value: dateValue,\n    };\n  },\n\n  /**\n   * Creates a TIMESTAMP type parameter\n   * Accepts Date objects, ISO timestamp strings, or Unix timestamp numbers\n   * @param value - Date object, ISO timestamp string, or Unix timestamp number\n   * @returns Marker object for TIMESTAMP type parameter\n   * @example\n   * ```typescript\n   * const params = { createdTime: sql.timestamp(new Date(\"2024-01-01T12:00:00Z\")) };\n   * params = { createdTime: \"2024-01-01T12:00:00Z\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { createdTime: sql.timestamp(\"2024-01-01T12:00:00Z\") };\n   * params = { createdTime: \"2024-01-01T12:00:00Z\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { createdTime: sql.timestamp(1704110400000) };\n   * params = { createdTime: \"2024-01-01T12:00:00Z\" }\n   * ```\n   */\n  timestamp(value: Date | string | number): SQLTimestampMarker {\n    let timestampValue: string = \"\";\n\n    if (value instanceof Date) {\n      timestampValue = value.toISOString().replace(/\\.000(Z|[+-])/, \"$1\");\n    } else if (typeof value === \"string\") {\n      const isoRegex =\n        /^\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}:\\d{2}(\\.\\d{1,9})?(Z|[+-]\\d{2}:\\d{2})?$/;\n      if (!isoRegex.test(value)) {\n        throw new Error(\n          `sql.timestamp() expects ISO timestamp string (YYYY-MM-DDTHH:MM:SS.mmmZ or YYYY-MM-DDTHH:MM:SS.mmm+HH:MM), got: ${value}`,\n        );\n      }\n      timestampValue = value;\n    } else if (typeof value === \"number\") {\n      const date = new Date(value > 1e12 ? value : value * 1000);\n      timestampValue = date.toISOString().replace(/\\.000(Z|[+-])/, \"$1\");\n    } else {\n      throw new Error(\n        `sql.timestamp() expects Date, ISO timestamp string, or Unix timestamp number, got: ${typeof value}`,\n      );\n    }\n\n    return {\n      __sql_type: \"TIMESTAMP\",\n      value: timestampValue,\n    };\n  },\n\n  /**\n   * Creates a NUMERIC type parameter\n   * Accepts numbers or numeric strings\n   * @param value - Number or numeric string\n   * @returns Marker object for NUMERIC type parameter\n   * @example\n   * ```typescript\n   * const params = { userId: sql.number(123) };\n   * params = { userId: \"123\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { userId: sql.number(\"123\") };\n   * params = { userId: \"123\" }\n   * ```\n   */\n  number(value: number | string): SQLNumberMarker {\n    let numValue: string = \"\";\n\n    // check if value is a number\n    if (typeof value === \"number\") {\n      numValue = value.toString();\n    }\n    // check if value is a string\n    else if (typeof value === \"string\") {\n      if (value === \"\" || Number.isNaN(Number(value))) {\n        throw new Error(\n          `sql.number() expects number or numeric string, got: ${value === \"\" ? \"empty string\" : value}`,\n        );\n      }\n      numValue = value;\n    }\n    // if value is not a number or string, throw an error\n    else {\n      throw new Error(\n        `sql.number() expects number or numeric string, got: ${typeof value}`,\n      );\n    }\n\n    return {\n      __sql_type: \"NUMERIC\",\n      value: numValue,\n    };\n  },\n\n  /**\n   * Creates a STRING type parameter\n   * Accepts strings, numbers, or booleans\n   * @param value - String, number, or boolean\n   * @returns Marker object for STRING type parameter\n   * @example\n   * ```typescript\n   * const params = { name: sql.string(\"John\") };\n   * params = { name: \"John\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { name: sql.string(123) };\n   * params = { name: \"123\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { name: sql.string(true) };\n   * params = { name: \"true\" }\n   * ```\n   */\n  string(value: string | number | boolean): SQLStringMarker {\n    if (\n      typeof value !== \"string\" &&\n      typeof value !== \"number\" &&\n      typeof value !== \"boolean\"\n    ) {\n      throw new Error(\n        `sql.string() expects string or number or boolean, got: ${typeof value}`,\n      );\n    }\n\n    let stringValue: string = \"\";\n\n    if (typeof value === \"string\") {\n      stringValue = value;\n    } else {\n      stringValue = value.toString();\n    }\n\n    return {\n      __sql_type: \"STRING\",\n      value: stringValue,\n    };\n  },\n\n  /**\n   * Create a BOOLEAN type parameter\n   * Accepts booleans, strings, or numbers\n   * @param value - Boolean, string, or number\n   * @returns Marker object for BOOLEAN type parameter\n   * @example\n   * ```typescript\n   * const params = { isActive: sql.boolean(true) };\n   * params = { isActive: \"true\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { isActive: sql.boolean(\"true\") };\n   * params = { isActive: \"true\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { isActive: sql.boolean(1) };\n   * params = { isActive: \"true\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { isActive: sql.boolean(\"false\") };\n   * params = { isActive: \"false\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { isActive: sql.boolean(0) };\n   * params = { isActive: \"false\" }\n   * ```\n   * @returns\n   */\n  boolean(value: boolean | string | number): SQLBooleanMarker {\n    if (\n      typeof value !== \"boolean\" &&\n      typeof value !== \"string\" &&\n      typeof value !== \"number\"\n    ) {\n      throw new Error(\n        `sql.boolean() expects boolean or string (true or false) or number (1 or 0), got: ${typeof value}`,\n      );\n    }\n\n    let booleanValue: string = \"\";\n\n    if (typeof value === \"boolean\") {\n      booleanValue = value.toString();\n    }\n    // check if value is a number\n    else if (typeof value === \"number\") {\n      if (value !== 1 && value !== 0) {\n        throw new Error(\n          `sql.boolean() expects boolean or string (true or false) or number (1 or 0), got: ${value}`,\n        );\n      }\n      booleanValue = value === 1 ? \"true\" : \"false\";\n    }\n    // check if value is a string\n    else if (typeof value === \"string\") {\n      if (value !== \"true\" && value !== \"false\") {\n        throw new Error(\n          `sql.boolean() expects boolean or string (true or false) or number (1 or 0), got: ${value}`,\n        );\n      }\n      booleanValue = value;\n    }\n\n    return {\n      __sql_type: \"BOOLEAN\",\n      value: booleanValue,\n    };\n  },\n\n  /**\n   * Creates a BINARY parameter as hex-encoded STRING\n   * Accepts Uint8Array, ArrayBuffer, or hex string\n   * Note: Databricks SQL Warehouse doesn't support BINARY as parameter type,\n   * so this helper returns a STRING with hex encoding. Use UNHEX(:param) in your SQL.\n   * @param value - Uint8Array, ArrayBuffer, or hex string\n   * @returns Marker object with STRING type and hex-encoded value\n   * @example\n   * ```typescript\n   * // From Uint8Array:\n   * const params = { data: sql.binary(new Uint8Array([0x53, 0x70, 0x61, 0x72, 0x6b])) };\n   * // Returns: { __sql_type: \"STRING\", value: \"537061726B\" }\n   * // SQL: SELECT UNHEX(:data) as binary_value\n   * ```\n   * @example\n   * ```typescript\n   * // From hex string:\n   * const params = { data: sql.binary(\"537061726B\") };\n   * // Returns: { __sql_type: \"STRING\", value: \"537061726B\" }\n   * ```\n   */\n  binary(value: Uint8Array | ArrayBuffer | string): SQLBinaryMarker {\n    let hexValue: string = \"\";\n\n    if (value instanceof Uint8Array) {\n      // if value is a Uint8Array, convert it to a hex string\n      hexValue = Array.from(value)\n        .map((b) => b.toString(16).padStart(2, \"0\").toUpperCase())\n        .join(\"\");\n    } else if (value instanceof ArrayBuffer) {\n      // if value is an ArrayBuffer, convert it to a hex string\n      hexValue = Array.from(new Uint8Array(value))\n        .map((b) => b.toString(16).padStart(2, \"0\").toUpperCase())\n        .join(\"\");\n    } else if (typeof value === \"string\") {\n      // validate hex string\n      if (!/^[0-9A-Fa-f]*$/.test(value)) {\n        throw new Error(\n          `sql.binary() expects Uint8Array, ArrayBuffer, or hex string, got invalid hex: ${value}`,\n        );\n      }\n      hexValue = value.toUpperCase();\n    } else {\n      throw new Error(\n        `sql.binary() expects Uint8Array, ArrayBuffer, or hex string, got: ${typeof value}`,\n      );\n    }\n\n    return {\n      __sql_type: \"STRING\",\n      value: hexValue,\n    };\n  },\n};\n\n/**\n * Type guard to check if a value is a SQL type marker\n * @param value - Value to check\n * @returns True if the value is a SQL type marker, false otherwise\n * @example\n * ```typescript\n * const value = {\n *   __sql_type: \"DATE\",\n *   value: \"2024-01-01\",\n * };\n * const isSQLTypeMarker = isSQLTypeMarker(value);\n * console.log(isSQLTypeMarker); // true\n * ```\n */\nexport function isSQLTypeMarker(value: any): value is SQLTypeMarker {\n  return (\n    value !== null &&\n    typeof value === \"object\" &&\n    \"__sql_type\" in value &&\n    \"value\" in value &&\n    typeof value.__sql_type === \"string\" &&\n    typeof value.value === \"string\"\n  );\n}\n"],"mappings":";;;;AAaA,MAAa,MAAM;CAiBjB,KAAK,OAAqC;EACxC,IAAI,YAAoB;AAGxB,MAAI,iBAAiB,KACnB,aAAY,MAAM,aAAa,CAAC,MAAM,IAAI,CAAC;WAGpC,OAAO,UAAU,UAAU;AAElC,OAAI,CAAC,sBAAsB,KAAK,MAAM,CACpC,OAAM,IAAI,MACR,wEAAwE,QACzE;AAEH,eAAY;QAIZ,OAAM,IAAI,MACR,wEAAwE,OAAO,QAChF;AAGH,SAAO;GACL,YAAY;GACZ,OAAO;GACR;;CAwBH,UAAU,OAAmD;EAC3D,IAAI,iBAAyB;AAE7B,MAAI,iBAAiB,KACnB,kBAAiB,MAAM,aAAa,CAAC,QAAQ,iBAAiB,KAAK;WAC1D,OAAO,UAAU,UAAU;AAGpC,OAAI,CADF,wEACY,KAAK,MAAM,CACvB,OAAM,IAAI,MACR,kHAAkH,QACnH;AAEH,oBAAiB;aACR,OAAO,UAAU,SAE1B,kBADa,IAAI,KAAK,QAAQ,eAAO,QAAQ,QAAQ,IAAK,CACpC,aAAa,CAAC,QAAQ,iBAAiB,KAAK;MAElE,OAAM,IAAI,MACR,sFAAsF,OAAO,QAC9F;AAGH,SAAO;GACL,YAAY;GACZ,OAAO;GACR;;CAmBH,OAAO,OAAyC;EAC9C,IAAI,WAAmB;AAGvB,MAAI,OAAO,UAAU,SACnB,YAAW,MAAM,UAAU;WAGpB,OAAO,UAAU,UAAU;AAClC,OAAI,UAAU,MAAM,OAAO,MAAM,OAAO,MAAM,CAAC,CAC7C,OAAM,IAAI,MACR,uDAAuD,UAAU,KAAK,iBAAiB,QACxF;AAEH,cAAW;QAIX,OAAM,IAAI,MACR,uDAAuD,OAAO,QAC/D;AAGH,SAAO;GACL,YAAY;GACZ,OAAO;GACR;;CAwBH,OAAO,OAAmD;AACxD,MACE,OAAO,UAAU,YACjB,OAAO,UAAU,YACjB,OAAO,UAAU,UAEjB,OAAM,IAAI,MACR,0DAA0D,OAAO,QAClE;EAGH,IAAI,cAAsB;AAE1B,MAAI,OAAO,UAAU,SACnB,eAAc;MAEd,eAAc,MAAM,UAAU;AAGhC,SAAO;GACL,YAAY;GACZ,OAAO;GACR;;CAmCH,QAAQ,OAAoD;AAC1D,MACE,OAAO,UAAU,aACjB,OAAO,UAAU,YACjB,OAAO,UAAU,SAEjB,OAAM,IAAI,MACR,oFAAoF,OAAO,QAC5F;EAGH,IAAI,eAAuB;AAE3B,MAAI,OAAO,UAAU,UACnB,gBAAe,MAAM,UAAU;WAGxB,OAAO,UAAU,UAAU;AAClC,OAAI,UAAU,KAAK,UAAU,EAC3B,OAAM,IAAI,MACR,oFAAoF,QACrF;AAEH,kBAAe,UAAU,IAAI,SAAS;aAG/B,OAAO,UAAU,UAAU;AAClC,OAAI,UAAU,UAAU,UAAU,QAChC,OAAM,IAAI,MACR,oFAAoF,QACrF;AAEH,kBAAe;;AAGjB,SAAO;GACL,YAAY;GACZ,OAAO;GACR;;CAwBH,OAAO,OAA2D;EAChE,IAAI,WAAmB;AAEvB,MAAI,iBAAiB,WAEnB,YAAW,MAAM,KAAK,MAAM,CACzB,KAAK,MAAM,EAAE,SAAS,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,CACzD,KAAK,GAAG;WACF,iBAAiB,YAE1B,YAAW,MAAM,KAAK,IAAI,WAAW,MAAM,CAAC,CACzC,KAAK,MAAM,EAAE,SAAS,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,CACzD,KAAK,GAAG;WACF,OAAO,UAAU,UAAU;AAEpC,OAAI,CAAC,iBAAiB,KAAK,MAAM,CAC/B,OAAM,IAAI,MACR,iFAAiF,QAClF;AAEH,cAAW,MAAM,aAAa;QAE9B,OAAM,IAAI,MACR,qEAAqE,OAAO,QAC7E;AAGH,SAAO;GACL,YAAY;GACZ,OAAO;GACR;;CAEJ;;;;;;;;;;;;;;;AAgBD,SAAgB,gBAAgB,OAAoC;AAClE,QACE,UAAU,QACV,OAAO,UAAU,YACjB,gBAAgB,SAChB,WAAW,SACX,OAAO,MAAM,eAAe,YAC5B,OAAO,MAAM,UAAU"}
+{"version":3,"file":"helpers.js","names":[],"sources":["../../../../../shared/src/sql/helpers.ts"],"sourcesContent":["import type {\n  SQLBinaryMarker,\n  SQLBooleanMarker,\n  SQLDateMarker,\n  SQLNumberMarker,\n  SQLStringMarker,\n  SQLTimestampMarker,\n  SQLTypeMarker,\n} from \"./types\";\n\n// Strict numeric-literal regex used by string-input paths. Rejects empty\n// strings, whitespace, hex/octal/binary, `NaN`, `Infinity`, and other forms\n// that JS `Number()` would silently coerce.\nconst NUMERIC_LITERAL_RE = /^-?(\\d+\\.?\\d*|\\.\\d+)([eE][+-]?\\d+)?$/;\nconst INTEGER_LITERAL_RE = /^-?\\d+$/;\n\n// 32-bit signed INT range\nconst INT_MIN = -(2n ** 31n);\nconst INT_MAX = 2n ** 31n - 1n;\n// 64-bit signed BIGINT range\nconst BIGINT_MIN = -(2n ** 63n);\nconst BIGINT_MAX = 2n ** 63n - 1n;\n\nfunction ensureFiniteNumber(value: number, fnName: string): void {\n  if (!Number.isFinite(value)) {\n    throw new Error(`${fnName}() expects a finite number, got: ${value}`);\n  }\n}\n\nfunction ensureSafeInteger(value: number, fnName: string): void {\n  if (!Number.isSafeInteger(value)) {\n    throw new Error(\n      `${fnName}() received an integer outside Number.MAX_SAFE_INTEGER ` +\n        `(${value}); JS numbers cannot represent it exactly. ` +\n        `Pass a bigint (sql.bigint(BigInt(\"...\"))) or an integer-shaped string instead.`,\n    );\n  }\n}\n\nfunction ensureInBigIntRange(\n  parsed: bigint,\n  min: bigint,\n  max: bigint,\n  typeName: string,\n  fnName: string,\n  hint: string,\n): void {\n  if (parsed < min || parsed > max) {\n    throw new Error(\n      `${fnName}() value ${parsed} is outside ${typeName} range [${min}, ${max}]. ${hint}`,\n    );\n  }\n}\n\nfunction coerceNumericLike(value: number | string, fnName: string): string {\n  if (typeof value === \"number\") {\n    ensureFiniteNumber(value, fnName);\n    return value.toString();\n  }\n  if (typeof value === \"string\") {\n    if (!NUMERIC_LITERAL_RE.test(value)) {\n      throw new Error(\n        `${fnName}() expects number or numeric string, got: ${value === \"\" ? \"empty string\" : value}`,\n      );\n    }\n    return value;\n  }\n  throw new Error(\n    `${fnName}() expects number or numeric string, got: ${typeof value}`,\n  );\n}\n\nfunction coerceIntegerLike(value: number | string, fnName: string): string {\n  if (typeof value === \"number\") {\n    ensureFiniteNumber(value, fnName);\n    if (!Number.isInteger(value)) {\n      throw new Error(\n        `${fnName}() expects an integer, got non-integer number: ${value}`,\n      );\n    }\n    ensureSafeInteger(value, fnName);\n    // BigInt(value).toString() emits canonical decimal-integer text;\n    // Number.prototype.toString emits exponent notation for values like 1e21.\n    return BigInt(value).toString();\n  }\n  if (typeof value === \"string\") {\n    if (!INTEGER_LITERAL_RE.test(value)) {\n      throw new Error(\n        `${fnName}() expects integer number or integer-shaped string, got: ${value === \"\" ? \"empty string\" : value}`,\n      );\n    }\n    return value;\n  }\n  throw new Error(\n    `${fnName}() expects integer number or integer-shaped string, got: ${typeof value}`,\n  );\n}\n\n/**\n * SQL helper namespace\n */\nexport const sql = {\n  /**\n   * Creates a DATE type parameter\n   * Accepts Date objects or ISO date strings (YYYY-MM-DD format)\n   * @param value - Date object or ISO date string\n   * @returns Marker object for DATE type parameter\n   * @example\n   * ```typescript\n   * const params = { startDate: sql.date(new Date(\"2024-01-01\")) };\n   * params = { startDate: \"2024-01-01\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { startDate: sql.date(\"2024-01-01\") };\n   * params = { startDate: \"2024-01-01\" }\n   * ```\n   */\n  date(value: Date | string): SQLDateMarker {\n    let dateValue: string = \"\";\n\n    // check if value is a Date object\n    if (value instanceof Date) {\n      dateValue = value.toISOString().split(\"T\")[0];\n    }\n    // check if value is a string\n    else if (typeof value === \"string\") {\n      // validate format\n      if (!/^\\d{4}-\\d{2}-\\d{2}$/.test(value)) {\n        throw new Error(\n          `sql.date() expects Date or ISO date string (YYYY-MM-DD format), got: ${value}`,\n        );\n      }\n      dateValue = value;\n    }\n    // if value is not a Date object or string, throw an error\n    else {\n      throw new Error(\n        `sql.date() expects Date or ISO date string (YYYY-MM-DD format), got: ${typeof value}`,\n      );\n    }\n\n    return {\n      __sql_type: \"DATE\",\n      value: dateValue,\n    };\n  },\n\n  /**\n   * Creates a TIMESTAMP type parameter\n   * Accepts Date objects, ISO timestamp strings, or Unix timestamp numbers\n   * @param value - Date object, ISO timestamp string, or Unix timestamp number\n   * @returns Marker object for TIMESTAMP type parameter\n   * @example\n   * ```typescript\n   * const params = { createdTime: sql.timestamp(new Date(\"2024-01-01T12:00:00Z\")) };\n   * params = { createdTime: \"2024-01-01T12:00:00Z\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { createdTime: sql.timestamp(\"2024-01-01T12:00:00Z\") };\n   * params = { createdTime: \"2024-01-01T12:00:00Z\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { createdTime: sql.timestamp(1704110400000) };\n   * params = { createdTime: \"2024-01-01T12:00:00Z\" }\n   * ```\n   */\n  timestamp(value: Date | string | number): SQLTimestampMarker {\n    let timestampValue: string = \"\";\n\n    if (value instanceof Date) {\n      timestampValue = value.toISOString().replace(/\\.000(Z|[+-])/, \"$1\");\n    } else if (typeof value === \"string\") {\n      const isoRegex =\n        /^\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}:\\d{2}(\\.\\d{1,9})?(Z|[+-]\\d{2}:\\d{2})?$/;\n      if (!isoRegex.test(value)) {\n        throw new Error(\n          `sql.timestamp() expects ISO timestamp string (YYYY-MM-DDTHH:MM:SS.mmmZ or YYYY-MM-DDTHH:MM:SS.mmm+HH:MM), got: ${value}`,\n        );\n      }\n      timestampValue = value;\n    } else if (typeof value === \"number\") {\n      const date = new Date(value > 1e12 ? value : value * 1000);\n      timestampValue = date.toISOString().replace(/\\.000(Z|[+-])/, \"$1\");\n    } else {\n      throw new Error(\n        `sql.timestamp() expects Date, ISO timestamp string, or Unix timestamp number, got: ${typeof value}`,\n      );\n    }\n\n    return {\n      __sql_type: \"TIMESTAMP\",\n      value: timestampValue,\n    };\n  },\n\n  /**\n   * Creates a numeric type parameter. The wire SQL type is inferred from the\n   * value so the parameter binds correctly in any context, including `LIMIT`\n   * and `OFFSET`:\n   *\n   * - JS integer in `[-2^31, 2^31 - 1]` → `INT`\n   * - JS integer outside `INT` but within `Number.MAX_SAFE_INTEGER` → `BIGINT`\n   * - JS non-integer (`3.14`) → `DOUBLE`\n   * - integer-shaped string in `INT` range → `INT` (common HTTP-input case)\n   * - integer-shaped string outside `INT` but within `BIGINT` → `BIGINT`\n   * - decimal-shaped string (`\"123.45\"`) → `NUMERIC` (preserves precision)\n   *\n   * Why default to `INT`? Spark's `LIMIT` and `OFFSET` operators require\n   * `IntegerType` specifically — `BIGINT` (`LongType`) is rejected with\n   * `INVALID_LIMIT_LIKE_EXPRESSION.DATA_TYPE`. Catalyst auto-widens `INT`\n   * to `BIGINT` / `DECIMAL` / `DOUBLE` for wider columns, so `INT` is a\n   * strictly better default than `BIGINT`.\n   *\n   * Throws on `NaN`, `Infinity`, JS integers outside `Number.MAX_SAFE_INTEGER`,\n   * integer-shaped strings outside the `BIGINT` range, or non-numeric strings.\n   * Reach for `sql.int()`, `sql.bigint()`, `sql.float()`, `sql.double()`, or\n   * `sql.numeric()` to override the inferred type.\n   *\n   * @param value - Number or numeric string\n   * @returns Marker for a numeric SQL parameter\n   * @example\n   * ```typescript\n   * sql.number(123);              // { __sql_type: \"INT\",    value: \"123\" }\n   * sql.number(3_000_000_000);    // { __sql_type: \"BIGINT\", value: \"3000000000\" }\n   * sql.number(0.5);              // { __sql_type: \"DOUBLE\", value: \"0.5\" }\n   * sql.number(\"10\");             // { __sql_type: \"INT\",    value: \"10\" }\n   * sql.number(\"123.45\");         // { __sql_type: \"NUMERIC\", value: \"123.45\" }\n   * ```\n   */\n  number(value: number | string): SQLNumberMarker {\n    if (typeof value === \"number\") {\n      ensureFiniteNumber(value, \"sql.number\");\n      if (Number.isInteger(value)) {\n        ensureSafeInteger(value, \"sql.number\");\n        const asBigInt = BigInt(value);\n        // INT (32-bit) is required by Spark for LIMIT/OFFSET; Catalyst\n        // widens INT → BIGINT/DECIMAL/DOUBLE automatically.\n        if (asBigInt >= INT_MIN && asBigInt <= INT_MAX) {\n          return { __sql_type: \"INT\", value: asBigInt.toString() };\n        }\n        return { __sql_type: \"BIGINT\", value: asBigInt.toString() };\n      }\n      return { __sql_type: \"DOUBLE\", value: value.toString() };\n    }\n    if (typeof value === \"string\") {\n      if (!NUMERIC_LITERAL_RE.test(value)) {\n        throw new Error(\n          `sql.number() expects number or numeric string, got: ${value === \"\" ? \"empty string\" : value}`,\n        );\n      }\n      // Integer-shaped strings get the same INT-preferring inference, so\n      // `sql.number(req.query.n)` (Express/URLSearchParams strings) works\n      // with LIMIT/OFFSET out of the box. Out-of-BIGINT-range throws —\n      // sql.numeric() is the right helper for arbitrary-precision integers.\n      if (INTEGER_LITERAL_RE.test(value)) {\n        const parsed = BigInt(value);\n        ensureInBigIntRange(\n          parsed,\n          BIGINT_MIN,\n          BIGINT_MAX,\n          \"BIGINT (64-bit signed)\",\n          \"sql.number\",\n          \"Use sql.numeric() with a string for arbitrary-precision integers.\",\n        );\n        if (parsed >= INT_MIN && parsed <= INT_MAX) {\n          return { __sql_type: \"INT\", value };\n        }\n        return { __sql_type: \"BIGINT\", value };\n      }\n      // Non-integer strings stay NUMERIC: the caller chose to pass a string,\n      // honour their precision intent rather than coercing through JS number.\n      return { __sql_type: \"NUMERIC\", value };\n    }\n    throw new Error(\n      `sql.number() expects number or numeric string, got: ${typeof value}`,\n    );\n  },\n\n  /**\n   * Creates an `INT` (32-bit signed integer) parameter. Use when the column\n   * or context requires `INT` specifically (e.g. legacy schemas, or to make\n   * the wire type explicit).\n   *\n   * Rejects non-integers, values outside `Number.MAX_SAFE_INTEGER` (for\n   * number inputs), and values outside the signed 32-bit range\n   * `[-2^31, 2^31 - 1]`.\n   *\n   * @param value - Integer number or integer-shaped string\n   * @returns Marker pinned to `INT`\n   * @example\n   * ```typescript\n   * sql.int(42);     // { __sql_type: \"INT\", value: \"42\" }\n   * sql.int(\"42\");   // { __sql_type: \"INT\", value: \"42\" }\n   * ```\n   */\n  int(value: number | string): SQLNumberMarker & { __sql_type: \"INT\" } {\n    const stringValue = coerceIntegerLike(value, \"sql.int\");\n    ensureInBigIntRange(\n      BigInt(stringValue),\n      INT_MIN,\n      INT_MAX,\n      \"INT (32-bit signed)\",\n      \"sql.int\",\n      \"Use sql.bigint() for 64-bit values.\",\n    );\n    return { __sql_type: \"INT\", value: stringValue };\n  },\n\n  /**\n   * Creates a `BIGINT` (64-bit signed integer) parameter. Accepts JS\n   * `bigint` so callers can round-trip values outside `Number.MAX_SAFE_INTEGER`\n   * without precision loss; for `number` inputs, requires\n   * `Number.isSafeInteger(value)`.\n   *\n   * Rejects values outside the signed 64-bit range `[-2^63, 2^63 - 1]`.\n   *\n   * @param value - Integer number, bigint, or integer-shaped string\n   * @returns Marker pinned to `BIGINT`\n   * @example\n   * ```typescript\n   * sql.bigint(42);                     // { __sql_type: \"BIGINT\", value: \"42\" }\n   * sql.bigint(9007199254740993n);      // { __sql_type: \"BIGINT\", value: \"9007199254740993\" }\n   * sql.bigint(\"9007199254740993\");     // { __sql_type: \"BIGINT\", value: \"9007199254740993\" }\n   * ```\n   */\n  bigint(\n    value: number | bigint | string,\n  ): SQLNumberMarker & { __sql_type: \"BIGINT\" } {\n    if (typeof value === \"bigint\") {\n      ensureInBigIntRange(\n        value,\n        BIGINT_MIN,\n        BIGINT_MAX,\n        \"BIGINT (64-bit signed)\",\n        \"sql.bigint\",\n        \"Use sql.numeric() with a string for arbitrary-precision integers.\",\n      );\n      return { __sql_type: \"BIGINT\", value: value.toString() };\n    }\n    const stringValue = coerceIntegerLike(value, \"sql.bigint\");\n    ensureInBigIntRange(\n      BigInt(stringValue),\n      BIGINT_MIN,\n      BIGINT_MAX,\n      \"BIGINT (64-bit signed)\",\n      \"sql.bigint\",\n      \"Use sql.numeric() with a string for arbitrary-precision integers.\",\n    );\n    return { __sql_type: \"BIGINT\", value: stringValue };\n  },\n\n  /**\n   * Creates a `FLOAT` (single-precision, 32-bit) parameter. Note that JS\n   * numbers are 64-bit doubles, so values may be rounded to fit FLOAT\n   * precision at bind time.\n   *\n   * @param value - Number or numeric string\n   * @returns Marker pinned to `FLOAT`\n   * @example\n   * ```typescript\n   * sql.float(3.14);   // { __sql_type: \"FLOAT\", value: \"3.14\" }\n   * ```\n   */\n  float(value: number | string): SQLNumberMarker & { __sql_type: \"FLOAT\" } {\n    return {\n      __sql_type: \"FLOAT\",\n      value: coerceNumericLike(value, \"sql.float\"),\n    };\n  },\n\n  /**\n   * Creates a `DOUBLE` (double-precision, 64-bit) parameter. Same precision\n   * as a JS `number`, so `sql.double(value)` is exact for any JS number.\n   *\n   * @param value - Number or numeric string\n   * @returns Marker pinned to `DOUBLE`\n   * @example\n   * ```typescript\n   * sql.double(3.14);   // { __sql_type: \"DOUBLE\", value: \"3.14\" }\n   * ```\n   */\n  double(value: number | string): SQLNumberMarker & { __sql_type: \"DOUBLE\" } {\n    return {\n      __sql_type: \"DOUBLE\",\n      value: coerceNumericLike(value, \"sql.double\"),\n    };\n  },\n\n  /**\n   * Creates a `NUMERIC` (fixed-point DECIMAL) parameter. Use when you need\n   * exact decimal arithmetic (currency, percentages) — pass values as\n   * strings to avoid JS-number precision loss.\n   *\n   * Note: passing a JS `number` is accepted but lossy for many values\n   * (e.g. `0.1 + 0.2` → `\"0.30000000000000004\"`). Prefer strings.\n   *\n   * @param value - Number or numeric string (strings preferred for precision)\n   * @returns Marker pinned to `NUMERIC`\n   * @example\n   * ```typescript\n   * sql.numeric(\"12345.6789\");   // { __sql_type: \"NUMERIC\", value: \"12345.6789\" }\n   * ```\n   */\n  numeric(value: number | string): SQLNumberMarker & { __sql_type: \"NUMERIC\" } {\n    return {\n      __sql_type: \"NUMERIC\",\n      value: coerceNumericLike(value, \"sql.numeric\"),\n    };\n  },\n\n  /**\n   * Creates a STRING type parameter\n   * Accepts strings, numbers, or booleans\n   * @param value - String, number, or boolean\n   * @returns Marker object for STRING type parameter\n   * @example\n   * ```typescript\n   * const params = { name: sql.string(\"John\") };\n   * params = { name: \"John\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { name: sql.string(123) };\n   * params = { name: \"123\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { name: sql.string(true) };\n   * params = { name: \"true\" }\n   * ```\n   */\n  string(value: string | number | boolean): SQLStringMarker {\n    if (\n      typeof value !== \"string\" &&\n      typeof value !== \"number\" &&\n      typeof value !== \"boolean\"\n    ) {\n      throw new Error(\n        `sql.string() expects string or number or boolean, got: ${typeof value}`,\n      );\n    }\n\n    let stringValue: string = \"\";\n\n    if (typeof value === \"string\") {\n      stringValue = value;\n    } else {\n      stringValue = value.toString();\n    }\n\n    return {\n      __sql_type: \"STRING\",\n      value: stringValue,\n    };\n  },\n\n  /**\n   * Create a BOOLEAN type parameter\n   * Accepts booleans, strings, or numbers\n   * @param value - Boolean, string, or number\n   * @returns Marker object for BOOLEAN type parameter\n   * @example\n   * ```typescript\n   * const params = { isActive: sql.boolean(true) };\n   * params = { isActive: \"true\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { isActive: sql.boolean(\"true\") };\n   * params = { isActive: \"true\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { isActive: sql.boolean(1) };\n   * params = { isActive: \"true\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { isActive: sql.boolean(\"false\") };\n   * params = { isActive: \"false\" }\n   * ```\n   * @example\n   * ```typescript\n   * const params = { isActive: sql.boolean(0) };\n   * params = { isActive: \"false\" }\n   * ```\n   * @returns\n   */\n  boolean(value: boolean | string | number): SQLBooleanMarker {\n    if (\n      typeof value !== \"boolean\" &&\n      typeof value !== \"string\" &&\n      typeof value !== \"number\"\n    ) {\n      throw new Error(\n        `sql.boolean() expects boolean or string (true or false) or number (1 or 0), got: ${typeof value}`,\n      );\n    }\n\n    let booleanValue: string = \"\";\n\n    if (typeof value === \"boolean\") {\n      booleanValue = value.toString();\n    }\n    // check if value is a number\n    else if (typeof value === \"number\") {\n      if (value !== 1 && value !== 0) {\n        throw new Error(\n          `sql.boolean() expects boolean or string (true or false) or number (1 or 0), got: ${value}`,\n        );\n      }\n      booleanValue = value === 1 ? \"true\" : \"false\";\n    }\n    // check if value is a string\n    else if (typeof value === \"string\") {\n      if (value !== \"true\" && value !== \"false\") {\n        throw new Error(\n          `sql.boolean() expects boolean or string (true or false) or number (1 or 0), got: ${value}`,\n        );\n      }\n      booleanValue = value;\n    }\n\n    return {\n      __sql_type: \"BOOLEAN\",\n      value: booleanValue,\n    };\n  },\n\n  /**\n   * Creates a BINARY parameter as hex-encoded STRING\n   * Accepts Uint8Array, ArrayBuffer, or hex string\n   * Note: Databricks SQL Warehouse doesn't support BINARY as parameter type,\n   * so this helper returns a STRING with hex encoding. Use UNHEX(:param) in your SQL.\n   * @param value - Uint8Array, ArrayBuffer, or hex string\n   * @returns Marker object with STRING type and hex-encoded value\n   * @example\n   * ```typescript\n   * // From Uint8Array:\n   * const params = { data: sql.binary(new Uint8Array([0x53, 0x70, 0x61, 0x72, 0x6b])) };\n   * // Returns: { __sql_type: \"STRING\", value: \"537061726B\" }\n   * // SQL: SELECT UNHEX(:data) as binary_value\n   * ```\n   * @example\n   * ```typescript\n   * // From hex string:\n   * const params = { data: sql.binary(\"537061726B\") };\n   * // Returns: { __sql_type: \"STRING\", value: \"537061726B\" }\n   * ```\n   */\n  binary(value: Uint8Array | ArrayBuffer | string): SQLBinaryMarker {\n    let hexValue: string = \"\";\n\n    if (value instanceof Uint8Array) {\n      // if value is a Uint8Array, convert it to a hex string\n      hexValue = Array.from(value)\n        .map((b) => b.toString(16).padStart(2, \"0\").toUpperCase())\n        .join(\"\");\n    } else if (value instanceof ArrayBuffer) {\n      // if value is an ArrayBuffer, convert it to a hex string\n      hexValue = Array.from(new Uint8Array(value))\n        .map((b) => b.toString(16).padStart(2, \"0\").toUpperCase())\n        .join(\"\");\n    } else if (typeof value === \"string\") {\n      // validate hex string\n      if (!/^[0-9A-Fa-f]*$/.test(value)) {\n        throw new Error(\n          `sql.binary() expects Uint8Array, ArrayBuffer, or hex string, got invalid hex: ${value}`,\n        );\n      }\n      hexValue = value.toUpperCase();\n    } else {\n      throw new Error(\n        `sql.binary() expects Uint8Array, ArrayBuffer, or hex string, got: ${typeof value}`,\n      );\n    }\n\n    return {\n      __sql_type: \"STRING\",\n      value: hexValue,\n    };\n  },\n};\n\n/**\n * Type guard to check if a value is a SQL type marker\n * @param value - Value to check\n * @returns True if the value is a SQL type marker, false otherwise\n * @example\n * ```typescript\n * const value = {\n *   __sql_type: \"DATE\",\n *   value: \"2024-01-01\",\n * };\n * const isSQLTypeMarker = isSQLTypeMarker(value);\n * console.log(isSQLTypeMarker); // true\n * ```\n */\nexport function isSQLTypeMarker(value: any): value is SQLTypeMarker {\n  return (\n    value !== null &&\n    typeof value === \"object\" &&\n    \"__sql_type\" in value &&\n    \"value\" in value &&\n    typeof value.__sql_type === \"string\" &&\n    typeof value.value === \"string\"\n  );\n}\n"],"mappings":";AAaA,MAAM,qBAAqB;AAC3B,MAAM,qBAAqB;AAG3B,MAAM,UAAU,EAAE,MAAM;AACxB,MAAM,UAAU,MAAM,MAAM;AAE5B,MAAM,aAAa,EAAE,MAAM;AAC3B,MAAM,aAAa,MAAM,MAAM;AAE/B,SAAS,mBAAmB,OAAe,QAAsB;AAC/D,KAAI,CAAC,OAAO,SAAS,MAAM,CACzB,OAAM,IAAI,MAAM,GAAG,OAAO,mCAAmC,QAAQ;;AAIzE,SAAS,kBAAkB,OAAe,QAAsB;AAC9D,KAAI,CAAC,OAAO,cAAc,MAAM,CAC9B,OAAM,IAAI,MACR,GAAG,OAAO,0DACJ,MAAM,2HAEb;;AAIL,SAAS,oBACP,QACA,KACA,KACA,UACA,QACA,MACM;AACN,KAAI,SAAS,OAAO,SAAS,IAC3B,OAAM,IAAI,MACR,GAAG,OAAO,WAAW,OAAO,cAAc,SAAS,UAAU,IAAI,IAAI,IAAI,KAAK,OAC/E;;AAIL,SAAS,kBAAkB,OAAwB,QAAwB;AACzE,KAAI,OAAO,UAAU,UAAU;AAC7B,qBAAmB,OAAO,OAAO;AACjC,SAAO,MAAM,UAAU;;AAEzB,KAAI,OAAO,UAAU,UAAU;AAC7B,MAAI,CAAC,mBAAmB,KAAK,MAAM,CACjC,OAAM,IAAI,MACR,GAAG,OAAO,4CAA4C,UAAU,KAAK,iBAAiB,QACvF;AAEH,SAAO;;AAET,OAAM,IAAI,MACR,GAAG,OAAO,4CAA4C,OAAO,QAC9D;;AAGH,SAAS,kBAAkB,OAAwB,QAAwB;AACzE,KAAI,OAAO,UAAU,UAAU;AAC7B,qBAAmB,OAAO,OAAO;AACjC,MAAI,CAAC,OAAO,UAAU,MAAM,CAC1B,OAAM,IAAI,MACR,GAAG,OAAO,iDAAiD,QAC5D;AAEH,oBAAkB,OAAO,OAAO;AAGhC,SAAO,OAAO,MAAM,CAAC,UAAU;;AAEjC,KAAI,OAAO,UAAU,UAAU;AAC7B,MAAI,CAAC,mBAAmB,KAAK,MAAM,CACjC,OAAM,IAAI,MACR,GAAG,OAAO,2DAA2D,UAAU,KAAK,iBAAiB,QACtG;AAEH,SAAO;;AAET,OAAM,IAAI,MACR,GAAG,OAAO,2DAA2D,OAAO,QAC7E;;;;;AAMH,MAAa,MAAM;CAiBjB,KAAK,OAAqC;EACxC,IAAI,YAAoB;AAGxB,MAAI,iBAAiB,KACnB,aAAY,MAAM,aAAa,CAAC,MAAM,IAAI,CAAC;WAGpC,OAAO,UAAU,UAAU;AAElC,OAAI,CAAC,sBAAsB,KAAK,MAAM,CACpC,OAAM,IAAI,MACR,wEAAwE,QACzE;AAEH,eAAY;QAIZ,OAAM,IAAI,MACR,wEAAwE,OAAO,QAChF;AAGH,SAAO;GACL,YAAY;GACZ,OAAO;GACR;;CAwBH,UAAU,OAAmD;EAC3D,IAAI,iBAAyB;AAE7B,MAAI,iBAAiB,KACnB,kBAAiB,MAAM,aAAa,CAAC,QAAQ,iBAAiB,KAAK;WAC1D,OAAO,UAAU,UAAU;AAGpC,OAAI,CADF,wEACY,KAAK,MAAM,CACvB,OAAM,IAAI,MACR,kHAAkH,QACnH;AAEH,oBAAiB;aACR,OAAO,UAAU,SAE1B,kBADa,IAAI,KAAK,QAAQ,eAAO,QAAQ,QAAQ,IAAK,CACpC,aAAa,CAAC,QAAQ,iBAAiB,KAAK;MAElE,OAAM,IAAI,MACR,sFAAsF,OAAO,QAC9F;AAGH,SAAO;GACL,YAAY;GACZ,OAAO;GACR;;CAqCH,OAAO,OAAyC;AAC9C,MAAI,OAAO,UAAU,UAAU;AAC7B,sBAAmB,OAAO,aAAa;AACvC,OAAI,OAAO,UAAU,MAAM,EAAE;AAC3B,sBAAkB,OAAO,aAAa;IACtC,MAAM,WAAW,OAAO,MAAM;AAG9B,QAAI,YAAY,WAAW,YAAY,QACrC,QAAO;KAAE,YAAY;KAAO,OAAO,SAAS,UAAU;KAAE;AAE1D,WAAO;KAAE,YAAY;KAAU,OAAO,SAAS,UAAU;KAAE;;AAE7D,UAAO;IAAE,YAAY;IAAU,OAAO,MAAM,UAAU;IAAE;;AAE1D,MAAI,OAAO,UAAU,UAAU;AAC7B,OAAI,CAAC,mBAAmB,KAAK,MAAM,CACjC,OAAM,IAAI,MACR,uDAAuD,UAAU,KAAK,iBAAiB,QACxF;AAMH,OAAI,mBAAmB,KAAK,MAAM,EAAE;IAClC,MAAM,SAAS,OAAO,MAAM;AAC5B,wBACE,QACA,YACA,YACA,0BACA,cACA,oEACD;AACD,QAAI,UAAU,WAAW,UAAU,QACjC,QAAO;KAAE,YAAY;KAAO;KAAO;AAErC,WAAO;KAAE,YAAY;KAAU;KAAO;;AAIxC,UAAO;IAAE,YAAY;IAAW;IAAO;;AAEzC,QAAM,IAAI,MACR,uDAAuD,OAAO,QAC/D;;CAoBH,IAAI,OAAiE;EACnE,MAAM,cAAc,kBAAkB,OAAO,UAAU;AACvD,sBACE,OAAO,YAAY,EACnB,SACA,SACA,uBACA,WACA,sCACD;AACD,SAAO;GAAE,YAAY;GAAO,OAAO;GAAa;;CAoBlD,OACE,OAC4C;AAC5C,MAAI,OAAO,UAAU,UAAU;AAC7B,uBACE,OACA,YACA,YACA,0BACA,cACA,oEACD;AACD,UAAO;IAAE,YAAY;IAAU,OAAO,MAAM,UAAU;IAAE;;EAE1D,MAAM,cAAc,kBAAkB,OAAO,aAAa;AAC1D,sBACE,OAAO,YAAY,EACnB,YACA,YACA,0BACA,cACA,oEACD;AACD,SAAO;GAAE,YAAY;GAAU,OAAO;GAAa;;CAerD,MAAM,OAAmE;AACvE,SAAO;GACL,YAAY;GACZ,OAAO,kBAAkB,OAAO,YAAY;GAC7C;;CAcH,OAAO,OAAoE;AACzE,SAAO;GACL,YAAY;GACZ,OAAO,kBAAkB,OAAO,aAAa;GAC9C;;CAkBH,QAAQ,OAAqE;AAC3E,SAAO;GACL,YAAY;GACZ,OAAO,kBAAkB,OAAO,cAAc;GAC/C;;CAwBH,OAAO,OAAmD;AACxD,MACE,OAAO,UAAU,YACjB,OAAO,UAAU,YACjB,OAAO,UAAU,UAEjB,OAAM,IAAI,MACR,0DAA0D,OAAO,QAClE;EAGH,IAAI,cAAsB;AAE1B,MAAI,OAAO,UAAU,SACnB,eAAc;MAEd,eAAc,MAAM,UAAU;AAGhC,SAAO;GACL,YAAY;GACZ,OAAO;GACR;;CAmCH,QAAQ,OAAoD;AAC1D,MACE,OAAO,UAAU,aACjB,OAAO,UAAU,YACjB,OAAO,UAAU,SAEjB,OAAM,IAAI,MACR,oFAAoF,OAAO,QAC5F;EAGH,IAAI,eAAuB;AAE3B,MAAI,OAAO,UAAU,UACnB,gBAAe,MAAM,UAAU;WAGxB,OAAO,UAAU,UAAU;AAClC,OAAI,UAAU,KAAK,UAAU,EAC3B,OAAM,IAAI,MACR,oFAAoF,QACrF;AAEH,kBAAe,UAAU,IAAI,SAAS;aAG/B,OAAO,UAAU,UAAU;AAClC,OAAI,UAAU,UAAU,UAAU,QAChC,OAAM,IAAI,MACR,oFAAoF,QACrF;AAEH,kBAAe;;AAGjB,SAAO;GACL,YAAY;GACZ,OAAO;GACR;;CAwBH,OAAO,OAA2D;EAChE,IAAI,WAAmB;AAEvB,MAAI,iBAAiB,WAEnB,YAAW,MAAM,KAAK,MAAM,CACzB,KAAK,MAAM,EAAE,SAAS,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,CACzD,KAAK,GAAG;WACF,iBAAiB,YAE1B,YAAW,MAAM,KAAK,IAAI,WAAW,MAAM,CAAC,CACzC,KAAK,MAAM,EAAE,SAAS,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,CACzD,KAAK,GAAG;WACF,OAAO,UAAU,UAAU;AAEpC,OAAI,CAAC,iBAAiB,KAAK,MAAM,CAC/B,OAAM,IAAI,MACR,iFAAiF,QAClF;AAEH,cAAW,MAAM,aAAa;QAE9B,OAAM,IAAI,MACR,qEAAqE,OAAO,QAC7E;AAGH,SAAO;GACL,YAAY;GACZ,OAAO;GACR;;CAEJ;;;;;;;;;;;;;;;AAgBD,SAAgB,gBAAgB,OAAoC;AAClE,QACE,UAAU,QACV,OAAO,UAAU,YACjB,gBAAgB,SAChB,WAAW,SACX,OAAO,MAAM,eAAe,YAC5B,OAAO,MAAM,UAAU"}

package/dist/shared/src/sql/index.js

@@ -0,0 +1,3 @@
+import { isSQLTypeMarker, sql } from "./helpers.js";
+
+export {  };

package/dist/shared/src/sql/types.d.ts

@@ -3,8 +3,20 @@ interface SQLStringMarker {
   __sql_type: "STRING";
   value: string;
 }
+/**
+ * SQL numeric parameter marker. The wire type controls how Databricks SQL
+ * binds the value — notably, only integer types satisfy the `LIMIT` and
+ * `OFFSET` clauses.
+ *
+ * - `BIGINT` / `INT` — integer columns, LIMIT/OFFSET, IDs
+ * - `FLOAT` / `DOUBLE` — floating-point columns
+ * - `NUMERIC` — fixed-point DECIMAL columns (preserves precision)
+ *
+ * Created by `sql.number()` (auto-inferred), or by typed variants
+ * `sql.int()`, `sql.bigint()`, `sql.float()`, `sql.double()`, `sql.numeric()`.
+ */
 interface SQLNumberMarker {
-  __sql_type: "NUMERIC";
+  __sql_type: "INT" | "BIGINT" | "FLOAT" | "DOUBLE" | "NUMERIC";
   value: string;
 }
 interface SQLBooleanMarker {
@@ -26,7 +38,9 @@ interface SQLTimestampMarker {
 }
 /**
  * Object that identifies a typed SQL parameter.
- * Created using sql.date(), sql.string(), sql.number(), sql.boolean(), sql.timestamp(), sql.binary(), or sql.interval().
+ * Created using sql.date(), sql.string(), sql.number() (or the typed numeric
+ * variants sql.int/bigint/float/double/numeric), sql.boolean(),
+ * sql.timestamp(), or sql.binary().
  */
 type SQLTypeMarker = SQLStringMarker | SQLNumberMarker | SQLBooleanMarker | SQLBinaryMarker | SQLDateMarker | SQLTimestampMarker;
 //#endregion

package/dist/shared/src/sql/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","names":[],"sources":["../../../../../shared/src/sql/types.ts"],"mappings":";UAAiB,eAAA;EACf,UAAA;EACA,KAAA;AAAA;AAAA,UAGe,eAAA;EACf,UAAA;EACA,KAAA;AAAA;AAAA,UAGe,gBAAA;EACf,UAAA;EACA,KAAA;AAAA;;UAIe,eAAA;EACf,UAAA;EACA,KAAA;AAAA;AAAA,UAGe,aAAA;EACf,UAAA;EACA,KAAA;AAAA;AAAA,UAGe,kBAAA;EACf,UAAA;EACA,KAAA;AAAA;;AAFF;;;KASY,aAAA,GACR,eAAA,GACA,eAAA,GACA,gBAAA,GACA,eAAA,GACA,aAAA,GACA,kBAAA"}
+{"version":3,"file":"types.d.ts","names":[],"sources":["../../../../../shared/src/sql/types.ts"],"mappings":";UAAiB,eAAA;EACf,UAAA;EACA,KAAA;AAAA;;;AAeF;;;;;AAKA;;;;;UALiB,eAAA;EACf,UAAA;EACA,KAAA;AAAA;AAAA,UAGe,gBAAA;EACf,UAAA;EACA,KAAA;AAAA;;UAIe,eAAA;EACf,UAAA;EACA,KAAA;AAAA;AAAA,UAGe,aAAA;EACf,UAAA;EACA,KAAA;AAAA;AAAA,UAGe,kBAAA;EACf,UAAA;EACA,KAAA;AAAA;;;;;;;KASU,aAAA,GACR,eAAA,GACA,eAAA,GACA,gBAAA,GACA,eAAA,GACA,aAAA,GACA,kBAAA"}

package/dist/type-generator/query-registry.js

@@ -122,7 +122,7 @@ function generateUnknownResultQuery(sql, queryName) {
 }
 function extractParameterTypes(sql) {
 	const paramTypes = {};
-	const matches = sql.matchAll(/--\s*@param\s+(\w+)\s+(STRING|NUMERIC|BOOLEAN|DATE|TIMESTAMP|BINARY)/gi);
+	const matches = sql.matchAll(/--\s*@param\s+(\w+)\s+(STRING|NUMERIC|DECIMAL|BIGINT|TINYINT|SMALLINT|INT|FLOAT|DOUBLE|BOOLEAN|DATE|TIMESTAMP_NTZ|TIMESTAMP|BINARY)\b/gi);
 	for (const match of matches) {
 		const [, paramName, paramType] = match;
 		paramTypes[paramName] = paramType.toUpperCase();
@@ -131,11 +131,19 @@ function extractParameterTypes(sql) {
 }
 function defaultForType(sqlType) {
 	switch (sqlType?.toUpperCase()) {
-		case "NUMERIC": return "0";
+		case "NUMERIC":
+		case "DECIMAL":
+		case "BIGINT":
+		case "TINYINT":
+		case "SMALLINT":
+		case "INT": return "0";
+		case "FLOAT":
+		case "DOUBLE": return "0.0";
 		case "STRING": return "''";
 		case "BOOLEAN": return "true";
 		case "DATE": return "'2000-01-01'";
 		case "TIMESTAMP": return "'2000-01-01T00:00:00Z'";
+		case "TIMESTAMP_NTZ": return "'2000-01-01T00:00:00'";
 		case "BINARY": return "X'00'";
 		default: return "''";
 	}

package/dist/type-generator/query-registry.js.map

@@ -1 +1 @@
-{"version":3,"file":"query-registry.js","names":[],"sources":["../../src/type-generator/query-registry.ts"],"sourcesContent":["import fs from \"node:fs/promises\";\nimport path from \"node:path\";\nimport { WorkspaceClient } from \"@databricks/sdk-experimental\";\nimport pc from \"picocolors\";\nimport { createLogger } from \"../logging/logger\";\nimport { CACHE_VERSION, hashSQL, loadCache, saveCache } from \"./cache\";\nimport { Spinner } from \"./spinner\";\nimport {\n  type DatabricksStatementExecutionResponse,\n  type QuerySchema,\n  sqlTypeToHelper,\n  sqlTypeToMarker,\n} from \"./types\";\n\nconst logger = createLogger(\"type-generator:query-registry\");\n\n/**\n * Regex breakdown:\n *   '(?:[^']|'')*'   — matches a SQL string literal, including escaped '' pairs\n *   |                 — alternation: whichever branch matches first at a position wins\n *   --[^\\n]*          — matches a single-line SQL comment\n *\n * Because the regex engine scans left-to-right, a `'` is consumed as a string\n * literal before any `--` inside it could match as a comment — giving us\n * correct single-pass ordering without a manual state machine.\n *\n * V1: no block-comment support (deferred to next PR).\n */\nconst PROTECTED_RANGE_RE = /'(?:[^']|'')*'|--[^\\n]*/g;\n\n/**\n * Numeric-context patterns for positional type inference.\n * Hoisted to module scope — safe because matchAll() clones the regex internally.\n */\nconst NUMERIC_PATTERNS: RegExp[] = [\n  /\\bLIMIT\\s+:([a-zA-Z_]\\w*)/gi,\n  /\\bOFFSET\\s+:([a-zA-Z_]\\w*)/gi,\n  /\\bTOP\\s+:([a-zA-Z_]\\w*)/gi,\n  /\\bFETCH\\s+FIRST\\s+:([a-zA-Z_]\\w*)\\s+ROWS/gi,\n  // V1 limitation: arithmetic operators may false-positive for date\n  // expressions like `:start_date - INTERVAL '1 day'`. A smarter\n  // heuristic (e.g. look-ahead for INTERVAL) is deferred to a future PR.\n  /[+\\-*/]\\s*:([a-zA-Z_]\\w*)/g,\n  /:([a-zA-Z_]\\w*)\\s*[+\\-*/]/g,\n];\n\nexport function getProtectedRanges(sql: string): Array<[number, number]> {\n  const ranges: Array<[number, number]> = [];\n  for (const m of sql.matchAll(PROTECTED_RANGE_RE)) {\n    ranges.push([m.index, m.index + m[0].length]);\n  }\n  return ranges;\n}\n\nfunction isInsideProtectedRange(\n  offset: number,\n  ranges: Array<[number, number]>,\n): boolean {\n  return ranges.some(([start, end]) => offset >= start && offset < end);\n}\n\n/**\n * Parse a raw API/SDK error into a structured code + message.\n * Handles Databricks-style JSON bodies embedded in the message string,\n * e.g. `Response from server (Bad Request) {\"error_code\":\"...\",\"message\":\"...\"}`.\n */\nfunction parseError(raw: string): { code?: string; message: string } {\n  const jsonMatch = raw.match(/\\{[\\s\\S]*\\}/);\n  if (jsonMatch) {\n    try {\n      const parsed = JSON.parse(jsonMatch[0]);\n      if (parsed.error_code || parsed.message) {\n        return {\n          code: parsed.error_code,\n          message: parsed.message || raw,\n        };\n      }\n    } catch {\n      // not valid JSON, fall through\n    }\n  }\n  return { message: raw };\n}\n\n/**\n * Extract parameters from a SQL query\n * @param sql - the SQL query to extract parameters from\n * @returns an array of parameter names\n */\nexport function extractParameters(\n  sql: string,\n  ranges?: Array<[number, number]>,\n): string[] {\n  const protectedRanges = ranges ?? getProtectedRanges(sql);\n  const matches = sql.matchAll(/(?<!:):([a-zA-Z_]\\w*)/g);\n  const params = new Set<string>();\n  for (const match of matches) {\n    if (!isInsideProtectedRange(match.index, protectedRanges)) {\n      params.add(match[1]);\n    }\n  }\n  return Array.from(params);\n}\n\n// parameters that are injected by the server\nexport const SERVER_INJECTED_PARAMS = [\"workspaceId\"];\n\n/**\n * Generates the TypeScript type literal for query parameters from SQL.\n * Shared by both the success and failure paths.\n */\nfunction formatParametersType(sql: string): string {\n  const params = extractParameters(sql).filter(\n    (p) => !SERVER_INJECTED_PARAMS.includes(p),\n  );\n  const paramTypes = extractParameterTypes(sql);\n\n  return params.length > 0\n    ? `{\\n      ${params\n        .map((p) => {\n          const sqlType = paramTypes[p];\n          const markerType = sqlType\n            ? sqlTypeToMarker[sqlType]\n            : \"SQLTypeMarker\";\n          const helper = sqlType ? sqlTypeToHelper[sqlType] : \"sql.*()\";\n          return `/** ${sqlType || \"any\"} - use ${helper} */\\n      ${p}: ${markerType}`;\n        })\n        .join(\";\\n      \")};\\n    }`\n    : \"Record<string, never>\";\n}\n\nexport function convertToQueryType(\n  result: DatabricksStatementExecutionResponse,\n  sql: string,\n  queryName: string,\n): { type: string; hasResults: boolean } {\n  const dataRows = result.result?.data_array || [];\n  const columns = dataRows.map((row) => ({\n    name: row[0] || \"\",\n    type_name: row[1]?.toUpperCase() || \"STRING\",\n    comment: row[2] || undefined,\n  }));\n\n  const paramsType = formatParametersType(sql);\n\n  // generate result fields with JSDoc\n  const resultFields = columns.map((column) => {\n    const normalizedType = normalizeTypeName(column.type_name);\n    const mappedType = typeMap[normalizedType] || \"unknown\";\n    // validate column name is a valid identifier\n    const name = /^[a-zA-Z_$][a-zA-Z0-9_$]*$/.test(column.name)\n      ? column.name\n      : `\"${column.name}\"`;\n\n    // generate comment for column\n    const comment = column.comment\n      ? `/** ${column.comment} */\\n      `\n      : `/** @sqlType ${column.type_name} */\\n      `;\n\n    return `${comment}${name}: ${mappedType}`;\n  });\n\n  const hasResults = resultFields.length > 0;\n\n  const type = `{\n    name: \"${queryName}\";\n    parameters: ${paramsType};\n    result: ${\n      hasResults\n        ? `Array<{\n      ${resultFields.join(\";\\n      \")};\n    }>`\n        : \"unknown\"\n    };\n  }`;\n\n  return { type, hasResults };\n}\n\n/**\n * Used when DESCRIBE QUERY fails so the query still appears in QueryRegistry.\n * Generates a type with unknown result from SQL alone (no warehouse call).\n */\nfunction generateUnknownResultQuery(sql: string, queryName: string): string {\n  const paramsType = formatParametersType(sql);\n\n  return `{\n    name: \"${queryName}\";\n    parameters: ${paramsType};\n    result: unknown;\n  }`;\n}\n\nexport function extractParameterTypes(sql: string): Record<string, string> {\n  const paramTypes: Record<string, string> = {};\n  const regex =\n    /--\\s*@param\\s+(\\w+)\\s+(STRING|NUMERIC|BOOLEAN|DATE|TIMESTAMP|BINARY)/gi;\n  const matches = sql.matchAll(regex);\n  for (const match of matches) {\n    const [, paramName, paramType] = match;\n    paramTypes[paramName] = paramType.toUpperCase();\n  }\n\n  return paramTypes;\n}\n\nexport function defaultForType(sqlType: string | undefined): string {\n  switch (sqlType?.toUpperCase()) {\n    case \"NUMERIC\":\n      return \"0\";\n    case \"STRING\":\n      return \"''\";\n    case \"BOOLEAN\":\n      return \"true\";\n    case \"DATE\":\n      return \"'2000-01-01'\";\n    case \"TIMESTAMP\":\n      return \"'2000-01-01T00:00:00Z'\";\n    case \"BINARY\":\n      return \"X'00'\";\n    default:\n      return \"''\";\n  }\n}\n\n/**\n * Infer parameter types from positional context in SQL.\n * V1 only infers NUMERIC from patterns like LIMIT, OFFSET, TOP,\n * FETCH FIRST ... ROWS, and arithmetic operators.\n * Parameters inside string literals or SQL comments are ignored.\n */\nexport function inferParameterTypes(\n  sql: string,\n  ranges?: Array<[number, number]>,\n): Record<string, string> {\n  const inferred: Record<string, string> = {};\n  const protectedRanges = ranges ?? getProtectedRanges(sql);\n\n  for (const pattern of NUMERIC_PATTERNS) {\n    for (const match of sql.matchAll(pattern)) {\n      if (!isInsideProtectedRange(match.index, protectedRanges)) {\n        inferred[match[1]] = \"NUMERIC\";\n      }\n    }\n  }\n\n  return inferred;\n}\n\n/**\n * Generate query schemas from a folder of SQL files\n * It uses DESCRIBE QUERY to get the schema without executing the query\n * @param queryFolder - the folder containing the SQL files\n * @param warehouseId - the warehouse id to use for schema analysis\n * @param options - options for the query generation\n * @param options.noCache - if true, skip the cache and regenerate all types\n * @returns an array of query schemas\n */\nexport async function generateQueriesFromDescribe(\n  queryFolder: string,\n  warehouseId: string,\n  options: { noCache?: boolean; concurrency?: number } = {},\n): Promise<QuerySchema[]> {\n  const { noCache = false, concurrency: rawConcurrency = 10 } = options;\n  const concurrency =\n    typeof rawConcurrency === \"number\" && Number.isFinite(rawConcurrency)\n      ? Math.max(1, Math.floor(rawConcurrency))\n      : 10;\n\n  // read all query files and cache in parallel\n  const [allFiles, cache] = await Promise.all([\n    fs.readdir(queryFolder),\n    noCache\n      ? ({ version: CACHE_VERSION, queries: {} } as Awaited<\n          ReturnType<typeof loadCache>\n        >)\n      : loadCache(),\n  ]);\n\n  const queryFiles = allFiles.filter((file) => file.endsWith(\".sql\"));\n  logger.debug(\"Found %d SQL queries\", queryFiles.length);\n\n  const client = new WorkspaceClient({});\n  const spinner = new Spinner();\n\n  // Read all SQL files in parallel\n  const sqlContents = await Promise.all(\n    queryFiles.map((file) => fs.readFile(path.join(queryFolder, file), \"utf8\")),\n  );\n\n  const startTime = performance.now();\n\n  // Phase 1: Check cache, separate cached vs uncached\n  const cachedResults: Array<{ index: number; schema: QuerySchema }> = [];\n  const uncachedQueries: Array<{\n    index: number;\n    queryName: string;\n    sql: string;\n    sqlHash: string;\n    cleanedSql: string;\n  }> = [];\n  const logEntries: Array<{\n    queryName: string;\n    status: \"HIT\" | \"MISS\";\n    failed?: boolean;\n    error?: { code?: string; message: string };\n  }> = [];\n\n  for (let i = 0; i < queryFiles.length; i++) {\n    const file = queryFiles[i];\n    const rawName = path.basename(file, \".sql\");\n    const queryName = normalizeQueryName(rawName);\n\n    const sql = sqlContents[i];\n    const sqlHash = hashSQL(sql);\n\n    const cached = cache.queries[queryName];\n    if (cached && cached.hash === sqlHash && !cached.retry) {\n      cachedResults.push({\n        index: i,\n        schema: { name: queryName, type: cached.type },\n      });\n      logEntries.push({ queryName, status: \"HIT\" });\n    } else {\n      const protectedRanges = getProtectedRanges(sql);\n      const annotatedTypes = extractParameterTypes(sql);\n      const inferredTypes = inferParameterTypes(sql, protectedRanges);\n      const parameterTypes = { ...inferredTypes, ...annotatedTypes };\n      const sqlWithDefaults = sql.replace(\n        /(?<!:):([a-zA-Z_]\\w*)/g,\n        (original, paramName, offset) => {\n          if (isInsideProtectedRange(offset, protectedRanges)) {\n            return original;\n          }\n          return defaultForType(parameterTypes[paramName]);\n        },\n      );\n\n      // Warn about unresolved parameters\n      const allParams = extractParameters(sql, protectedRanges);\n      for (const param of allParams) {\n        if (SERVER_INJECTED_PARAMS.includes(param)) continue;\n        if (parameterTypes[param]) continue;\n        logger.warn(\n          '%s: parameter \":%s\" has no type annotation or inference. Add %s to the query file.',\n          queryFiles[i],\n          param,\n          `-- @param ${param} <TYPE>`,\n        );\n      }\n\n      const cleanedSql = sqlWithDefaults.trim().replace(/;\\s*$/, \"\");\n      uncachedQueries.push({ index: i, queryName, sql, sqlHash, cleanedSql });\n    }\n  }\n\n  // Phase 2: Execute all uncached DESCRIBE calls in parallel\n  type DescribeResult =\n    | {\n        status: \"ok\";\n        index: number;\n        schema: QuerySchema;\n        cacheEntry: { hash: string; type: string; retry: boolean };\n      }\n    | {\n        status: \"fail\";\n        index: number;\n        schema: QuerySchema;\n        cacheEntry: { hash: string; type: string; retry: boolean };\n        error: { code?: string; message: string };\n      };\n\n  const freshResults: Array<{ index: number; schema: QuerySchema }> = [];\n\n  if (uncachedQueries.length > 0) {\n    let completed = 0;\n    const total = uncachedQueries.length;\n    spinner.start(\n      `Describing ${total} ${total === 1 ? \"query\" : \"queries\"} (0/${total})`,\n    );\n\n    const describeOne = async ({\n      index,\n      queryName,\n      sql,\n      sqlHash,\n      cleanedSql,\n    }: (typeof uncachedQueries)[number]): Promise<DescribeResult> => {\n      const result = (await client.statementExecution.executeStatement({\n        statement: `DESCRIBE QUERY ${cleanedSql}`,\n        warehouse_id: warehouseId,\n      })) as DatabricksStatementExecutionResponse;\n\n      completed++;\n      spinner.update(\n        `Describing ${total} ${total === 1 ? \"query\" : \"queries\"} (${completed}/${total})`,\n      );\n\n      logger.debug(\n        \"DESCRIBE result for %s: state=%s, rows=%d\",\n        queryName,\n        result.status.state,\n        result.result?.data_array?.length ?? 0,\n      );\n\n      if (result.status.state === \"FAILED\") {\n        const sqlError =\n          result.status.error?.message || \"Query execution failed\";\n        logger.warn(\"DESCRIBE failed for %s: %s\", queryName, sqlError);\n        const type = generateUnknownResultQuery(sql, queryName);\n        return {\n          status: \"fail\",\n          index,\n          schema: { name: queryName, type },\n          cacheEntry: { hash: sqlHash, type, retry: true },\n          error: parseError(sqlError),\n        };\n      }\n\n      const { type, hasResults } = convertToQueryType(result, sql, queryName);\n      return {\n        status: \"ok\",\n        index,\n        schema: { name: queryName, type },\n        cacheEntry: { hash: sqlHash, type, retry: !hasResults },\n      };\n    };\n\n    // Process in chunks, saving cache after each chunk\n    const processBatchResults = (\n      settled: PromiseSettledResult<DescribeResult>[],\n      batchOffset: number,\n    ) => {\n      for (let i = 0; i < settled.length; i++) {\n        const entry = settled[i];\n        const { queryName } = uncachedQueries[batchOffset + i];\n\n        if (entry.status === \"fulfilled\") {\n          const res = entry.value;\n          freshResults.push({ index: res.index, schema: res.schema });\n          cache.queries[queryName] = res.cacheEntry;\n          logEntries.push({\n            queryName,\n            status: \"MISS\",\n            failed: res.status === \"fail\",\n            error: res.status === \"fail\" ? res.error : undefined,\n          });\n        } else {\n          const { sql, sqlHash, index } = uncachedQueries[batchOffset + i];\n          const reason =\n            entry.reason instanceof Error\n              ? entry.reason.message\n              : String(entry.reason);\n          logger.warn(\"DESCRIBE rejected for %s: %s\", queryName, reason);\n          const type = generateUnknownResultQuery(sql, queryName);\n          freshResults.push({ index, schema: { name: queryName, type } });\n          cache.queries[queryName] = { hash: sqlHash, type, retry: true };\n          logEntries.push({\n            queryName,\n            status: \"MISS\",\n            failed: true,\n            error: parseError(reason),\n          });\n        }\n      }\n    };\n\n    if (uncachedQueries.length > concurrency) {\n      for (let b = 0; b < uncachedQueries.length; b += concurrency) {\n        const batch = uncachedQueries.slice(b, b + concurrency);\n        const batchResults = await Promise.allSettled(batch.map(describeOne));\n        processBatchResults(batchResults, b);\n        await saveCache(cache);\n      }\n    } else {\n      const settled = await Promise.allSettled(\n        uncachedQueries.map(describeOne),\n      );\n      processBatchResults(settled, 0);\n      await saveCache(cache);\n    }\n\n    spinner.stop(\"\");\n  }\n\n  const elapsed = ((performance.now() - startTime) / 1000).toFixed(2);\n\n  // Print formatted table\n  if (logEntries.length > 0) {\n    const maxNameLen = Math.max(...logEntries.map((e) => e.queryName.length));\n    const separator = pc.dim(\"─\".repeat(50));\n    console.log(\"\");\n    console.log(\n      `  ${pc.bold(\"Typegen Queries\")} ${pc.dim(`(${logEntries.length})`)}`,\n    );\n    console.log(`  ${separator}`);\n    for (const entry of logEntries) {\n      const tag = entry.failed\n        ? pc.bold(pc.red(\"ERROR\"))\n        : entry.status === \"HIT\"\n          ? `cache ${pc.bold(pc.green(\"HIT  \"))}`\n          : `cache ${pc.bold(pc.yellow(\"MISS \"))}`;\n      const rawName = entry.queryName.padEnd(maxNameLen);\n      const name = entry.failed ? pc.dim(pc.strikethrough(rawName)) : rawName;\n      const errorCode = entry.error?.message.match(/\\[([^\\]]+)\\]/)?.[1];\n      const reason = errorCode ? `  ${pc.dim(errorCode)}` : \"\";\n      console.log(`  ${tag}  ${name}${reason}`);\n    }\n    const newCount = logEntries.filter(\n      (e) => e.status === \"MISS\" && !e.failed,\n    ).length;\n    const cacheCount = logEntries.filter(\n      (e) => e.status === \"HIT\" && !e.failed,\n    ).length;\n    const errorCount = logEntries.filter((e) => e.failed).length;\n    console.log(`  ${separator}`);\n    const parts = [`${newCount} new`, `${cacheCount} from cache`];\n    if (errorCount > 0)\n      parts.push(`${errorCount} ${errorCount === 1 ? \"error\" : \"errors\"}`);\n    console.log(`  ${parts.join(\", \")}. ${pc.dim(`${elapsed}s`)}`);\n    console.log(\"\");\n  }\n\n  // Merge and sort by original file index for deterministic output\n  return [...cachedResults, ...freshResults]\n    .sort((a, b) => a.index - b.index)\n    .map((r) => r.schema);\n}\n\n/**\n * Normalize query name by removing the .obo extension\n * @param queryName - the query name to normalize\n * @returns the normalized query name\n */\nfunction normalizeQueryName(fileName: string): string {\n  return fileName.replace(/\\.obo$/, \"\");\n}\n\n/**\n * Normalize SQL type name by removing parameters/generics\n * Examples:\n *   DECIMAL(38,6) -> DECIMAL\n *   ARRAY<STRING> -> ARRAY\n *   MAP<STRING,INT> -> MAP\n *   STRUCT<name:STRING> -> STRUCT\n *   INTERVAL DAY TO SECOND -> INTERVAL\n *   GEOGRAPHY(4326) -> GEOGRAPHY\n */\nexport function normalizeTypeName(typeName: string): string {\n  return typeName\n    .replace(/\\(.*\\)$/, \"\") // remove (p, s) eg: DECIMAL(38,6) -> DECIMAL\n    .replace(/<.*>$/, \"\") // remove <T> eg: ARRAY<STRING> -> ARRAY\n    .split(\" \")[0]; // take first word eg: INTERVAL DAY TO SECOND -> INTERVAL\n}\n\n/** Type Map for Databricks data types to JavaScript types */\nconst typeMap: Record<string, string> = {\n  // string types\n  STRING: \"string\",\n  BINARY: \"string\",\n  // boolean\n  BOOLEAN: \"boolean\",\n  // numeric types\n  TINYINT: \"number\",\n  SMALLINT: \"number\",\n  INT: \"number\",\n  BIGINT: \"number\",\n  FLOAT: \"number\",\n  DOUBLE: \"number\",\n  DECIMAL: \"number\",\n  // date/time types\n  DATE: \"string\",\n  TIMESTAMP: \"string\",\n  TIMESTAMP_NTZ: \"string\",\n  INTERVAL: \"string\",\n  // complex types\n  ARRAY: \"unknown[]\",\n  MAP: \"Record<string, unknown>\",\n  STRUCT: \"Record<string, unknown>\",\n  OBJECT: \"Record<string, unknown>\",\n  VARIANT: \"unknown\",\n  // spatial types\n  GEOGRAPHY: \"unknown\",\n  GEOMETRY: \"unknown\",\n  // null type\n  VOID: \"null\",\n};\n"],"mappings":";;;;;;;;;;AAcA,MAAM,SAAS,aAAa,gCAAgC;;;;;;;;;;;;;AAc5D,MAAM,qBAAqB;;;;;AAM3B,MAAM,mBAA6B;CACjC;CACA;CACA;CACA;CAIA;CACA;CACD;AAED,SAAgB,mBAAmB,KAAsC;CACvE,MAAM,SAAkC,EAAE;AAC1C,MAAK,MAAM,KAAK,IAAI,SAAS,mBAAmB,CAC9C,QAAO,KAAK,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;AAE/C,QAAO;;AAGT,SAAS,uBACP,QACA,QACS;AACT,QAAO,OAAO,MAAM,CAAC,OAAO,SAAS,UAAU,SAAS,SAAS,IAAI;;;;;;;AAQvE,SAAS,WAAW,KAAiD;CACnE,MAAM,YAAY,IAAI,MAAM,cAAc;AAC1C,KAAI,UACF,KAAI;EACF,MAAM,SAAS,KAAK,MAAM,UAAU,GAAG;AACvC,MAAI,OAAO,cAAc,OAAO,QAC9B,QAAO;GACL,MAAM,OAAO;GACb,SAAS,OAAO,WAAW;GAC5B;SAEG;AAIV,QAAO,EAAE,SAAS,KAAK;;;;;;;AAQzB,SAAgB,kBACd,KACA,QACU;CACV,MAAM,kBAAkB,UAAU,mBAAmB,IAAI;CACzD,MAAM,UAAU,IAAI,SAAS,yBAAyB;CACtD,MAAM,yBAAS,IAAI,KAAa;AAChC,MAAK,MAAM,SAAS,QAClB,KAAI,CAAC,uBAAuB,MAAM,OAAO,gBAAgB,CACvD,QAAO,IAAI,MAAM,GAAG;AAGxB,QAAO,MAAM,KAAK,OAAO;;AAI3B,MAAa,yBAAyB,CAAC,cAAc;;;;;AAMrD,SAAS,qBAAqB,KAAqB;CACjD,MAAM,SAAS,kBAAkB,IAAI,CAAC,QACnC,MAAM,CAAC,uBAAuB,SAAS,EAAE,CAC3C;CACD,MAAM,aAAa,sBAAsB,IAAI;AAE7C,QAAO,OAAO,SAAS,IACnB,YAAY,OACT,KAAK,MAAM;EACV,MAAM,UAAU,WAAW;EAC3B,MAAM,aAAa,UACf,gBAAgB,WAChB;EACJ,MAAM,SAAS,UAAU,gBAAgB,WAAW;AACpD,SAAO,OAAO,WAAW,MAAM,SAAS,OAAO,aAAa,EAAE,IAAI;GAClE,CACD,KAAK,YAAY,CAAC,YACrB;;AAGN,SAAgB,mBACd,QACA,KACA,WACuC;CAEvC,MAAM,WADW,OAAO,QAAQ,cAAc,EAAE,EACvB,KAAK,SAAS;EACrC,MAAM,IAAI,MAAM;EAChB,WAAW,IAAI,IAAI,aAAa,IAAI;EACpC,SAAS,IAAI,MAAM;EACpB,EAAE;CAEH,MAAM,aAAa,qBAAqB,IAAI;CAG5C,MAAM,eAAe,QAAQ,KAAK,WAAW;EAE3C,MAAM,aAAa,QADI,kBAAkB,OAAO,UAAU,KACZ;EAE9C,MAAM,OAAO,6BAA6B,KAAK,OAAO,KAAK,GACvD,OAAO,OACP,IAAI,OAAO,KAAK;AAOpB,SAAO,GAJS,OAAO,UACnB,OAAO,OAAO,QAAQ,eACtB,gBAAgB,OAAO,UAAU,eAEjB,KAAK,IAAI;GAC7B;CAEF,MAAM,aAAa,aAAa,SAAS;AAczC,QAAO;EAAE,MAZI;aACF,UAAU;kBACL,WAAW;cAEvB,aACI;QACF,aAAa,KAAK,YAAY,CAAC;UAE7B,UACL;;EAGY;EAAY;;;;;;AAO7B,SAAS,2BAA2B,KAAa,WAA2B;AAG1E,QAAO;aACI,UAAU;kBAHF,qBAAqB,IAAI,CAIjB;;;;AAK7B,SAAgB,sBAAsB,KAAqC;CACzE,MAAM,aAAqC,EAAE;CAG7C,MAAM,UAAU,IAAI,SADlB,yEACiC;AACnC,MAAK,MAAM,SAAS,SAAS;EAC3B,MAAM,GAAG,WAAW,aAAa;AACjC,aAAW,aAAa,UAAU,aAAa;;AAGjD,QAAO;;AAGT,SAAgB,eAAe,SAAqC;AAClE,SAAQ,SAAS,aAAa,EAA9B;EACE,KAAK,UACH,QAAO;EACT,KAAK,SACH,QAAO;EACT,KAAK,UACH,QAAO;EACT,KAAK,OACH,QAAO;EACT,KAAK,YACH,QAAO;EACT,KAAK,SACH,QAAO;EACT,QACE,QAAO;;;;;;;;;AAUb,SAAgB,oBACd,KACA,QACwB;CACxB,MAAM,WAAmC,EAAE;CAC3C,MAAM,kBAAkB,UAAU,mBAAmB,IAAI;AAEzD,MAAK,MAAM,WAAW,iBACpB,MAAK,MAAM,SAAS,IAAI,SAAS,QAAQ,CACvC,KAAI,CAAC,uBAAuB,MAAM,OAAO,gBAAgB,CACvD,UAAS,MAAM,MAAM;AAK3B,QAAO;;;;;;;;;;;AAYT,eAAsB,4BACpB,aACA,aACA,UAAuD,EAAE,EACjC;CACxB,MAAM,EAAE,UAAU,OAAO,aAAa,iBAAiB,OAAO;CAC9D,MAAM,cACJ,OAAO,mBAAmB,YAAY,OAAO,SAAS,eAAe,GACjE,KAAK,IAAI,GAAG,KAAK,MAAM,eAAe,CAAC,GACvC;CAGN,MAAM,CAAC,UAAU,SAAS,MAAM,QAAQ,IAAI,CAC1C,GAAG,QAAQ,YAAY,EACvB,UACK;EAAE,SAAS;EAAe,SAAS,EAAE;EAAE,GAGxC,WAAW,CAChB,CAAC;CAEF,MAAM,aAAa,SAAS,QAAQ,SAAS,KAAK,SAAS,OAAO,CAAC;AACnE,QAAO,MAAM,wBAAwB,WAAW,OAAO;CAEvD,MAAM,SAAS,IAAI,gBAAgB,EAAE,CAAC;CACtC,MAAM,UAAU,IAAI,SAAS;CAG7B,MAAM,cAAc,MAAM,QAAQ,IAChC,WAAW,KAAK,SAAS,GAAG,SAAS,KAAK,KAAK,aAAa,KAAK,EAAE,OAAO,CAAC,CAC5E;CAED,MAAM,YAAY,YAAY,KAAK;CAGnC,MAAM,gBAA+D,EAAE;CACvE,MAAM,kBAMD,EAAE;CACP,MAAM,aAKD,EAAE;AAEP,MAAK,IAAI,IAAI,GAAG,IAAI,WAAW,QAAQ,KAAK;EAC1C,MAAM,OAAO,WAAW;EAExB,MAAM,YAAY,mBADF,KAAK,SAAS,MAAM,OAAO,CACE;EAE7C,MAAM,MAAM,YAAY;EACxB,MAAM,UAAU,QAAQ,IAAI;EAE5B,MAAM,SAAS,MAAM,QAAQ;AAC7B,MAAI,UAAU,OAAO,SAAS,WAAW,CAAC,OAAO,OAAO;AACtD,iBAAc,KAAK;IACjB,OAAO;IACP,QAAQ;KAAE,MAAM;KAAW,MAAM,OAAO;KAAM;IAC/C,CAAC;AACF,cAAW,KAAK;IAAE;IAAW,QAAQ;IAAO,CAAC;SACxC;GACL,MAAM,kBAAkB,mBAAmB,IAAI;GAC/C,MAAM,iBAAiB,sBAAsB,IAAI;GAEjD,MAAM,iBAAiB;IAAE,GADH,oBAAoB,KAAK,gBAAgB;IACpB,GAAG;IAAgB;GAC9D,MAAM,kBAAkB,IAAI,QAC1B,2BACC,UAAU,WAAW,WAAW;AAC/B,QAAI,uBAAuB,QAAQ,gBAAgB,CACjD,QAAO;AAET,WAAO,eAAe,eAAe,WAAW;KAEnD;GAGD,MAAM,YAAY,kBAAkB,KAAK,gBAAgB;AACzD,QAAK,MAAM,SAAS,WAAW;AAC7B,QAAI,uBAAuB,SAAS,MAAM,CAAE;AAC5C,QAAI,eAAe,OAAQ;AAC3B,WAAO,KACL,wFACA,WAAW,IACX,OACA,aAAa,MAAM,SACpB;;GAGH,MAAM,aAAa,gBAAgB,MAAM,CAAC,QAAQ,SAAS,GAAG;AAC9D,mBAAgB,KAAK;IAAE,OAAO;IAAG;IAAW;IAAK;IAAS;IAAY,CAAC;;;CAoB3E,MAAM,eAA8D,EAAE;AAEtE,KAAI,gBAAgB,SAAS,GAAG;EAC9B,IAAI,YAAY;EAChB,MAAM,QAAQ,gBAAgB;AAC9B,UAAQ,MACN,cAAc,MAAM,GAAG,UAAU,IAAI,UAAU,UAAU,MAAM,MAAM,GACtE;EAED,MAAM,cAAc,OAAO,EACzB,OACA,WACA,KACA,SACA,iBAC+D;GAC/D,MAAM,SAAU,MAAM,OAAO,mBAAmB,iBAAiB;IAC/D,WAAW,kBAAkB;IAC7B,cAAc;IACf,CAAC;AAEF;AACA,WAAQ,OACN,cAAc,MAAM,GAAG,UAAU,IAAI,UAAU,UAAU,IAAI,UAAU,GAAG,MAAM,GACjF;AAED,UAAO,MACL,6CACA,WACA,OAAO,OAAO,OACd,OAAO,QAAQ,YAAY,UAAU,EACtC;AAED,OAAI,OAAO,OAAO,UAAU,UAAU;IACpC,MAAM,WACJ,OAAO,OAAO,OAAO,WAAW;AAClC,WAAO,KAAK,8BAA8B,WAAW,SAAS;IAC9D,MAAM,OAAO,2BAA2B,KAAK,UAAU;AACvD,WAAO;KACL,QAAQ;KACR;KACA,QAAQ;MAAE,MAAM;MAAW;MAAM;KACjC,YAAY;MAAE,MAAM;MAAS;MAAM,OAAO;MAAM;KAChD,OAAO,WAAW,SAAS;KAC5B;;GAGH,MAAM,EAAE,MAAM,eAAe,mBAAmB,QAAQ,KAAK,UAAU;AACvE,UAAO;IACL,QAAQ;IACR;IACA,QAAQ;KAAE,MAAM;KAAW;KAAM;IACjC,YAAY;KAAE,MAAM;KAAS;KAAM,OAAO,CAAC;KAAY;IACxD;;EAIH,MAAM,uBACJ,SACA,gBACG;AACH,QAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAAK;IACvC,MAAM,QAAQ,QAAQ;IACtB,MAAM,EAAE,cAAc,gBAAgB,cAAc;AAEpD,QAAI,MAAM,WAAW,aAAa;KAChC,MAAM,MAAM,MAAM;AAClB,kBAAa,KAAK;MAAE,OAAO,IAAI;MAAO,QAAQ,IAAI;MAAQ,CAAC;AAC3D,WAAM,QAAQ,aAAa,IAAI;AAC/B,gBAAW,KAAK;MACd;MACA,QAAQ;MACR,QAAQ,IAAI,WAAW;MACvB,OAAO,IAAI,WAAW,SAAS,IAAI,QAAQ;MAC5C,CAAC;WACG;KACL,MAAM,EAAE,KAAK,SAAS,UAAU,gBAAgB,cAAc;KAC9D,MAAM,SACJ,MAAM,kBAAkB,QACpB,MAAM,OAAO,UACb,OAAO,MAAM,OAAO;AAC1B,YAAO,KAAK,gCAAgC,WAAW,OAAO;KAC9D,MAAM,OAAO,2BAA2B,KAAK,UAAU;AACvD,kBAAa,KAAK;MAAE;MAAO,QAAQ;OAAE,MAAM;OAAW;OAAM;MAAE,CAAC;AAC/D,WAAM,QAAQ,aAAa;MAAE,MAAM;MAAS;MAAM,OAAO;MAAM;AAC/D,gBAAW,KAAK;MACd;MACA,QAAQ;MACR,QAAQ;MACR,OAAO,WAAW,OAAO;MAC1B,CAAC;;;;AAKR,MAAI,gBAAgB,SAAS,YAC3B,MAAK,IAAI,IAAI,GAAG,IAAI,gBAAgB,QAAQ,KAAK,aAAa;GAC5D,MAAM,QAAQ,gBAAgB,MAAM,GAAG,IAAI,YAAY;AAEvD,uBADqB,MAAM,QAAQ,WAAW,MAAM,IAAI,YAAY,CAAC,EACnC,EAAE;AACpC,SAAM,UAAU,MAAM;;OAEnB;AAIL,uBAHgB,MAAM,QAAQ,WAC5B,gBAAgB,IAAI,YAAY,CACjC,EAC4B,EAAE;AAC/B,SAAM,UAAU,MAAM;;AAGxB,UAAQ,KAAK,GAAG;;CAGlB,MAAM,YAAY,YAAY,KAAK,GAAG,aAAa,KAAM,QAAQ,EAAE;AAGnE,KAAI,WAAW,SAAS,GAAG;EACzB,MAAM,aAAa,KAAK,IAAI,GAAG,WAAW,KAAK,MAAM,EAAE,UAAU,OAAO,CAAC;EACzE,MAAM,YAAY,GAAG,IAAI,IAAI,OAAO,GAAG,CAAC;AACxC,UAAQ,IAAI,GAAG;AACf,UAAQ,IACN,KAAK,GAAG,KAAK,kBAAkB,CAAC,GAAG,GAAG,IAAI,IAAI,WAAW,OAAO,GAAG,GACpE;AACD,UAAQ,IAAI,KAAK,YAAY;AAC7B,OAAK,MAAM,SAAS,YAAY;GAC9B,MAAM,MAAM,MAAM,SACd,GAAG,KAAK,GAAG,IAAI,QAAQ,CAAC,GACxB,MAAM,WAAW,QACf,SAAS,GAAG,KAAK,GAAG,MAAM,QAAQ,CAAC,KACnC,SAAS,GAAG,KAAK,GAAG,OAAO,QAAQ,CAAC;GAC1C,MAAM,UAAU,MAAM,UAAU,OAAO,WAAW;GAClD,MAAM,OAAO,MAAM,SAAS,GAAG,IAAI,GAAG,cAAc,QAAQ,CAAC,GAAG;GAChE,MAAM,YAAY,MAAM,OAAO,QAAQ,MAAM,eAAe,GAAG;GAC/D,MAAM,SAAS,YAAY,KAAK,GAAG,IAAI,UAAU,KAAK;AACtD,WAAQ,IAAI,KAAK,IAAI,IAAI,OAAO,SAAS;;EAE3C,MAAM,WAAW,WAAW,QACzB,MAAM,EAAE,WAAW,UAAU,CAAC,EAAE,OAClC,CAAC;EACF,MAAM,aAAa,WAAW,QAC3B,MAAM,EAAE,WAAW,SAAS,CAAC,EAAE,OACjC,CAAC;EACF,MAAM,aAAa,WAAW,QAAQ,MAAM,EAAE,OAAO,CAAC;AACtD,UAAQ,IAAI,KAAK,YAAY;EAC7B,MAAM,QAAQ,CAAC,GAAG,SAAS,OAAO,GAAG,WAAW,aAAa;AAC7D,MAAI,aAAa,EACf,OAAM,KAAK,GAAG,WAAW,GAAG,eAAe,IAAI,UAAU,WAAW;AACtE,UAAQ,IAAI,KAAK,MAAM,KAAK,KAAK,CAAC,IAAI,GAAG,IAAI,GAAG,QAAQ,GAAG,GAAG;AAC9D,UAAQ,IAAI,GAAG;;AAIjB,QAAO,CAAC,GAAG,eAAe,GAAG,aAAa,CACvC,MAAM,GAAG,MAAM,EAAE,QAAQ,EAAE,MAAM,CACjC,KAAK,MAAM,EAAE,OAAO;;;;;;;AAQzB,SAAS,mBAAmB,UAA0B;AACpD,QAAO,SAAS,QAAQ,UAAU,GAAG;;;;;;;;;;;;AAavC,SAAgB,kBAAkB,UAA0B;AAC1D,QAAO,SACJ,QAAQ,WAAW,GAAG,CACtB,QAAQ,SAAS,GAAG,CACpB,MAAM,IAAI,CAAC;;;AAIhB,MAAM,UAAkC;CAEtC,QAAQ;CACR,QAAQ;CAER,SAAS;CAET,SAAS;CACT,UAAU;CACV,KAAK;CACL,QAAQ;CACR,OAAO;CACP,QAAQ;CACR,SAAS;CAET,MAAM;CACN,WAAW;CACX,eAAe;CACf,UAAU;CAEV,OAAO;CACP,KAAK;CACL,QAAQ;CACR,QAAQ;CACR,SAAS;CAET,WAAW;CACX,UAAU;CAEV,MAAM;CACP"}
+{"version":3,"file":"query-registry.js","names":[],"sources":["../../src/type-generator/query-registry.ts"],"sourcesContent":["import fs from \"node:fs/promises\";\nimport path from \"node:path\";\nimport { WorkspaceClient } from \"@databricks/sdk-experimental\";\nimport pc from \"picocolors\";\nimport { createLogger } from \"../logging/logger\";\nimport { CACHE_VERSION, hashSQL, loadCache, saveCache } from \"./cache\";\nimport { Spinner } from \"./spinner\";\nimport {\n  type DatabricksStatementExecutionResponse,\n  type QuerySchema,\n  sqlTypeToHelper,\n  sqlTypeToMarker,\n} from \"./types\";\n\nconst logger = createLogger(\"type-generator:query-registry\");\n\n/**\n * Regex breakdown:\n *   '(?:[^']|'')*'   — matches a SQL string literal, including escaped '' pairs\n *   |                 — alternation: whichever branch matches first at a position wins\n *   --[^\\n]*          — matches a single-line SQL comment\n *\n * Because the regex engine scans left-to-right, a `'` is consumed as a string\n * literal before any `--` inside it could match as a comment — giving us\n * correct single-pass ordering without a manual state machine.\n *\n * V1: no block-comment support (deferred to next PR).\n */\nconst PROTECTED_RANGE_RE = /'(?:[^']|'')*'|--[^\\n]*/g;\n\n/**\n * Numeric-context patterns for positional type inference.\n * Hoisted to module scope — safe because matchAll() clones the regex internally.\n */\nconst NUMERIC_PATTERNS: RegExp[] = [\n  /\\bLIMIT\\s+:([a-zA-Z_]\\w*)/gi,\n  /\\bOFFSET\\s+:([a-zA-Z_]\\w*)/gi,\n  /\\bTOP\\s+:([a-zA-Z_]\\w*)/gi,\n  /\\bFETCH\\s+FIRST\\s+:([a-zA-Z_]\\w*)\\s+ROWS/gi,\n  // V1 limitation: arithmetic operators may false-positive for date\n  // expressions like `:start_date - INTERVAL '1 day'`. A smarter\n  // heuristic (e.g. look-ahead for INTERVAL) is deferred to a future PR.\n  /[+\\-*/]\\s*:([a-zA-Z_]\\w*)/g,\n  /:([a-zA-Z_]\\w*)\\s*[+\\-*/]/g,\n];\n\nexport function getProtectedRanges(sql: string): Array<[number, number]> {\n  const ranges: Array<[number, number]> = [];\n  for (const m of sql.matchAll(PROTECTED_RANGE_RE)) {\n    ranges.push([m.index, m.index + m[0].length]);\n  }\n  return ranges;\n}\n\nfunction isInsideProtectedRange(\n  offset: number,\n  ranges: Array<[number, number]>,\n): boolean {\n  return ranges.some(([start, end]) => offset >= start && offset < end);\n}\n\n/**\n * Parse a raw API/SDK error into a structured code + message.\n * Handles Databricks-style JSON bodies embedded in the message string,\n * e.g. `Response from server (Bad Request) {\"error_code\":\"...\",\"message\":\"...\"}`.\n */\nfunction parseError(raw: string): { code?: string; message: string } {\n  const jsonMatch = raw.match(/\\{[\\s\\S]*\\}/);\n  if (jsonMatch) {\n    try {\n      const parsed = JSON.parse(jsonMatch[0]);\n      if (parsed.error_code || parsed.message) {\n        return {\n          code: parsed.error_code,\n          message: parsed.message || raw,\n        };\n      }\n    } catch {\n      // not valid JSON, fall through\n    }\n  }\n  return { message: raw };\n}\n\n/**\n * Extract parameters from a SQL query\n * @param sql - the SQL query to extract parameters from\n * @returns an array of parameter names\n */\nexport function extractParameters(\n  sql: string,\n  ranges?: Array<[number, number]>,\n): string[] {\n  const protectedRanges = ranges ?? getProtectedRanges(sql);\n  const matches = sql.matchAll(/(?<!:):([a-zA-Z_]\\w*)/g);\n  const params = new Set<string>();\n  for (const match of matches) {\n    if (!isInsideProtectedRange(match.index, protectedRanges)) {\n      params.add(match[1]);\n    }\n  }\n  return Array.from(params);\n}\n\n// parameters that are injected by the server\nexport const SERVER_INJECTED_PARAMS = [\"workspaceId\"];\n\n/**\n * Generates the TypeScript type literal for query parameters from SQL.\n * Shared by both the success and failure paths.\n */\nfunction formatParametersType(sql: string): string {\n  const params = extractParameters(sql).filter(\n    (p) => !SERVER_INJECTED_PARAMS.includes(p),\n  );\n  const paramTypes = extractParameterTypes(sql);\n\n  return params.length > 0\n    ? `{\\n      ${params\n        .map((p) => {\n          const sqlType = paramTypes[p];\n          const markerType = sqlType\n            ? sqlTypeToMarker[sqlType]\n            : \"SQLTypeMarker\";\n          const helper = sqlType ? sqlTypeToHelper[sqlType] : \"sql.*()\";\n          return `/** ${sqlType || \"any\"} - use ${helper} */\\n      ${p}: ${markerType}`;\n        })\n        .join(\";\\n      \")};\\n    }`\n    : \"Record<string, never>\";\n}\n\nexport function convertToQueryType(\n  result: DatabricksStatementExecutionResponse,\n  sql: string,\n  queryName: string,\n): { type: string; hasResults: boolean } {\n  const dataRows = result.result?.data_array || [];\n  const columns = dataRows.map((row) => ({\n    name: row[0] || \"\",\n    type_name: row[1]?.toUpperCase() || \"STRING\",\n    comment: row[2] || undefined,\n  }));\n\n  const paramsType = formatParametersType(sql);\n\n  // generate result fields with JSDoc\n  const resultFields = columns.map((column) => {\n    const normalizedType = normalizeTypeName(column.type_name);\n    const mappedType = typeMap[normalizedType] || \"unknown\";\n    // validate column name is a valid identifier\n    const name = /^[a-zA-Z_$][a-zA-Z0-9_$]*$/.test(column.name)\n      ? column.name\n      : `\"${column.name}\"`;\n\n    // generate comment for column\n    const comment = column.comment\n      ? `/** ${column.comment} */\\n      `\n      : `/** @sqlType ${column.type_name} */\\n      `;\n\n    return `${comment}${name}: ${mappedType}`;\n  });\n\n  const hasResults = resultFields.length > 0;\n\n  const type = `{\n    name: \"${queryName}\";\n    parameters: ${paramsType};\n    result: ${\n      hasResults\n        ? `Array<{\n      ${resultFields.join(\";\\n      \")};\n    }>`\n        : \"unknown\"\n    };\n  }`;\n\n  return { type, hasResults };\n}\n\n/**\n * Used when DESCRIBE QUERY fails so the query still appears in QueryRegistry.\n * Generates a type with unknown result from SQL alone (no warehouse call).\n */\nfunction generateUnknownResultQuery(sql: string, queryName: string): string {\n  const paramsType = formatParametersType(sql);\n\n  return `{\n    name: \"${queryName}\";\n    parameters: ${paramsType};\n    result: unknown;\n  }`;\n}\n\nexport function extractParameterTypes(sql: string): Record<string, string> {\n  const paramTypes: Record<string, string> = {};\n  // Alternation order matters: TIMESTAMP_NTZ must precede TIMESTAMP so the\n  // regex engine doesn't greedy-match TIMESTAMP and leave `_NTZ` unconsumed.\n  const regex =\n    /--\\s*@param\\s+(\\w+)\\s+(STRING|NUMERIC|DECIMAL|BIGINT|TINYINT|SMALLINT|INT|FLOAT|DOUBLE|BOOLEAN|DATE|TIMESTAMP_NTZ|TIMESTAMP|BINARY)\\b/gi;\n  const matches = sql.matchAll(regex);\n  for (const match of matches) {\n    const [, paramName, paramType] = match;\n    paramTypes[paramName] = paramType.toUpperCase();\n  }\n\n  return paramTypes;\n}\n\nexport function defaultForType(sqlType: string | undefined): string {\n  switch (sqlType?.toUpperCase()) {\n    case \"NUMERIC\":\n    case \"DECIMAL\":\n    case \"BIGINT\":\n    case \"TINYINT\":\n    case \"SMALLINT\":\n    case \"INT\":\n      return \"0\";\n    case \"FLOAT\":\n    case \"DOUBLE\":\n      return \"0.0\";\n    case \"STRING\":\n      return \"''\";\n    case \"BOOLEAN\":\n      return \"true\";\n    case \"DATE\":\n      return \"'2000-01-01'\";\n    case \"TIMESTAMP\":\n      return \"'2000-01-01T00:00:00Z'\";\n    case \"TIMESTAMP_NTZ\":\n      return \"'2000-01-01T00:00:00'\";\n    case \"BINARY\":\n      return \"X'00'\";\n    default:\n      return \"''\";\n  }\n}\n\n/**\n * Infer parameter types from positional context in SQL.\n * V1 only infers NUMERIC from patterns like LIMIT, OFFSET, TOP,\n * FETCH FIRST ... ROWS, and arithmetic operators.\n * Parameters inside string literals or SQL comments are ignored.\n */\nexport function inferParameterTypes(\n  sql: string,\n  ranges?: Array<[number, number]>,\n): Record<string, string> {\n  const inferred: Record<string, string> = {};\n  const protectedRanges = ranges ?? getProtectedRanges(sql);\n\n  for (const pattern of NUMERIC_PATTERNS) {\n    for (const match of sql.matchAll(pattern)) {\n      if (!isInsideProtectedRange(match.index, protectedRanges)) {\n        inferred[match[1]] = \"NUMERIC\";\n      }\n    }\n  }\n\n  return inferred;\n}\n\n/**\n * Generate query schemas from a folder of SQL files\n * It uses DESCRIBE QUERY to get the schema without executing the query\n * @param queryFolder - the folder containing the SQL files\n * @param warehouseId - the warehouse id to use for schema analysis\n * @param options - options for the query generation\n * @param options.noCache - if true, skip the cache and regenerate all types\n * @returns an array of query schemas\n */\nexport async function generateQueriesFromDescribe(\n  queryFolder: string,\n  warehouseId: string,\n  options: { noCache?: boolean; concurrency?: number } = {},\n): Promise<QuerySchema[]> {\n  const { noCache = false, concurrency: rawConcurrency = 10 } = options;\n  const concurrency =\n    typeof rawConcurrency === \"number\" && Number.isFinite(rawConcurrency)\n      ? Math.max(1, Math.floor(rawConcurrency))\n      : 10;\n\n  // read all query files and cache in parallel\n  const [allFiles, cache] = await Promise.all([\n    fs.readdir(queryFolder),\n    noCache\n      ? ({ version: CACHE_VERSION, queries: {} } as Awaited<\n          ReturnType<typeof loadCache>\n        >)\n      : loadCache(),\n  ]);\n\n  const queryFiles = allFiles.filter((file) => file.endsWith(\".sql\"));\n  logger.debug(\"Found %d SQL queries\", queryFiles.length);\n\n  const client = new WorkspaceClient({});\n  const spinner = new Spinner();\n\n  // Read all SQL files in parallel\n  const sqlContents = await Promise.all(\n    queryFiles.map((file) => fs.readFile(path.join(queryFolder, file), \"utf8\")),\n  );\n\n  const startTime = performance.now();\n\n  // Phase 1: Check cache, separate cached vs uncached\n  const cachedResults: Array<{ index: number; schema: QuerySchema }> = [];\n  const uncachedQueries: Array<{\n    index: number;\n    queryName: string;\n    sql: string;\n    sqlHash: string;\n    cleanedSql: string;\n  }> = [];\n  const logEntries: Array<{\n    queryName: string;\n    status: \"HIT\" | \"MISS\";\n    failed?: boolean;\n    error?: { code?: string; message: string };\n  }> = [];\n\n  for (let i = 0; i < queryFiles.length; i++) {\n    const file = queryFiles[i];\n    const rawName = path.basename(file, \".sql\");\n    const queryName = normalizeQueryName(rawName);\n\n    const sql = sqlContents[i];\n    const sqlHash = hashSQL(sql);\n\n    const cached = cache.queries[queryName];\n    if (cached && cached.hash === sqlHash && !cached.retry) {\n      cachedResults.push({\n        index: i,\n        schema: { name: queryName, type: cached.type },\n      });\n      logEntries.push({ queryName, status: \"HIT\" });\n    } else {\n      const protectedRanges = getProtectedRanges(sql);\n      const annotatedTypes = extractParameterTypes(sql);\n      const inferredTypes = inferParameterTypes(sql, protectedRanges);\n      const parameterTypes = { ...inferredTypes, ...annotatedTypes };\n      const sqlWithDefaults = sql.replace(\n        /(?<!:):([a-zA-Z_]\\w*)/g,\n        (original, paramName, offset) => {\n          if (isInsideProtectedRange(offset, protectedRanges)) {\n            return original;\n          }\n          return defaultForType(parameterTypes[paramName]);\n        },\n      );\n\n      // Warn about unresolved parameters\n      const allParams = extractParameters(sql, protectedRanges);\n      for (const param of allParams) {\n        if (SERVER_INJECTED_PARAMS.includes(param)) continue;\n        if (parameterTypes[param]) continue;\n        logger.warn(\n          '%s: parameter \":%s\" has no type annotation or inference. Add %s to the query file.',\n          queryFiles[i],\n          param,\n          `-- @param ${param} <TYPE>`,\n        );\n      }\n\n      const cleanedSql = sqlWithDefaults.trim().replace(/;\\s*$/, \"\");\n      uncachedQueries.push({ index: i, queryName, sql, sqlHash, cleanedSql });\n    }\n  }\n\n  // Phase 2: Execute all uncached DESCRIBE calls in parallel\n  type DescribeResult =\n    | {\n        status: \"ok\";\n        index: number;\n        schema: QuerySchema;\n        cacheEntry: { hash: string; type: string; retry: boolean };\n      }\n    | {\n        status: \"fail\";\n        index: number;\n        schema: QuerySchema;\n        cacheEntry: { hash: string; type: string; retry: boolean };\n        error: { code?: string; message: string };\n      };\n\n  const freshResults: Array<{ index: number; schema: QuerySchema }> = [];\n\n  if (uncachedQueries.length > 0) {\n    let completed = 0;\n    const total = uncachedQueries.length;\n    spinner.start(\n      `Describing ${total} ${total === 1 ? \"query\" : \"queries\"} (0/${total})`,\n    );\n\n    const describeOne = async ({\n      index,\n      queryName,\n      sql,\n      sqlHash,\n      cleanedSql,\n    }: (typeof uncachedQueries)[number]): Promise<DescribeResult> => {\n      const result = (await client.statementExecution.executeStatement({\n        statement: `DESCRIBE QUERY ${cleanedSql}`,\n        warehouse_id: warehouseId,\n      })) as DatabricksStatementExecutionResponse;\n\n      completed++;\n      spinner.update(\n        `Describing ${total} ${total === 1 ? \"query\" : \"queries\"} (${completed}/${total})`,\n      );\n\n      logger.debug(\n        \"DESCRIBE result for %s: state=%s, rows=%d\",\n        queryName,\n        result.status.state,\n        result.result?.data_array?.length ?? 0,\n      );\n\n      if (result.status.state === \"FAILED\") {\n        const sqlError =\n          result.status.error?.message || \"Query execution failed\";\n        logger.warn(\"DESCRIBE failed for %s: %s\", queryName, sqlError);\n        const type = generateUnknownResultQuery(sql, queryName);\n        return {\n          status: \"fail\",\n          index,\n          schema: { name: queryName, type },\n          cacheEntry: { hash: sqlHash, type, retry: true },\n          error: parseError(sqlError),\n        };\n      }\n\n      const { type, hasResults } = convertToQueryType(result, sql, queryName);\n      return {\n        status: \"ok\",\n        index,\n        schema: { name: queryName, type },\n        cacheEntry: { hash: sqlHash, type, retry: !hasResults },\n      };\n    };\n\n    // Process in chunks, saving cache after each chunk\n    const processBatchResults = (\n      settled: PromiseSettledResult<DescribeResult>[],\n      batchOffset: number,\n    ) => {\n      for (let i = 0; i < settled.length; i++) {\n        const entry = settled[i];\n        const { queryName } = uncachedQueries[batchOffset + i];\n\n        if (entry.status === \"fulfilled\") {\n          const res = entry.value;\n          freshResults.push({ index: res.index, schema: res.schema });\n          cache.queries[queryName] = res.cacheEntry;\n          logEntries.push({\n            queryName,\n            status: \"MISS\",\n            failed: res.status === \"fail\",\n            error: res.status === \"fail\" ? res.error : undefined,\n          });\n        } else {\n          const { sql, sqlHash, index } = uncachedQueries[batchOffset + i];\n          const reason =\n            entry.reason instanceof Error\n              ? entry.reason.message\n              : String(entry.reason);\n          logger.warn(\"DESCRIBE rejected for %s: %s\", queryName, reason);\n          const type = generateUnknownResultQuery(sql, queryName);\n          freshResults.push({ index, schema: { name: queryName, type } });\n          cache.queries[queryName] = { hash: sqlHash, type, retry: true };\n          logEntries.push({\n            queryName,\n            status: \"MISS\",\n            failed: true,\n            error: parseError(reason),\n          });\n        }\n      }\n    };\n\n    if (uncachedQueries.length > concurrency) {\n      for (let b = 0; b < uncachedQueries.length; b += concurrency) {\n        const batch = uncachedQueries.slice(b, b + concurrency);\n        const batchResults = await Promise.allSettled(batch.map(describeOne));\n        processBatchResults(batchResults, b);\n        await saveCache(cache);\n      }\n    } else {\n      const settled = await Promise.allSettled(\n        uncachedQueries.map(describeOne),\n      );\n      processBatchResults(settled, 0);\n      await saveCache(cache);\n    }\n\n    spinner.stop(\"\");\n  }\n\n  const elapsed = ((performance.now() - startTime) / 1000).toFixed(2);\n\n  // Print formatted table\n  if (logEntries.length > 0) {\n    const maxNameLen = Math.max(...logEntries.map((e) => e.queryName.length));\n    const separator = pc.dim(\"─\".repeat(50));\n    console.log(\"\");\n    console.log(\n      `  ${pc.bold(\"Typegen Queries\")} ${pc.dim(`(${logEntries.length})`)}`,\n    );\n    console.log(`  ${separator}`);\n    for (const entry of logEntries) {\n      const tag = entry.failed\n        ? pc.bold(pc.red(\"ERROR\"))\n        : entry.status === \"HIT\"\n          ? `cache ${pc.bold(pc.green(\"HIT  \"))}`\n          : `cache ${pc.bold(pc.yellow(\"MISS \"))}`;\n      const rawName = entry.queryName.padEnd(maxNameLen);\n      const name = entry.failed ? pc.dim(pc.strikethrough(rawName)) : rawName;\n      const errorCode = entry.error?.message.match(/\\[([^\\]]+)\\]/)?.[1];\n      const reason = errorCode ? `  ${pc.dim(errorCode)}` : \"\";\n      console.log(`  ${tag}  ${name}${reason}`);\n    }\n    const newCount = logEntries.filter(\n      (e) => e.status === \"MISS\" && !e.failed,\n    ).length;\n    const cacheCount = logEntries.filter(\n      (e) => e.status === \"HIT\" && !e.failed,\n    ).length;\n    const errorCount = logEntries.filter((e) => e.failed).length;\n    console.log(`  ${separator}`);\n    const parts = [`${newCount} new`, `${cacheCount} from cache`];\n    if (errorCount > 0)\n      parts.push(`${errorCount} ${errorCount === 1 ? \"error\" : \"errors\"}`);\n    console.log(`  ${parts.join(\", \")}. ${pc.dim(`${elapsed}s`)}`);\n    console.log(\"\");\n  }\n\n  // Merge and sort by original file index for deterministic output\n  return [...cachedResults, ...freshResults]\n    .sort((a, b) => a.index - b.index)\n    .map((r) => r.schema);\n}\n\n/**\n * Normalize query name by removing the .obo extension\n * @param queryName - the query name to normalize\n * @returns the normalized query name\n */\nfunction normalizeQueryName(fileName: string): string {\n  return fileName.replace(/\\.obo$/, \"\");\n}\n\n/**\n * Normalize SQL type name by removing parameters/generics\n * Examples:\n *   DECIMAL(38,6) -> DECIMAL\n *   ARRAY<STRING> -> ARRAY\n *   MAP<STRING,INT> -> MAP\n *   STRUCT<name:STRING> -> STRUCT\n *   INTERVAL DAY TO SECOND -> INTERVAL\n *   GEOGRAPHY(4326) -> GEOGRAPHY\n */\nexport function normalizeTypeName(typeName: string): string {\n  return typeName\n    .replace(/\\(.*\\)$/, \"\") // remove (p, s) eg: DECIMAL(38,6) -> DECIMAL\n    .replace(/<.*>$/, \"\") // remove <T> eg: ARRAY<STRING> -> ARRAY\n    .split(\" \")[0]; // take first word eg: INTERVAL DAY TO SECOND -> INTERVAL\n}\n\n/** Type Map for Databricks data types to JavaScript types */\nconst typeMap: Record<string, string> = {\n  // string types\n  STRING: \"string\",\n  BINARY: \"string\",\n  // boolean\n  BOOLEAN: \"boolean\",\n  // numeric types\n  TINYINT: \"number\",\n  SMALLINT: \"number\",\n  INT: \"number\",\n  BIGINT: \"number\",\n  FLOAT: \"number\",\n  DOUBLE: \"number\",\n  DECIMAL: \"number\",\n  // date/time types\n  DATE: \"string\",\n  TIMESTAMP: \"string\",\n  TIMESTAMP_NTZ: \"string\",\n  INTERVAL: \"string\",\n  // complex types\n  ARRAY: \"unknown[]\",\n  MAP: \"Record<string, unknown>\",\n  STRUCT: \"Record<string, unknown>\",\n  OBJECT: \"Record<string, unknown>\",\n  VARIANT: \"unknown\",\n  // spatial types\n  GEOGRAPHY: \"unknown\",\n  GEOMETRY: \"unknown\",\n  // null type\n  VOID: \"null\",\n};\n"],"mappings":";;;;;;;;;;AAcA,MAAM,SAAS,aAAa,gCAAgC;;;;;;;;;;;;;AAc5D,MAAM,qBAAqB;;;;;AAM3B,MAAM,mBAA6B;CACjC;CACA;CACA;CACA;CAIA;CACA;CACD;AAED,SAAgB,mBAAmB,KAAsC;CACvE,MAAM,SAAkC,EAAE;AAC1C,MAAK,MAAM,KAAK,IAAI,SAAS,mBAAmB,CAC9C,QAAO,KAAK,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;AAE/C,QAAO;;AAGT,SAAS,uBACP,QACA,QACS;AACT,QAAO,OAAO,MAAM,CAAC,OAAO,SAAS,UAAU,SAAS,SAAS,IAAI;;;;;;;AAQvE,SAAS,WAAW,KAAiD;CACnE,MAAM,YAAY,IAAI,MAAM,cAAc;AAC1C,KAAI,UACF,KAAI;EACF,MAAM,SAAS,KAAK,MAAM,UAAU,GAAG;AACvC,MAAI,OAAO,cAAc,OAAO,QAC9B,QAAO;GACL,MAAM,OAAO;GACb,SAAS,OAAO,WAAW;GAC5B;SAEG;AAIV,QAAO,EAAE,SAAS,KAAK;;;;;;;AAQzB,SAAgB,kBACd,KACA,QACU;CACV,MAAM,kBAAkB,UAAU,mBAAmB,IAAI;CACzD,MAAM,UAAU,IAAI,SAAS,yBAAyB;CACtD,MAAM,yBAAS,IAAI,KAAa;AAChC,MAAK,MAAM,SAAS,QAClB,KAAI,CAAC,uBAAuB,MAAM,OAAO,gBAAgB,CACvD,QAAO,IAAI,MAAM,GAAG;AAGxB,QAAO,MAAM,KAAK,OAAO;;AAI3B,MAAa,yBAAyB,CAAC,cAAc;;;;;AAMrD,SAAS,qBAAqB,KAAqB;CACjD,MAAM,SAAS,kBAAkB,IAAI,CAAC,QACnC,MAAM,CAAC,uBAAuB,SAAS,EAAE,CAC3C;CACD,MAAM,aAAa,sBAAsB,IAAI;AAE7C,QAAO,OAAO,SAAS,IACnB,YAAY,OACT,KAAK,MAAM;EACV,MAAM,UAAU,WAAW;EAC3B,MAAM,aAAa,UACf,gBAAgB,WAChB;EACJ,MAAM,SAAS,UAAU,gBAAgB,WAAW;AACpD,SAAO,OAAO,WAAW,MAAM,SAAS,OAAO,aAAa,EAAE,IAAI;GAClE,CACD,KAAK,YAAY,CAAC,YACrB;;AAGN,SAAgB,mBACd,QACA,KACA,WACuC;CAEvC,MAAM,WADW,OAAO,QAAQ,cAAc,EAAE,EACvB,KAAK,SAAS;EACrC,MAAM,IAAI,MAAM;EAChB,WAAW,IAAI,IAAI,aAAa,IAAI;EACpC,SAAS,IAAI,MAAM;EACpB,EAAE;CAEH,MAAM,aAAa,qBAAqB,IAAI;CAG5C,MAAM,eAAe,QAAQ,KAAK,WAAW;EAE3C,MAAM,aAAa,QADI,kBAAkB,OAAO,UAAU,KACZ;EAE9C,MAAM,OAAO,6BAA6B,KAAK,OAAO,KAAK,GACvD,OAAO,OACP,IAAI,OAAO,KAAK;AAOpB,SAAO,GAJS,OAAO,UACnB,OAAO,OAAO,QAAQ,eACtB,gBAAgB,OAAO,UAAU,eAEjB,KAAK,IAAI;GAC7B;CAEF,MAAM,aAAa,aAAa,SAAS;AAczC,QAAO;EAAE,MAZI;aACF,UAAU;kBACL,WAAW;cAEvB,aACI;QACF,aAAa,KAAK,YAAY,CAAC;UAE7B,UACL;;EAGY;EAAY;;;;;;AAO7B,SAAS,2BAA2B,KAAa,WAA2B;AAG1E,QAAO;aACI,UAAU;kBAHF,qBAAqB,IAAI,CAIjB;;;;AAK7B,SAAgB,sBAAsB,KAAqC;CACzE,MAAM,aAAqC,EAAE;CAK7C,MAAM,UAAU,IAAI,SADlB,0IACiC;AACnC,MAAK,MAAM,SAAS,SAAS;EAC3B,MAAM,GAAG,WAAW,aAAa;AACjC,aAAW,aAAa,UAAU,aAAa;;AAGjD,QAAO;;AAGT,SAAgB,eAAe,SAAqC;AAClE,SAAQ,SAAS,aAAa,EAA9B;EACE,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK,MACH,QAAO;EACT,KAAK;EACL,KAAK,SACH,QAAO;EACT,KAAK,SACH,QAAO;EACT,KAAK,UACH,QAAO;EACT,KAAK,OACH,QAAO;EACT,KAAK,YACH,QAAO;EACT,KAAK,gBACH,QAAO;EACT,KAAK,SACH,QAAO;EACT,QACE,QAAO;;;;;;;;;AAUb,SAAgB,oBACd,KACA,QACwB;CACxB,MAAM,WAAmC,EAAE;CAC3C,MAAM,kBAAkB,UAAU,mBAAmB,IAAI;AAEzD,MAAK,MAAM,WAAW,iBACpB,MAAK,MAAM,SAAS,IAAI,SAAS,QAAQ,CACvC,KAAI,CAAC,uBAAuB,MAAM,OAAO,gBAAgB,CACvD,UAAS,MAAM,MAAM;AAK3B,QAAO;;;;;;;;;;;AAYT,eAAsB,4BACpB,aACA,aACA,UAAuD,EAAE,EACjC;CACxB,MAAM,EAAE,UAAU,OAAO,aAAa,iBAAiB,OAAO;CAC9D,MAAM,cACJ,OAAO,mBAAmB,YAAY,OAAO,SAAS,eAAe,GACjE,KAAK,IAAI,GAAG,KAAK,MAAM,eAAe,CAAC,GACvC;CAGN,MAAM,CAAC,UAAU,SAAS,MAAM,QAAQ,IAAI,CAC1C,GAAG,QAAQ,YAAY,EACvB,UACK;EAAE,SAAS;EAAe,SAAS,EAAE;EAAE,GAGxC,WAAW,CAChB,CAAC;CAEF,MAAM,aAAa,SAAS,QAAQ,SAAS,KAAK,SAAS,OAAO,CAAC;AACnE,QAAO,MAAM,wBAAwB,WAAW,OAAO;CAEvD,MAAM,SAAS,IAAI,gBAAgB,EAAE,CAAC;CACtC,MAAM,UAAU,IAAI,SAAS;CAG7B,MAAM,cAAc,MAAM,QAAQ,IAChC,WAAW,KAAK,SAAS,GAAG,SAAS,KAAK,KAAK,aAAa,KAAK,EAAE,OAAO,CAAC,CAC5E;CAED,MAAM,YAAY,YAAY,KAAK;CAGnC,MAAM,gBAA+D,EAAE;CACvE,MAAM,kBAMD,EAAE;CACP,MAAM,aAKD,EAAE;AAEP,MAAK,IAAI,IAAI,GAAG,IAAI,WAAW,QAAQ,KAAK;EAC1C,MAAM,OAAO,WAAW;EAExB,MAAM,YAAY,mBADF,KAAK,SAAS,MAAM,OAAO,CACE;EAE7C,MAAM,MAAM,YAAY;EACxB,MAAM,UAAU,QAAQ,IAAI;EAE5B,MAAM,SAAS,MAAM,QAAQ;AAC7B,MAAI,UAAU,OAAO,SAAS,WAAW,CAAC,OAAO,OAAO;AACtD,iBAAc,KAAK;IACjB,OAAO;IACP,QAAQ;KAAE,MAAM;KAAW,MAAM,OAAO;KAAM;IAC/C,CAAC;AACF,cAAW,KAAK;IAAE;IAAW,QAAQ;IAAO,CAAC;SACxC;GACL,MAAM,kBAAkB,mBAAmB,IAAI;GAC/C,MAAM,iBAAiB,sBAAsB,IAAI;GAEjD,MAAM,iBAAiB;IAAE,GADH,oBAAoB,KAAK,gBAAgB;IACpB,GAAG;IAAgB;GAC9D,MAAM,kBAAkB,IAAI,QAC1B,2BACC,UAAU,WAAW,WAAW;AAC/B,QAAI,uBAAuB,QAAQ,gBAAgB,CACjD,QAAO;AAET,WAAO,eAAe,eAAe,WAAW;KAEnD;GAGD,MAAM,YAAY,kBAAkB,KAAK,gBAAgB;AACzD,QAAK,MAAM,SAAS,WAAW;AAC7B,QAAI,uBAAuB,SAAS,MAAM,CAAE;AAC5C,QAAI,eAAe,OAAQ;AAC3B,WAAO,KACL,wFACA,WAAW,IACX,OACA,aAAa,MAAM,SACpB;;GAGH,MAAM,aAAa,gBAAgB,MAAM,CAAC,QAAQ,SAAS,GAAG;AAC9D,mBAAgB,KAAK;IAAE,OAAO;IAAG;IAAW;IAAK;IAAS;IAAY,CAAC;;;CAoB3E,MAAM,eAA8D,EAAE;AAEtE,KAAI,gBAAgB,SAAS,GAAG;EAC9B,IAAI,YAAY;EAChB,MAAM,QAAQ,gBAAgB;AAC9B,UAAQ,MACN,cAAc,MAAM,GAAG,UAAU,IAAI,UAAU,UAAU,MAAM,MAAM,GACtE;EAED,MAAM,cAAc,OAAO,EACzB,OACA,WACA,KACA,SACA,iBAC+D;GAC/D,MAAM,SAAU,MAAM,OAAO,mBAAmB,iBAAiB;IAC/D,WAAW,kBAAkB;IAC7B,cAAc;IACf,CAAC;AAEF;AACA,WAAQ,OACN,cAAc,MAAM,GAAG,UAAU,IAAI,UAAU,UAAU,IAAI,UAAU,GAAG,MAAM,GACjF;AAED,UAAO,MACL,6CACA,WACA,OAAO,OAAO,OACd,OAAO,QAAQ,YAAY,UAAU,EACtC;AAED,OAAI,OAAO,OAAO,UAAU,UAAU;IACpC,MAAM,WACJ,OAAO,OAAO,OAAO,WAAW;AAClC,WAAO,KAAK,8BAA8B,WAAW,SAAS;IAC9D,MAAM,OAAO,2BAA2B,KAAK,UAAU;AACvD,WAAO;KACL,QAAQ;KACR;KACA,QAAQ;MAAE,MAAM;MAAW;MAAM;KACjC,YAAY;MAAE,MAAM;MAAS;MAAM,OAAO;MAAM;KAChD,OAAO,WAAW,SAAS;KAC5B;;GAGH,MAAM,EAAE,MAAM,eAAe,mBAAmB,QAAQ,KAAK,UAAU;AACvE,UAAO;IACL,QAAQ;IACR;IACA,QAAQ;KAAE,MAAM;KAAW;KAAM;IACjC,YAAY;KAAE,MAAM;KAAS;KAAM,OAAO,CAAC;KAAY;IACxD;;EAIH,MAAM,uBACJ,SACA,gBACG;AACH,QAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,KAAK;IACvC,MAAM,QAAQ,QAAQ;IACtB,MAAM,EAAE,cAAc,gBAAgB,cAAc;AAEpD,QAAI,MAAM,WAAW,aAAa;KAChC,MAAM,MAAM,MAAM;AAClB,kBAAa,KAAK;MAAE,OAAO,IAAI;MAAO,QAAQ,IAAI;MAAQ,CAAC;AAC3D,WAAM,QAAQ,aAAa,IAAI;AAC/B,gBAAW,KAAK;MACd;MACA,QAAQ;MACR,QAAQ,IAAI,WAAW;MACvB,OAAO,IAAI,WAAW,SAAS,IAAI,QAAQ;MAC5C,CAAC;WACG;KACL,MAAM,EAAE,KAAK,SAAS,UAAU,gBAAgB,cAAc;KAC9D,MAAM,SACJ,MAAM,kBAAkB,QACpB,MAAM,OAAO,UACb,OAAO,MAAM,OAAO;AAC1B,YAAO,KAAK,gCAAgC,WAAW,OAAO;KAC9D,MAAM,OAAO,2BAA2B,KAAK,UAAU;AACvD,kBAAa,KAAK;MAAE;MAAO,QAAQ;OAAE,MAAM;OAAW;OAAM;MAAE,CAAC;AAC/D,WAAM,QAAQ,aAAa;MAAE,MAAM;MAAS;MAAM,OAAO;MAAM;AAC/D,gBAAW,KAAK;MACd;MACA,QAAQ;MACR,QAAQ;MACR,OAAO,WAAW,OAAO;MAC1B,CAAC;;;;AAKR,MAAI,gBAAgB,SAAS,YAC3B,MAAK,IAAI,IAAI,GAAG,IAAI,gBAAgB,QAAQ,KAAK,aAAa;GAC5D,MAAM,QAAQ,gBAAgB,MAAM,GAAG,IAAI,YAAY;AAEvD,uBADqB,MAAM,QAAQ,WAAW,MAAM,IAAI,YAAY,CAAC,EACnC,EAAE;AACpC,SAAM,UAAU,MAAM;;OAEnB;AAIL,uBAHgB,MAAM,QAAQ,WAC5B,gBAAgB,IAAI,YAAY,CACjC,EAC4B,EAAE;AAC/B,SAAM,UAAU,MAAM;;AAGxB,UAAQ,KAAK,GAAG;;CAGlB,MAAM,YAAY,YAAY,KAAK,GAAG,aAAa,KAAM,QAAQ,EAAE;AAGnE,KAAI,WAAW,SAAS,GAAG;EACzB,MAAM,aAAa,KAAK,IAAI,GAAG,WAAW,KAAK,MAAM,EAAE,UAAU,OAAO,CAAC;EACzE,MAAM,YAAY,GAAG,IAAI,IAAI,OAAO,GAAG,CAAC;AACxC,UAAQ,IAAI,GAAG;AACf,UAAQ,IACN,KAAK,GAAG,KAAK,kBAAkB,CAAC,GAAG,GAAG,IAAI,IAAI,WAAW,OAAO,GAAG,GACpE;AACD,UAAQ,IAAI,KAAK,YAAY;AAC7B,OAAK,MAAM,SAAS,YAAY;GAC9B,MAAM,MAAM,MAAM,SACd,GAAG,KAAK,GAAG,IAAI,QAAQ,CAAC,GACxB,MAAM,WAAW,QACf,SAAS,GAAG,KAAK,GAAG,MAAM,QAAQ,CAAC,KACnC,SAAS,GAAG,KAAK,GAAG,OAAO,QAAQ,CAAC;GAC1C,MAAM,UAAU,MAAM,UAAU,OAAO,WAAW;GAClD,MAAM,OAAO,MAAM,SAAS,GAAG,IAAI,GAAG,cAAc,QAAQ,CAAC,GAAG;GAChE,MAAM,YAAY,MAAM,OAAO,QAAQ,MAAM,eAAe,GAAG;GAC/D,MAAM,SAAS,YAAY,KAAK,GAAG,IAAI,UAAU,KAAK;AACtD,WAAQ,IAAI,KAAK,IAAI,IAAI,OAAO,SAAS;;EAE3C,MAAM,WAAW,WAAW,QACzB,MAAM,EAAE,WAAW,UAAU,CAAC,EAAE,OAClC,CAAC;EACF,MAAM,aAAa,WAAW,QAC3B,MAAM,EAAE,WAAW,SAAS,CAAC,EAAE,OACjC,CAAC;EACF,MAAM,aAAa,WAAW,QAAQ,MAAM,EAAE,OAAO,CAAC;AACtD,UAAQ,IAAI,KAAK,YAAY;EAC7B,MAAM,QAAQ,CAAC,GAAG,SAAS,OAAO,GAAG,WAAW,aAAa;AAC7D,MAAI,aAAa,EACf,OAAM,KAAK,GAAG,WAAW,GAAG,eAAe,IAAI,UAAU,WAAW;AACtE,UAAQ,IAAI,KAAK,MAAM,KAAK,KAAK,CAAC,IAAI,GAAG,IAAI,GAAG,QAAQ,GAAG,GAAG;AAC9D,UAAQ,IAAI,GAAG;;AAIjB,QAAO,CAAC,GAAG,eAAe,GAAG,aAAa,CACvC,MAAM,GAAG,MAAM,EAAE,QAAQ,EAAE,MAAM,CACjC,KAAK,MAAM,EAAE,OAAO;;;;;;;AAQzB,SAAS,mBAAmB,UAA0B;AACpD,QAAO,SAAS,QAAQ,UAAU,GAAG;;;;;;;;;;;;AAavC,SAAgB,kBAAkB,UAA0B;AAC1D,QAAO,SACJ,QAAQ,WAAW,GAAG,CACtB,QAAQ,SAAS,GAAG,CACpB,MAAM,IAAI,CAAC;;;AAIhB,MAAM,UAAkC;CAEtC,QAAQ;CACR,QAAQ;CAER,SAAS;CAET,SAAS;CACT,UAAU;CACV,KAAK;CACL,QAAQ;CACR,OAAO;CACP,QAAQ;CACR,SAAS;CAET,MAAM;CACN,WAAW;CACX,eAAe;CACf,UAAU;CAEV,OAAO;CACP,KAAK;CACL,QAAQ;CACR,QAAQ;CACR,SAAS;CAET,WAAW;CACX,UAAU;CAEV,MAAM;CACP"}

package/dist/type-generator/types.js

@@ -27,14 +27,14 @@ const sqlTypeToHelper = {
 	STRING: "sql.string()",
 	BINARY: "sql.binary()",
 	BOOLEAN: "sql.boolean()",
-	NUMERIC: "sql.number()",
-	INT: "sql.number()",
-	BIGINT: "sql.number()",
-	TINYINT: "sql.number()",
-	SMALLINT: "sql.number()",
-	FLOAT: "sql.number()",
-	DOUBLE: "sql.number()",
-	DECIMAL: "sql.number()",
+	NUMERIC: "sql.numeric()",
+	DECIMAL: "sql.numeric()",
+	BIGINT: "sql.bigint()",
+	INT: "sql.int()",
+	TINYINT: "sql.int()",
+	SMALLINT: "sql.int()",
+	FLOAT: "sql.float()",
+	DOUBLE: "sql.double()",
 	DATE: "sql.date()",
 	TIMESTAMP: "sql.timestamp()",
 	TIMESTAMP_NTZ: "sql.timestamp()"

package/dist/type-generator/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","names":[],"sources":["../../src/type-generator/types.ts"],"sourcesContent":["/**\n * Databricks statement execution response interface for DESCRIBE QUERY\n * @property statement_id - the id of the statement\n * @property status - the status of the statement\n * @property result - the result containing column schema as rows [col_name, data_type, comment]\n */\nexport interface DatabricksStatementExecutionResponse {\n  statement_id: string;\n  status: {\n    state: string;\n    error?: { error_code?: string; message?: string };\n  };\n  result?: {\n    data_array?: (string | null)[][];\n  };\n}\n\n/**\n * Map of SQL types to their corresponding marker types\n * Used to convert SQL types to their corresponding marker types\n */\nexport const sqlTypeToMarker: Record<string, string> = {\n  // string\n  STRING: \"SQLStringMarker\",\n  BINARY: \"SQLBinaryMarker\",\n  // boolean\n  BOOLEAN: \"SQLBooleanMarker\",\n  // numeric\n  NUMERIC: \"SQLNumberMarker\",\n  INT: \"SQLNumberMarker\",\n  BIGINT: \"SQLNumberMarker\",\n  TINYINT: \"SQLNumberMarker\",\n  SMALLINT: \"SQLNumberMarker\",\n  FLOAT: \"SQLNumberMarker\",\n  DOUBLE: \"SQLNumberMarker\",\n  DECIMAL: \"SQLNumberMarker\",\n  // date/time\n  DATE: \"SQLDateMarker\",\n  TIMESTAMP: \"SQLTimestampMarker\",\n  TIMESTAMP_NTZ: \"SQLTimestampMarker\",\n};\n\n/**\n * Map of SQL types to their corresponding helper function names\n * Used to generate JSDoc hints for parameters\n */\nexport const sqlTypeToHelper: Record<string, string> = {\n  // string\n  STRING: \"sql.string()\",\n  BINARY: \"sql.binary()\",\n  // boolean\n  BOOLEAN: \"sql.boolean()\",\n  // numeric\n  NUMERIC: \"sql.number()\",\n  INT: \"sql.number()\",\n  BIGINT: \"sql.number()\",\n  TINYINT: \"sql.number()\",\n  SMALLINT: \"sql.number()\",\n  FLOAT: \"sql.number()\",\n  DOUBLE: \"sql.number()\",\n  DECIMAL: \"sql.number()\",\n  // date/time\n  DATE: \"sql.date()\",\n  TIMESTAMP: \"sql.timestamp()\",\n  TIMESTAMP_NTZ: \"sql.timestamp()\",\n};\n\n/**\n * Query schema interface\n * @property name - the name of the query\n * @property type - the type of the query (string, number, boolean, object, array, etc.)\n */\nexport interface QuerySchema {\n  name: string;\n  type: string;\n}\n"],"mappings":";;;;;AAqBA,MAAa,kBAA0C;CAErD,QAAQ;CACR,QAAQ;CAER,SAAS;CAET,SAAS;CACT,KAAK;CACL,QAAQ;CACR,SAAS;CACT,UAAU;CACV,OAAO;CACP,QAAQ;CACR,SAAS;CAET,MAAM;CACN,WAAW;CACX,eAAe;CAChB;;;;;AAMD,MAAa,kBAA0C;CAErD,QAAQ;CACR,QAAQ;CAER,SAAS;CAET,SAAS;CACT,KAAK;CACL,QAAQ;CACR,SAAS;CACT,UAAU;CACV,OAAO;CACP,QAAQ;CACR,SAAS;CAET,MAAM;CACN,WAAW;CACX,eAAe;CAChB"}
+{"version":3,"file":"types.js","names":[],"sources":["../../src/type-generator/types.ts"],"sourcesContent":["/**\n * Databricks statement execution response interface for DESCRIBE QUERY\n * @property statement_id - the id of the statement\n * @property status - the status of the statement\n * @property result - the result containing column schema as rows [col_name, data_type, comment]\n */\nexport interface DatabricksStatementExecutionResponse {\n  statement_id: string;\n  status: {\n    state: string;\n    error?: { error_code?: string; message?: string };\n  };\n  result?: {\n    data_array?: (string | null)[][];\n  };\n}\n\n/**\n * Map of SQL types to their corresponding marker types\n * Used to convert SQL types to their corresponding marker types\n */\nexport const sqlTypeToMarker: Record<string, string> = {\n  // string\n  STRING: \"SQLStringMarker\",\n  BINARY: \"SQLBinaryMarker\",\n  // boolean\n  BOOLEAN: \"SQLBooleanMarker\",\n  // numeric\n  NUMERIC: \"SQLNumberMarker\",\n  INT: \"SQLNumberMarker\",\n  BIGINT: \"SQLNumberMarker\",\n  TINYINT: \"SQLNumberMarker\",\n  SMALLINT: \"SQLNumberMarker\",\n  FLOAT: \"SQLNumberMarker\",\n  DOUBLE: \"SQLNumberMarker\",\n  DECIMAL: \"SQLNumberMarker\",\n  // date/time\n  DATE: \"SQLDateMarker\",\n  TIMESTAMP: \"SQLTimestampMarker\",\n  TIMESTAMP_NTZ: \"SQLTimestampMarker\",\n};\n\n/**\n * Map of SQL types to their corresponding helper function names\n * Used to generate JSDoc hints for parameters\n */\nexport const sqlTypeToHelper: Record<string, string> = {\n  // string\n  STRING: \"sql.string()\",\n  BINARY: \"sql.binary()\",\n  // boolean\n  BOOLEAN: \"sql.boolean()\",\n  // numeric — route each SQL type to its closest typed helper. INT/BIGINT\n  // are critical for LIMIT/OFFSET; FLOAT/DOUBLE preserve precision intent;\n  // NUMERIC/DECIMAL route to sql.numeric() for exact-decimal columns.\n  NUMERIC: \"sql.numeric()\",\n  DECIMAL: \"sql.numeric()\",\n  BIGINT: \"sql.bigint()\",\n  INT: \"sql.int()\",\n  TINYINT: \"sql.int()\",\n  SMALLINT: \"sql.int()\",\n  FLOAT: \"sql.float()\",\n  DOUBLE: \"sql.double()\",\n  // date/time\n  DATE: \"sql.date()\",\n  TIMESTAMP: \"sql.timestamp()\",\n  TIMESTAMP_NTZ: \"sql.timestamp()\",\n};\n\n/**\n * Query schema interface\n * @property name - the name of the query\n * @property type - the type of the query (string, number, boolean, object, array, etc.)\n */\nexport interface QuerySchema {\n  name: string;\n  type: string;\n}\n"],"mappings":";;;;;AAqBA,MAAa,kBAA0C;CAErD,QAAQ;CACR,QAAQ;CAER,SAAS;CAET,SAAS;CACT,KAAK;CACL,QAAQ;CACR,SAAS;CACT,UAAU;CACV,OAAO;CACP,QAAQ;CACR,SAAS;CAET,MAAM;CACN,WAAW;CACX,eAAe;CAChB;;;;;AAMD,MAAa,kBAA0C;CAErD,QAAQ;CACR,QAAQ;CAER,SAAS;CAIT,SAAS;CACT,SAAS;CACT,QAAQ;CACR,KAAK;CACL,SAAS;CACT,UAAU;CACV,OAAO;CACP,QAAQ;CAER,MAAM;CACN,WAAW;CACX,eAAe;CAChB"}