@databricks/appkit 0.1.4 → 0.2.0

package/dist/utils/path-exclusions.js

@@ -0,0 +1,66 @@
+//#region src/utils/path-exclusions.ts
+/**
+* Paths and patterns to exclude from tracing and logging.
+* Requests matching these will not create spans or WideEvents.
+*/
+const EXCLUDED_PATH_PREFIXES = [
+	"/@fs/",
+	"/@vite/",
+	"/@id/",
+	"/@react-refresh",
+	"/src/",
+	"/node_modules/",
+	"/favicon.ico",
+	"/_next/",
+	"/static/",
+	"/health",
+	"/metrics"
+];
+/**
+* File extensions to exclude from tracing.
+* These are typically static assets that don't need tracing.
+*/
+const EXCLUDED_EXTENSIONS = [
+	".svg",
+	".png",
+	".jpg",
+	".jpeg",
+	".gif",
+	".webp",
+	".ico",
+	".css",
+	".woff",
+	".woff2",
+	".ttf",
+	".eot",
+	".map",
+	".js"
+];
+/**
+* Check if a request should be ignored for tracing.
+* This is the primary filter used by HttpInstrumentation.
+*/
+function shouldIgnoreRequest(request) {
+	const url = request.url;
+	if (!url) return false;
+	const path = url.split("?")[0];
+	return shouldExcludePath(path);
+}
+/**
+* Check if a path should be excluded from tracing/logging.
+* Returns true if path should be excluded, false otherwise.
+*/
+function shouldExcludePath(path) {
+	if (typeof path !== "string") return false;
+	const cleanPath = path.split("?")[0];
+	const lowerPath = cleanPath.toLowerCase();
+	for (const prefix of EXCLUDED_PATH_PREFIXES) if (cleanPath.includes(prefix)) return true;
+	if (!cleanPath.startsWith("/api/")) {
+		for (const ext of EXCLUDED_EXTENSIONS) if (lowerPath.endsWith(ext)) return true;
+	}
+	return false;
+}
+
+//#endregion
+export { shouldExcludePath, shouldIgnoreRequest };
+//# sourceMappingURL=path-exclusions.js.map

package/dist/utils/path-exclusions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"path-exclusions.js","names":[],"sources":["../../src/utils/path-exclusions.ts"],"sourcesContent":["import type { IncomingMessage } from \"node:http\";\n\n/**\n * Paths and patterns to exclude from tracing and logging.\n * Requests matching these will not create spans or WideEvents.\n */\nexport const EXCLUDED_PATH_PREFIXES = [\n  // Vite dev server internals\n  \"/@fs/\",\n  \"/@vite/\",\n  \"/@id/\",\n  \"/@react-refresh\",\n  \"/src/\", // Vite HMR source files\n  \"/node_modules/\",\n\n  // Static assets and common paths\n  \"/favicon.ico\",\n  \"/_next/\",\n  \"/static/\",\n\n  // Health checks\n  \"/health\",\n  \"/metrics\",\n];\n\n/**\n * File extensions to exclude from tracing.\n * These are typically static assets that don't need tracing.\n */\nexport const EXCLUDED_EXTENSIONS = [\n  \".svg\",\n  \".png\",\n  \".jpg\",\n  \".jpeg\",\n  \".gif\",\n  \".webp\",\n  \".ico\",\n  \".css\",\n  \".woff\",\n  \".woff2\",\n  \".ttf\",\n  \".eot\",\n  \".map\", // Source maps\n  \".js\", // Static JS files (not API endpoints)\n];\n\n/**\n * Check if a request should be ignored for tracing.\n * This is the primary filter used by HttpInstrumentation.\n */\nexport function shouldIgnoreRequest(request: IncomingMessage): boolean {\n  const url = request.url;\n  if (!url) return false;\n\n  // Remove query string for path matching\n  const path = url.split(\"?\")[0];\n\n  return shouldExcludePath(path);\n}\n\n/**\n * Check if a path should be excluded from tracing/logging.\n * Returns true if path should be excluded, false otherwise.\n */\nexport function shouldExcludePath(path: string | undefined): boolean {\n  if (typeof path !== \"string\") return false;\n\n  // Remove query string\n  const cleanPath = path.split(\"?\")[0];\n  const lowerPath = cleanPath.toLowerCase();\n\n  // Check path prefixes\n  for (const prefix of EXCLUDED_PATH_PREFIXES) {\n    if (cleanPath.includes(prefix)) {\n      return true;\n    }\n  }\n\n  // Check file extensions (but not for /api/ routes)\n  if (!cleanPath.startsWith(\"/api/\")) {\n    for (const ext of EXCLUDED_EXTENSIONS) {\n      if (lowerPath.endsWith(ext)) {\n        return true;\n      }\n    }\n  }\n\n  return false;\n}\n"],"mappings":";;;;;AAMA,MAAa,yBAAyB;CAEpC;CACA;CACA;CACA;CACA;CACA;CAGA;CACA;CACA;CAGA;CACA;CACD;;;;;AAMD,MAAa,sBAAsB;CACjC;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACD;;;;;AAMD,SAAgB,oBAAoB,SAAmC;CACrE,MAAM,MAAM,QAAQ;AACpB,KAAI,CAAC,IAAK,QAAO;CAGjB,MAAM,OAAO,IAAI,MAAM,IAAI,CAAC;AAE5B,QAAO,kBAAkB,KAAK;;;;;;AAOhC,SAAgB,kBAAkB,MAAmC;AACnE,KAAI,OAAO,SAAS,SAAU,QAAO;CAGrC,MAAM,YAAY,KAAK,MAAM,IAAI,CAAC;CAClC,MAAM,YAAY,UAAU,aAAa;AAGzC,MAAK,MAAM,UAAU,uBACnB,KAAI,UAAU,SAAS,OAAO,CAC5B,QAAO;AAKX,KAAI,CAAC,UAAU,WAAW,QAAQ,EAChC;OAAK,MAAM,OAAO,oBAChB,KAAI,UAAU,SAAS,IAAI,CACzB,QAAO;;AAKb,QAAO"}

package/dist/utils/vite-config-merge.js

@@ -1,5 +1,3 @@
-import { __esmMin } from "../_virtual/rolldown_runtime.js";
-
 //#region src/utils/vite-config-merge.ts
 function mergeConfigDedup(base, override, mergeFn) {
 	const merged = mergeFn(base, override);
@@ -14,9 +12,7 @@ function mergeConfigDedup(base, override, mergeFn) {
 	}
 	return merged;
 }
-var init_vite_config_merge = __esmMin((() => {}));
 
 //#endregion
-init_vite_config_merge();
-export { init_vite_config_merge, mergeConfigDedup };
+export { mergeConfigDedup };
 //# sourceMappingURL=vite-config-merge.js.map

package/dist/utils/vite-config-merge.js.map

@@ -1 +1 @@
-{"version":3,"file":"vite-config-merge.js","names":[],"sources":["../../src/utils/vite-config-merge.ts"],"sourcesContent":["import type { Plugin } from \"vite\";\n\nexport function mergeConfigDedup(\n  base: any,\n  override: any,\n  mergeFn: (a: any, b: any) => any,\n) {\n  const merged = mergeFn(base, override);\n  if (base.plugins && override.plugins) {\n    const seen = new Set<string>();\n    merged.plugins = [...base.plugins, ...override.plugins].filter(\n      (p: Plugin) => {\n        const name = p.name;\n        if (seen.has(name)) return false;\n        seen.add(name);\n        return true;\n      },\n    );\n  }\n  return merged;\n}\n"],"mappings":";;;AAEA,SAAgB,iBACd,MACA,UACA,SACA;CACA,MAAM,SAAS,QAAQ,MAAM,SAAS;AACtC,KAAI,KAAK,WAAW,SAAS,SAAS;EACpC,MAAM,uBAAO,IAAI,KAAa;AAC9B,SAAO,UAAU,CAAC,GAAG,KAAK,SAAS,GAAG,SAAS,QAAQ,CAAC,QACrD,MAAc;GACb,MAAM,OAAO,EAAE;AACf,OAAI,KAAK,IAAI,KAAK,CAAE,QAAO;AAC3B,QAAK,IAAI,KAAK;AACd,UAAO;IAEV;;AAEH,QAAO"}
+{"version":3,"file":"vite-config-merge.js","names":[],"sources":["../../src/utils/vite-config-merge.ts"],"sourcesContent":["import type { Plugin } from \"vite\";\n\nexport function mergeConfigDedup(\n  base: any,\n  override: any,\n  mergeFn: (a: any, b: any) => any,\n) {\n  const merged = mergeFn(base, override);\n  if (base.plugins && override.plugins) {\n    const seen = new Set<string>();\n    merged.plugins = [...base.plugins, ...override.plugins].filter(\n      (p: Plugin) => {\n        const name = p.name;\n        if (seen.has(name)) return false;\n        seen.add(name);\n        return true;\n      },\n    );\n  }\n  return merged;\n}\n"],"mappings":";AAEA,SAAgB,iBACd,MACA,UACA,SACA;CACA,MAAM,SAAS,QAAQ,MAAM,SAAS;AACtC,KAAI,KAAK,WAAW,SAAS,SAAS;EACpC,MAAM,uBAAO,IAAI,KAAa;AAC9B,SAAO,UAAU,CAAC,GAAG,KAAK,SAAS,GAAG,SAAS,QAAQ,CAAC,QACrD,MAAc;GACb,MAAM,OAAO,EAAE;AACf,OAAI,KAAK,IAAI,KAAK,CAAE,QAAO;AAC3B,QAAK,IAAI,KAAK;AACd,UAAO;IAEV;;AAEH,QAAO"}

package/llms.txt

@@ -440,23 +440,49 @@ Formats:
 - `format: "JSON"` (default) returns JSON rows
 - `format: "ARROW"` returns an Arrow “statement_id” payload over SSE, then the client fetches binary Arrow from `/api/analytics/arrow-result/:jobId`
 
-### Request context (`getRequestContext()`)
+### Execution context and `asUser(req)`
 
-If a plugin sets `requiresDatabricksClient = true`, AppKit adds middleware that provides request context.
+AppKit manages Databricks authentication via two contexts:
 
-Headers used:
+- **ServiceContext** (singleton): Initialized at app startup with service principal credentials
+- **ExecutionContext**: Determined at runtime - either service principal or user context
+
+**Headers used for user context:**
 
 - `x-forwarded-user`: required in production; identifies the user
-- `x-forwarded-access-token`: optional; enables **user token passthrough** if `DATABRICKS_HOST` is set
+- `x-forwarded-access-token`: required for user token passthrough
+
+**Using `asUser(req)` for user-scoped operations:**
+
+The `asUser(req)` pattern allows plugins to execute operations using the requesting user's credentials:
+
+```ts
+// In a custom plugin route handler
+router.post("/users/me/data", async (req, res) => {
+  // Execute as the user (uses their Databricks permissions)
+  const result = await this.asUser(req).query("SELECT ...");
+  res.json(result);
+});
+
+// Service principal execution (default)
+router.post("/system/data", async (req, res) => {
+  const result = await this.query("SELECT ...");
+  res.json(result);
+});
+```
+
+**Context helper functions (exported from `@databricks/appkit`):**
 
-Context fields (real behavior):
+- `getExecutionContext()`: Returns current context (user or service)
+- `getCurrentUserId()`: Returns user ID in user context, service user ID otherwise
+- `getWorkspaceClient()`: Returns the appropriate WorkspaceClient for current context
+- `getWarehouseId()`: `Promise<string>` (from `DATABRICKS_WAREHOUSE_ID` or auto-selected in dev)
+- `getWorkspaceId()`: `Promise<string>` (from `DATABRICKS_WORKSPACE_ID` or fetched)
+- `isInUserContext()`: Returns `true` if currently executing in user context
 
-- `userId`: derived from `x-forwarded-user` (in development it falls back to `serviceUserId`)
-- `serviceUserId`: service principal/user ID
-- `warehouseId`: `Promise<string>` (from `DATABRICKS_WAREHOUSE_ID`, or auto-selected in development)
-- `workspaceId`: `Promise<string>` (from `DATABRICKS_WORKSPACE_ID` or fetched)
-- `userDatabricksClient`: present only when passthrough is available (or in dev it equals service client)
-- `serviceDatabricksClient`: always present
+**Development mode behavior:**
+
+In local development (`NODE_ENV=development`), if `asUser(req)` is called without a user token, it logs a warning and falls back to the service principal.
 
 ### Custom plugins (backend)
 
@@ -469,7 +495,6 @@ import type express from "express";
 class MyPlugin extends Plugin {
   name = "my-plugin";
   envVars = [];                 // list required env vars here
-  requiresDatabricksClient = false; // set true if you need getRequestContext()
 
   injectRoutes(router: express.Router) {
     this.route(router, {
@@ -643,6 +668,56 @@ export function SpendChart() {
 }
 ```
 
+**Chart props reference (important):**
+
+Charts are **self-contained ECharts components**. Configure via props, NOT children:
+
+```tsx
+// ✅ Correct: use props for customization
+<BarChart
+  queryKey="sales_by_region"
+  parameters={{}}
+  xKey="region"                    // X-axis field
+  yKey={["revenue", "expenses"]}   // Y-axis field(s) - string or string[]
+  colors={['#40d1f5', '#4462c9']}  // Custom colors
+  stacked                          // Stack bars (BarChart, AreaChart)
+  orientation="horizontal"         // "vertical" (default) | "horizontal"
+  showLegend                       // Show legend
+  height={400}                     // Height in pixels (default: 300)
+/>
+
+<LineChart
+  queryKey="trend_data"
+  parameters={{}}
+  xKey="date"
+  yKey="value"
+  smooth                           // Smooth curves (default: true)
+  showSymbol={false}               // Hide data point markers
+/>
+```
+
+**❌ CRITICAL: Charts do NOT accept Recharts children**
+
+```tsx
+// ❌ WRONG - AppKit charts are NOT Recharts wrappers
+import { BarChart } from "@databricks/appkit-ui/react";
+import { Bar, XAxis, YAxis, CartesianGrid } from "recharts";
+
+<BarChart queryKey="data" parameters={{}}>
+  <CartesianGrid />  // ❌ This will cause TypeScript errors
+  <XAxis dataKey="x" />  // ❌ Not supported
+  <Bar dataKey="y" />  // ❌ Not supported
+</BarChart>
+
+// ✅ CORRECT - use props instead
+<BarChart
+  queryKey="data"
+  parameters={{}}
+  xKey="x"
+  yKey="y"
+/>
+```
+
 ### SQL helpers (`sql.*`)
 
 Use these to build typed parameters (they return marker objects: `{ __sql_type, value }`):
@@ -1144,6 +1219,7 @@ env:
   - `useMemo` wraps parameters objects
   - Loading/error/empty states are explicit
   - Charts use `format="auto"` unless you have a reason to force `"json"`/`"arrow"`
+  - Charts use props (`xKey`, `yKey`, `colors`) NOT children (they're ECharts-based, not Recharts)
   - If using tooltips: root is wrapped with `<TooltipProvider>`
 
 - **Never**
@@ -1151,4 +1227,5 @@ env:
   - Don't pass untyped raw params for annotated queries
   - Don't ignore `createApp()`'s promise
   - Don't invent UI components not listed in this file
+  - Don't pass Recharts children (`<Bar>`, `<XAxis>`, etc.) to AppKit chart components
 

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@databricks/appkit",
   "type": "module",
-  "version": "0.1.4",
+  "version": "0.2.0",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
   "packageManager": "pnpm@10.21.0",
@@ -26,6 +26,7 @@
   },
   "bin": {
     "appkit-generate-types": "./bin/generate-types.js",
+    "appkit-lint": "./bin/appkit-lint.js",
     "appkit-setup": "./bin/setup-claude.js"
   },
   "scripts": {
@@ -39,6 +40,7 @@
     "postinstall": "node scripts/postinstall.js"
   },
   "dependencies": {
+    "@ast-grep/napi": "^0.37.0",
     "@databricks/sdk-experimental": "^0.15.0",
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/api-logs": "^0.208.0",
@@ -54,9 +56,12 @@
     "@opentelemetry/sdk-metrics": "^2.2.0",
     "@opentelemetry/sdk-node": "^0.208.0",
     "@opentelemetry/semantic-conventions": "^1.38.0",
+    "@types/semver": "^7.7.1",
     "dotenv": "^16.6.1",
     "express": "^4.22.0",
+    "obug": "^2.1.1",
     "pg": "^8.16.3",
+    "semver": "^7.7.3",
     "vite": "npm:rolldown-vite@7.1.14",
     "ws": "^8.18.3",
     "zod-to-ts": "^2.0.0"

package/dist/utils/databricks-client-middleware.d.ts

@@ -1,17 +0,0 @@
-import { WorkspaceClient } from "@databricks/sdk-experimental";
-import express from "express";
-
-//#region src/utils/databricks-client-middleware.d.ts
-type RequestContext = {
-  userDatabricksClient?: WorkspaceClient;
-  serviceDatabricksClient: WorkspaceClient;
-  userId: string;
-  userName?: string;
-  serviceUserId: string;
-  warehouseId: Promise<string>;
-  workspaceId: Promise<string>;
-};
-declare function getRequestContext(): RequestContext;
-//#endregion
-export { getRequestContext };
-//# sourceMappingURL=databricks-client-middleware.d.ts.map

package/dist/utils/databricks-client-middleware.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"databricks-client-middleware.d.ts","names":[],"sources":["../../src/utils/databricks-client-middleware.ts"],"sourcesContent":[],"mappings":";;;;KAYY,cAAA;yBACa;EADb,uBAAc,EAEC,eAFD;EAAA,MAAA,EAAA,MAAA;UACD,CAAA,EAAA,MAAA;eACE,EAAA,MAAA;aAIZ,EAAA,OAAA,CAAA,MAAA,CAAA;aACA,EAAA,OAAA,CAAA,MAAA,CAAA;CAAO;iBA4EN,iBAAA,CAAA,GAAqB"}

package/dist/utils/databricks-client-middleware.js

@@ -1,117 +0,0 @@
-import { __esmMin } from "../_virtual/rolldown_runtime.js";
-import { name, version } from "../appkit/package.js";
-import { WorkspaceClient } from "@databricks/sdk-experimental";
-import { AsyncLocalStorage } from "node:async_hooks";
-
-//#region src/utils/databricks-client-middleware.ts
-function getClientOptions() {
-	const isDev = process.env.NODE_ENV === "development";
-	return {
-		product: name,
-		productVersion: version.split(".").slice(0, 3).join("."),
-		...isDev && { userAgentExtra: { mode: "dev" } }
-	};
-}
-async function databricksClientMiddleware() {
-	const serviceDatabricksClient = new WorkspaceClient({}, getClientOptions());
-	const warehouseId = getWarehouseId(serviceDatabricksClient);
-	const workspaceId = getWorkspaceId(serviceDatabricksClient);
-	const serviceUserId = (await serviceDatabricksClient.currentUser.me()).id;
-	if (!serviceUserId) throw new Error("Service user ID not found");
-	return async (req, res, next) => {
-		const userToken = req.headers["x-forwarded-access-token"];
-		let userDatabricksClient;
-		const host = process.env.DATABRICKS_HOST;
-		if (userToken && host) userDatabricksClient = new WorkspaceClient({
-			token: userToken,
-			host,
-			authType: "pat"
-		}, getClientOptions());
-		else if (process.env.NODE_ENV === "development") userDatabricksClient = serviceDatabricksClient;
-		let userName = req.headers["x-forwarded-user"];
-		if (!userName && process.env.NODE_ENV !== "development") {
-			res.status(401).json({ error: "Unauthorized" });
-			return;
-		} else userName = serviceUserId;
-		return asyncLocalStorage.run({
-			userDatabricksClient,
-			serviceDatabricksClient,
-			warehouseId,
-			workspaceId,
-			userId: userName,
-			serviceUserId
-		}, async () => {
-			return next();
-		});
-	};
-}
-function getRequestContext() {
-	const store = asyncLocalStorage.getStore();
-	if (!store) throw new Error("Request context not found");
-	return store;
-}
-/**
-* Get the appropriate WorkspaceClient based on whether the request
-* should be executed as the user or as the service principal.
-*
-* @param asUser - If true, returns user's WorkspaceClient (requires token passthrough)
-* @throws Error if asUser is true but user token passthrough is not enabled
-*/
-function getWorkspaceClient(asUser) {
-	const context = getRequestContext();
-	if (asUser) {
-		if (!context.userDatabricksClient) throw new Error(`User token passthrough is not enabled for this workspace.`);
-		return context.userDatabricksClient;
-	}
-	return context.serviceDatabricksClient;
-}
-async function getWorkspaceId(workspaceClient) {
-	if (process.env.DATABRICKS_WORKSPACE_ID) return process.env.DATABRICKS_WORKSPACE_ID;
-	const response = await workspaceClient.apiClient.request({
-		path: "/api/2.0/preview/scim/v2/Me",
-		method: "GET",
-		headers: new Headers(),
-		raw: false,
-		query: {},
-		responseHeaders: ["x-databricks-org-id"]
-	});
-	if (!response["x-databricks-org-id"]) throw new Error("Workspace ID not found");
-	return response["x-databricks-org-id"];
-}
-async function getWarehouseId(workspaceClient) {
-	if (process.env.DATABRICKS_WAREHOUSE_ID) return process.env.DATABRICKS_WAREHOUSE_ID;
-	if (process.env.NODE_ENV === "development") {
-		const response = await workspaceClient.apiClient.request({
-			path: "/api/2.0/sql/warehouses",
-			method: "GET",
-			headers: new Headers(),
-			raw: false,
-			query: { skip_cannot_use: "true" }
-		});
-		const priorities = {
-			RUNNING: 0,
-			STOPPED: 1,
-			STARTING: 2,
-			STOPPING: 3,
-			DELETED: 99,
-			DELETING: 99
-		};
-		const warehouses = (response.warehouses || []).sort((a, b) => {
-			return priorities[a.state] - priorities[b.state];
-		});
-		if (response.warehouses.length === 0) throw new Error("Warehouse ID not found. Please configure the DATABRICKS_WAREHOUSE_ID environment variable.");
-		const firstWarehouse = warehouses[0];
-		if (firstWarehouse.state === "DELETED" || firstWarehouse.state === "DELETING" || !firstWarehouse.id) throw new Error("Warehouse ID not found. Please configure the DATABRICKS_WAREHOUSE_ID environment variable.");
-		return firstWarehouse.id;
-	}
-	throw new Error("Warehouse ID not found. Please configure the DATABRICKS_WAREHOUSE_ID environment variable.");
-}
-var asyncLocalStorage;
-var init_databricks_client_middleware = __esmMin((() => {
-	asyncLocalStorage = new AsyncLocalStorage();
-}));
-
-//#endregion
-init_databricks_client_middleware();
-export { databricksClientMiddleware, getRequestContext, getWorkspaceClient, init_databricks_client_middleware };
-//# sourceMappingURL=databricks-client-middleware.js.map

package/dist/utils/databricks-client-middleware.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"databricks-client-middleware.js","names":["productName","productVersion","userDatabricksClient: WorkspaceClient | undefined","priorities: Record<sql.State, number>"],"sources":["../../src/utils/databricks-client-middleware.ts"],"sourcesContent":["import { AsyncLocalStorage } from \"node:async_hooks\";\nimport {\n  type ClientOptions,\n  type sql,\n  WorkspaceClient,\n} from \"@databricks/sdk-experimental\";\nimport type express from \"express\";\nimport {\n  name as productName,\n  version as productVersion,\n} from \"../../package.json\";\n\nexport type RequestContext = {\n  userDatabricksClient?: WorkspaceClient;\n  serviceDatabricksClient: WorkspaceClient;\n  userId: string;\n  userName?: string;\n  serviceUserId: string;\n  warehouseId: Promise<string>;\n  workspaceId: Promise<string>;\n};\n\nconst asyncLocalStorage = new AsyncLocalStorage<RequestContext>();\n\nfunction getClientOptions(): ClientOptions {\n  const isDev = process.env.NODE_ENV === \"development\";\n  const normalizedVersion = productVersion\n    .split(\".\")\n    .slice(0, 3)\n    .join(\".\") as ClientOptions[\"productVersion\"];\n\n  return {\n    product: productName,\n    productVersion: normalizedVersion,\n    ...(isDev && { userAgentExtra: { mode: \"dev\" } }),\n  };\n}\n\nexport async function databricksClientMiddleware(): Promise<express.RequestHandler> {\n  const serviceDatabricksClient = new WorkspaceClient({}, getClientOptions());\n  const warehouseId = getWarehouseId(serviceDatabricksClient);\n  const workspaceId = getWorkspaceId(serviceDatabricksClient);\n  const serviceUserId = (await serviceDatabricksClient.currentUser.me()).id;\n\n  if (!serviceUserId) {\n    throw new Error(\"Service user ID not found\");\n  }\n\n  return async (\n    req: express.Request,\n    res: express.Response,\n    next: express.NextFunction,\n  ) => {\n    const userToken = req.headers[\"x-forwarded-access-token\"] as string;\n    let userDatabricksClient: WorkspaceClient | undefined;\n    const host = process.env.DATABRICKS_HOST;\n    if (userToken && host) {\n      userDatabricksClient = new WorkspaceClient(\n        {\n          token: userToken,\n          host,\n          authType: \"pat\",\n        },\n        getClientOptions(),\n      );\n    } else if (process.env.NODE_ENV === \"development\") {\n      // in local development service and no user token are the same\n      // TODO: use `databricks apps run-local` to fix this\n      userDatabricksClient = serviceDatabricksClient;\n    }\n\n    let userName = req.headers[\"x-forwarded-user\"] as string;\n    if (!userName && process.env.NODE_ENV !== \"development\") {\n      res.status(401).json({ error: \"Unauthorized\" });\n      return;\n    } else {\n      userName = serviceUserId;\n    }\n\n    return asyncLocalStorage.run(\n      {\n        userDatabricksClient,\n        serviceDatabricksClient,\n        warehouseId,\n        workspaceId,\n        userId: userName,\n        serviceUserId,\n      },\n      async () => {\n        return next();\n      },\n    );\n  };\n}\n\nexport function getRequestContext(): RequestContext {\n  const store = asyncLocalStorage.getStore();\n  if (!store) {\n    throw new Error(\"Request context not found\");\n  }\n  return store;\n}\n\n/**\n * Get the appropriate WorkspaceClient based on whether the request\n * should be executed as the user or as the service principal.\n *\n * @param asUser - If true, returns user's WorkspaceClient (requires token passthrough)\n * @throws Error if asUser is true but user token passthrough is not enabled\n */\nexport function getWorkspaceClient(asUser: boolean): WorkspaceClient {\n  const context = getRequestContext();\n\n  if (asUser) {\n    if (!context.userDatabricksClient) {\n      throw new Error(\n        `User token passthrough is not enabled for this workspace.`,\n      );\n    }\n    return context.userDatabricksClient;\n  }\n\n  return context.serviceDatabricksClient;\n}\n\nasync function getWorkspaceId(\n  workspaceClient: WorkspaceClient,\n): Promise<string> {\n  if (process.env.DATABRICKS_WORKSPACE_ID) {\n    return process.env.DATABRICKS_WORKSPACE_ID;\n  }\n\n  const response = (await workspaceClient.apiClient.request({\n    path: \"/api/2.0/preview/scim/v2/Me\",\n    method: \"GET\",\n    headers: new Headers(),\n    raw: false,\n    query: {},\n    responseHeaders: [\"x-databricks-org-id\"],\n  })) as { \"x-databricks-org-id\": string };\n\n  if (!response[\"x-databricks-org-id\"]) {\n    throw new Error(\"Workspace ID not found\");\n  }\n\n  return response[\"x-databricks-org-id\"];\n}\n\nasync function getWarehouseId(\n  workspaceClient: WorkspaceClient,\n): Promise<string> {\n  if (process.env.DATABRICKS_WAREHOUSE_ID) {\n    return process.env.DATABRICKS_WAREHOUSE_ID;\n  }\n\n  if (process.env.NODE_ENV === \"development\") {\n    const response = (await workspaceClient.apiClient.request({\n      path: \"/api/2.0/sql/warehouses\",\n      method: \"GET\",\n      headers: new Headers(),\n      raw: false,\n      query: {\n        skip_cannot_use: \"true\",\n      },\n    })) as { warehouses: sql.EndpointInfo[] };\n\n    const priorities: Record<sql.State, number> = {\n      RUNNING: 0,\n      STOPPED: 1,\n      STARTING: 2,\n      STOPPING: 3,\n      DELETED: 99,\n      DELETING: 99,\n    };\n\n    const warehouses = (response.warehouses || []).sort((a, b) => {\n      return (\n        priorities[a.state as sql.State] - priorities[b.state as sql.State]\n      );\n    });\n\n    if (response.warehouses.length === 0) {\n      throw new Error(\n        \"Warehouse ID not found. Please configure the DATABRICKS_WAREHOUSE_ID environment variable.\",\n      );\n    }\n\n    const firstWarehouse = warehouses[0];\n    if (\n      firstWarehouse.state === \"DELETED\" ||\n      firstWarehouse.state === \"DELETING\" ||\n      !firstWarehouse.id\n    ) {\n      throw new Error(\n        \"Warehouse ID not found. Please configure the DATABRICKS_WAREHOUSE_ID environment variable.\",\n      );\n    }\n\n    return firstWarehouse.id;\n  }\n\n  throw new Error(\n    \"Warehouse ID not found. Please configure the DATABRICKS_WAREHOUSE_ID environment variable.\",\n  );\n}\n\nexport type Request = express.Request;\nexport type Response = express.Response;\n"],"mappings":";;;;;;AAwBA,SAAS,mBAAkC;CACzC,MAAM,QAAQ,QAAQ,IAAI,aAAa;AAMvC,QAAO;EACL,SAASA;EACT,gBAPwBC,QACvB,MAAM,IAAI,CACV,MAAM,GAAG,EAAE,CACX,KAAK,IAAI;EAKV,GAAI,SAAS,EAAE,gBAAgB,EAAE,MAAM,OAAO,EAAE;EACjD;;AAGH,eAAsB,6BAA8D;CAClF,MAAM,0BAA0B,IAAI,gBAAgB,EAAE,EAAE,kBAAkB,CAAC;CAC3E,MAAM,cAAc,eAAe,wBAAwB;CAC3D,MAAM,cAAc,eAAe,wBAAwB;CAC3D,MAAM,iBAAiB,MAAM,wBAAwB,YAAY,IAAI,EAAE;AAEvE,KAAI,CAAC,cACH,OAAM,IAAI,MAAM,4BAA4B;AAG9C,QAAO,OACL,KACA,KACA,SACG;EACH,MAAM,YAAY,IAAI,QAAQ;EAC9B,IAAIC;EACJ,MAAM,OAAO,QAAQ,IAAI;AACzB,MAAI,aAAa,KACf,wBAAuB,IAAI,gBACzB;GACE,OAAO;GACP;GACA,UAAU;GACX,EACD,kBAAkB,CACnB;WACQ,QAAQ,IAAI,aAAa,cAGlC,wBAAuB;EAGzB,IAAI,WAAW,IAAI,QAAQ;AAC3B,MAAI,CAAC,YAAY,QAAQ,IAAI,aAAa,eAAe;AACvD,OAAI,OAAO,IAAI,CAAC,KAAK,EAAE,OAAO,gBAAgB,CAAC;AAC/C;QAEA,YAAW;AAGb,SAAO,kBAAkB,IACvB;GACE;GACA;GACA;GACA;GACA,QAAQ;GACR;GACD,EACD,YAAY;AACV,UAAO,MAAM;IAEhB;;;AAIL,SAAgB,oBAAoC;CAClD,MAAM,QAAQ,kBAAkB,UAAU;AAC1C,KAAI,CAAC,MACH,OAAM,IAAI,MAAM,4BAA4B;AAE9C,QAAO;;;;;;;;;AAUT,SAAgB,mBAAmB,QAAkC;CACnE,MAAM,UAAU,mBAAmB;AAEnC,KAAI,QAAQ;AACV,MAAI,CAAC,QAAQ,qBACX,OAAM,IAAI,MACR,4DACD;AAEH,SAAO,QAAQ;;AAGjB,QAAO,QAAQ;;AAGjB,eAAe,eACb,iBACiB;AACjB,KAAI,QAAQ,IAAI,wBACd,QAAO,QAAQ,IAAI;CAGrB,MAAM,WAAY,MAAM,gBAAgB,UAAU,QAAQ;EACxD,MAAM;EACN,QAAQ;EACR,SAAS,IAAI,SAAS;EACtB,KAAK;EACL,OAAO,EAAE;EACT,iBAAiB,CAAC,sBAAsB;EACzC,CAAC;AAEF,KAAI,CAAC,SAAS,uBACZ,OAAM,IAAI,MAAM,yBAAyB;AAG3C,QAAO,SAAS;;AAGlB,eAAe,eACb,iBACiB;AACjB,KAAI,QAAQ,IAAI,wBACd,QAAO,QAAQ,IAAI;AAGrB,KAAI,QAAQ,IAAI,aAAa,eAAe;EAC1C,MAAM,WAAY,MAAM,gBAAgB,UAAU,QAAQ;GACxD,MAAM;GACN,QAAQ;GACR,SAAS,IAAI,SAAS;GACtB,KAAK;GACL,OAAO,EACL,iBAAiB,QAClB;GACF,CAAC;EAEF,MAAMC,aAAwC;GAC5C,SAAS;GACT,SAAS;GACT,UAAU;GACV,UAAU;GACV,SAAS;GACT,UAAU;GACX;EAED,MAAM,cAAc,SAAS,cAAc,EAAE,EAAE,MAAM,GAAG,MAAM;AAC5D,UACE,WAAW,EAAE,SAAsB,WAAW,EAAE;IAElD;AAEF,MAAI,SAAS,WAAW,WAAW,EACjC,OAAM,IAAI,MACR,6FACD;EAGH,MAAM,iBAAiB,WAAW;AAClC,MACE,eAAe,UAAU,aACzB,eAAe,UAAU,cACzB,CAAC,eAAe,GAEhB,OAAM,IAAI,MACR,6FACD;AAGH,SAAO,eAAe;;AAGxB,OAAM,IAAI,MACR,6FACD;;;;CArLG,oBAAoB,IAAI,mBAAmC"}

package/dist/utils/index.js

@@ -1,26 +0,0 @@
-import { __esmMin, __exportAll } from "../_virtual/rolldown_runtime.js";
-import { databricksClientMiddleware, getRequestContext, getWorkspaceClient, init_databricks_client_middleware } from "./databricks-client-middleware.js";
-import { init_env_validator, validateEnv } from "./env-validator.js";
-import { deepMerge, init_merge } from "./merge.js";
-import { init_vite_config_merge, mergeConfigDedup } from "./vite-config-merge.js";
-
-//#region src/utils/index.ts
-var utils_exports = /* @__PURE__ */ __exportAll({
-	databricksClientMiddleware: () => databricksClientMiddleware,
-	deepMerge: () => deepMerge,
-	getRequestContext: () => getRequestContext,
-	getWorkspaceClient: () => getWorkspaceClient,
-	mergeConfigDedup: () => mergeConfigDedup,
-	validateEnv: () => validateEnv
-});
-var init_utils = __esmMin((() => {
-	init_databricks_client_middleware();
-	init_env_validator();
-	init_merge();
-	init_vite_config_merge();
-}));
-
-//#endregion
-init_utils();
-export { init_utils, utils_exports };
-//# sourceMappingURL=index.js.map

package/dist/utils/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","names":[],"sources":["../../src/utils/index.ts"],"sourcesContent":["export * from \"./databricks-client-middleware\";\nexport * from \"./env-validator\";\nexport * from \"./merge\";\nexport * from \"./vite-config-merge\";\n"],"mappings":""}