@databricks/appkit 0.25.1 → 0.26.1

package/dist/appkit/package.js

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

package/dist/cli/commands/codemod/index.js

@@ -0,0 +1,16 @@
+import { onPluginsReadyCommand } from "./on-plugins-ready.js";
+import { Command } from "commander";
+
+//#region src/cli/commands/codemod/index.ts
+/**
+* Parent command for codemod operations.
+* Subcommands:
+*   - on-plugins-ready: Migrate from autoStart/extend/start to onPluginsReady callback
+*/
+const codemodCommand = new Command("codemod").description("Run codemods to migrate to newer AppKit APIs").addCommand(onPluginsReadyCommand).addHelpText("after", `
+Examples:
+  $ appkit codemod on-plugins-ready --write`);
+
+//#endregion
+export { codemodCommand };
+//# sourceMappingURL=index.js.map

package/dist/cli/commands/codemod/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","names":[],"sources":["../../../../src/cli/commands/codemod/index.ts"],"sourcesContent":["import { Command } from \"commander\";\nimport { onPluginsReadyCommand } from \"./on-plugins-ready\";\n\n/**\n * Parent command for codemod operations.\n * Subcommands:\n *   - on-plugins-ready: Migrate from autoStart/extend/start to onPluginsReady callback\n */\nexport const codemodCommand = new Command(\"codemod\")\n  .description(\"Run codemods to migrate to newer AppKit APIs\")\n  .addCommand(onPluginsReadyCommand)\n  .addHelpText(\n    \"after\",\n    `\nExamples:\n  $ appkit codemod on-plugins-ready --write`,\n  );\n"],"mappings":";;;;;;;;;AAQA,MAAa,iBAAiB,IAAI,QAAQ,UAAU,CACjD,YAAY,+CAA+C,CAC3D,WAAW,sBAAsB,CACjC,YACC,SACA;;6CAGD"}

package/dist/cli/commands/codemod/on-plugins-ready.js

@@ -0,0 +1,364 @@
+import fs from "node:fs";
+import path from "node:path";
+import { Command } from "commander";
+import { Lang, parse } from "@ast-grep/napi";
+
+//#region src/cli/commands/codemod/on-plugins-ready.ts
+const SEARCH_DIRS = [
+	"server",
+	"src",
+	"."
+];
+const CANDIDATE_NAMES = ["server.ts", "index.ts"];
+const SKIP_DIRS = new Set([
+	"node_modules",
+	"dist",
+	"build",
+	".git"
+]);
+function findServerEntryFiles(rootDir) {
+	const results = [];
+	for (const dir of SEARCH_DIRS) {
+		const absDir = path.resolve(rootDir, dir);
+		if (!fs.existsSync(absDir)) continue;
+		const files = dir === "." ? CANDIDATE_NAMES.map((n) => path.join(absDir, n)).filter(fs.existsSync) : findTsFiles(absDir);
+		for (const file of files) {
+			const content = fs.readFileSync(file, "utf-8");
+			if (content.includes("createApp") && content.includes("@databricks/appkit")) results.push(file);
+		}
+	}
+	return [...new Set(results)];
+}
+function findTsFiles(dir, files = []) {
+	let entries;
+	try {
+		entries = fs.readdirSync(dir, { withFileTypes: true });
+	} catch {
+		return files;
+	}
+	for (const entry of entries) {
+		const fullPath = path.join(dir, entry.name);
+		if (entry.isDirectory()) {
+			if (SKIP_DIRS.has(entry.name)) continue;
+			findTsFiles(fullPath, files);
+		} else if (entry.isFile() && entry.name.endsWith(".ts")) files.push(fullPath);
+	}
+	return files;
+}
+function isAlreadyMigrated(content) {
+	return parse(Lang.TypeScript, content).root().findAll("createApp({ $$$PROPS })").some((match) => {
+		const text = match.text();
+		return /\bonPluginsReady\s*[(:]/.test(text);
+	});
+}
+/**
+* Find the index of the matching closing delimiter for an opening one.
+* Supports (), {}, and [].
+*/
+function findMatchingClose(content, openIdx) {
+	const open = content[openIdx];
+	const close = {
+		"(": ")",
+		"{": "}",
+		"[": "]"
+	}[open];
+	if (!close) return -1;
+	let depth = 1;
+	let i = openIdx + 1;
+	while (i < content.length && depth > 0) {
+		const ch = content[i];
+		if (ch === open) depth++;
+		else if (ch === close) depth--;
+		if (ch === "\"" || ch === "'" || ch === "`") {
+			i = skipString(content, i);
+			continue;
+		}
+		i++;
+	}
+	return depth === 0 ? i - 1 : -1;
+}
+function skipString(content, startIdx) {
+	const quote = content[startIdx];
+	let i = startIdx + 1;
+	while (i < content.length) {
+		if (content[i] === "\\") {
+			i += 2;
+			continue;
+		}
+		if (content[i] === quote) return i + 1;
+		i++;
+	}
+	return i;
+}
+function stripAutoStartFromServerCalls(content) {
+	return content.replace(/server\(\{([^}]*)\}\)/g, (_fullMatch, propsStr) => {
+		const cleaned = propsStr.replace(/autoStart\s*:\s*(true|false)\s*,?\s*/g, "").replace(/,\s*$/, "").trim();
+		if (!cleaned) return "server()";
+		return `server({ ${cleaned} })`;
+	});
+}
+function migratePatternA(content) {
+	const warnings = [];
+	const createAppIdx = content.indexOf("createApp(");
+	if (createAppIdx === -1) return {
+		migrated: false,
+		content,
+		warnings
+	};
+	const configOpenParen = content.indexOf("(", createAppIdx);
+	const configCloseParen = findMatchingClose(content, configOpenParen);
+	if (configCloseParen === -1) return {
+		migrated: false,
+		content,
+		warnings
+	};
+	const afterCreateApp = content.slice(configCloseParen + 1);
+	if (!afterCreateApp.match(/^\s*\.then\s*\(/)) return {
+		migrated: false,
+		content,
+		warnings
+	};
+	const thenStart = configCloseParen + 1 + afterCreateApp.indexOf(".then");
+	const thenOpenParen = content.indexOf("(", thenStart + 4);
+	const thenCloseParen = findMatchingClose(content, thenOpenParen);
+	if (thenCloseParen === -1) return {
+		migrated: false,
+		content,
+		warnings
+	};
+	const thenRaw = content.slice(thenOpenParen + 1, thenCloseParen);
+	const thenInner = thenRaw.trim();
+	const callbackMatch = thenInner.match(/^(?:async\s+)?\(\s*(\w+)\s*\)\s*=>\s*\{/);
+	if (!callbackMatch) return {
+		migrated: false,
+		content,
+		warnings
+	};
+	const paramName = callbackMatch[1];
+	const bodyOpenBrace = thenOpenParen + 1 + thenRaw.indexOf("{");
+	const bodyCloseBrace = findMatchingClose(content, bodyOpenBrace);
+	if (bodyCloseBrace === -1) return {
+		migrated: false,
+		content,
+		warnings
+	};
+	let callbackBody = content.slice(bodyOpenBrace + 1, bodyCloseBrace).trim();
+	callbackBody = callbackBody.replace(/^\s*(?:await\s+)?\w+\.server\s*\.\s*start\(\s*\)\s*;?\s*$/gm, "").replace(/\n\s*\.start\(\s*\)\s*;?/g, ";").replace(/\.start\(\s*\)/g, "").replace(/\n\s*\n\s*\n/g, "\n\n").trim();
+	if (callbackBody.endsWith(";")) {} else if (!callbackBody.endsWith("}")) callbackBody += ";";
+	const isAsync = /^async\s/.test(thenInner.trim());
+	const catchPatternMatch = content.slice(thenCloseParen + 1).match(/^\s*(?:\)\s*)?\.catch\s*\(/);
+	let catchSuffix;
+	let consumeAfterThen;
+	if (catchPatternMatch) {
+		const catchOpenParen = thenCloseParen + 1 + catchPatternMatch[0].length - 1;
+		const catchCloseParen = findMatchingClose(content, catchOpenParen);
+		if (catchCloseParen !== -1) {
+			catchSuffix = `.catch(${content.slice(catchOpenParen + 1, catchCloseParen).trim()})`;
+			consumeAfterThen = catchCloseParen + 1 - (thenCloseParen + 1);
+		} else {
+			catchSuffix = ".catch(console.error)";
+			consumeAfterThen = 0;
+		}
+	} else {
+		catchSuffix = ".catch(console.error)";
+		consumeAfterThen = 0;
+	}
+	const configStr = content.slice(configOpenParen + 1, configCloseParen);
+	const lastBraceIdx = configStr.lastIndexOf("}");
+	if (lastBraceIdx === -1) return {
+		migrated: false,
+		content,
+		warnings
+	};
+	const beforeLastBrace = configStr.slice(0, lastBraceIdx).trimEnd();
+	const needsComma = beforeLastBrace.endsWith(",") ? "" : ",";
+	const indentedBody = callbackBody.split("\n").map((line) => `    ${line.trimStart()}`).join("\n");
+	const newConfig = `${beforeLastBrace}${`${needsComma}\n  ${isAsync ? "async " : ""}onPluginsReady(${paramName}) {\n${indentedBody}\n  },`}\n}`;
+	let finalEnd = thenCloseParen + 1 + consumeAfterThen;
+	const trailing = content.slice(finalEnd).match(/^\s*\)?\s*;?\s*/);
+	if (trailing) finalEnd += trailing[0].length;
+	return {
+		migrated: true,
+		content: content.slice(0, createAppIdx) + `createApp(${newConfig})${catchSuffix};` + content.slice(finalEnd),
+		warnings
+	};
+}
+function migratePatternB(content) {
+	const warnings = [];
+	const match = content.match(/(?:const|let)\s+(\w+)\s*=\s*await\s+createApp\s*\(/);
+	if (!match) return {
+		migrated: false,
+		content,
+		warnings
+	};
+	const varName = match[1];
+	const matchIdx = content.indexOf(match[0]);
+	const configOpenParen = matchIdx + match[0].length - 1;
+	const configCloseParen = findMatchingClose(content, configOpenParen);
+	if (configCloseParen === -1) return {
+		migrated: false,
+		content,
+		warnings
+	};
+	const semiMatch = content.slice(configCloseParen + 1).match(/^\s*;/);
+	const createAppEnd = configCloseParen + 1 + (semiMatch ? semiMatch[0].length : 0);
+	const afterCreateApp = content.slice(createAppEnd);
+	const varUsagePattern = new RegExp(`\\b${varName}\\.(\\w+)`, "g");
+	const usages = [];
+	for (const usageMatch of afterCreateApp.matchAll(varUsagePattern)) usages.push({
+		plugin: usageMatch[1],
+		index: usageMatch.index
+	});
+	if (usages.filter((u) => u.plugin !== "server").length > 0) {
+		warnings.push(`Found additional usage of '${varName}' handle outside server.extend/start. Please migrate manually.`);
+		return {
+			migrated: false,
+			content,
+			warnings
+		};
+	}
+	const extendPattern = new RegExp(`\\b${varName}\\.server\\.extend\\s*\\(`, "g");
+	const startPattern = new RegExp(`(?:await\\s+)?${varName}\\.server\\.start\\s*\\(\\s*\\)\\s*;`);
+	const extendMatches = [...afterCreateApp.matchAll(extendPattern)];
+	if (extendMatches.length > 1) {
+		warnings.push(`Found ${extendMatches.length} server.extend() calls. Please migrate manually.`);
+		return {
+			migrated: false,
+			content,
+			warnings
+		};
+	}
+	const extendExec = extendMatches[0] ?? null;
+	const startExec = startPattern.exec(afterCreateApp);
+	if (!startExec) return {
+		migrated: false,
+		content,
+		warnings
+	};
+	let extendArg = "";
+	let extendFullStatement = "";
+	if (extendExec) {
+		const extendOpenParen = createAppEnd + extendExec.index + extendExec[0].length - 1;
+		const extendCloseParen = findMatchingClose(content, extendOpenParen);
+		if (extendCloseParen !== -1) {
+			extendArg = content.slice(extendOpenParen + 1, extendCloseParen).trim();
+			const stmtStart = createAppEnd + extendExec.index;
+			let stmtEnd = extendCloseParen + 1;
+			const trailingSemi = content.slice(stmtEnd).match(/^\s*;/);
+			if (trailingSemi) stmtEnd += trailingSemi[0].length;
+			extendFullStatement = content.slice(stmtStart, stmtEnd);
+		}
+	}
+	const startFullStatement = startExec[0];
+	const configStr = content.slice(configOpenParen + 1, configCloseParen);
+	const lastBraceIdx = configStr.lastIndexOf("}");
+	if (lastBraceIdx === -1) return {
+		migrated: false,
+		content,
+		warnings
+	};
+	const beforeLastBrace = configStr.slice(0, lastBraceIdx).trimEnd();
+	const needsComma = beforeLastBrace.endsWith(",") ? "" : ",";
+	let onPluginsReadyProp;
+	if (extendArg) onPluginsReadyProp = `${needsComma}\n  onPluginsReady(${varName}) {\n    ${varName}.server.extend(${extendArg});\n  },`;
+	else onPluginsReadyProp = "";
+	const newCreateApp = `await createApp(${`${beforeLastBrace}${onPluginsReadyProp}\n}`});`;
+	let result = content.slice(0, matchIdx) + newCreateApp;
+	let remaining = afterCreateApp;
+	if (extendFullStatement) remaining = remaining.replace(extendFullStatement, "");
+	remaining = remaining.replace(startFullStatement, "");
+	remaining = remaining.replace(/\n\s*\n\s*\n/g, "\n\n");
+	result += remaining;
+	return {
+		migrated: true,
+		content: result,
+		warnings
+	};
+}
+function migrateFile(filePath) {
+	const original = fs.readFileSync(filePath, "utf-8");
+	if (isAlreadyMigrated(original)) return {
+		migrated: false,
+		content: original,
+		warnings: ["Already migrated -- no changes needed."]
+	};
+	const content = stripAutoStartFromServerCalls(original);
+	const allWarnings = [];
+	const patternA = migratePatternA(content);
+	if (patternA.migrated) {
+		allWarnings.push(...patternA.warnings);
+		return {
+			migrated: true,
+			content: patternA.content,
+			warnings: allWarnings
+		};
+	}
+	allWarnings.push(...patternA.warnings);
+	const patternB = migratePatternB(content);
+	if (patternB.migrated) {
+		allWarnings.push(...patternB.warnings);
+		return {
+			migrated: true,
+			content: patternB.content,
+			warnings: allWarnings
+		};
+	}
+	allWarnings.push(...patternB.warnings);
+	if (content !== original) return {
+		migrated: true,
+		content,
+		warnings: allWarnings
+	};
+	return {
+		migrated: false,
+		content: original,
+		warnings: allWarnings
+	};
+}
+function runCodemod(options) {
+	const rootDir = process.cwd();
+	const write = options.write ?? false;
+	let files;
+	if (options.path) {
+		const absPath = path.resolve(rootDir, options.path);
+		if (!fs.existsSync(absPath)) {
+			console.error(`File not found: ${absPath}`);
+			process.exit(1);
+		}
+		files = [absPath];
+	} else files = findServerEntryFiles(rootDir);
+	if (files.length === 0) {
+		console.log("No files found importing createApp from @databricks/appkit.");
+		console.log("Use --path to specify a file explicitly.");
+		process.exit(0);
+	}
+	let hasChanges = false;
+	for (const file of files) {
+		const relPath = path.relative(rootDir, file);
+		const result = migrateFile(file);
+		for (const warning of result.warnings) console.log(`  ${relPath}: ${warning}`);
+		if (!result.migrated) {
+			if (result.warnings.length === 0) console.log(`  ${relPath}: No migration needed.`);
+			continue;
+		}
+		hasChanges = true;
+		if (write) {
+			fs.writeFileSync(file, result.content, "utf-8");
+			console.log(`  ${relPath}: Migrated successfully.`);
+		} else {
+			console.log(`\n--- ${relPath} (dry run) ---`);
+			console.log(result.content);
+			console.log("---");
+		}
+	}
+	if (hasChanges && !write) console.log("\nDry run complete. Run with --write to apply changes.");
+}
+const onPluginsReadyCommand = new Command("on-plugins-ready").description("Migrate createApp usage from autoStart/extend/start pattern to onPluginsReady callback").option("--path <file>", "Path to the server entry file to migrate").option("--write", "Apply changes (default: dry-run)", false).addHelpText("after", `
+Examples:
+  $ appkit codemod on-plugins-ready                    # dry-run, auto-detect files
+  $ appkit codemod on-plugins-ready --write            # apply changes
+  $ appkit codemod on-plugins-ready --path server.ts   # migrate a specific file`).action(runCodemod);
+
+//#endregion
+export { onPluginsReadyCommand };
+//# sourceMappingURL=on-plugins-ready.js.map

package/dist/cli/commands/codemod/on-plugins-ready.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"on-plugins-ready.js","names":[],"sources":["../../../../src/cli/commands/codemod/on-plugins-ready.ts"],"sourcesContent":["import fs from \"node:fs\";\nimport path from \"node:path\";\nimport { Lang, parse } from \"@ast-grep/napi\";\nimport { Command } from \"commander\";\n\nconst SEARCH_DIRS = [\"server\", \"src\", \".\"];\nconst CANDIDATE_NAMES = [\"server.ts\", \"index.ts\"];\nconst SKIP_DIRS = new Set([\"node_modules\", \"dist\", \"build\", \".git\"]);\n\nfunction findServerEntryFiles(rootDir: string): string[] {\n  const results: string[] = [];\n\n  for (const dir of SEARCH_DIRS) {\n    const absDir = path.resolve(rootDir, dir);\n    if (!fs.existsSync(absDir)) continue;\n\n    const files =\n      dir === \".\"\n        ? CANDIDATE_NAMES.map((n) => path.join(absDir, n)).filter(fs.existsSync)\n        : findTsFiles(absDir);\n\n    for (const file of files) {\n      const content = fs.readFileSync(file, \"utf-8\");\n      if (\n        content.includes(\"createApp\") &&\n        content.includes(\"@databricks/appkit\")\n      ) {\n        results.push(file);\n      }\n    }\n  }\n\n  return [...new Set(results)];\n}\n\nfunction findTsFiles(dir: string, files: string[] = []): string[] {\n  let entries: fs.Dirent[];\n  try {\n    entries = fs.readdirSync(dir, { withFileTypes: true });\n  } catch {\n    return files;\n  }\n\n  for (const entry of entries) {\n    const fullPath = path.join(dir, entry.name);\n    if (entry.isDirectory()) {\n      if (SKIP_DIRS.has(entry.name)) continue;\n      findTsFiles(fullPath, files);\n    } else if (entry.isFile() && entry.name.endsWith(\".ts\")) {\n      files.push(fullPath);\n    }\n  }\n\n  return files;\n}\n\nfunction isAlreadyMigrated(content: string): boolean {\n  const ast = parse(Lang.TypeScript, content);\n  const root = ast.root();\n  return root.findAll(\"createApp({ $$$PROPS })\").some((match) => {\n    const text = match.text();\n    return /\\bonPluginsReady\\s*[(:]/.test(text);\n  });\n}\n\n/**\n * Find the index of the matching closing delimiter for an opening one.\n * Supports (), {}, and [].\n */\nfunction findMatchingClose(content: string, openIdx: number): number {\n  const open = content[openIdx];\n  const closeMap: Record<string, string> = {\n    \"(\": \")\",\n    \"{\": \"}\",\n    \"[\": \"]\",\n  };\n  const close = closeMap[open];\n  if (!close) return -1;\n\n  let depth = 1;\n  let i = openIdx + 1;\n  while (i < content.length && depth > 0) {\n    const ch = content[i];\n    if (ch === open) depth++;\n    else if (ch === close) depth--;\n\n    // skip string literals\n    if (ch === '\"' || ch === \"'\" || ch === \"`\") {\n      i = skipString(content, i);\n      continue;\n    }\n    i++;\n  }\n  return depth === 0 ? i - 1 : -1;\n}\n\nfunction skipString(content: string, startIdx: number): number {\n  const quote = content[startIdx];\n  let i = startIdx + 1;\n  while (i < content.length) {\n    if (content[i] === \"\\\\\") {\n      i += 2;\n      continue;\n    }\n    if (content[i] === quote) return i + 1;\n    i++;\n  }\n  return i;\n}\n\nfunction stripAutoStartFromServerCalls(content: string): string {\n  return content.replace(\n    /server\\(\\{([^}]*)\\}\\)/g,\n    (_fullMatch, propsStr: string) => {\n      const cleaned = propsStr\n        .replace(/autoStart\\s*:\\s*(true|false)\\s*,?\\s*/g, \"\")\n        .replace(/,\\s*$/, \"\")\n        .trim();\n      if (!cleaned) return \"server()\";\n      return `server({ ${cleaned} })`;\n    },\n  );\n}\n\ninterface MigrationResult {\n  migrated: boolean;\n  content: string;\n  warnings: string[];\n}\n\nfunction migratePatternA(content: string): MigrationResult {\n  const warnings: string[] = [];\n\n  // Find createApp(...).then(\n  const createAppIdx = content.indexOf(\"createApp(\");\n  if (createAppIdx === -1) return { migrated: false, content, warnings };\n\n  // Find the opening paren of createApp(\n  const configOpenParen = content.indexOf(\"(\", createAppIdx);\n  const configCloseParen = findMatchingClose(content, configOpenParen);\n  if (configCloseParen === -1) return { migrated: false, content, warnings };\n\n  // Check for .then( after the closing paren\n  const afterCreateApp = content.slice(configCloseParen + 1);\n  const thenMatch = afterCreateApp.match(/^\\s*\\.then\\s*\\(/);\n  if (!thenMatch) return { migrated: false, content, warnings };\n\n  const thenStart = configCloseParen + 1 + afterCreateApp.indexOf(\".then\");\n  const thenOpenParen = content.indexOf(\"(\", thenStart + 4);\n  const thenCloseParen = findMatchingClose(content, thenOpenParen);\n  if (thenCloseParen === -1) return { migrated: false, content, warnings };\n\n  // Extract the callback inside .then(...)\n  const thenRaw = content.slice(thenOpenParen + 1, thenCloseParen);\n  const thenInner = thenRaw.trim();\n\n  // Parse callback: (param) => { body } or async (param) => { body }\n  const callbackMatch = thenInner.match(\n    /^(?:async\\s+)?\\(\\s*(\\w+)\\s*\\)\\s*=>\\s*\\{/,\n  );\n  if (!callbackMatch) return { migrated: false, content, warnings };\n\n  const paramName = callbackMatch[1];\n  const bodyOpenBrace = thenOpenParen + 1 + thenRaw.indexOf(\"{\");\n  const bodyCloseBrace = findMatchingClose(content, bodyOpenBrace);\n  if (bodyCloseBrace === -1) return { migrated: false, content, warnings };\n\n  let callbackBody = content.slice(bodyOpenBrace + 1, bodyCloseBrace).trim();\n\n  // Remove entire statements that are just .start() calls (e.g. `await appkit.server.start();`)\n  callbackBody = callbackBody\n    .replace(/^\\s*(?:await\\s+)?\\w+\\.server\\s*\\.\\s*start\\(\\s*\\)\\s*;?\\s*$/gm, \"\")\n    .replace(/\\n\\s*\\.start\\(\\s*\\)\\s*;?/g, \";\")\n    .replace(/\\.start\\(\\s*\\)/g, \"\")\n    .replace(/\\n\\s*\\n\\s*\\n/g, \"\\n\\n\")\n    .trim();\n\n  // Clean up trailing semicolons\n  if (callbackBody.endsWith(\";\")) {\n    // fine\n  } else if (!callbackBody.endsWith(\"}\")) {\n    callbackBody += \";\";\n  }\n\n  // Detect if the callback was async\n  const isAsync = /^async\\s/.test(thenInner.trim());\n\n  // Check for .catch() after .then(...) using brace-aware parsing\n  const afterThenClose = content.slice(thenCloseParen + 1);\n  const catchPatternMatch = afterThenClose.match(/^\\s*(?:\\)\\s*)?\\.catch\\s*\\(/);\n\n  let catchSuffix: string;\n  let consumeAfterThen: number;\n\n  if (catchPatternMatch) {\n    const catchOpenParen = thenCloseParen + 1 + catchPatternMatch[0].length - 1;\n    const catchCloseParen = findMatchingClose(content, catchOpenParen);\n    if (catchCloseParen !== -1) {\n      const catchArg = content\n        .slice(catchOpenParen + 1, catchCloseParen)\n        .trim();\n      catchSuffix = `.catch(${catchArg})`;\n      consumeAfterThen = catchCloseParen + 1 - (thenCloseParen + 1);\n    } else {\n      catchSuffix = \".catch(console.error)\";\n      consumeAfterThen = 0;\n    }\n  } else {\n    catchSuffix = \".catch(console.error)\";\n    consumeAfterThen = 0;\n  }\n\n  // Build the onPluginsReady property\n  const configStr = content.slice(configOpenParen + 1, configCloseParen);\n  const lastBraceIdx = configStr.lastIndexOf(\"}\");\n  if (lastBraceIdx === -1) return { migrated: false, content, warnings };\n\n  const beforeLastBrace = configStr.slice(0, lastBraceIdx).trimEnd();\n  const needsComma = beforeLastBrace.endsWith(\",\") ? \"\" : \",\";\n\n  // Indent the body properly\n  const bodyLines = callbackBody.split(\"\\n\");\n  const indentedBody = bodyLines\n    .map((line) => `    ${line.trimStart()}`)\n    .join(\"\\n\");\n\n  const asyncPrefix = isAsync ? \"async \" : \"\";\n  const onPluginsReadyProp = `${needsComma}\\n  ${asyncPrefix}onPluginsReady(${paramName}) {\\n${indentedBody}\\n  },`;\n  const newConfig = `${beforeLastBrace}${onPluginsReadyProp}\\n}`;\n\n  // Build the replacement\n  const endIdx = thenCloseParen + 1 + consumeAfterThen;\n  // Consume trailing ) ; and whitespace\n  let finalEnd = endIdx;\n  const trailing = content.slice(finalEnd).match(/^\\s*\\)?\\s*;?\\s*/);\n  if (trailing) finalEnd += trailing[0].length;\n\n  const newContent =\n    content.slice(0, createAppIdx) +\n    `createApp(${newConfig})${catchSuffix};` +\n    content.slice(finalEnd);\n\n  return { migrated: true, content: newContent, warnings };\n}\n\nfunction migratePatternB(content: string): MigrationResult {\n  const warnings: string[] = [];\n\n  // Match: const/let varName = await createApp({...});\n  const awaitPattern = /(?:const|let)\\s+(\\w+)\\s*=\\s*await\\s+createApp\\s*\\(/;\n\n  const match = content.match(awaitPattern);\n  if (!match) return { migrated: false, content, warnings };\n\n  const varName = match[1];\n  const matchIdx = content.indexOf(match[0]);\n\n  // Find the createApp(...) closing paren\n  const configOpenParen = matchIdx + match[0].length - 1;\n  const configCloseParen = findMatchingClose(content, configOpenParen);\n  if (configCloseParen === -1) return { migrated: false, content, warnings };\n\n  // Find the semicolon after the createApp call\n  const afterCall = content.slice(configCloseParen + 1);\n  const semiMatch = afterCall.match(/^\\s*;/);\n  const createAppEnd =\n    configCloseParen + 1 + (semiMatch ? semiMatch[0].length : 0);\n\n  // Find all uses of varName after the createApp call\n  const afterCreateApp = content.slice(createAppEnd);\n  const varUsagePattern = new RegExp(`\\\\b${varName}\\\\.(\\\\w+)`, \"g\");\n\n  const usages: { plugin: string; index: number }[] = [];\n  for (const usageMatch of afterCreateApp.matchAll(varUsagePattern)) {\n    usages.push({ plugin: usageMatch[1], index: usageMatch.index });\n  }\n\n  // Check for non-server usage\n  const nonServerUsage = usages.filter((u) => u.plugin !== \"server\");\n  if (nonServerUsage.length > 0) {\n    warnings.push(\n      `Found additional usage of '${varName}' handle outside server.extend/start. Please migrate manually.`,\n    );\n    return { migrated: false, content, warnings };\n  }\n\n  // Find the extend call(s) and start call in the after-createApp region\n  const extendPattern = new RegExp(\n    `\\\\b${varName}\\\\.server\\\\.extend\\\\s*\\\\(`,\n    \"g\",\n  );\n  const startPattern = new RegExp(\n    `(?:await\\\\s+)?${varName}\\\\.server\\\\.start\\\\s*\\\\(\\\\s*\\\\)\\\\s*;`,\n  );\n\n  const extendMatches = [...afterCreateApp.matchAll(extendPattern)];\n  if (extendMatches.length > 1) {\n    warnings.push(\n      `Found ${extendMatches.length} server.extend() calls. Please migrate manually.`,\n    );\n    return { migrated: false, content, warnings };\n  }\n\n  const extendExec = extendMatches[0] ?? null;\n  const startExec = startPattern.exec(afterCreateApp);\n\n  if (!startExec) return { migrated: false, content, warnings };\n\n  // Extract the extend call's argument\n  let extendArg = \"\";\n  let extendFullStatement = \"\";\n  if (extendExec) {\n    const extendOpenParen =\n      createAppEnd + extendExec.index + extendExec[0].length - 1;\n    const extendCloseParen = findMatchingClose(content, extendOpenParen);\n    if (extendCloseParen !== -1) {\n      extendArg = content.slice(extendOpenParen + 1, extendCloseParen).trim();\n      // Find the full statement including trailing semicolon\n      const stmtStart = createAppEnd + extendExec.index;\n      let stmtEnd = extendCloseParen + 1;\n      const afterExtend = content.slice(stmtEnd);\n      const trailingSemi = afterExtend.match(/^\\s*;/);\n      if (trailingSemi) stmtEnd += trailingSemi[0].length;\n      extendFullStatement = content.slice(stmtStart, stmtEnd);\n    }\n  }\n\n  const startFullStatement = startExec[0];\n\n  // Build the onPluginsReady callback\n  const configStr = content.slice(configOpenParen + 1, configCloseParen);\n  const lastBraceIdx = configStr.lastIndexOf(\"}\");\n  if (lastBraceIdx === -1) return { migrated: false, content, warnings };\n\n  const beforeLastBrace = configStr.slice(0, lastBraceIdx).trimEnd();\n  const needsComma = beforeLastBrace.endsWith(\",\") ? \"\" : \",\";\n\n  let onPluginsReadyProp: string;\n  if (extendArg) {\n    onPluginsReadyProp =\n      `${needsComma}\\n  onPluginsReady(${varName}) {\\n` +\n      `    ${varName}.server.extend(${extendArg});\\n` +\n      \"  },\";\n  } else {\n    onPluginsReadyProp = \"\";\n  }\n\n  const newConfig = `${beforeLastBrace}${onPluginsReadyProp}\\n}`;\n  const newCreateApp = `await createApp(${newConfig});`;\n\n  // Replace: remove const declaration, replace with plain await, remove extend + start\n  let result = content.slice(0, matchIdx) + newCreateApp;\n  let remaining = afterCreateApp;\n\n  if (extendFullStatement) {\n    remaining = remaining.replace(extendFullStatement, \"\");\n  }\n  remaining = remaining.replace(startFullStatement, \"\");\n\n  // Clean up consecutive blank lines\n  remaining = remaining.replace(/\\n\\s*\\n\\s*\\n/g, \"\\n\\n\");\n\n  result += remaining;\n\n  return { migrated: true, content: result, warnings };\n}\n\nexport function migrateFile(filePath: string): MigrationResult {\n  const original = fs.readFileSync(filePath, \"utf-8\");\n\n  if (isAlreadyMigrated(original)) {\n    return {\n      migrated: false,\n      content: original,\n      warnings: [\"Already migrated -- no changes needed.\"],\n    };\n  }\n\n  const content = stripAutoStartFromServerCalls(original);\n  const allWarnings: string[] = [];\n\n  // Try Pattern A first\n  const patternA = migratePatternA(content);\n  if (patternA.migrated) {\n    allWarnings.push(...patternA.warnings);\n    return {\n      migrated: true,\n      content: patternA.content,\n      warnings: allWarnings,\n    };\n  }\n  allWarnings.push(...patternA.warnings);\n\n  // Try Pattern B\n  const patternB = migratePatternB(content);\n  if (patternB.migrated) {\n    allWarnings.push(...patternB.warnings);\n    return {\n      migrated: true,\n      content: patternB.content,\n      warnings: allWarnings,\n    };\n  }\n  allWarnings.push(...patternB.warnings);\n\n  // Check if autoStart was stripped (content changed but no pattern matched)\n  if (content !== original) {\n    return { migrated: true, content, warnings: allWarnings };\n  }\n\n  return { migrated: false, content: original, warnings: allWarnings };\n}\n\nfunction runCodemod(options: { path?: string; write?: boolean }) {\n  const rootDir = process.cwd();\n  const write = options.write ?? false;\n\n  let files: string[];\n  if (options.path) {\n    const absPath = path.resolve(rootDir, options.path);\n    if (!fs.existsSync(absPath)) {\n      console.error(`File not found: ${absPath}`);\n      process.exit(1);\n    }\n    files = [absPath];\n  } else {\n    files = findServerEntryFiles(rootDir);\n  }\n\n  if (files.length === 0) {\n    console.log(\"No files found importing createApp from @databricks/appkit.\");\n    console.log(\"Use --path to specify a file explicitly.\");\n    process.exit(0);\n  }\n\n  let hasChanges = false;\n\n  for (const file of files) {\n    const relPath = path.relative(rootDir, file);\n    const result = migrateFile(file);\n\n    for (const warning of result.warnings) {\n      console.log(`  ${relPath}: ${warning}`);\n    }\n\n    if (!result.migrated) {\n      if (result.warnings.length === 0) {\n        console.log(`  ${relPath}: No migration needed.`);\n      }\n      continue;\n    }\n\n    hasChanges = true;\n\n    if (write) {\n      fs.writeFileSync(file, result.content, \"utf-8\");\n      console.log(`  ${relPath}: Migrated successfully.`);\n    } else {\n      console.log(`\\n--- ${relPath} (dry run) ---`);\n      console.log(result.content);\n      console.log(\"---\");\n    }\n  }\n\n  if (hasChanges && !write) {\n    console.log(\"\\nDry run complete. Run with --write to apply changes.\");\n  }\n}\n\nexport const onPluginsReadyCommand = new Command(\"on-plugins-ready\")\n  .description(\n    \"Migrate createApp usage from autoStart/extend/start pattern to onPluginsReady callback\",\n  )\n  .option(\"--path <file>\", \"Path to the server entry file to migrate\")\n  .option(\"--write\", \"Apply changes (default: dry-run)\", false)\n  .addHelpText(\n    \"after\",\n    `\nExamples:\n  $ appkit codemod on-plugins-ready                    # dry-run, auto-detect files\n  $ appkit codemod on-plugins-ready --write            # apply changes\n  $ appkit codemod on-plugins-ready --path server.ts   # migrate a specific file`,\n  )\n  .action(runCodemod);\n"],"mappings":";;;;;;AAKA,MAAM,cAAc;CAAC;CAAU;CAAO;CAAI;AAC1C,MAAM,kBAAkB,CAAC,aAAa,WAAW;AACjD,MAAM,YAAY,IAAI,IAAI;CAAC;CAAgB;CAAQ;CAAS;CAAO,CAAC;AAEpE,SAAS,qBAAqB,SAA2B;CACvD,MAAM,UAAoB,EAAE;AAE5B,MAAK,MAAM,OAAO,aAAa;EAC7B,MAAM,SAAS,KAAK,QAAQ,SAAS,IAAI;AACzC,MAAI,CAAC,GAAG,WAAW,OAAO,CAAE;EAE5B,MAAM,QACJ,QAAQ,MACJ,gBAAgB,KAAK,MAAM,KAAK,KAAK,QAAQ,EAAE,CAAC,CAAC,OAAO,GAAG,WAAW,GACtE,YAAY,OAAO;AAEzB,OAAK,MAAM,QAAQ,OAAO;GACxB,MAAM,UAAU,GAAG,aAAa,MAAM,QAAQ;AAC9C,OACE,QAAQ,SAAS,YAAY,IAC7B,QAAQ,SAAS,qBAAqB,CAEtC,SAAQ,KAAK,KAAK;;;AAKxB,QAAO,CAAC,GAAG,IAAI,IAAI,QAAQ,CAAC;;AAG9B,SAAS,YAAY,KAAa,QAAkB,EAAE,EAAY;CAChE,IAAI;AACJ,KAAI;AACF,YAAU,GAAG,YAAY,KAAK,EAAE,eAAe,MAAM,CAAC;SAChD;AACN,SAAO;;AAGT,MAAK,MAAM,SAAS,SAAS;EAC3B,MAAM,WAAW,KAAK,KAAK,KAAK,MAAM,KAAK;AAC3C,MAAI,MAAM,aAAa,EAAE;AACvB,OAAI,UAAU,IAAI,MAAM,KAAK,CAAE;AAC/B,eAAY,UAAU,MAAM;aACnB,MAAM,QAAQ,IAAI,MAAM,KAAK,SAAS,MAAM,CACrD,OAAM,KAAK,SAAS;;AAIxB,QAAO;;AAGT,SAAS,kBAAkB,SAA0B;AAGnD,QAFY,MAAM,KAAK,YAAY,QAAQ,CAC1B,MAAM,CACX,QAAQ,0BAA0B,CAAC,MAAM,UAAU;EAC7D,MAAM,OAAO,MAAM,MAAM;AACzB,SAAO,0BAA0B,KAAK,KAAK;GAC3C;;;;;;AAOJ,SAAS,kBAAkB,SAAiB,SAAyB;CACnE,MAAM,OAAO,QAAQ;CAMrB,MAAM,QALmC;EACvC,KAAK;EACL,KAAK;EACL,KAAK;EACN,CACsB;AACvB,KAAI,CAAC,MAAO,QAAO;CAEnB,IAAI,QAAQ;CACZ,IAAI,IAAI,UAAU;AAClB,QAAO,IAAI,QAAQ,UAAU,QAAQ,GAAG;EACtC,MAAM,KAAK,QAAQ;AACnB,MAAI,OAAO,KAAM;WACR,OAAO,MAAO;AAGvB,MAAI,OAAO,QAAO,OAAO,OAAO,OAAO,KAAK;AAC1C,OAAI,WAAW,SAAS,EAAE;AAC1B;;AAEF;;AAEF,QAAO,UAAU,IAAI,IAAI,IAAI;;AAG/B,SAAS,WAAW,SAAiB,UAA0B;CAC7D,MAAM,QAAQ,QAAQ;CACtB,IAAI,IAAI,WAAW;AACnB,QAAO,IAAI,QAAQ,QAAQ;AACzB,MAAI,QAAQ,OAAO,MAAM;AACvB,QAAK;AACL;;AAEF,MAAI,QAAQ,OAAO,MAAO,QAAO,IAAI;AACrC;;AAEF,QAAO;;AAGT,SAAS,8BAA8B,SAAyB;AAC9D,QAAO,QAAQ,QACb,2BACC,YAAY,aAAqB;EAChC,MAAM,UAAU,SACb,QAAQ,yCAAyC,GAAG,CACpD,QAAQ,SAAS,GAAG,CACpB,MAAM;AACT,MAAI,CAAC,QAAS,QAAO;AACrB,SAAO,YAAY,QAAQ;GAE9B;;AASH,SAAS,gBAAgB,SAAkC;CACzD,MAAM,WAAqB,EAAE;CAG7B,MAAM,eAAe,QAAQ,QAAQ,aAAa;AAClD,KAAI,iBAAiB,GAAI,QAAO;EAAE,UAAU;EAAO;EAAS;EAAU;CAGtE,MAAM,kBAAkB,QAAQ,QAAQ,KAAK,aAAa;CAC1D,MAAM,mBAAmB,kBAAkB,SAAS,gBAAgB;AACpE,KAAI,qBAAqB,GAAI,QAAO;EAAE,UAAU;EAAO;EAAS;EAAU;CAG1E,MAAM,iBAAiB,QAAQ,MAAM,mBAAmB,EAAE;AAE1D,KAAI,CADc,eAAe,MAAM,kBAAkB,CACzC,QAAO;EAAE,UAAU;EAAO;EAAS;EAAU;CAE7D,MAAM,YAAY,mBAAmB,IAAI,eAAe,QAAQ,QAAQ;CACxE,MAAM,gBAAgB,QAAQ,QAAQ,KAAK,YAAY,EAAE;CACzD,MAAM,iBAAiB,kBAAkB,SAAS,cAAc;AAChE,KAAI,mBAAmB,GAAI,QAAO;EAAE,UAAU;EAAO;EAAS;EAAU;CAGxE,MAAM,UAAU,QAAQ,MAAM,gBAAgB,GAAG,eAAe;CAChE,MAAM,YAAY,QAAQ,MAAM;CAGhC,MAAM,gBAAgB,UAAU,MAC9B,0CACD;AACD,KAAI,CAAC,cAAe,QAAO;EAAE,UAAU;EAAO;EAAS;EAAU;CAEjE,MAAM,YAAY,cAAc;CAChC,MAAM,gBAAgB,gBAAgB,IAAI,QAAQ,QAAQ,IAAI;CAC9D,MAAM,iBAAiB,kBAAkB,SAAS,cAAc;AAChE,KAAI,mBAAmB,GAAI,QAAO;EAAE,UAAU;EAAO;EAAS;EAAU;CAExE,IAAI,eAAe,QAAQ,MAAM,gBAAgB,GAAG,eAAe,CAAC,MAAM;AAG1E,gBAAe,aACZ,QAAQ,+DAA+D,GAAG,CAC1E,QAAQ,6BAA6B,IAAI,CACzC,QAAQ,mBAAmB,GAAG,CAC9B,QAAQ,iBAAiB,OAAO,CAChC,MAAM;AAGT,KAAI,aAAa,SAAS,IAAI,EAAE,YAErB,CAAC,aAAa,SAAS,IAAI,CACpC,iBAAgB;CAIlB,MAAM,UAAU,WAAW,KAAK,UAAU,MAAM,CAAC;CAIjD,MAAM,oBADiB,QAAQ,MAAM,iBAAiB,EAAE,CACf,MAAM,6BAA6B;CAE5E,IAAI;CACJ,IAAI;AAEJ,KAAI,mBAAmB;EACrB,MAAM,iBAAiB,iBAAiB,IAAI,kBAAkB,GAAG,SAAS;EAC1E,MAAM,kBAAkB,kBAAkB,SAAS,eAAe;AAClE,MAAI,oBAAoB,IAAI;AAI1B,iBAAc,UAHG,QACd,MAAM,iBAAiB,GAAG,gBAAgB,CAC1C,MAAM,CACwB;AACjC,sBAAmB,kBAAkB,KAAK,iBAAiB;SACtD;AACL,iBAAc;AACd,sBAAmB;;QAEhB;AACL,gBAAc;AACd,qBAAmB;;CAIrB,MAAM,YAAY,QAAQ,MAAM,kBAAkB,GAAG,iBAAiB;CACtE,MAAM,eAAe,UAAU,YAAY,IAAI;AAC/C,KAAI,iBAAiB,GAAI,QAAO;EAAE,UAAU;EAAO;EAAS;EAAU;CAEtE,MAAM,kBAAkB,UAAU,MAAM,GAAG,aAAa,CAAC,SAAS;CAClE,MAAM,aAAa,gBAAgB,SAAS,IAAI,GAAG,KAAK;CAIxD,MAAM,eADY,aAAa,MAAM,KAAK,CAEvC,KAAK,SAAS,OAAO,KAAK,WAAW,GAAG,CACxC,KAAK,KAAK;CAIb,MAAM,YAAY,GAAG,kBADM,GAAG,WAAW,MADrB,UAAU,WAAW,GACkB,iBAAiB,UAAU,OAAO,aAAa,QAChD;CAK1D,IAAI,WAFW,iBAAiB,IAAI;CAGpC,MAAM,WAAW,QAAQ,MAAM,SAAS,CAAC,MAAM,kBAAkB;AACjE,KAAI,SAAU,aAAY,SAAS,GAAG;AAOtC,QAAO;EAAE,UAAU;EAAM,SAJvB,QAAQ,MAAM,GAAG,aAAa,GAC9B,aAAa,UAAU,GAAG,YAAY,KACtC,QAAQ,MAAM,SAAS;EAEqB;EAAU;;AAG1D,SAAS,gBAAgB,SAAkC;CACzD,MAAM,WAAqB,EAAE;CAK7B,MAAM,QAAQ,QAAQ,MAFD,qDAEoB;AACzC,KAAI,CAAC,MAAO,QAAO;EAAE,UAAU;EAAO;EAAS;EAAU;CAEzD,MAAM,UAAU,MAAM;CACtB,MAAM,WAAW,QAAQ,QAAQ,MAAM,GAAG;CAG1C,MAAM,kBAAkB,WAAW,MAAM,GAAG,SAAS;CACrD,MAAM,mBAAmB,kBAAkB,SAAS,gBAAgB;AACpE,KAAI,qBAAqB,GAAI,QAAO;EAAE,UAAU;EAAO;EAAS;EAAU;CAI1E,MAAM,YADY,QAAQ,MAAM,mBAAmB,EAAE,CACzB,MAAM,QAAQ;CAC1C,MAAM,eACJ,mBAAmB,KAAK,YAAY,UAAU,GAAG,SAAS;CAG5D,MAAM,iBAAiB,QAAQ,MAAM,aAAa;CAClD,MAAM,kBAAkB,IAAI,OAAO,MAAM,QAAQ,YAAY,IAAI;CAEjE,MAAM,SAA8C,EAAE;AACtD,MAAK,MAAM,cAAc,eAAe,SAAS,gBAAgB,CAC/D,QAAO,KAAK;EAAE,QAAQ,WAAW;EAAI,OAAO,WAAW;EAAO,CAAC;AAKjE,KADuB,OAAO,QAAQ,MAAM,EAAE,WAAW,SAAS,CAC/C,SAAS,GAAG;AAC7B,WAAS,KACP,8BAA8B,QAAQ,gEACvC;AACD,SAAO;GAAE,UAAU;GAAO;GAAS;GAAU;;CAI/C,MAAM,gBAAgB,IAAI,OACxB,MAAM,QAAQ,4BACd,IACD;CACD,MAAM,eAAe,IAAI,OACvB,iBAAiB,QAAQ,sCAC1B;CAED,MAAM,gBAAgB,CAAC,GAAG,eAAe,SAAS,cAAc,CAAC;AACjE,KAAI,cAAc,SAAS,GAAG;AAC5B,WAAS,KACP,SAAS,cAAc,OAAO,kDAC/B;AACD,SAAO;GAAE,UAAU;GAAO;GAAS;GAAU;;CAG/C,MAAM,aAAa,cAAc,MAAM;CACvC,MAAM,YAAY,aAAa,KAAK,eAAe;AAEnD,KAAI,CAAC,UAAW,QAAO;EAAE,UAAU;EAAO;EAAS;EAAU;CAG7D,IAAI,YAAY;CAChB,IAAI,sBAAsB;AAC1B,KAAI,YAAY;EACd,MAAM,kBACJ,eAAe,WAAW,QAAQ,WAAW,GAAG,SAAS;EAC3D,MAAM,mBAAmB,kBAAkB,SAAS,gBAAgB;AACpE,MAAI,qBAAqB,IAAI;AAC3B,eAAY,QAAQ,MAAM,kBAAkB,GAAG,iBAAiB,CAAC,MAAM;GAEvE,MAAM,YAAY,eAAe,WAAW;GAC5C,IAAI,UAAU,mBAAmB;GAEjC,MAAM,eADc,QAAQ,MAAM,QAAQ,CACT,MAAM,QAAQ;AAC/C,OAAI,aAAc,YAAW,aAAa,GAAG;AAC7C,yBAAsB,QAAQ,MAAM,WAAW,QAAQ;;;CAI3D,MAAM,qBAAqB,UAAU;CAGrC,MAAM,YAAY,QAAQ,MAAM,kBAAkB,GAAG,iBAAiB;CACtE,MAAM,eAAe,UAAU,YAAY,IAAI;AAC/C,KAAI,iBAAiB,GAAI,QAAO;EAAE,UAAU;EAAO;EAAS;EAAU;CAEtE,MAAM,kBAAkB,UAAU,MAAM,GAAG,aAAa,CAAC,SAAS;CAClE,MAAM,aAAa,gBAAgB,SAAS,IAAI,GAAG,KAAK;CAExD,IAAI;AACJ,KAAI,UACF,sBACE,GAAG,WAAW,qBAAqB,QAAQ,WACpC,QAAQ,iBAAiB,UAAU;KAG5C,sBAAqB;CAIvB,MAAM,eAAe,mBADH,GAAG,kBAAkB,mBAAmB,KACR;CAGlD,IAAI,SAAS,QAAQ,MAAM,GAAG,SAAS,GAAG;CAC1C,IAAI,YAAY;AAEhB,KAAI,oBACF,aAAY,UAAU,QAAQ,qBAAqB,GAAG;AAExD,aAAY,UAAU,QAAQ,oBAAoB,GAAG;AAGrD,aAAY,UAAU,QAAQ,iBAAiB,OAAO;AAEtD,WAAU;AAEV,QAAO;EAAE,UAAU;EAAM,SAAS;EAAQ;EAAU;;AAGtD,SAAgB,YAAY,UAAmC;CAC7D,MAAM,WAAW,GAAG,aAAa,UAAU,QAAQ;AAEnD,KAAI,kBAAkB,SAAS,CAC7B,QAAO;EACL,UAAU;EACV,SAAS;EACT,UAAU,CAAC,yCAAyC;EACrD;CAGH,MAAM,UAAU,8BAA8B,SAAS;CACvD,MAAM,cAAwB,EAAE;CAGhC,MAAM,WAAW,gBAAgB,QAAQ;AACzC,KAAI,SAAS,UAAU;AACrB,cAAY,KAAK,GAAG,SAAS,SAAS;AACtC,SAAO;GACL,UAAU;GACV,SAAS,SAAS;GAClB,UAAU;GACX;;AAEH,aAAY,KAAK,GAAG,SAAS,SAAS;CAGtC,MAAM,WAAW,gBAAgB,QAAQ;AACzC,KAAI,SAAS,UAAU;AACrB,cAAY,KAAK,GAAG,SAAS,SAAS;AACtC,SAAO;GACL,UAAU;GACV,SAAS,SAAS;GAClB,UAAU;GACX;;AAEH,aAAY,KAAK,GAAG,SAAS,SAAS;AAGtC,KAAI,YAAY,SACd,QAAO;EAAE,UAAU;EAAM;EAAS,UAAU;EAAa;AAG3D,QAAO;EAAE,UAAU;EAAO,SAAS;EAAU,UAAU;EAAa;;AAGtE,SAAS,WAAW,SAA6C;CAC/D,MAAM,UAAU,QAAQ,KAAK;CAC7B,MAAM,QAAQ,QAAQ,SAAS;CAE/B,IAAI;AACJ,KAAI,QAAQ,MAAM;EAChB,MAAM,UAAU,KAAK,QAAQ,SAAS,QAAQ,KAAK;AACnD,MAAI,CAAC,GAAG,WAAW,QAAQ,EAAE;AAC3B,WAAQ,MAAM,mBAAmB,UAAU;AAC3C,WAAQ,KAAK,EAAE;;AAEjB,UAAQ,CAAC,QAAQ;OAEjB,SAAQ,qBAAqB,QAAQ;AAGvC,KAAI,MAAM,WAAW,GAAG;AACtB,UAAQ,IAAI,8DAA8D;AAC1E,UAAQ,IAAI,2CAA2C;AACvD,UAAQ,KAAK,EAAE;;CAGjB,IAAI,aAAa;AAEjB,MAAK,MAAM,QAAQ,OAAO;EACxB,MAAM,UAAU,KAAK,SAAS,SAAS,KAAK;EAC5C,MAAM,SAAS,YAAY,KAAK;AAEhC,OAAK,MAAM,WAAW,OAAO,SAC3B,SAAQ,IAAI,KAAK,QAAQ,IAAI,UAAU;AAGzC,MAAI,CAAC,OAAO,UAAU;AACpB,OAAI,OAAO,SAAS,WAAW,EAC7B,SAAQ,IAAI,KAAK,QAAQ,wBAAwB;AAEnD;;AAGF,eAAa;AAEb,MAAI,OAAO;AACT,MAAG,cAAc,MAAM,OAAO,SAAS,QAAQ;AAC/C,WAAQ,IAAI,KAAK,QAAQ,0BAA0B;SAC9C;AACL,WAAQ,IAAI,SAAS,QAAQ,gBAAgB;AAC7C,WAAQ,IAAI,OAAO,QAAQ;AAC3B,WAAQ,IAAI,MAAM;;;AAItB,KAAI,cAAc,CAAC,MACjB,SAAQ,IAAI,yDAAyD;;AAIzE,MAAa,wBAAwB,IAAI,QAAQ,mBAAmB,CACjE,YACC,yFACD,CACA,OAAO,iBAAiB,2CAA2C,CACnE,OAAO,WAAW,oCAAoC,MAAM,CAC5D,YACC,SACA;;;;kFAKD,CACA,OAAO,WAAW"}

package/dist/cli/index.js

@@ -1,4 +1,5 @@
 #!/usr/bin/env node
+import { codemodCommand } from "./commands/codemod/index.js";
 import { docsCommand } from "./commands/docs.js";
 import { generateTypesCommand } from "./commands/generate-types.js";
 import { lintCommand } from "./commands/lint.js";
@@ -20,6 +21,7 @@ cmd.addCommand(generateTypesCommand);
 cmd.addCommand(lintCommand);
 cmd.addCommand(docsCommand);
 cmd.addCommand(pluginCommand);
+cmd.addCommand(codemodCommand);
 cmd.parse();
 
 //#endregion

package/dist/cli/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","names":[],"sources":["../../src/cli/index.ts"],"sourcesContent":["#!/usr/bin/env node\nimport \"dotenv/config\";\nimport { readFileSync } from \"node:fs\";\nimport { dirname, join } from \"node:path\";\nimport { fileURLToPath } from \"node:url\";\nimport { Command } from \"commander\";\nimport { docsCommand } from \"./commands/docs.js\";\nimport { generateTypesCommand } from \"./commands/generate-types.js\";\nimport { lintCommand } from \"./commands/lint.js\";\nimport { pluginCommand } from \"./commands/plugin/index.js\";\nimport { setupCommand } from \"./commands/setup.js\";\n\nconst __dirname = dirname(fileURLToPath(import.meta.url));\nconst pkgPath = join(__dirname, \"../../package.json\");\nconst pkg = JSON.parse(readFileSync(pkgPath, \"utf-8\"));\n\nconst cmd = new Command();\n\ncmd\n  .name(\"appkit\")\n  .description(\"CLI tools for Databricks AppKit\")\n  .version(pkg.version);\n\ncmd.addCommand(setupCommand);\ncmd.addCommand(generateTypesCommand);\ncmd.addCommand(lintCommand);\ncmd.addCommand(docsCommand);\ncmd.addCommand(pluginCommand);\n\ncmd.parse();\n"],"mappings":";;;;;;;;;;;;;AAaA,MAAM,UAAU,KADE,QAAQ,cAAc,OAAO,KAAK,IAAI,CAAC,EACzB,qBAAqB;AACrD,MAAM,MAAM,KAAK,MAAM,aAAa,SAAS,QAAQ,CAAC;AAEtD,MAAM,MAAM,IAAI,SAAS;AAEzB,IACG,KAAK,SAAS,CACd,YAAY,kCAAkC,CAC9C,QAAQ,IAAI,QAAQ;AAEvB,IAAI,WAAW,aAAa;AAC5B,IAAI,WAAW,qBAAqB;AACpC,IAAI,WAAW,YAAY;AAC3B,IAAI,WAAW,YAAY;AAC3B,IAAI,WAAW,cAAc;AAE7B,IAAI,OAAO"}
+{"version":3,"file":"index.js","names":[],"sources":["../../src/cli/index.ts"],"sourcesContent":["#!/usr/bin/env node\nimport \"dotenv/config\";\nimport { readFileSync } from \"node:fs\";\nimport { dirname, join } from \"node:path\";\nimport { fileURLToPath } from \"node:url\";\nimport { Command } from \"commander\";\nimport { codemodCommand } from \"./commands/codemod/index.js\";\nimport { docsCommand } from \"./commands/docs.js\";\nimport { generateTypesCommand } from \"./commands/generate-types.js\";\nimport { lintCommand } from \"./commands/lint.js\";\nimport { pluginCommand } from \"./commands/plugin/index.js\";\nimport { setupCommand } from \"./commands/setup.js\";\n\nconst __dirname = dirname(fileURLToPath(import.meta.url));\nconst pkgPath = join(__dirname, \"../../package.json\");\nconst pkg = JSON.parse(readFileSync(pkgPath, \"utf-8\"));\n\nconst cmd = new Command();\n\ncmd\n  .name(\"appkit\")\n  .description(\"CLI tools for Databricks AppKit\")\n  .version(pkg.version);\n\ncmd.addCommand(setupCommand);\ncmd.addCommand(generateTypesCommand);\ncmd.addCommand(lintCommand);\ncmd.addCommand(docsCommand);\ncmd.addCommand(pluginCommand);\ncmd.addCommand(codemodCommand);\n\ncmd.parse();\n"],"mappings":";;;;;;;;;;;;;;AAcA,MAAM,UAAU,KADE,QAAQ,cAAc,OAAO,KAAK,IAAI,CAAC,EACzB,qBAAqB;AACrD,MAAM,MAAM,KAAK,MAAM,aAAa,SAAS,QAAQ,CAAC;AAEtD,MAAM,MAAM,IAAI,SAAS;AAEzB,IACG,KAAK,SAAS,CACd,YAAY,kCAAkC,CAC9C,QAAQ,IAAI,QAAQ;AAEvB,IAAI,WAAW,aAAa;AAC5B,IAAI,WAAW,qBAAqB;AACpC,IAAI,WAAW,YAAY;AAC3B,IAAI,WAAW,YAAY;AAC3B,IAAI,WAAW,cAAc;AAC7B,IAAI,WAAW,eAAe;AAE9B,IAAI,OAAO"}

package/dist/connectors/lakebase-v1/client.js.map

@@ -1 +1 @@
-{"version":3,"file":"client.js","names":[],"sources":["../../../src/connectors/lakebase-v1/client.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type { WorkspaceClient } from \"@databricks/sdk-experimental\";\nimport { ApiClient, Config } from \"@databricks/sdk-experimental\";\nimport pg from \"pg\";\nimport {\n  AppKitError,\n  AuthenticationError,\n  ConfigurationError,\n  ConnectionError,\n  ValidationError,\n} from \"../../errors\";\nimport { createLogger } from \"../../logging/logger\";\nimport {\n  type Counter,\n  type Histogram,\n  SpanStatusCode,\n  TelemetryManager,\n  type TelemetryProvider,\n} from \"../../telemetry\";\nimport { deepMerge } from \"../../utils\";\nimport { lakebaseV1Defaults } from \"./defaults\";\nimport type {\n  LakebaseV1Config,\n  LakebaseV1ConnectionConfig,\n  LakebaseV1Credentials,\n} from \"./types\";\n\nconst logger = createLogger(\"connectors:lakebase-v1\");\n\n/**\n * Enterprise-grade connector for Databricks Lakebase Provisioned\n *\n * @deprecated This connector is for Lakebase Provisioned only.\n * For new projects, use Lakebase Autoscaling instead: https://docs.databricks.com/aws/en/oltp/projects/\n *\n * This connector is compatible with Lakebase Provisioned: https://docs.databricks.com/aws/en/oltp/instances/\n *\n * Lakebase Autoscaling offers:\n * - Automatic compute scaling\n * - Scale-to-zero for cost optimization\n * - Database branching for development\n * - Instant restore capabilities\n *\n * Use the new LakebaseConnector (coming in a future release) for Lakebase Autoscaling support.\n *\n * @example Simplest - everything from env/context\n * ```typescript\n * const connector = new LakebaseV1Connector();\n * await connector.query('SELECT * FROM users');\n * ```\n *\n * @example With explicit connection string\n * ```typescript\n * const connector = new LakebaseV1Connector({\n *   connectionString: 'postgresql://...'\n * });\n * ```\n */\nexport class LakebaseV1Connector {\n  private readonly name: string = \"lakebase-v1\";\n  private readonly CACHE_BUFFER_MS = 2 * 60 * 1000;\n  private readonly config: LakebaseV1Config;\n  private readonly connectionConfig: LakebaseV1ConnectionConfig;\n  private pool: pg.Pool | null = null;\n  private credentials: LakebaseV1Credentials | null = null;\n\n  // telemetry\n  private readonly telemetry: TelemetryProvider;\n  private readonly telemetryMetrics: {\n    queryCount: Counter;\n    queryDuration: Histogram;\n  };\n\n  constructor(userConfig?: Partial<LakebaseV1Config>) {\n    this.config = deepMerge(lakebaseV1Defaults, userConfig);\n    this.connectionConfig = this.parseConnectionConfig();\n\n    this.telemetry = TelemetryManager.getProvider(\n      this.name,\n      this.config.telemetry,\n    );\n    this.telemetryMetrics = {\n      queryCount: this.telemetry\n        .getMeter()\n        .createCounter(\"lakebase.v1.query.count\", {\n          description: \"Total number of queries executed\",\n          unit: \"1\",\n        }),\n      queryDuration: this.telemetry\n        .getMeter()\n        .createHistogram(\"lakebase.v1.query.duration\", {\n          description: \"Duration of queries executed\",\n          unit: \"ms\",\n        }),\n    };\n\n    // validate configuration\n    if (this.config.maxPoolSize < 1) {\n      throw ValidationError.invalidValue(\n        \"maxPoolSize\",\n        this.config.maxPoolSize,\n        \"at least 1\",\n      );\n    }\n  }\n\n  /**\n   * Execute a SQL query\n   *\n   * @example\n   * ```typescript\n   * const users = await connector.query('SELECT * FROM users');\n   * const user = await connector.query('SELECT * FROM users WHERE id = $1', [123]);\n   * ```\n   */\n  async query<T extends pg.QueryResultRow>(\n    sql: string,\n    params?: any[],\n    retryCount: number = 0,\n  ): Promise<pg.QueryResult<T>> {\n    const startTime = Date.now();\n\n    return this.telemetry.startActiveSpan(\n      \"lakebase.v1.query\",\n      {\n        attributes: {\n          \"db.system\": \"lakebase-v1\",\n          \"db.statement\": sql.substring(0, 500),\n          \"db.retry_count\": retryCount,\n        },\n      },\n      async (span) => {\n        try {\n          const pool = await this.getPool();\n          const result = await pool.query<T>(sql, params);\n          span.setAttribute(\"db.rows_affected\", result.rowCount ?? 0);\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\n        } catch (error) {\n          // retry on auth failure\n          if (this.isAuthError(error)) {\n            span.addEvent(\"auth_error_retry\");\n            await this.rotateCredentials();\n            const newPool = await this.getPool();\n            const result = await newPool.query<T>(sql, params);\n            span.setAttribute(\"db.rows_affected\", result.rowCount ?? 0);\n            span.setStatus({ code: SpanStatusCode.OK });\n            return result;\n          }\n\n          // retry on transient errors, but only once\n          if (this.isTransientError(error) && retryCount < 1) {\n            span.addEvent(\"transient_error_retry\");\n            await new Promise((resolve) => setTimeout(resolve, 100));\n            return await this.query<T>(sql, params, retryCount + 1);\n          }\n\n          span.recordException(error as Error);\n          span.setStatus({ code: SpanStatusCode.ERROR });\n\n          if (error instanceof AppKitError) {\n            throw error;\n          }\n          throw ConnectionError.queryFailed(error as Error);\n        } finally {\n          const duration = Date.now() - startTime;\n          this.telemetryMetrics.queryCount.add(1);\n          this.telemetryMetrics.queryDuration.record(duration);\n          span.end();\n        }\n      },\n    );\n  }\n\n  /**\n   * Execute a transaction\n   *\n   * COMMIT and ROLLBACK are automatically managed by the transaction function.\n   *\n   * @param callback - Callback function to execute within the transaction context\n   * @example\n   * ```typescript\n   * await connector.transaction(async (client) => {\n   *   await client.query('INSERT INTO accounts (name) VALUES ($1)', ['Alice']);\n   *   await client.query('INSERT INTO logs (action) VALUES ($1)', ['Created Alice']);\n   * });\n   * ```\n   */\n  async transaction<T>(\n    callback: (client: pg.PoolClient) => Promise<T>,\n    retryCount: number = 0,\n  ): Promise<T> {\n    const startTime = Date.now();\n    return this.telemetry.startActiveSpan(\n      \"lakebase.v1.transaction\",\n      {\n        attributes: {\n          \"db.system\": \"lakebase-v1\",\n          \"db.retry_count\": retryCount,\n        },\n      },\n      async (span) => {\n        const pool = await this.getPool();\n        const client = await pool.connect();\n        try {\n          await client.query(\"BEGIN\");\n          const result = await callback(client);\n          await client.query(\"COMMIT\");\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\n        } catch (error) {\n          try {\n            await client.query(\"ROLLBACK\");\n          } catch {}\n          // retry on auth failure\n          if (this.isAuthError(error)) {\n            span.addEvent(\"auth_error_retry\");\n            client.release();\n            await this.rotateCredentials();\n            const newPool = await this.getPool();\n            const retryClient = await newPool.connect();\n            try {\n              await client.query(\"BEGIN\");\n              const result = await callback(retryClient);\n              await client.query(\"COMMIT\");\n              span.setStatus({ code: SpanStatusCode.OK });\n              return result;\n            } catch (retryError) {\n              try {\n                await retryClient.query(\"ROLLBACK\");\n              } catch {}\n              throw retryError;\n            } finally {\n              retryClient.release();\n            }\n          }\n\n          // retry on transient errors, but only once\n          if (this.isTransientError(error) && retryCount < 1) {\n            span.addEvent(\"transaction_error_retry\");\n            client.release();\n            await new Promise((resolve) => setTimeout(resolve, 100));\n            return await this.transaction<T>(callback, retryCount + 1);\n          }\n          span.recordException(error as Error);\n          span.setStatus({ code: SpanStatusCode.ERROR });\n\n          if (error instanceof AppKitError) {\n            throw error;\n          }\n          throw ConnectionError.transactionFailed(error as Error);\n        } finally {\n          client.release();\n          const duration = Date.now() - startTime;\n          this.telemetryMetrics.queryCount.add(1);\n          this.telemetryMetrics.queryDuration.record(duration);\n          span.end();\n        }\n      },\n    );\n  }\n\n  /** Check if database connection is healthy */\n  async healthCheck(): Promise<boolean> {\n    return this.telemetry.startActiveSpan(\n      \"lakebase.v1.healthCheck\",\n      {},\n      async (span) => {\n        try {\n          const result = await this.query<{ result: number }>(\n            \"SELECT 1 as result\",\n          );\n          const healthy = result.rows[0]?.result === 1;\n          span.setAttribute(\"db.healthy\", healthy);\n          span.setStatus({ code: SpanStatusCode.OK });\n          return healthy;\n        } catch {\n          span.setAttribute(\"db.healthy\", false);\n          span.setStatus({ code: SpanStatusCode.ERROR });\n          return false;\n        } finally {\n          span.end();\n        }\n      },\n    );\n  }\n\n  /** Close connection pool (call on shutdown) */\n  async close(): Promise<void> {\n    if (this.pool) {\n      await this.pool.end().catch((error: unknown) => {\n        logger.error(\"Error closing connection pool: %O\", error);\n      });\n      this.pool = null;\n    }\n    this.credentials = null;\n  }\n\n  /** Setup graceful shutdown to close connection pools */\n  shutdown(): void {\n    process.on(\"SIGTERM\", () => this.close());\n    process.on(\"SIGINT\", () => this.close());\n    this.close();\n  }\n\n  /** Get Databricks workspace client - from config or execution context */\n  private getWorkspaceClient(): WorkspaceClient {\n    if (this.config.workspaceClient) {\n      return this.config.workspaceClient;\n    }\n\n    try {\n      const { getWorkspaceClient: getClient } = require(\"../../context\");\n      const client = getClient();\n\n      // cache it for subsequent calls\n      this.config.workspaceClient = client;\n      return client;\n    } catch (_error) {\n      throw ConnectionError.clientUnavailable(\n        \"Databricks workspace client\",\n        \"Either pass it in config or ensure ServiceContext is initialized\",\n      );\n    }\n  }\n\n  /** Get or create connection pool */\n  private async getPool(): Promise<pg.Pool> {\n    if (!this.connectionConfig) {\n      throw ConfigurationError.invalidConnection(\n        \"Lakebase\",\n        \"Set PGHOST, PGDATABASE, PGAPPNAME env vars, provide a connectionString, or pass explicit config\",\n      );\n    }\n\n    if (!this.pool) {\n      const creds = await this.getCredentials();\n      this.pool = this.createPool(creds);\n    }\n    return this.pool;\n  }\n\n  /** Create PostgreSQL pool */\n  private createPool(credentials: {\n    username: string;\n    password: string;\n  }): pg.Pool {\n    const { host, database, port, sslMode } = this.connectionConfig;\n\n    const pool = new pg.Pool({\n      host,\n      port,\n      database,\n      user: credentials.username,\n      password: credentials.password,\n      max: this.config.maxPoolSize,\n      idleTimeoutMillis: this.config.idleTimeoutMs,\n      connectionTimeoutMillis: this.config.connectionTimeoutMs,\n      ssl: sslMode === \"require\" ? { rejectUnauthorized: true } : false,\n    });\n\n    pool.on(\"error\", (error: Error & { code?: string }) => {\n      logger.error(\n        \"Connection pool error: %s (code: %s)\",\n        error.message,\n        error.code,\n      );\n    });\n\n    return pool;\n  }\n\n  /** Get or fetch credentials with caching */\n  private async getCredentials(): Promise<{\n    username: string;\n    password: string;\n  }> {\n    const now = Date.now();\n\n    // return cached if still valid\n    if (\n      this.credentials &&\n      now < this.credentials.expiresAt - this.CACHE_BUFFER_MS\n    ) {\n      return this.credentials;\n    }\n\n    // fetch new credentials\n    const username = await this.fetchUsername();\n    const { token, expiresAt } = await this.fetchPassword();\n\n    this.credentials = {\n      username,\n      password: token,\n      expiresAt,\n    };\n\n    return { username, password: token };\n  }\n\n  /** Rotate credentials and recreate pool */\n  private async rotateCredentials(): Promise<void> {\n    // clear cached credentials\n    this.credentials = null;\n\n    if (this.pool) {\n      const oldPool = this.pool;\n      this.pool = null;\n      oldPool.end().catch((error: unknown) => {\n        logger.error(\n          \"Error closing old connection pool during rotation: %O\",\n          error,\n        );\n      });\n    }\n  }\n\n  /** Fetch username from Databricks */\n  private async fetchUsername(): Promise<string> {\n    const workspaceClient = this.getWorkspaceClient();\n    const user = await workspaceClient.currentUser.me();\n    if (!user.userName) {\n      throw AuthenticationError.userLookupFailed();\n    }\n    return user.userName;\n  }\n\n  /** Fetch password (OAuth token) from Databricks */\n  private async fetchPassword(): Promise<{ token: string; expiresAt: number }> {\n    const workspaceClient = this.getWorkspaceClient();\n    const config = new Config({ host: workspaceClient.config.host });\n    const apiClient = new ApiClient(config);\n\n    if (!this.connectionConfig.appName) {\n      throw ConfigurationError.resourceNotFound(\"Database app name\");\n    }\n\n    const credentials = await apiClient.request({\n      path: `/api/2.0/database/credentials`,\n      method: \"POST\",\n      headers: new Headers(),\n      raw: false,\n      payload: {\n        instance_names: [this.connectionConfig.appName],\n        request_id: randomUUID(),\n      },\n    });\n\n    if (!this.validateCredentials(credentials)) {\n      throw AuthenticationError.credentialsFailed(\n        this.connectionConfig.appName,\n      );\n    }\n\n    const expiresAt = new Date(credentials.expiration_time).getTime();\n\n    return { token: credentials.token, expiresAt };\n  }\n\n  /** Check if error is auth failure */\n  private isAuthError(error: unknown): boolean {\n    return (\n      typeof error === \"object\" &&\n      error !== null &&\n      \"code\" in error &&\n      (error as any).code === \"28P01\"\n    );\n  }\n\n  /** Check if error is transient */\n  private isTransientError(error: unknown): boolean {\n    if (typeof error !== \"object\" || error === null || !(\"code\" in error)) {\n      return false;\n    }\n\n    const code = (error as any).code;\n    return (\n      code === \"ECONNRESET\" ||\n      code === \"ECONNREFUSED\" ||\n      code === \"ETIMEDOUT\" ||\n      code === \"57P01\" || // admin_shutdown\n      code === \"57P03\" || // cannot_connect_now\n      code === \"08006\" || // connection_failure\n      code === \"08003\" || // connection_does_not_exist\n      code === \"08000\" // connection_exception\n    );\n  }\n\n  /** Type guard for credentials */\n  private validateCredentials(\n    value: unknown,\n  ): value is { token: string; expiration_time: string } {\n    if (typeof value !== \"object\" || value === null) {\n      return false;\n    }\n\n    const credentials = value as { token: string; expiration_time: string };\n    return (\n      \"token\" in credentials &&\n      typeof credentials.token === \"string\" &&\n      \"expiration_time\" in credentials &&\n      typeof credentials.expiration_time === \"string\" &&\n      new Date(credentials.expiration_time).getTime() > Date.now()\n    );\n  }\n\n  /** Parse connection configuration from config or environment */\n  private parseConnectionConfig(): LakebaseV1ConnectionConfig {\n    if (this.config.connectionString) {\n      return this.parseConnectionString(this.config.connectionString);\n    }\n\n    // get connection from config\n    if (this.config.host && this.config.database && this.config.appName) {\n      return {\n        host: this.config.host,\n        database: this.config.database,\n        port: this.config.port ?? 5432,\n        sslMode: this.config.sslMode ?? \"require\",\n        appName: this.config.appName,\n      };\n    }\n\n    // get connection from environment variables\n    const pgHost = process.env.PGHOST;\n    const pgDatabase = process.env.PGDATABASE;\n    const pgAppName = process.env.PGAPPNAME;\n    if (!pgHost || !pgDatabase || !pgAppName) {\n      throw ConfigurationError.invalidConnection(\n        \"Lakebase\",\n        \"Required env vars: PGHOST, PGDATABASE, PGAPPNAME. Optional: PGPORT (default: 5432), PGSSLMODE (default: require)\",\n      );\n    }\n    const pgPort = process.env.PGPORT;\n    const port = pgPort ? parseInt(pgPort, 10) : 5432;\n\n    if (Number.isNaN(port)) {\n      throw ValidationError.invalidValue(\"port\", pgPort, \"a number\");\n    }\n\n    const pgSSLMode = process.env.PGSSLMODE;\n    const sslMode =\n      (pgSSLMode as \"require\" | \"disable\" | \"prefer\") || \"require\";\n\n    return {\n      host: pgHost,\n      database: pgDatabase,\n      port,\n      sslMode,\n      appName: pgAppName,\n    };\n  }\n\n  private parseConnectionString(\n    connectionString: string,\n  ): LakebaseV1ConnectionConfig {\n    const url = new URL(connectionString);\n    const appName = url.searchParams.get(\"appName\");\n    if (!appName) {\n      throw ConfigurationError.missingConnectionParam(\"appName\");\n    }\n\n    return {\n      host: url.hostname,\n      database: url.pathname.slice(1), // remove leading slash\n      port: url.port ? parseInt(url.port, 10) : 5432,\n      sslMode:\n        (url.searchParams.get(\"sslmode\") as \"require\" | \"disable\" | \"prefer\") ??\n        \"require\",\n      appName: appName,\n    };\n  }\n}\n"],"mappings":";;;;;;;;AA2BA,MAAM,SAAS,aAAa,yBAAyB"}
+{"version":3,"file":"client.js","names":[],"sources":["../../../src/connectors/lakebase-v1/client.ts"],"sourcesContent":["import { randomUUID } from \"node:crypto\";\nimport type { WorkspaceClient } from \"@databricks/sdk-experimental\";\nimport { ApiClient, Config } from \"@databricks/sdk-experimental\";\nimport pg from \"pg\";\nimport {\n  AppKitError,\n  AuthenticationError,\n  ConfigurationError,\n  ConnectionError,\n  ValidationError,\n} from \"../../errors\";\nimport { createLogger } from \"../../logging/logger\";\nimport {\n  type Counter,\n  type Histogram,\n  SpanStatusCode,\n  TelemetryManager,\n  type TelemetryProvider,\n} from \"../../telemetry\";\nimport { deepMerge } from \"../../utils\";\nimport { lakebaseV1Defaults } from \"./defaults\";\nimport type {\n  LakebaseV1Config,\n  LakebaseV1ConnectionConfig,\n  LakebaseV1Credentials,\n} from \"./types\";\n\nconst logger = createLogger(\"connectors:lakebase-v1\");\n\n/**\n * Enterprise-grade connector for Databricks Lakebase Provisioned\n *\n * @deprecated This connector is for Lakebase Provisioned only.\n * For new projects, use Lakebase Autoscaling instead: https://docs.databricks.com/aws/en/oltp/projects/\n *\n * This connector is compatible with Lakebase Provisioned: https://docs.databricks.com/aws/en/oltp/instances/\n *\n * Lakebase Autoscaling offers:\n * - Automatic compute scaling\n * - Scale-to-zero for cost optimization\n * - Database branching for development\n * - Instant restore capabilities\n *\n * Use the new LakebaseConnector (coming in a future release) for Lakebase Autoscaling support.\n *\n * @example Simplest - everything from env/context\n * ```typescript\n * const connector = new LakebaseV1Connector();\n * await connector.query('SELECT * FROM users');\n * ```\n *\n * @example With explicit connection string\n * ```typescript\n * const connector = new LakebaseV1Connector({\n *   connectionString: 'postgresql://...'\n * });\n * ```\n */\nexport class LakebaseV1Connector {\n  private readonly name: string = \"lakebase-v1\";\n  private readonly CACHE_BUFFER_MS = 2 * 60 * 1000;\n  private readonly config: LakebaseV1Config;\n  private readonly connectionConfig: LakebaseV1ConnectionConfig;\n  private pool: pg.Pool | null = null;\n  private credentials: LakebaseV1Credentials | null = null;\n\n  // telemetry\n  private readonly telemetry: TelemetryProvider;\n  private readonly telemetryMetrics: {\n    queryCount: Counter;\n    queryDuration: Histogram;\n  };\n\n  constructor(userConfig?: Partial<LakebaseV1Config>) {\n    this.config = deepMerge(lakebaseV1Defaults, userConfig);\n    this.connectionConfig = this.parseConnectionConfig();\n\n    this.telemetry = TelemetryManager.getProvider(\n      this.name,\n      this.config.telemetry,\n    );\n    this.telemetryMetrics = {\n      queryCount: this.telemetry\n        .getMeter()\n        .createCounter(\"lakebase.v1.query.count\", {\n          description: \"Total number of queries executed\",\n          unit: \"1\",\n        }),\n      queryDuration: this.telemetry\n        .getMeter()\n        .createHistogram(\"lakebase.v1.query.duration\", {\n          description: \"Duration of queries executed\",\n          unit: \"ms\",\n        }),\n    };\n\n    // validate configuration\n    if (this.config.maxPoolSize < 1) {\n      throw ValidationError.invalidValue(\n        \"maxPoolSize\",\n        this.config.maxPoolSize,\n        \"at least 1\",\n      );\n    }\n  }\n\n  /**\n   * Execute a SQL query\n   *\n   * @example\n   * ```typescript\n   * const users = await connector.query('SELECT * FROM users');\n   * const user = await connector.query('SELECT * FROM users WHERE id = $1', [123]);\n   * ```\n   */\n  async query<T extends pg.QueryResultRow>(\n    sql: string,\n    params?: any[],\n    retryCount: number = 0,\n  ): Promise<pg.QueryResult<T>> {\n    const startTime = Date.now();\n\n    return this.telemetry.startActiveSpan(\n      \"lakebase.v1.query\",\n      {\n        attributes: {\n          \"db.system\": \"lakebase-v1\",\n          \"db.statement\": sql.substring(0, 500),\n          \"db.retry_count\": retryCount,\n        },\n      },\n      async (span) => {\n        try {\n          const pool = await this.getPool();\n          const result = await pool.query<T>(sql, params);\n          span.setAttribute(\"db.rows_affected\", result.rowCount ?? 0);\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\n        } catch (error) {\n          // retry on auth failure\n          if (this.isAuthError(error)) {\n            span.addEvent(\"auth_error_retry\");\n            await this.rotateCredentials();\n            const newPool = await this.getPool();\n            const result = await newPool.query<T>(sql, params);\n            span.setAttribute(\"db.rows_affected\", result.rowCount ?? 0);\n            span.setStatus({ code: SpanStatusCode.OK });\n            return result;\n          }\n\n          // retry on transient errors, but only once\n          if (this.isTransientError(error) && retryCount < 1) {\n            span.addEvent(\"transient_error_retry\");\n            await new Promise((resolve) => setTimeout(resolve, 100));\n            return await this.query<T>(sql, params, retryCount + 1);\n          }\n\n          span.recordException(error as Error);\n          span.setStatus({ code: SpanStatusCode.ERROR });\n\n          logger.error(\n            \"Query execution failed: %s (code=%s)\",\n            error instanceof Error ? error.message : String(error),\n            (error as any)?.code,\n          );\n\n          if (error instanceof AppKitError) {\n            throw error;\n          }\n          throw ConnectionError.queryFailed(error as Error);\n        } finally {\n          const duration = Date.now() - startTime;\n          this.telemetryMetrics.queryCount.add(1);\n          this.telemetryMetrics.queryDuration.record(duration);\n          span.end();\n        }\n      },\n    );\n  }\n\n  /**\n   * Execute a transaction\n   *\n   * COMMIT and ROLLBACK are automatically managed by the transaction function.\n   *\n   * @param callback - Callback function to execute within the transaction context\n   * @example\n   * ```typescript\n   * await connector.transaction(async (client) => {\n   *   await client.query('INSERT INTO accounts (name) VALUES ($1)', ['Alice']);\n   *   await client.query('INSERT INTO logs (action) VALUES ($1)', ['Created Alice']);\n   * });\n   * ```\n   */\n  async transaction<T>(\n    callback: (client: pg.PoolClient) => Promise<T>,\n    retryCount: number = 0,\n  ): Promise<T> {\n    const startTime = Date.now();\n    return this.telemetry.startActiveSpan(\n      \"lakebase.v1.transaction\",\n      {\n        attributes: {\n          \"db.system\": \"lakebase-v1\",\n          \"db.retry_count\": retryCount,\n        },\n      },\n      async (span) => {\n        const pool = await this.getPool();\n        const client = await pool.connect();\n        try {\n          await client.query(\"BEGIN\");\n          const result = await callback(client);\n          await client.query(\"COMMIT\");\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\n        } catch (error) {\n          try {\n            await client.query(\"ROLLBACK\");\n          } catch {}\n          // retry on auth failure\n          if (this.isAuthError(error)) {\n            span.addEvent(\"auth_error_retry\");\n            client.release();\n            await this.rotateCredentials();\n            const newPool = await this.getPool();\n            const retryClient = await newPool.connect();\n            try {\n              await client.query(\"BEGIN\");\n              const result = await callback(retryClient);\n              await client.query(\"COMMIT\");\n              span.setStatus({ code: SpanStatusCode.OK });\n              return result;\n            } catch (retryError) {\n              try {\n                await retryClient.query(\"ROLLBACK\");\n              } catch {}\n              throw retryError;\n            } finally {\n              retryClient.release();\n            }\n          }\n\n          // retry on transient errors, but only once\n          if (this.isTransientError(error) && retryCount < 1) {\n            span.addEvent(\"transaction_error_retry\");\n            client.release();\n            await new Promise((resolve) => setTimeout(resolve, 100));\n            return await this.transaction<T>(callback, retryCount + 1);\n          }\n          span.recordException(error as Error);\n          span.setStatus({ code: SpanStatusCode.ERROR });\n\n          logger.error(\n            \"Transaction execution failed: %s (code=%s)\",\n            error instanceof Error ? error.message : String(error),\n            (error as any)?.code,\n          );\n\n          if (error instanceof AppKitError) {\n            throw error;\n          }\n          throw ConnectionError.transactionFailed(error as Error);\n        } finally {\n          client.release();\n          const duration = Date.now() - startTime;\n          this.telemetryMetrics.queryCount.add(1);\n          this.telemetryMetrics.queryDuration.record(duration);\n          span.end();\n        }\n      },\n    );\n  }\n\n  /** Check if database connection is healthy */\n  async healthCheck(): Promise<boolean> {\n    return this.telemetry.startActiveSpan(\n      \"lakebase.v1.healthCheck\",\n      {},\n      async (span) => {\n        try {\n          const result = await this.query<{ result: number }>(\n            \"SELECT 1 as result\",\n          );\n          const healthy = result.rows[0]?.result === 1;\n          span.setAttribute(\"db.healthy\", healthy);\n          span.setStatus({ code: SpanStatusCode.OK });\n          return healthy;\n        } catch {\n          span.setAttribute(\"db.healthy\", false);\n          span.setStatus({ code: SpanStatusCode.ERROR });\n          return false;\n        } finally {\n          span.end();\n        }\n      },\n    );\n  }\n\n  /** Close connection pool (call on shutdown) */\n  async close(): Promise<void> {\n    if (this.pool) {\n      await this.pool.end().catch((error: unknown) => {\n        logger.error(\"Error closing connection pool: %O\", error);\n      });\n      this.pool = null;\n    }\n    this.credentials = null;\n  }\n\n  /** Setup graceful shutdown to close connection pools */\n  shutdown(): void {\n    process.on(\"SIGTERM\", () => this.close());\n    process.on(\"SIGINT\", () => this.close());\n    this.close();\n  }\n\n  /** Get Databricks workspace client - from config or execution context */\n  private getWorkspaceClient(): WorkspaceClient {\n    if (this.config.workspaceClient) {\n      return this.config.workspaceClient;\n    }\n\n    try {\n      const { getWorkspaceClient: getClient } = require(\"../../context\");\n      const client = getClient();\n\n      // cache it for subsequent calls\n      this.config.workspaceClient = client;\n      return client;\n    } catch (_error) {\n      throw ConnectionError.clientUnavailable(\n        \"Databricks workspace client\",\n        \"Either pass it in config or ensure ServiceContext is initialized\",\n      );\n    }\n  }\n\n  /** Get or create connection pool */\n  private async getPool(): Promise<pg.Pool> {\n    if (!this.connectionConfig) {\n      throw ConfigurationError.invalidConnection(\n        \"Lakebase\",\n        \"Set PGHOST, PGDATABASE, PGAPPNAME env vars, provide a connectionString, or pass explicit config\",\n      );\n    }\n\n    if (!this.pool) {\n      const creds = await this.getCredentials();\n      this.pool = this.createPool(creds);\n    }\n    return this.pool;\n  }\n\n  /** Create PostgreSQL pool */\n  private createPool(credentials: {\n    username: string;\n    password: string;\n  }): pg.Pool {\n    const { host, database, port, sslMode } = this.connectionConfig;\n\n    const pool = new pg.Pool({\n      host,\n      port,\n      database,\n      user: credentials.username,\n      password: credentials.password,\n      max: this.config.maxPoolSize,\n      idleTimeoutMillis: this.config.idleTimeoutMs,\n      connectionTimeoutMillis: this.config.connectionTimeoutMs,\n      ssl: sslMode === \"require\" ? { rejectUnauthorized: true } : false,\n    });\n\n    pool.on(\"error\", (error: Error & { code?: string }) => {\n      logger.error(\n        \"Connection pool error: %s (code: %s)\",\n        error.message,\n        error.code,\n      );\n    });\n\n    return pool;\n  }\n\n  /** Get or fetch credentials with caching */\n  private async getCredentials(): Promise<{\n    username: string;\n    password: string;\n  }> {\n    const now = Date.now();\n\n    // return cached if still valid\n    if (\n      this.credentials &&\n      now < this.credentials.expiresAt - this.CACHE_BUFFER_MS\n    ) {\n      return this.credentials;\n    }\n\n    // fetch new credentials\n    const username = await this.fetchUsername();\n    const { token, expiresAt } = await this.fetchPassword();\n\n    this.credentials = {\n      username,\n      password: token,\n      expiresAt,\n    };\n\n    return { username, password: token };\n  }\n\n  /** Rotate credentials and recreate pool */\n  private async rotateCredentials(): Promise<void> {\n    // clear cached credentials\n    this.credentials = null;\n\n    if (this.pool) {\n      const oldPool = this.pool;\n      this.pool = null;\n      oldPool.end().catch((error: unknown) => {\n        logger.error(\n          \"Error closing old connection pool during rotation: %O\",\n          error,\n        );\n      });\n    }\n  }\n\n  /** Fetch username from Databricks */\n  private async fetchUsername(): Promise<string> {\n    const workspaceClient = this.getWorkspaceClient();\n    const user = await workspaceClient.currentUser.me();\n    if (!user.userName) {\n      throw AuthenticationError.userLookupFailed();\n    }\n    return user.userName;\n  }\n\n  /** Fetch password (OAuth token) from Databricks */\n  private async fetchPassword(): Promise<{ token: string; expiresAt: number }> {\n    const workspaceClient = this.getWorkspaceClient();\n    const config = new Config({ host: workspaceClient.config.host });\n    const apiClient = new ApiClient(config);\n\n    if (!this.connectionConfig.appName) {\n      throw ConfigurationError.resourceNotFound(\"Database app name\");\n    }\n\n    const credentials = await apiClient.request({\n      path: `/api/2.0/database/credentials`,\n      method: \"POST\",\n      headers: new Headers(),\n      raw: false,\n      payload: {\n        instance_names: [this.connectionConfig.appName],\n        request_id: randomUUID(),\n      },\n    });\n\n    if (!this.validateCredentials(credentials)) {\n      throw AuthenticationError.credentialsFailed(\n        this.connectionConfig.appName,\n      );\n    }\n\n    const expiresAt = new Date(credentials.expiration_time).getTime();\n\n    return { token: credentials.token, expiresAt };\n  }\n\n  /** Check if error is auth failure */\n  private isAuthError(error: unknown): boolean {\n    return (\n      typeof error === \"object\" &&\n      error !== null &&\n      \"code\" in error &&\n      (error as any).code === \"28P01\"\n    );\n  }\n\n  /** Check if error is transient */\n  private isTransientError(error: unknown): boolean {\n    if (typeof error !== \"object\" || error === null || !(\"code\" in error)) {\n      return false;\n    }\n\n    const code = (error as any).code;\n    return (\n      code === \"ECONNRESET\" ||\n      code === \"ECONNREFUSED\" ||\n      code === \"ETIMEDOUT\" ||\n      code === \"57P01\" || // admin_shutdown\n      code === \"57P03\" || // cannot_connect_now\n      code === \"08006\" || // connection_failure\n      code === \"08003\" || // connection_does_not_exist\n      code === \"08000\" // connection_exception\n    );\n  }\n\n  /** Type guard for credentials */\n  private validateCredentials(\n    value: unknown,\n  ): value is { token: string; expiration_time: string } {\n    if (typeof value !== \"object\" || value === null) {\n      return false;\n    }\n\n    const credentials = value as { token: string; expiration_time: string };\n    return (\n      \"token\" in credentials &&\n      typeof credentials.token === \"string\" &&\n      \"expiration_time\" in credentials &&\n      typeof credentials.expiration_time === \"string\" &&\n      new Date(credentials.expiration_time).getTime() > Date.now()\n    );\n  }\n\n  /** Parse connection configuration from config or environment */\n  private parseConnectionConfig(): LakebaseV1ConnectionConfig {\n    if (this.config.connectionString) {\n      return this.parseConnectionString(this.config.connectionString);\n    }\n\n    // get connection from config\n    if (this.config.host && this.config.database && this.config.appName) {\n      return {\n        host: this.config.host,\n        database: this.config.database,\n        port: this.config.port ?? 5432,\n        sslMode: this.config.sslMode ?? \"require\",\n        appName: this.config.appName,\n      };\n    }\n\n    // get connection from environment variables\n    const pgHost = process.env.PGHOST;\n    const pgDatabase = process.env.PGDATABASE;\n    const pgAppName = process.env.PGAPPNAME;\n    if (!pgHost || !pgDatabase || !pgAppName) {\n      throw ConfigurationError.invalidConnection(\n        \"Lakebase\",\n        \"Required env vars: PGHOST, PGDATABASE, PGAPPNAME. Optional: PGPORT (default: 5432), PGSSLMODE (default: require)\",\n      );\n    }\n    const pgPort = process.env.PGPORT;\n    const port = pgPort ? parseInt(pgPort, 10) : 5432;\n\n    if (Number.isNaN(port)) {\n      throw ValidationError.invalidValue(\"port\", pgPort, \"a number\");\n    }\n\n    const pgSSLMode = process.env.PGSSLMODE;\n    const sslMode =\n      (pgSSLMode as \"require\" | \"disable\" | \"prefer\") || \"require\";\n\n    return {\n      host: pgHost,\n      database: pgDatabase,\n      port,\n      sslMode,\n      appName: pgAppName,\n    };\n  }\n\n  private parseConnectionString(\n    connectionString: string,\n  ): LakebaseV1ConnectionConfig {\n    const url = new URL(connectionString);\n    const appName = url.searchParams.get(\"appName\");\n    if (!appName) {\n      throw ConfigurationError.missingConnectionParam(\"appName\");\n    }\n\n    return {\n      host: url.hostname,\n      database: url.pathname.slice(1), // remove leading slash\n      port: url.port ? parseInt(url.port, 10) : 5432,\n      sslMode:\n        (url.searchParams.get(\"sslmode\") as \"require\" | \"disable\" | \"prefer\") ??\n        \"require\",\n      appName: appName,\n    };\n  }\n}\n"],"mappings":";;;;;;;;AA2BA,MAAM,SAAS,aAAa,yBAAyB"}

package/dist/connectors/sql-warehouse/client.js

@@ -135,6 +135,7 @@ var SQLWarehouseConnector = class {
 						code: SpanStatusCode.ERROR,
 						message: error instanceof Error ? error.message : String(error)
 					});
+					logger.error("Statement execution failed: %s", error instanceof Error ? error.message : String(error));
 				}
 				if (error instanceof AppKitError) throw error;
 				throw ExecutionError.statementFailed(error instanceof Error ? error.message : String(error));