@databricks/appkit 0.1.1 → 0.1.3

package/bin/generate-types.js

@@ -0,0 +1,27 @@
+#!/usr/bin/env node
+import path from "node:path";
+
+import { generateFromEntryPoint } from "../dist/type-generator/index.js";
+
+// Parse arguments
+const args = process.argv.slice(2);
+const noCache = args.includes("--no-cache");
+const positionalArgs = args.filter((arg) => !arg.startsWith("--"));
+
+const rootDir = positionalArgs[0] || process.cwd();
+const outFile =
+  positionalArgs[1] || path.join(process.cwd(), "client/src/appKitTypes.d.ts");
+
+const queryFolder = path.join(rootDir, "config/queries");
+
+const warehouseId = positionalArgs[2] || process.env.DATABRICKS_WAREHOUSE_ID;
+if (!warehouseId) {
+  throw new Error("DATABRICKS_WAREHOUSE_ID is not set");
+}
+
+await generateFromEntryPoint({
+  queryFolder,
+  outFile,
+  warehouseId,
+  noCache,
+});

package/dist/appkit/package.js

@@ -1,6 +1,6 @@
 //#region package.json
 var name = "@databricks/appkit";
-var version = "0.1.1";
+var version = "0.1.3";
 
 //#endregion
 export { name, version };

package/dist/type-generator/cache.js

@@ -0,0 +1,49 @@
+import crypto from "node:crypto";
+import path from "node:path";
+import fs from "node:fs";
+
+//#region src/type-generator/cache.ts
+const CACHE_VERSION = "1";
+const CACHE_FILE = ".appkit-types-cache.json";
+const CACHE_DIR = path.join(process.cwd(), "node_modules", ".databricks", "appkit");
+/**
+* Hash the SQL query
+* Uses MD5 to hash the SQL query
+* @param sql - the SQL query to hash
+* @returns - the hash of the SQL query
+*/
+function hashSQL(sql) {
+	return crypto.createHash("md5").update(sql).digest("hex");
+}
+/**
+* Load the cache from the file system
+* If the cache is not found, run the query explain
+* @returns - the cache
+*/
+function loadCache() {
+	const cachePath = path.join(CACHE_DIR, CACHE_FILE);
+	try {
+		if (!fs.existsSync(CACHE_DIR)) fs.mkdirSync(CACHE_DIR, { recursive: true });
+		if (fs.existsSync(cachePath)) {
+			const cache = JSON.parse(fs.readFileSync(cachePath, "utf8"));
+			if (cache.version === CACHE_VERSION) return cache;
+		}
+	} catch {}
+	return {
+		version: CACHE_VERSION,
+		queries: {}
+	};
+}
+/**
+* Save the cache to the file system
+* The cache is saved as a JSON file, it is used to avoid running the query explain multiple times
+* @param cache - cache object to save
+*/
+function saveCache(cache) {
+	const cachePath = path.join(CACHE_DIR, CACHE_FILE);
+	fs.writeFileSync(cachePath, JSON.stringify(cache, null, 2), "utf8");
+}
+
+//#endregion
+export { CACHE_VERSION, hashSQL, loadCache, saveCache };
+//# sourceMappingURL=cache.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"cache.js","names":[],"sources":["../../src/type-generator/cache.ts"],"sourcesContent":["import crypto from \"node:crypto\";\nimport fs from \"node:fs\";\nimport path from \"node:path\";\n\n/**\n * Cache types\n * @property hash - the hash of the SQL query\n * @property type - the type of the query\n */\ninterface CacheEntry {\n  hash: string;\n  type: string;\n}\n\n/**\n * Cache interface\n * @property version - the version of the cache\n * @property queries - the queries in the cache\n */\ninterface Cache {\n  version: string;\n  queries: Record<string, CacheEntry>;\n}\n\nexport const CACHE_VERSION = \"1\";\nconst CACHE_FILE = \".appkit-types-cache.json\";\nconst CACHE_DIR = path.join(\n  process.cwd(),\n  \"node_modules\",\n  \".databricks\",\n  \"appkit\",\n);\n\n/**\n * Hash the SQL query\n * Uses MD5 to hash the SQL query\n * @param sql - the SQL query to hash\n * @returns - the hash of the SQL query\n */\nexport function hashSQL(sql: string): string {\n  return crypto.createHash(\"md5\").update(sql).digest(\"hex\");\n}\n\n/**\n * Load the cache from the file system\n * If the cache is not found, run the query explain\n * @returns - the cache\n */\nexport function loadCache(): Cache {\n  const cachePath = path.join(CACHE_DIR, CACHE_FILE);\n  try {\n    if (!fs.existsSync(CACHE_DIR)) {\n      fs.mkdirSync(CACHE_DIR, { recursive: true });\n    }\n\n    if (fs.existsSync(cachePath)) {\n      const cache = JSON.parse(fs.readFileSync(cachePath, \"utf8\")) as Cache;\n      if (cache.version === CACHE_VERSION) {\n        return cache;\n      }\n    }\n  } catch {\n    // ignore cache errors\n  }\n  return { version: CACHE_VERSION, queries: {} };\n}\n\n/**\n * Save the cache to the file system\n * The cache is saved as a JSON file, it is used to avoid running the query explain multiple times\n * @param cache - cache object to save\n */\nexport function saveCache(cache: Cache): void {\n  const cachePath = path.join(CACHE_DIR, CACHE_FILE);\n  fs.writeFileSync(cachePath, JSON.stringify(cache, null, 2), \"utf8\");\n}\n"],"mappings":";;;;;AAwBA,MAAa,gBAAgB;AAC7B,MAAM,aAAa;AACnB,MAAM,YAAY,KAAK,KACrB,QAAQ,KAAK,EACb,gBACA,eACA,SACD;;;;;;;AAQD,SAAgB,QAAQ,KAAqB;AAC3C,QAAO,OAAO,WAAW,MAAM,CAAC,OAAO,IAAI,CAAC,OAAO,MAAM;;;;;;;AAQ3D,SAAgB,YAAmB;CACjC,MAAM,YAAY,KAAK,KAAK,WAAW,WAAW;AAClD,KAAI;AACF,MAAI,CAAC,GAAG,WAAW,UAAU,CAC3B,IAAG,UAAU,WAAW,EAAE,WAAW,MAAM,CAAC;AAG9C,MAAI,GAAG,WAAW,UAAU,EAAE;GAC5B,MAAM,QAAQ,KAAK,MAAM,GAAG,aAAa,WAAW,OAAO,CAAC;AAC5D,OAAI,MAAM,YAAY,cACpB,QAAO;;SAGL;AAGR,QAAO;EAAE,SAAS;EAAe,SAAS,EAAE;EAAE;;;;;;;AAQhD,SAAgB,UAAU,OAAoB;CAC5C,MAAM,YAAY,KAAK,KAAK,WAAW,WAAW;AAClD,IAAG,cAAc,WAAW,KAAK,UAAU,OAAO,MAAM,EAAE,EAAE,OAAO"}

package/dist/type-generator/index.js

@@ -0,0 +1,46 @@
+import { generateQueriesFromDescribe } from "./query-registry.js";
+import fs from "node:fs";
+import dotenv from "dotenv";
+
+//#region src/type-generator/index.ts
+dotenv.config();
+/**
+* Generate type declarations for QueryRegistry
+* Create the d.ts file from the plugin routes and query schemas
+* @param querySchemas - the list of query schemas
+* @returns - the type declarations as a string
+*/
+function generateTypeDeclarations(querySchemas = []) {
+	const queryEntries = querySchemas.map(({ name, type }) => {
+		return `    ${name}: ${type.split("\n").map((line, i) => i === 0 ? line : `    ${line}`).join("\n")}`;
+	}).join(";\n");
+	return `// Auto-generated by AppKit - DO NOT EDIT
+// Generated by 'npx appkit-generate-types' or Vite plugin during build
+import "@databricks/appkit-ui/react";
+import type { SQLTypeMarker, SQLStringMarker, SQLNumberMarker, SQLBooleanMarker, SQLBinaryMarker, SQLDateMarker, SQLTimestampMarker } from "@databricks/appkit-ui/js";
+
+declare module "@databricks/appkit-ui/react" {
+  interface QueryRegistry {${queryEntries ? `\n${queryEntries};\n  ` : ""}}
+}
+`;
+}
+/**
+* Entry point for generating type declarations from all imported files
+* @param options - the options for the generation
+* @param options.entryPoint - the entry point file
+* @param options.outFile - the output file
+* @param options.querySchemaFile - optional path to query schema file (e.g. config/queries/schema.ts)
+*/
+async function generateFromEntryPoint(options) {
+	const { outFile, queryFolder, warehouseId, noCache } = options;
+	console.log("\n[AppKit] Starting type generation...\n");
+	let queryRegistry = [];
+	if (queryFolder) queryRegistry = await generateQueriesFromDescribe(queryFolder, warehouseId, { noCache });
+	const typeDeclarations = generateTypeDeclarations(queryRegistry);
+	fs.writeFileSync(outFile, typeDeclarations, "utf-8");
+	console.log("\n[AppKit] Type generation complete!\n");
+}
+
+//#endregion
+export { generateFromEntryPoint };
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","names":["queryRegistry: QuerySchema[]"],"sources":["../../src/type-generator/index.ts"],"sourcesContent":["import fs from \"node:fs\";\nimport dotenv from \"dotenv\";\nimport { generateQueriesFromDescribe } from \"./query-registry\";\nimport type { QuerySchema } from \"./types\";\n\ndotenv.config();\n\n/**\n * Generate type declarations for QueryRegistry\n * Create the d.ts file from the plugin routes and query schemas\n * @param querySchemas - the list of query schemas\n * @returns - the type declarations as a string\n */\nfunction generateTypeDeclarations(querySchemas: QuerySchema[] = []): string {\n  const queryEntries = querySchemas\n    .map(({ name, type }) => {\n      const indentedType = type\n        .split(\"\\n\")\n        .map((line, i) => (i === 0 ? line : `    ${line}`))\n        .join(\"\\n\");\n      return `    ${name}: ${indentedType}`;\n    })\n    .join(\";\\n\");\n\n  const querySection = queryEntries ? `\\n${queryEntries};\\n  ` : \"\";\n\n  return `// Auto-generated by AppKit - DO NOT EDIT\n// Generated by 'npx appkit-generate-types' or Vite plugin during build\nimport \"@databricks/appkit-ui/react\";\nimport type { SQLTypeMarker, SQLStringMarker, SQLNumberMarker, SQLBooleanMarker, SQLBinaryMarker, SQLDateMarker, SQLTimestampMarker } from \"@databricks/appkit-ui/js\";\n\ndeclare module \"@databricks/appkit-ui/react\" {\n  interface QueryRegistry {${querySection}}\n}\n`;\n}\n\n/**\n * Entry point for generating type declarations from all imported files\n * @param options - the options for the generation\n * @param options.entryPoint - the entry point file\n * @param options.outFile - the output file\n * @param options.querySchemaFile - optional path to query schema file (e.g. config/queries/schema.ts)\n */\nexport async function generateFromEntryPoint(options: {\n  outFile: string;\n  queryFolder?: string;\n  warehouseId: string;\n  noCache?: boolean;\n}) {\n  const { outFile, queryFolder, warehouseId, noCache } = options;\n\n  console.log(\"\\n[AppKit] Starting type generation...\\n\");\n\n  let queryRegistry: QuerySchema[] = [];\n  if (queryFolder)\n    queryRegistry = await generateQueriesFromDescribe(\n      queryFolder,\n      warehouseId,\n      {\n        noCache,\n      },\n    );\n\n  const typeDeclarations = generateTypeDeclarations(queryRegistry);\n\n  fs.writeFileSync(outFile, typeDeclarations, \"utf-8\");\n\n  console.log(\"\\n[AppKit] Type generation complete!\\n\");\n}\n"],"mappings":";;;;;AAKA,OAAO,QAAQ;;;;;;;AAQf,SAAS,yBAAyB,eAA8B,EAAE,EAAU;CAC1E,MAAM,eAAe,aAClB,KAAK,EAAE,MAAM,WAAW;AAKvB,SAAO,OAAO,KAAK,IAJE,KAClB,MAAM,KAAK,CACX,KAAK,MAAM,MAAO,MAAM,IAAI,OAAO,OAAO,OAAQ,CAClD,KAAK,KAAK;GAEb,CACD,KAAK,MAAM;AAId,QAAO;;;;;;6BAFc,eAAe,KAAK,aAAa,SAAS,GAQvB;;;;;;;;;;;AAY1C,eAAsB,uBAAuB,SAK1C;CACD,MAAM,EAAE,SAAS,aAAa,aAAa,YAAY;AAEvD,SAAQ,IAAI,2CAA2C;CAEvD,IAAIA,gBAA+B,EAAE;AACrC,KAAI,YACF,iBAAgB,MAAM,4BACpB,aACA,aACA,EACE,SACD,CACF;CAEH,MAAM,mBAAmB,yBAAyB,cAAc;AAEhE,IAAG,cAAc,SAAS,kBAAkB,QAAQ;AAEpD,SAAQ,IAAI,yCAAyC"}

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

@@ -0,0 +1,170 @@
+import { CACHE_VERSION, hashSQL, loadCache, saveCache } from "./cache.js";
+import { Spinner } from "./spinner.js";
+import { sqlTypeToHelper, sqlTypeToMarker } from "./types.js";
+import { WorkspaceClient } from "@databricks/sdk-experimental";
+import path from "node:path";
+import fs from "node:fs";
+
+//#region src/type-generator/query-registry.ts
+/**
+* Extract parameters from a SQL query
+* @param sql - the SQL query to extract parameters from
+* @returns an array of parameter names
+*/
+function extractParameters(sql) {
+	const matches = sql.matchAll(/:([a-zA-Z_]\w*)/g);
+	const params = /* @__PURE__ */ new Set();
+	for (const match of matches) params.add(match[1]);
+	return Array.from(params);
+}
+const SERVER_INJECTED_PARAMS = ["workspaceId"];
+function convertToQueryType(result, sql, queryName) {
+	const columns = (result.result?.data_array || []).map((row) => ({
+		name: row[0] || "",
+		type_name: row[1]?.toUpperCase() || "STRING",
+		comment: row[2] || void 0
+	}));
+	const params = extractParameters(sql).filter((p) => !SERVER_INJECTED_PARAMS.includes(p));
+	const paramTypes = extractParameterTypes(sql);
+	return `{
+    name: "${queryName}";
+    parameters: ${params.length > 0 ? `{\n      ${params.map((p) => {
+		const sqlType = paramTypes[p];
+		const markerType = sqlType ? sqlTypeToMarker[sqlType] : "SQLTypeMarker";
+		const helper = sqlType ? sqlTypeToHelper[sqlType] : "sql.*()";
+		return `/** ${sqlType || "any"} - use ${helper} */\n      ${p}: ${markerType}`;
+	}).join(";\n      ")};\n    }` : "Record<string, never>"};
+    result: Array<{
+      ${columns.map((column) => {
+		const mappedType = typeMap[normalizeTypeName(column.type_name)] || "unknown";
+		const name = /^[a-zA-Z_$][a-zA-Z0-9_$]*$/.test(column.name) ? column.name : `"${column.name}"`;
+		return `${column.comment ? `/** ${column.comment} */\n      ` : `/** @sqlType ${column.type_name} */\n      `}${name}: ${mappedType}`;
+	}).join(";\n      ")};
+    }>;
+  }`;
+}
+function extractParameterTypes(sql) {
+	const paramTypes = {};
+	const matches = sql.matchAll(/--\s*@param\s+(\w+)\s+(STRING|NUMERIC|BOOLEAN|DATE|TIMESTAMP|BINARY)/gi);
+	for (const match of matches) {
+		const [, paramName, paramType] = match;
+		paramTypes[paramName] = paramType.toUpperCase();
+	}
+	return paramTypes;
+}
+/**
+* Generate query schemas from a folder of SQL files
+* It uses DESCRIBE QUERY to get the schema without executing the query
+* @param queryFolder - the folder containing the SQL files
+* @param warehouseId - the warehouse id to use for schema analysis
+* @param options - options for the query generation
+* @param options.noCache - if true, skip the cache and regenerate all types
+* @returns an array of query schemas
+*/
+async function generateQueriesFromDescribe(queryFolder, warehouseId, options = {}) {
+	const { noCache = false } = options;
+	const queryFiles = fs.readdirSync(queryFolder).filter((file) => file.endsWith(".sql"));
+	console.log(`  Found ${queryFiles.length} SQL queries\n`);
+	const cache = noCache ? {
+		version: CACHE_VERSION,
+		queries: {}
+	} : loadCache();
+	const client = new WorkspaceClient({});
+	const querySchemas = [];
+	const failedQueries = [];
+	const spinner = new Spinner();
+	for (let i = 0; i < queryFiles.length; i++) {
+		const file = queryFiles[i];
+		const queryName = path.basename(file, ".sql");
+		const sql = fs.readFileSync(path.join(queryFolder, file), "utf8");
+		const sqlHash = hashSQL(sql);
+		const cached = cache.queries[queryName];
+		if (cached && cached.hash === sqlHash) {
+			querySchemas.push({
+				name: queryName,
+				type: cached.type
+			});
+			spinner.start(`Processing ${queryName} (${i + 1}/${queryFiles.length})`);
+			spinner.stop(`✓ ${queryName} (cached)`);
+			continue;
+		}
+		spinner.start(`Processing ${queryName} (${i + 1}/${queryFiles.length})`);
+		const cleanedSql = sql.replace(/:([a-zA-Z_]\w*)/g, "''").trim().replace(/;\s*$/, "");
+		try {
+			const result = await client.statementExecution.executeStatement({
+				statement: `DESCRIBE QUERY ${cleanedSql}`,
+				warehouse_id: warehouseId
+			});
+			if (result.status.state === "FAILED") {
+				spinner.stop(`✗ ${queryName} - failed`);
+				failedQueries.push({
+					name: queryName,
+					error: "Query execution failed"
+				});
+				continue;
+			}
+			const type = convertToQueryType(result, sql, queryName);
+			querySchemas.push({
+				name: queryName,
+				type
+			});
+			cache.queries[queryName] = {
+				hash: sqlHash,
+				type
+			};
+			spinner.stop(`✓ ${queryName}`);
+		} catch (error) {
+			const errorMessage = error instanceof Error ? error.message : "Unknown error";
+			spinner.stop(`✗ ${queryName} - ${errorMessage}`);
+			failedQueries.push({
+				name: queryName,
+				error: errorMessage
+			});
+		}
+	}
+	saveCache(cache);
+	if (failedQueries.length > 0) console.warn(`  Warning: ${failedQueries.length} queries failed\n`);
+	return querySchemas;
+}
+/**
+* Normalize SQL type name by removing parameters/generics
+* Examples:
+*   DECIMAL(38,6) -> DECIMAL
+*   ARRAY<STRING> -> ARRAY
+*   MAP<STRING,INT> -> MAP
+*   STRUCT<name:STRING> -> STRUCT
+*   INTERVAL DAY TO SECOND -> INTERVAL
+*   GEOGRAPHY(4326) -> GEOGRAPHY
+*/
+function normalizeTypeName(typeName) {
+	return typeName.replace(/\(.*\)$/, "").replace(/<.*>$/, "").split(" ")[0];
+}
+/** Type Map for Databricks data types to JavaScript types */
+const typeMap = {
+	STRING: "string",
+	BINARY: "string",
+	BOOLEAN: "boolean",
+	TINYINT: "number",
+	SMALLINT: "number",
+	INT: "number",
+	BIGINT: "number",
+	FLOAT: "number",
+	DOUBLE: "number",
+	DECIMAL: "number",
+	DATE: "string",
+	TIMESTAMP: "string",
+	TIMESTAMP_NTZ: "string",
+	INTERVAL: "string",
+	ARRAY: "unknown[]",
+	MAP: "Record<string, unknown>",
+	STRUCT: "Record<string, unknown>",
+	OBJECT: "Record<string, unknown>",
+	VARIANT: "unknown",
+	GEOGRAPHY: "unknown",
+	GEOMETRY: "unknown",
+	VOID: "null"
+};
+
+//#endregion
+export { generateQueriesFromDescribe };
+//# sourceMappingURL=query-registry.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"query-registry.js","names":["paramTypes: Record<string, string>","querySchemas: QuerySchema[]","failedQueries: { name: string; error: string }[]","typeMap: Record<string, string>"],"sources":["../../src/type-generator/query-registry.ts"],"sourcesContent":["import fs from \"node:fs\";\nimport path from \"node:path\";\nimport { WorkspaceClient } from \"@databricks/sdk-experimental\";\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\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(sql: string): string[] {\n  const matches = sql.matchAll(/:([a-zA-Z_]\\w*)/g);\n  const params = new Set<string>();\n  for (const match of matches) {\n    params.add(match[1]);\n  }\n  return Array.from(params);\n}\n\n// parameters that are injected by the server\nexport const SERVER_INJECTED_PARAMS = [\"workspaceId\"];\n\nexport function convertToQueryType(\n  result: DatabricksStatementExecutionResponse,\n  sql: string,\n  queryName: string,\n): string {\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 params = extractParameters(sql).filter(\n    (p) => !SERVER_INJECTED_PARAMS.includes(p),\n  );\n\n  const paramTypes = extractParameterTypes(sql);\n\n  // generate parameters types with JSDoc hints\n  const paramsType =\n    params.length > 0\n      ? `{\\n      ${params\n          .map((p) => {\n            const sqlType = paramTypes[p];\n            // if no type annotation, use SQLTypeMarker (union type)\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  // 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  return `{\n    name: \"${queryName}\";\n    parameters: ${paramsType};\n    result: Array<{\n      ${resultFields.join(\";\\n      \")};\n    }>;\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\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 } = {},\n): Promise<QuerySchema[]> {\n  const { noCache = false } = options;\n\n  // read all query files in the folder\n  const queryFiles = fs\n    .readdirSync(queryFolder)\n    .filter((file) => file.endsWith(\".sql\"));\n\n  console.log(`  Found ${queryFiles.length} SQL queries\\n`);\n\n  // load cache\n  const cache = noCache ? { version: CACHE_VERSION, queries: {} } : loadCache();\n\n  const client = new WorkspaceClient({});\n  const querySchemas: QuerySchema[] = [];\n  const failedQueries: { name: string; error: string }[] = [];\n  const spinner = new Spinner();\n\n  // process each query file\n  for (let i = 0; i < queryFiles.length; i++) {\n    const file = queryFiles[i];\n    const queryName = path.basename(file, \".sql\");\n\n    // read query file content\n    const sql = fs.readFileSync(path.join(queryFolder, file), \"utf8\");\n    const sqlHash = hashSQL(sql);\n\n    // check cache\n    const cached = cache.queries[queryName];\n    if (cached && cached.hash === sqlHash) {\n      querySchemas.push({ name: queryName, type: cached.type });\n      spinner.start(`Processing ${queryName} (${i + 1}/${queryFiles.length})`);\n      spinner.stop(`✓ ${queryName} (cached)`);\n      continue;\n    }\n\n    spinner.start(`Processing ${queryName} (${i + 1}/${queryFiles.length})`);\n\n    const sqlWithDefaults = sql.replace(/:([a-zA-Z_]\\w*)/g, \"''\");\n\n    // strip trailing semicolon for DESCRIBE QUERY\n    const cleanedSql = sqlWithDefaults.trim().replace(/;\\s*$/, \"\");\n\n    // execute DESCRIBE QUERY to get schema without running the actual query\n    try {\n      const result = (await client.statementExecution.executeStatement({\n        statement: `DESCRIBE QUERY ${cleanedSql}`,\n        warehouse_id: warehouseId,\n      })) as DatabricksStatementExecutionResponse;\n\n      if (result.status.state === \"FAILED\") {\n        spinner.stop(`✗ ${queryName} - failed`);\n        failedQueries.push({\n          name: queryName,\n          error: \"Query execution failed\",\n        });\n        continue;\n      }\n\n      // convert result to query schema\n      const type = convertToQueryType(result, sql, queryName);\n      querySchemas.push({ name: queryName, type });\n\n      // update cache\n      cache.queries[queryName] = { hash: sqlHash, type };\n\n      spinner.stop(`✓ ${queryName}`);\n    } catch (error) {\n      const errorMessage =\n        error instanceof Error ? error.message : \"Unknown error\";\n      spinner.stop(`✗ ${queryName} - ${errorMessage}`);\n      failedQueries.push({ name: queryName, error: errorMessage });\n    }\n  }\n\n  // save cache\n  saveCache(cache);\n\n  // log warning if there are failed queries\n  if (failedQueries.length > 0) {\n    console.warn(`  Warning: ${failedQueries.length} queries failed\\n`);\n  }\n\n  return querySchemas;\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":";;;;;;;;;;;;;AAiBA,SAAgB,kBAAkB,KAAuB;CACvD,MAAM,UAAU,IAAI,SAAS,mBAAmB;CAChD,MAAM,yBAAS,IAAI,KAAa;AAChC,MAAK,MAAM,SAAS,QAClB,QAAO,IAAI,MAAM,GAAG;AAEtB,QAAO,MAAM,KAAK,OAAO;;AAI3B,MAAa,yBAAyB,CAAC,cAAc;AAErD,SAAgB,mBACd,QACA,KACA,WACQ;CAER,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,SAAS,kBAAkB,IAAI,CAAC,QACnC,MAAM,CAAC,uBAAuB,SAAS,EAAE,CAC3C;CAED,MAAM,aAAa,sBAAsB,IAAI;AAmC7C,QAAO;aACI,UAAU;kBAhCnB,OAAO,SAAS,IACZ,YAAY,OACT,KAAK,MAAM;EACV,MAAM,UAAU,WAAW;EAE3B,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,wBAqBqB;;QAlBN,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,CAMiB,KAAK,YAAY,CAAC;;;;AAKvC,SAAgB,sBAAsB,KAAqC;CACzE,MAAMA,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;;;;;;;;;;;AAYT,eAAsB,4BACpB,aACA,aACA,UAAiC,EAAE,EACX;CACxB,MAAM,EAAE,UAAU,UAAU;CAG5B,MAAM,aAAa,GAChB,YAAY,YAAY,CACxB,QAAQ,SAAS,KAAK,SAAS,OAAO,CAAC;AAE1C,SAAQ,IAAI,WAAW,WAAW,OAAO,gBAAgB;CAGzD,MAAM,QAAQ,UAAU;EAAE,SAAS;EAAe,SAAS,EAAE;EAAE,GAAG,WAAW;CAE7E,MAAM,SAAS,IAAI,gBAAgB,EAAE,CAAC;CACtC,MAAMC,eAA8B,EAAE;CACtC,MAAMC,gBAAmD,EAAE;CAC3D,MAAM,UAAU,IAAI,SAAS;AAG7B,MAAK,IAAI,IAAI,GAAG,IAAI,WAAW,QAAQ,KAAK;EAC1C,MAAM,OAAO,WAAW;EACxB,MAAM,YAAY,KAAK,SAAS,MAAM,OAAO;EAG7C,MAAM,MAAM,GAAG,aAAa,KAAK,KAAK,aAAa,KAAK,EAAE,OAAO;EACjE,MAAM,UAAU,QAAQ,IAAI;EAG5B,MAAM,SAAS,MAAM,QAAQ;AAC7B,MAAI,UAAU,OAAO,SAAS,SAAS;AACrC,gBAAa,KAAK;IAAE,MAAM;IAAW,MAAM,OAAO;IAAM,CAAC;AACzD,WAAQ,MAAM,cAAc,UAAU,IAAI,IAAI,EAAE,GAAG,WAAW,OAAO,GAAG;AACxE,WAAQ,KAAK,KAAK,UAAU,WAAW;AACvC;;AAGF,UAAQ,MAAM,cAAc,UAAU,IAAI,IAAI,EAAE,GAAG,WAAW,OAAO,GAAG;EAKxE,MAAM,aAHkB,IAAI,QAAQ,oBAAoB,KAAK,CAG1B,MAAM,CAAC,QAAQ,SAAS,GAAG;AAG9D,MAAI;GACF,MAAM,SAAU,MAAM,OAAO,mBAAmB,iBAAiB;IAC/D,WAAW,kBAAkB;IAC7B,cAAc;IACf,CAAC;AAEF,OAAI,OAAO,OAAO,UAAU,UAAU;AACpC,YAAQ,KAAK,KAAK,UAAU,WAAW;AACvC,kBAAc,KAAK;KACjB,MAAM;KACN,OAAO;KACR,CAAC;AACF;;GAIF,MAAM,OAAO,mBAAmB,QAAQ,KAAK,UAAU;AACvD,gBAAa,KAAK;IAAE,MAAM;IAAW;IAAM,CAAC;AAG5C,SAAM,QAAQ,aAAa;IAAE,MAAM;IAAS;IAAM;AAElD,WAAQ,KAAK,KAAK,YAAY;WACvB,OAAO;GACd,MAAM,eACJ,iBAAiB,QAAQ,MAAM,UAAU;AAC3C,WAAQ,KAAK,KAAK,UAAU,KAAK,eAAe;AAChD,iBAAc,KAAK;IAAE,MAAM;IAAW,OAAO;IAAc,CAAC;;;AAKhE,WAAU,MAAM;AAGhB,KAAI,cAAc,SAAS,EACzB,SAAQ,KAAK,cAAc,cAAc,OAAO,mBAAmB;AAGrE,QAAO;;;;;;;;;;;;AAaT,SAAgB,kBAAkB,UAA0B;AAC1D,QAAO,SACJ,QAAQ,WAAW,GAAG,CACtB,QAAQ,SAAS,GAAG,CACpB,MAAM,IAAI,CAAC;;;AAIhB,MAAMC,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/spinner.js

@@ -0,0 +1,37 @@
+//#region src/type-generator/spinner.ts
+/**
+* Simple loading spinner for CLI
+*/
+var Spinner = class {
+	constructor() {
+		this.frames = [
+			"   ",
+			".  ",
+			".. ",
+			"..."
+		];
+		this.current = 0;
+		this.interval = null;
+		this.text = "";
+	}
+	start(text) {
+		this.text = text;
+		this.current = 0;
+		process.stdout.write(`  ${this.text}${this.frames[0]}`);
+		this.interval = setInterval(() => {
+			this.current = (this.current + 1) % this.frames.length;
+			process.stdout.write(`\r  ${this.text}${this.frames[this.current]}`);
+		}, 300);
+	}
+	stop(finalText) {
+		if (this.interval) {
+			clearInterval(this.interval);
+			this.interval = null;
+		}
+		process.stdout.write(`\x1b[2K\r  ${finalText || this.text}\n`);
+	}
+};
+
+//#endregion
+export { Spinner };
+//# sourceMappingURL=spinner.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"spinner.js","names":[],"sources":["../../src/type-generator/spinner.ts"],"sourcesContent":["/**\n * Simple loading spinner for CLI\n */\nexport class Spinner {\n  private frames = [\"   \", \".  \", \".. \", \"...\"];\n  private current = 0;\n  private interval: NodeJS.Timeout | null = null;\n  private text = \"\";\n\n  start(text: string) {\n    this.text = text;\n    this.current = 0;\n    process.stdout.write(`  ${this.text}${this.frames[0]}`);\n    this.interval = setInterval(() => {\n      this.current = (this.current + 1) % this.frames.length;\n      process.stdout.write(`\\r  ${this.text}${this.frames[this.current]}`);\n    }, 300);\n  }\n\n  stop(finalText?: string) {\n    if (this.interval) {\n      clearInterval(this.interval);\n      this.interval = null;\n    }\n    // clear the line and write the final text\n    process.stdout.write(`\\x1b[2K\\r  ${finalText || this.text}\\n`);\n  }\n}\n"],"mappings":";;;;AAGA,IAAa,UAAb,MAAqB;;gBACF;GAAC;GAAO;GAAO;GAAO;GAAM;iBAC3B;kBACwB;cAC3B;;CAEf,MAAM,MAAc;AAClB,OAAK,OAAO;AACZ,OAAK,UAAU;AACf,UAAQ,OAAO,MAAM,KAAK,KAAK,OAAO,KAAK,OAAO,KAAK;AACvD,OAAK,WAAW,kBAAkB;AAChC,QAAK,WAAW,KAAK,UAAU,KAAK,KAAK,OAAO;AAChD,WAAQ,OAAO,MAAM,OAAO,KAAK,OAAO,KAAK,OAAO,KAAK,WAAW;KACnE,IAAI;;CAGT,KAAK,WAAoB;AACvB,MAAI,KAAK,UAAU;AACjB,iBAAc,KAAK,SAAS;AAC5B,QAAK,WAAW;;AAGlB,UAAQ,OAAO,MAAM,cAAc,aAAa,KAAK,KAAK,IAAI"}

package/dist/type-generator/types.js

@@ -0,0 +1,45 @@
+//#region src/type-generator/types.ts
+/**
+* Map of SQL types to their corresponding marker types
+* Used to convert SQL types to their corresponding marker types
+*/
+const sqlTypeToMarker = {
+	STRING: "SQLStringMarker",
+	BINARY: "SQLBinaryMarker",
+	BOOLEAN: "SQLBooleanMarker",
+	NUMERIC: "SQLNumberMarker",
+	INT: "SQLNumberMarker",
+	BIGINT: "SQLNumberMarker",
+	TINYINT: "SQLNumberMarker",
+	SMALLINT: "SQLNumberMarker",
+	FLOAT: "SQLNumberMarker",
+	DOUBLE: "SQLNumberMarker",
+	DECIMAL: "SQLNumberMarker",
+	DATE: "SQLDateMarker",
+	TIMESTAMP: "SQLTimestampMarker",
+	TIMESTAMP_NTZ: "SQLTimestampMarker"
+};
+/**
+* Map of SQL types to their corresponding helper function names
+* Used to generate JSDoc hints for parameters
+*/
+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()",
+	DATE: "sql.date()",
+	TIMESTAMP: "sql.timestamp()",
+	TIMESTAMP_NTZ: "sql.timestamp()"
+};
+
+//#endregion
+export { sqlTypeToHelper, sqlTypeToMarker };
+//# sourceMappingURL=types.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","names":["sqlTypeToMarker: Record<string, string>","sqlTypeToHelper: Record<string, string>"],"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: { state: string };\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":";;;;;AAkBA,MAAaA,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,MAAaC,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"}

package/dist/type-generator/vite-plugin.d.ts

@@ -12,7 +12,7 @@ interface AppKitTypesPluginOptions {
 }
 /**
  * Vite plugin to generate types for AppKit queries.
- * Calls `npx appkit-generate-types` under the hood.
+ * Calls generateFromEntryPoint under the hood.
  * @param options - Options to override default values.
  * @returns Vite plugin to generate types for AppKit queries.
  */

package/dist/type-generator/vite-plugin.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"vite-plugin.d.ts","names":[],"sources":["../../src/type-generator/vite-plugin.ts"],"sourcesContent":[],"mappings":";;;;;;AAEmC;AAkBnC,UAbU,wBAAA,CAauB;EAAA,OAAA,CAAA,EAAA,MAAA;;cAAsC,CAAA,EAAA,MAAA,EAAA;;;;;;;;iBAAvD,iBAAA,WAA4B,2BAA2B"}
+{"version":3,"file":"vite-plugin.d.ts","names":[],"sources":["../../src/type-generator/vite-plugin.ts"],"sourcesContent":[],"mappings":";;;;;;AACmC;AAoBnC,UAbU,wBAAA,CAauB;EAAA,OAAA,CAAA,EAAA,MAAA;;cAAsC,CAAA,EAAA,MAAA,EAAA;;;;;;;;iBAAvD,iBAAA,WAA4B,2BAA2B"}

package/dist/type-generator/vite-plugin.js

@@ -1,23 +1,30 @@
+import { generateFromEntryPoint } from "./index.js";
 import path from "node:path";
-import { execSync } from "node:child_process";
+import fs from "node:fs";
 
 //#region src/type-generator/vite-plugin.ts
 /**
 * Vite plugin to generate types for AppKit queries.
-* Calls `npx appkit-generate-types` under the hood.
+* Calls generateFromEntryPoint under the hood.
 * @param options - Options to override default values.
 * @returns Vite plugin to generate types for AppKit queries.
 */
 function appKitTypesPlugin(options) {
 	let root;
-	let appRoot;
 	let outFile;
 	let watchFolders;
-	function generate() {
+	async function generate() {
 		try {
-			execSync(`npx appkit-generate-types ${[appRoot, outFile].join(" ")}`, {
-				cwd: appRoot,
-				stdio: "inherit"
+			const warehouseId = process.env.DATABRICKS_WAREHOUSE_ID || "";
+			if (!warehouseId) {
+				console.warn("[AppKit] Warehouse ID not found. Skipping type generation.");
+				return;
+			}
+			await generateFromEntryPoint({
+				outFile,
+				queryFolder: watchFolders[0],
+				warehouseId,
+				noCache: false
 			});
 		} catch (error) {
 			if (process.env.NODE_ENV === "production") throw error;
@@ -26,11 +33,18 @@ function appKitTypesPlugin(options) {
 	}
 	return {
 		name: "appkit-types",
+		apply() {
+			if (!(process.env.DATABRICKS_WAREHOUSE_ID || "")) {
+				console.warn("[AppKit] Warehouse ID not found. Skipping type generation.");
+				return false;
+			}
+			if (!fs.existsSync(path.join(process.cwd(), "config", "queries"))) return false;
+			return true;
+		},
 		configResolved(config) {
 			root = config.root;
-			appRoot = path.resolve(root, "..");
 			outFile = path.resolve(root, options?.outFile ?? "src/appKitTypes.d.ts");
-			watchFolders = (options?.watchFolders ?? ["../config/queries"]).map((folder) => path.resolve(root, folder));
+			watchFolders = options?.watchFolders ?? [path.join(process.cwd(), "config", "queries")];
 		},
 		buildStart() {
 			generate();

package/dist/type-generator/vite-plugin.js.map

@@ -1 +1 @@
-{"version":3,"file":"vite-plugin.js","names":["root: string","appRoot: string","outFile: string","watchFolders: string[]"],"sources":["../../src/type-generator/vite-plugin.ts"],"sourcesContent":["import { execSync } from \"node:child_process\";\nimport path from \"node:path\";\nimport type { Plugin } from \"vite\";\n\n/**\n * Options for the AppKit types plugin.\n */\ninterface AppKitTypesPluginOptions {\n  /* Path to the output d.ts file (relative to client folder). */\n  outFile?: string;\n  /** Folders to watch for changes. */\n  watchFolders?: string[];\n}\n\n/**\n * Vite plugin to generate types for AppKit queries.\n * Calls `npx appkit-generate-types` under the hood.\n * @param options - Options to override default values.\n * @returns Vite plugin to generate types for AppKit queries.\n */\nexport function appKitTypesPlugin(options?: AppKitTypesPluginOptions): Plugin {\n  let root: string;\n  let appRoot: string;\n  let outFile: string;\n  let watchFolders: string[];\n\n  function generate() {\n    try {\n      const args = [appRoot, outFile].join(\" \");\n      execSync(`npx appkit-generate-types ${args}`, {\n        cwd: appRoot,\n        stdio: \"inherit\",\n      });\n    } catch (error) {\n      // throw in production to fail the build\n      if (process.env.NODE_ENV === \"production\") {\n        throw error;\n      }\n      console.error(\"[AppKit] Error generating types:\", error);\n    }\n  }\n\n  return {\n    name: \"appkit-types\",\n    configResolved(config) {\n      root = config.root;\n      appRoot = path.resolve(root, \"..\");\n\n      outFile = path.resolve(root, options?.outFile ?? \"src/appKitTypes.d.ts\");\n\n      watchFolders = (options?.watchFolders ?? [\"../config/queries\"]).map(\n        (folder) => path.resolve(root, folder),\n      );\n    },\n    buildStart() {\n      generate();\n    },\n\n    configureServer(server) {\n      server.watcher.add(watchFolders);\n\n      server.watcher.on(\"change\", (changedFile) => {\n        const isWatchedFile = watchFolders.some((folder) =>\n          changedFile.startsWith(folder),\n        );\n\n        if (isWatchedFile && changedFile.endsWith(\".sql\")) {\n          generate();\n        }\n      });\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;AAoBA,SAAgB,kBAAkB,SAA4C;CAC5E,IAAIA;CACJ,IAAIC;CACJ,IAAIC;CACJ,IAAIC;CAEJ,SAAS,WAAW;AAClB,MAAI;AAEF,YAAS,6BADI,CAAC,SAAS,QAAQ,CAAC,KAAK,IAAI,IACK;IAC5C,KAAK;IACL,OAAO;IACR,CAAC;WACK,OAAO;AAEd,OAAI,QAAQ,IAAI,aAAa,aAC3B,OAAM;AAER,WAAQ,MAAM,oCAAoC,MAAM;;;AAI5D,QAAO;EACL,MAAM;EACN,eAAe,QAAQ;AACrB,UAAO,OAAO;AACd,aAAU,KAAK,QAAQ,MAAM,KAAK;AAElC,aAAU,KAAK,QAAQ,MAAM,SAAS,WAAW,uBAAuB;AAExE,mBAAgB,SAAS,gBAAgB,CAAC,oBAAoB,EAAE,KAC7D,WAAW,KAAK,QAAQ,MAAM,OAAO,CACvC;;EAEH,aAAa;AACX,aAAU;;EAGZ,gBAAgB,QAAQ;AACtB,UAAO,QAAQ,IAAI,aAAa;AAEhC,UAAO,QAAQ,GAAG,WAAW,gBAAgB;AAK3C,QAJsB,aAAa,MAAM,WACvC,YAAY,WAAW,OAAO,CAC/B,IAEoB,YAAY,SAAS,OAAO,CAC/C,WAAU;KAEZ;;EAEL"}
+{"version":3,"file":"vite-plugin.js","names":["root: string","outFile: string","watchFolders: string[]"],"sources":["../../src/type-generator/vite-plugin.ts"],"sourcesContent":["import path from \"node:path\";\nimport type { Plugin } from \"vite\";\nimport fs from \"node:fs\";\nimport { generateFromEntryPoint } from \"./index\";\n\n/**\n * Options for the AppKit types plugin.\n */\ninterface AppKitTypesPluginOptions {\n  /* Path to the output d.ts file (relative to client folder). */\n  outFile?: string;\n  /** Folders to watch for changes. */\n  watchFolders?: string[];\n}\n\n/**\n * Vite plugin to generate types for AppKit queries.\n * Calls generateFromEntryPoint under the hood.\n * @param options - Options to override default values.\n * @returns Vite plugin to generate types for AppKit queries.\n */\nexport function appKitTypesPlugin(options?: AppKitTypesPluginOptions): Plugin {\n  let root: string;\n  let outFile: string;\n  let watchFolders: string[];\n\n  async function generate() {\n    try {\n      const warehouseId = process.env.DATABRICKS_WAREHOUSE_ID || \"\";\n\n      if (!warehouseId) {\n        console.warn(\n          \"[AppKit] Warehouse ID not found. Skipping type generation.\",\n        );\n        return;\n      }\n\n      await generateFromEntryPoint({\n        outFile,\n        queryFolder: watchFolders[0],\n        warehouseId,\n        noCache: false,\n      });\n    } catch (error) {\n      // throw in production to fail the build\n      if (process.env.NODE_ENV === \"production\") {\n        throw error;\n      }\n      console.error(\"[AppKit] Error generating types:\", error);\n    }\n  }\n\n  return {\n    name: \"appkit-types\",\n\n    apply() {\n      const warehouseId = process.env.DATABRICKS_WAREHOUSE_ID || \"\";\n\n      if (!warehouseId) {\n        console.warn(\n          \"[AppKit] Warehouse ID not found. Skipping type generation.\",\n        );\n        return false;\n      }\n\n      if (!fs.existsSync(path.join(process.cwd(), \"config\", \"queries\"))) {\n        return false;\n      }\n\n      return true;\n    },\n\n    configResolved(config) {\n      root = config.root;\n      outFile = path.resolve(root, options?.outFile ?? \"src/appKitTypes.d.ts\");\n      watchFolders = options?.watchFolders ?? [\n        path.join(process.cwd(), \"config\", \"queries\"),\n      ];\n    },\n\n    buildStart() {\n      generate();\n    },\n\n    configureServer(server) {\n      server.watcher.add(watchFolders);\n\n      server.watcher.on(\"change\", (changedFile) => {\n        const isWatchedFile = watchFolders.some((folder) =>\n          changedFile.startsWith(folder),\n        );\n\n        if (isWatchedFile && changedFile.endsWith(\".sql\")) {\n          generate();\n        }\n      });\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;AAqBA,SAAgB,kBAAkB,SAA4C;CAC5E,IAAIA;CACJ,IAAIC;CACJ,IAAIC;CAEJ,eAAe,WAAW;AACxB,MAAI;GACF,MAAM,cAAc,QAAQ,IAAI,2BAA2B;AAE3D,OAAI,CAAC,aAAa;AAChB,YAAQ,KACN,6DACD;AACD;;AAGF,SAAM,uBAAuB;IAC3B;IACA,aAAa,aAAa;IAC1B;IACA,SAAS;IACV,CAAC;WACK,OAAO;AAEd,OAAI,QAAQ,IAAI,aAAa,aAC3B,OAAM;AAER,WAAQ,MAAM,oCAAoC,MAAM;;;AAI5D,QAAO;EACL,MAAM;EAEN,QAAQ;AAGN,OAAI,EAFgB,QAAQ,IAAI,2BAA2B,KAEzC;AAChB,YAAQ,KACN,6DACD;AACD,WAAO;;AAGT,OAAI,CAAC,GAAG,WAAW,KAAK,KAAK,QAAQ,KAAK,EAAE,UAAU,UAAU,CAAC,CAC/D,QAAO;AAGT,UAAO;;EAGT,eAAe,QAAQ;AACrB,UAAO,OAAO;AACd,aAAU,KAAK,QAAQ,MAAM,SAAS,WAAW,uBAAuB;AACxE,kBAAe,SAAS,gBAAgB,CACtC,KAAK,KAAK,QAAQ,KAAK,EAAE,UAAU,UAAU,CAC9C;;EAGH,aAAa;AACX,aAAU;;EAGZ,gBAAgB,QAAQ;AACtB,UAAO,QAAQ,IAAI,aAAa;AAEhC,UAAO,QAAQ,GAAG,WAAW,gBAAgB;AAK3C,QAJsB,aAAa,MAAM,WACvC,YAAY,WAAW,OAAO,CAC/B,IAEoB,YAAY,SAAS,OAAO,CAC/C,WAAU;KAEZ;;EAEL"}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@databricks/appkit",
   "type": "module",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
   "packageManager": "pnpm@10.21.0",
@@ -24,7 +24,7 @@
     "./package.json": "./package.json"
   },
   "bin": {
-    "appkit-generate-types": "./bin/generate-types.ts",
+    "appkit-generate-types": "./bin/generate-types.js",
     "appkit-setup": "./bin/setup-claude.js"
   },
   "scripts": {