@einlogic/mcp-fabric-api 2.0.0 → 2.1.1

package/build/tools/git-integration.js

@@ -0,0 +1,159 @@
+import { z } from "zod";
+import { formatToolError } from "../core/errors.js";
+import { pollOperation } from "../core/lro.js";
+export function registerGitIntegrationTools(server, fabricClient, workspaceGuard) {
+    server.tool("git_get_connection", "Get the Git connection details for a workspace", { workspaceId: z.string().describe("The workspace ID") }, async ({ workspaceId }) => {
+        try {
+            const response = await fabricClient.get(`/workspaces/${workspaceId}/git/connection`);
+            return { content: [{ type: "text", text: JSON.stringify(response.data, null, 2) }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("git_get_status", "Get the Git status of items in a workspace (shows sync state between workspace and remote)", { workspaceId: z.string().describe("The workspace ID") }, async ({ workspaceId }) => {
+        try {
+            const response = await fabricClient.get(`/workspaces/${workspaceId}/git/status`);
+            return { content: [{ type: "text", text: JSON.stringify(response.data, null, 2) }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("git_connect", "Connect a workspace to a Git repository (Azure DevOps or GitHub)", {
+        workspaceId: z.string().describe("The workspace ID"),
+        gitProviderDetails: z.object({
+            gitProviderType: z.enum(["AzureDevOps", "GitHub"]).describe("Git provider type"),
+            organizationName: z.string().describe("Organization or account name"),
+            projectName: z.string().optional().describe("Project name (required for Azure DevOps)"),
+            repositoryName: z.string().describe("Repository name"),
+            branchName: z.string().describe("Branch name to connect to"),
+            directoryName: z.string().optional().describe("Directory path within the repository"),
+        }).describe("Git provider connection details"),
+    }, async ({ workspaceId, gitProviderDetails }) => {
+        try {
+            await workspaceGuard.assertWorkspaceAllowed(fabricClient, workspaceId);
+            const response = await fabricClient.post(`/workspaces/${workspaceId}/git/connect`, { gitProviderDetails });
+            return { content: [{ type: "text", text: response.data ? JSON.stringify(response.data, null, 2) : "Git connection established successfully" }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("git_disconnect", "Disconnect a workspace from its Git repository", { workspaceId: z.string().describe("The workspace ID") }, async ({ workspaceId }) => {
+        try {
+            await workspaceGuard.assertWorkspaceAllowed(fabricClient, workspaceId);
+            await fabricClient.post(`/workspaces/${workspaceId}/git/disconnect`);
+            return { content: [{ type: "text", text: "Git connection disconnected successfully" }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("git_initialize_connection", "Initialize a Git connection after connecting (long-running). Chooses whether workspace or remote content takes precedence.", {
+        workspaceId: z.string().describe("The workspace ID"),
+        initializationStrategy: z.enum(["PreferWorkspace", "PreferRemote"]).describe("Strategy for resolving conflicts during initialization"),
+    }, async ({ workspaceId, initializationStrategy }) => {
+        try {
+            await workspaceGuard.assertWorkspaceAllowed(fabricClient, workspaceId);
+            const response = await fabricClient.post(`/workspaces/${workspaceId}/git/initializeConnection`, { initializationStrategy });
+            if (response.lro) {
+                const state = await pollOperation(fabricClient, response.lro.operationId);
+                return { content: [{ type: "text", text: JSON.stringify(state, null, 2) }] };
+            }
+            return { content: [{ type: "text", text: response.data ? JSON.stringify(response.data, null, 2) : "Git connection initialized successfully" }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("git_commit_to_git", "Commit workspace changes to the connected Git repository (long-running)", {
+        workspaceId: z.string().describe("The workspace ID"),
+        mode: z.enum(["All", "Selective"]).describe("Commit mode: All items or Selective"),
+        comment: z.string().optional().describe("Commit comment/message"),
+        items: z.array(z.object({
+            objectId: z.string().describe("The item object ID"),
+            logicalId: z.string().optional().describe("The item logical ID"),
+        })).optional().describe("Items to commit (required when mode is Selective)"),
+        workspaceHead: z.string().optional().describe("Workspace head object ID for concurrency control"),
+    }, async ({ workspaceId, mode, comment, items, workspaceHead }) => {
+        try {
+            await workspaceGuard.assertWorkspaceAllowed(fabricClient, workspaceId);
+            const body = { mode };
+            if (comment)
+                body.comment = comment;
+            if (items)
+                body.items = items;
+            if (workspaceHead)
+                body.workspaceHead = workspaceHead;
+            const response = await fabricClient.post(`/workspaces/${workspaceId}/git/commitToGit`, body);
+            if (response.lro) {
+                const state = await pollOperation(fabricClient, response.lro.operationId);
+                return { content: [{ type: "text", text: JSON.stringify(state, null, 2) }] };
+            }
+            return { content: [{ type: "text", text: response.data ? JSON.stringify(response.data, null, 2) : "Changes committed to Git successfully" }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("git_update_from_git", "Update workspace from the connected Git repository (long-running)", {
+        workspaceId: z.string().describe("The workspace ID"),
+        remoteCommitHash: z.string().describe("The remote commit hash to update to"),
+        conflictResolution: z.object({
+            conflictResolutionType: z.enum(["Workspace", "Remote"]).describe("How to resolve conflicts"),
+            conflictResolutionPolicy: z.enum(["PreferWorkspace", "PreferRemote"]).optional().describe("Conflict resolution policy"),
+        }).optional().describe("Conflict resolution settings"),
+        workspaceHead: z.string().optional().describe("Workspace head object ID for concurrency control"),
+        options: z.object({
+            allowOverrideItems: z.boolean().optional().describe("Allow overriding workspace items"),
+        }).optional().describe("Update options"),
+    }, async ({ workspaceId, remoteCommitHash, conflictResolution, workspaceHead, options }) => {
+        try {
+            await workspaceGuard.assertWorkspaceAllowed(fabricClient, workspaceId);
+            const body = { remoteCommitHash };
+            if (conflictResolution)
+                body.conflictResolution = conflictResolution;
+            if (workspaceHead)
+                body.workspaceHead = workspaceHead;
+            if (options)
+                body.options = options;
+            const response = await fabricClient.post(`/workspaces/${workspaceId}/git/updateFromGit`, body);
+            if (response.lro) {
+                const state = await pollOperation(fabricClient, response.lro.operationId);
+                return { content: [{ type: "text", text: JSON.stringify(state, null, 2) }] };
+            }
+            return { content: [{ type: "text", text: response.data ? JSON.stringify(response.data, null, 2) : "Workspace updated from Git successfully" }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("git_get_credentials", "Get the Git credentials configuration for the current user in a workspace", { workspaceId: z.string().describe("The workspace ID") }, async ({ workspaceId }) => {
+        try {
+            const response = await fabricClient.get(`/workspaces/${workspaceId}/git/myGitCredentials`);
+            return { content: [{ type: "text", text: JSON.stringify(response.data, null, 2) }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("git_update_credentials", "Update the Git credentials configuration for the current user in a workspace", {
+        workspaceId: z.string().describe("The workspace ID"),
+        source: z.enum(["Automatic", "None", "ConfiguredConnection"]).describe("Credential source type"),
+        connectionId: z.string().optional().describe("Connection ID (required when source is ConfiguredConnection)"),
+    }, async ({ workspaceId, source, connectionId }) => {
+        try {
+            await workspaceGuard.assertWorkspaceAllowed(fabricClient, workspaceId);
+            const body = { source };
+            if (connectionId)
+                body.connectionId = connectionId;
+            const response = await fabricClient.patch(`/workspaces/${workspaceId}/git/myGitCredentials`, body);
+            return { content: [{ type: "text", text: response.data ? JSON.stringify(response.data, null, 2) : "Git credentials updated successfully" }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+}
+//# sourceMappingURL=git-integration.js.map

package/build/tools/git-integration.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"git-integration.js","sourceRoot":"","sources":["../../src/tools/git-integration.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAG/C,MAAM,UAAU,2BAA2B,CAAC,MAAiB,EAAE,YAA0B,EAAE,cAA8B;IACvH,MAAM,CAAC,IAAI,CACT,oBAAoB,EACpB,gDAAgD,EAChD,EAAE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,EACxD,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;QACxB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,GAAG,CAAC,eAAe,WAAW,iBAAiB,CAAC,CAAC;YACrF,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;QACvF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,gBAAgB,EAChB,4FAA4F,EAC5F,EAAE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,EACxD,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;QACxB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,GAAG,CAAC,eAAe,WAAW,aAAa,CAAC,CAAC;YACjF,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;QACvF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,aAAa,EACb,kEAAkE,EAClE;QACE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QACpD,kBAAkB,EAAE,CAAC,CAAC,MAAM,CAAC;YAC3B,eAAe,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,mBAAmB,CAAC;YAChF,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;YACrE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,0CAA0C,CAAC;YACvF,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;YACtD,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;YAC5D,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,sCAAsC,CAAC;SACtF,CAAC,CAAC,QAAQ,CAAC,iCAAiC,CAAC;KAC/C,EACD,KAAK,EAAE,EAAE,WAAW,EAAE,kBAAkB,EAAE,EAAE,EAAE;QAC5C,IAAI,CAAC;YACH,MAAM,cAAc,CAAC,sBAAsB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;YACvE,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,eAAe,WAAW,cAAc,EAAE,EAAE,kBAAkB,EAAE,CAAC,CAAC;YAC3G,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,yCAAyC,EAAE,CAAC,EAAE,CAAC;QACnJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,gBAAgB,EAChB,gDAAgD,EAChD,EAAE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,EACxD,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;QACxB,IAAI,CAAC;YACH,MAAM,cAAc,CAAC,sBAAsB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;YACvE,MAAM,YAAY,CAAC,IAAI,CAAC,eAAe,WAAW,iBAAiB,CAAC,CAAC;YACrE,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,0CAA0C,EAAE,CAAC,EAAE,CAAC;QAC3F,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,2BAA2B,EAC3B,4HAA4H,EAC5H;QACE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QACpD,sBAAsB,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC,CAAC,QAAQ,CAAC,wDAAwD,CAAC;KACvI,EACD,KAAK,EAAE,EAAE,WAAW,EAAE,sBAAsB,EAAE,EAAE,EAAE;QAChD,IAAI,CAAC;YACH,MAAM,cAAc,CAAC,sBAAsB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;YACvE,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,eAAe,WAAW,2BAA2B,EAAE,EAAE,sBAAsB,EAAE,CAAC,CAAC;YAC5H,IAAI,QAAQ,CAAC,GAAG,EAAE,CAAC;gBACjB,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,YAAY,EAAE,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAC1E,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;YAC/E,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,yCAAyC,EAAE,CAAC,EAAE,CAAC;QACnJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,mBAAmB,EACnB,yEAAyE,EACzE;QACE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QACpD,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,qCAAqC,CAAC;QAClF,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;QACjE,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC;YACtB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,oBAAoB,CAAC;YACnD,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;SACjE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mDAAmD,CAAC;QAC5E,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kDAAkD,CAAC;KAClG,EACD,KAAK,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE,EAAE;QAC7D,IAAI,CAAC;YACH,MAAM,cAAc,CAAC,sBAAsB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;YACvE,MAAM,IAAI,GAA4B,EAAE,IAAI,EAAE,CAAC;YAC/C,IAAI,OAAO;gBAAE,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;YACpC,IAAI,KAAK;gBAAE,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;YAC9B,IAAI,aAAa;gBAAE,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;YACtD,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,eAAe,WAAW,kBAAkB,EAAE,IAAI,CAAC,CAAC;YAC7F,IAAI,QAAQ,CAAC,GAAG,EAAE,CAAC;gBACjB,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,YAAY,EAAE,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAC1E,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;YAC/E,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,uCAAuC,EAAE,CAAC,EAAE,CAAC;QACjJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,qBAAqB,EACrB,mEAAmE,EACnE;QACE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QACpD,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qCAAqC,CAAC;QAC5E,kBAAkB,EAAE,CAAC,CAAC,MAAM,CAAC;YAC3B,sBAAsB,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,0BAA0B,CAAC;YAC5F,wBAAwB,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4BAA4B,CAAC;SACxH,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;QACtD,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kDAAkD,CAAC;QACjG,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC;YAChB,kBAAkB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kCAAkC,CAAC;SACxF,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC;KACzC,EACD,KAAK,EAAE,EAAE,WAAW,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,aAAa,EAAE,OAAO,EAAE,EAAE,EAAE;QACtF,IAAI,CAAC;YACH,MAAM,cAAc,CAAC,sBAAsB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;YACvE,MAAM,IAAI,GAA4B,EAAE,gBAAgB,EAAE,CAAC;YAC3D,IAAI,kBAAkB;gBAAE,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;YACrE,IAAI,aAAa;gBAAE,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;YACtD,IAAI,OAAO;gBAAE,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;YACpC,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,eAAe,WAAW,oBAAoB,EAAE,IAAI,CAAC,CAAC;YAC/F,IAAI,QAAQ,CAAC,GAAG,EAAE,CAAC;gBACjB,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,YAAY,EAAE,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAC1E,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;YAC/E,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,yCAAyC,EAAE,CAAC,EAAE,CAAC;QACnJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,qBAAqB,EACrB,2EAA2E,EAC3E,EAAE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,EACxD,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;QACxB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,GAAG,CAAC,eAAe,WAAW,uBAAuB,CAAC,CAAC;YAC3F,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;QACvF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,wBAAwB,EACxB,8EAA8E,EAC9E;QACE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QACpD,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,MAAM,EAAE,sBAAsB,CAAC,CAAC,CAAC,QAAQ,CAAC,wBAAwB,CAAC;QAChG,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8DAA8D,CAAC;KAC7G,EACD,KAAK,EAAE,EAAE,WAAW,EAAE,MAAM,EAAE,YAAY,EAAE,EAAE,EAAE;QAC9C,IAAI,CAAC;YACH,MAAM,cAAc,CAAC,sBAAsB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;YACvE,MAAM,IAAI,GAA4B,EAAE,MAAM,EAAE,CAAC;YACjD,IAAI,YAAY;gBAAE,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;YACnD,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,KAAK,CAAC,eAAe,WAAW,uBAAuB,EAAE,IAAI,CAAC,CAAC;YACnG,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,sCAAsC,EAAE,CAAC,EAAE,CAAC;QAChJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC"}

package/build/tools/kql-database.d.ts

@@ -0,0 +1,5 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { FabricClient } from "../client/fabric-client.js";
+import { WorkspaceGuard } from "../core/workspace-guard.js";
+export declare function registerKqlDatabaseTools(server: McpServer, fabricClient: FabricClient, workspaceGuard: WorkspaceGuard): void;
+//# sourceMappingURL=kql-database.d.ts.map

package/build/tools/kql-database.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"kql-database.d.ts","sourceRoot":"","sources":["../../src/tools/kql-database.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEpE,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAK1D,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAI5D,wBAAgB,wBAAwB,CAAC,MAAM,EAAE,SAAS,EAAE,YAAY,EAAE,YAAY,EAAE,cAAc,EAAE,cAAc,QAuLrH"}

package/build/tools/kql-database.js

@@ -0,0 +1,154 @@
+import { z } from "zod";
+import { formatToolError } from "../core/errors.js";
+import { paginateAll } from "../core/pagination.js";
+import { pollOperation, getOperationResult } from "../core/lro.js";
+import { encodeBase64, decodeBase64 } from "../utils/base64.js";
+import { resolveFilesOrDirectory, writeFilesToDirectory } from "../utils/file-utils.js";
+export function registerKqlDatabaseTools(server, fabricClient, workspaceGuard) {
+    server.tool("kql_database_list", "List all KQL databases in a workspace", { workspaceId: z.string().describe("The workspace ID") }, async ({ workspaceId }) => {
+        try {
+            const items = await paginateAll(fabricClient, `/workspaces/${workspaceId}/kqlDatabases`);
+            return { content: [{ type: "text", text: JSON.stringify(items, null, 2) }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("kql_database_get", "Get details of a specific KQL database", {
+        workspaceId: z.string().describe("The workspace ID"),
+        kqlDatabaseId: z.string().describe("The KQL database ID"),
+    }, async ({ workspaceId, kqlDatabaseId }) => {
+        try {
+            const response = await fabricClient.get(`/workspaces/${workspaceId}/kqlDatabases/${kqlDatabaseId}`);
+            return { content: [{ type: "text", text: JSON.stringify(response.data, null, 2) }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("kql_database_create", "Create a new KQL database in a workspace (long-running). Requires a parent eventhouse.", {
+        workspaceId: z.string().describe("The workspace ID"),
+        displayName: z.string().describe("Display name for the KQL database"),
+        description: z.string().optional().describe("Description of the KQL database"),
+        databaseType: z.enum(["ReadWrite", "ReadOnlyShortcut"]).describe("Database type"),
+        parentEventhouseItemId: z.string().describe("Parent eventhouse item ID"),
+    }, async ({ workspaceId, displayName, description, databaseType, parentEventhouseItemId }) => {
+        try {
+            await workspaceGuard.assertWorkspaceAllowed(fabricClient, workspaceId);
+            const body = {
+                displayName,
+                creationPayload: { databaseType, parentEventhouseItemId },
+            };
+            if (description)
+                body.description = description;
+            const response = await fabricClient.post(`/workspaces/${workspaceId}/kqlDatabases`, body);
+            if (response.lro) {
+                const state = await pollOperation(fabricClient, response.lro.operationId);
+                const result = await getOperationResult(fabricClient, response.lro.operationId);
+                return { content: [{ type: "text", text: JSON.stringify({ operation: state, item: result ?? response.data }, null, 2) }] };
+            }
+            return { content: [{ type: "text", text: JSON.stringify(response.data, null, 2) }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("kql_database_update", "Update a KQL database's name or description", {
+        workspaceId: z.string().describe("The workspace ID"),
+        kqlDatabaseId: z.string().describe("The KQL database ID"),
+        displayName: z.string().optional().describe("New display name"),
+        description: z.string().optional().describe("New description"),
+    }, async ({ workspaceId, kqlDatabaseId, displayName, description }) => {
+        try {
+            await workspaceGuard.assertWorkspaceAllowed(fabricClient, workspaceId);
+            const body = {};
+            if (displayName !== undefined)
+                body.displayName = displayName;
+            if (description !== undefined)
+                body.description = description;
+            const response = await fabricClient.patch(`/workspaces/${workspaceId}/kqlDatabases/${kqlDatabaseId}`, body);
+            return { content: [{ type: "text", text: JSON.stringify(response.data, null, 2) }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("kql_database_delete", "Delete a KQL database", {
+        workspaceId: z.string().describe("The workspace ID"),
+        kqlDatabaseId: z.string().describe("The KQL database ID"),
+    }, async ({ workspaceId, kqlDatabaseId }) => {
+        try {
+            await workspaceGuard.assertWorkspaceAllowed(fabricClient, workspaceId);
+            await fabricClient.delete(`/workspaces/${workspaceId}/kqlDatabases/${kqlDatabaseId}`);
+            return { content: [{ type: "text", text: `KQL database ${kqlDatabaseId} deleted successfully` }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("kql_database_get_definition", "Get the definition of a KQL database (long-running). Writes definition files to the specified output directory.", {
+        workspaceId: z.string().describe("The workspace ID"),
+        kqlDatabaseId: z.string().describe("The KQL database ID"),
+        outputDirectoryPath: z.string().describe("Directory path where definition files will be written"),
+    }, async ({ workspaceId, kqlDatabaseId, outputDirectoryPath }) => {
+        try {
+            const response = await fabricClient.post(`/workspaces/${workspaceId}/kqlDatabases/${kqlDatabaseId}/getDefinition`);
+            let parts;
+            if (response.lro) {
+                await pollOperation(fabricClient, response.lro.operationId);
+                const result = await getOperationResult(fabricClient, response.lro.operationId);
+                if (result?.definition) {
+                    parts = result.definition.parts;
+                }
+            }
+            if (!parts && response.data?.definition) {
+                parts = response.data.definition.parts;
+            }
+            if (!parts) {
+                return { content: [{ type: "text", text: "No definition returned from Fabric API" }], isError: true };
+            }
+            const files = parts.map((part) => ({
+                path: part.path,
+                content: part.payloadType === "InlineBase64" ? decodeBase64(part.payload) : part.payload,
+            }));
+            const written = await writeFilesToDirectory(outputDirectoryPath, files);
+            return { content: [{ type: "text", text: `KQL database definition written to: ${outputDirectoryPath}\nFiles:\n${written.map((f) => `  ${f}`).join("\n")}` }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("kql_database_update_definition", "Update a KQL database's definition (long-running). Accepts definition parts inline or a directory path.", {
+        workspaceId: z.string().describe("The workspace ID"),
+        kqlDatabaseId: z.string().describe("The KQL database ID"),
+        parts: z.array(z.object({
+            path: z.string().describe("The definition part path"),
+            content: z.string().describe("The file content as a string"),
+        })).optional().describe("Array of definition parts to upload"),
+        partsDirectoryPath: z.string().optional().describe("Path to a directory containing definition files"),
+    }, async ({ workspaceId, kqlDatabaseId, parts, partsDirectoryPath }) => {
+        try {
+            await workspaceGuard.assertWorkspaceAllowed(fabricClient, workspaceId);
+            const resolved = await resolveFilesOrDirectory(parts, partsDirectoryPath);
+            const body = {
+                definition: {
+                    parts: resolved.map((part) => ({
+                        path: part.path,
+                        payload: encodeBase64(part.content),
+                        payloadType: "InlineBase64",
+                    })),
+                },
+            };
+            const response = await fabricClient.post(`/workspaces/${workspaceId}/kqlDatabases/${kqlDatabaseId}/updateDefinition`, body);
+            if (response.lro) {
+                const state = await pollOperation(fabricClient, response.lro.operationId);
+                return { content: [{ type: "text", text: JSON.stringify(state, null, 2) }] };
+            }
+            return { content: [{ type: "text", text: "KQL database definition updated successfully" }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+}
+//# sourceMappingURL=kql-database.js.map

package/build/tools/kql-database.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"kql-database.js","sourceRoot":"","sources":["../../src/tools/kql-database.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACnE,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAEhE,OAAO,EAAE,uBAAuB,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC;AAGxF,MAAM,UAAU,wBAAwB,CAAC,MAAiB,EAAE,YAA0B,EAAE,cAA8B;IACpH,MAAM,CAAC,IAAI,CACT,mBAAmB,EACnB,uCAAuC,EACvC,EAAE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,EACxD,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;QACxB,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,WAAW,CAAC,YAAY,EAAE,eAAe,WAAW,eAAe,CAAC,CAAC;YACzF,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;QAC/E,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,kBAAkB,EAClB,wCAAwC,EACxC;QACE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QACpD,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;KAC1D,EACD,KAAK,EAAE,EAAE,WAAW,EAAE,aAAa,EAAE,EAAE,EAAE;QACvC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,GAAG,CAAC,eAAe,WAAW,iBAAiB,aAAa,EAAE,CAAC,CAAC;YACpG,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;QACvF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,qBAAqB,EACrB,wFAAwF,EACxF;QACE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QACpD,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;QACrE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iCAAiC,CAAC;QAC9E,YAAY,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,kBAAkB,CAAC,CAAC,CAAC,QAAQ,CAAC,eAAe,CAAC;QACjF,sBAAsB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;KACzE,EACD,KAAK,EAAE,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,YAAY,EAAE,sBAAsB,EAAE,EAAE,EAAE;QACxF,IAAI,CAAC;YACH,MAAM,cAAc,CAAC,sBAAsB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;YACvE,MAAM,IAAI,GAA4B;gBACpC,WAAW;gBACX,eAAe,EAAE,EAAE,YAAY,EAAE,sBAAsB,EAAE;aAC1D,CAAC;YACF,IAAI,WAAW;gBAAE,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;YAChD,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,eAAe,WAAW,eAAe,EAAE,IAAI,CAAC,CAAC;YAC1F,IAAI,QAAQ,CAAC,GAAG,EAAE,CAAC;gBACjB,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,YAAY,EAAE,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAC1E,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,YAAY,EAAE,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAChF,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,IAAI,QAAQ,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;YAC7H,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;QACvF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,qBAAqB,EACrB,6CAA6C,EAC7C;QACE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QACpD,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;QACzD,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QAC/D,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;KAC/D,EACD,KAAK,EAAE,EAAE,WAAW,EAAE,aAAa,EAAE,WAAW,EAAE,WAAW,EAAE,EAAE,EAAE;QACjE,IAAI,CAAC;YACH,MAAM,cAAc,CAAC,sBAAsB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;YACvE,MAAM,IAAI,GAA4B,EAAE,CAAC;YACzC,IAAI,WAAW,KAAK,SAAS;gBAAE,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;YAC9D,IAAI,WAAW,KAAK,SAAS;gBAAE,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;YAC9D,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,KAAK,CAAC,eAAe,WAAW,iBAAiB,aAAa,EAAE,EAAE,IAAI,CAAC,CAAC;YAC5G,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;QACvF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,qBAAqB,EACrB,uBAAuB,EACvB;QACE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QACpD,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;KAC1D,EACD,KAAK,EAAE,EAAE,WAAW,EAAE,aAAa,EAAE,EAAE,EAAE;QACvC,IAAI,CAAC;YACH,MAAM,cAAc,CAAC,sBAAsB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;YACvE,MAAM,YAAY,CAAC,MAAM,CAAC,eAAe,WAAW,iBAAiB,aAAa,EAAE,CAAC,CAAC;YACtF,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,gBAAgB,aAAa,uBAAuB,EAAE,CAAC,EAAE,CAAC;QACrG,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,6BAA6B,EAC7B,iHAAiH,EACjH;QACE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QACpD,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;QACzD,mBAAmB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uDAAuD,CAAC;KAClG,EACD,KAAK,EAAE,EAAE,WAAW,EAAE,aAAa,EAAE,mBAAmB,EAAE,EAAE,EAAE;QAC5D,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,IAAI,CACtC,eAAe,WAAW,iBAAiB,aAAa,gBAAgB,CACzE,CAAC;YAEF,IAAI,KAA4B,CAAC;YACjC,IAAI,QAAQ,CAAC,GAAG,EAAE,CAAC;gBACjB,MAAM,aAAa,CAAC,YAAY,EAAE,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAC5D,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAA0B,YAAY,EAAE,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBACzG,IAAI,MAAM,EAAE,UAAU,EAAE,CAAC;oBACvB,KAAK,GAAI,MAAM,CAAC,UAAmC,CAAC,KAAK,CAAC;gBAC5D,CAAC;YACH,CAAC;YACD,IAAI,CAAC,KAAK,IAAI,QAAQ,CAAC,IAAI,EAAE,UAAU,EAAE,CAAC;gBACxC,KAAK,GAAI,QAAQ,CAAC,IAAI,CAAC,UAAmC,CAAC,KAAK,CAAC;YACnE,CAAC;YACD,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,wCAAwC,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YACxG,CAAC;YACD,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACjC,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,OAAO,EAAE,IAAI,CAAC,WAAW,KAAK,cAAc,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO;aACzF,CAAC,CAAC,CAAC;YACJ,MAAM,OAAO,GAAG,MAAM,qBAAqB,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAC;YACxE,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,uCAAuC,mBAAmB,aAAa,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;QACjK,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,IAAI,CACT,gCAAgC,EAChC,yGAAyG,EACzG;QACE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QACpD,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;QACzD,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC;YACtB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,0BAA0B,CAAC;YACrD,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;SAC7D,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,qCAAqC,CAAC;QAC9D,kBAAkB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iDAAiD,CAAC;KACtG,EACD,KAAK,EAAE,EAAE,WAAW,EAAE,aAAa,EAAE,KAAK,EAAE,kBAAkB,EAAE,EAAE,EAAE;QAClE,IAAI,CAAC;YACH,MAAM,cAAc,CAAC,sBAAsB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;YACvE,MAAM,QAAQ,GAAgB,MAAM,uBAAuB,CAAC,KAAK,EAAE,kBAAkB,CAAC,CAAC;YACvF,MAAM,IAAI,GAAG;gBACX,UAAU,EAAE;oBACV,KAAK,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;wBAC7B,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,OAAO,EAAE,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC;wBACnC,WAAW,EAAE,cAAc;qBAC5B,CAAC,CAAC;iBACJ;aACF,CAAC;YACF,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,IAAI,CACtC,eAAe,WAAW,iBAAiB,aAAa,mBAAmB,EAC3E,IAAI,CACL,CAAC;YACF,IAAI,QAAQ,CAAC,GAAG,EAAE,CAAC;gBACjB,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,YAAY,EAAE,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAC1E,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;YAC/E,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,8CAA8C,EAAE,CAAC,EAAE,CAAC;QAC/F,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,eAAe,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC"}

package/build/tools/lakehouse.d.ts

@@ -1,5 +1,6 @@
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { FabricClient } from "../client/fabric-client.js";
+import { SqlClient } from "../client/sql-client.js";
 import { WorkspaceGuard } from "../core/workspace-guard.js";
-export declare function registerLakehouseTools(server: McpServer, fabricClient: FabricClient, workspaceGuard: WorkspaceGuard): void;
+export declare function registerLakehouseTools(server: McpServer, fabricClient: FabricClient, sqlClient: SqlClient, workspaceGuard: WorkspaceGuard): void;
 //# sourceMappingURL=lakehouse.d.ts.map

package/build/tools/lakehouse.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"lakehouse.d.ts","sourceRoot":"","sources":["../../src/tools/lakehouse.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEpE,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAI1D,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAE5D,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,SAAS,EAAE,YAAY,EAAE,YAAY,EAAE,cAAc,EAAE,cAAc,QA4QnH"}
+{"version":3,"file":"lakehouse.d.ts","sourceRoot":"","sources":["../../src/tools/lakehouse.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEpE,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AAKpD,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAS5D,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,SAAS,EAAE,YAAY,EAAE,YAAY,EAAE,SAAS,EAAE,SAAS,EAAE,cAAc,EAAE,cAAc,QAiczI"}

package/build/tools/lakehouse.js

@@ -1,8 +1,14 @@
 import { z } from "zod";
-import { formatToolError } from "../core/errors.js";
+import { FabricApiError, formatToolError } from "../core/errors.js";
 import { paginateAll } from "../core/pagination.js";
-import { pollOperation } from "../core/lro.js";
-export function registerLakehouseTools(server, fabricClient, workspaceGuard) {
+import { pollOperation, getOperationResult } from "../core/lro.js";
+import { encodeBase64, decodeBase64 } from "../utils/base64.js";
+import { resolveFilesOrDirectory, writeFilesToDirectory } from "../utils/file-utils.js";
+function isSchemaEnabledError(error) {
+    return error instanceof FabricApiError &&
+        error.errorCode === "UnsupportedOperationForSchemasEnabledLakehouse";
+}
+export function registerLakehouseTools(server, fabricClient, sqlClient, workspaceGuard) {
     server.tool("lakehouse_list", "List all lakehouses in a workspace", { workspaceId: z.string().describe("The workspace ID") }, async ({ workspaceId }) => {
         try {
             const lakehouses = await paginateAll(fabricClient, `/workspaces/${workspaceId}/lakehouses`);
@@ -81,7 +87,7 @@ export function registerLakehouseTools(server, fabricClient, workspaceGuard) {
             return formatToolError(error);
         }
     });
-    server.tool("lakehouse_list_tables", "List all tables in a lakehouse", {
+    server.tool("lakehouse_list_tables", "List all tables in a lakehouse. For schema-enabled lakehouses, automatically falls back to querying INFORMATION_SCHEMA via the SQL endpoint.", {
         workspaceId: z.string().describe("The workspace ID"),
         lakehouseId: z.string().describe("The lakehouse ID"),
     }, async ({ workspaceId, lakehouseId }) => {
@@ -90,10 +96,38 @@ export function registerLakehouseTools(server, fabricClient, workspaceGuard) {
             return { content: [{ type: "text", text: JSON.stringify(tables, null, 2) }] };
         }
         catch (error) {
-            return formatToolError(error);
+            if (!isSchemaEnabledError(error)) {
+                return formatToolError(error);
+            }
+            // Fallback: query INFORMATION_SCHEMA via SQL endpoint for schema-enabled lakehouses
+            try {
+                const response = await fabricClient.get(`/workspaces/${workspaceId}/lakehouses/${lakehouseId}`);
+                const displayName = response.data.displayName;
+                const properties = response.data.properties;
+                const sqlEndpointProps = properties?.sqlEndpointProperties;
+                const connectionString = sqlEndpointProps?.connectionString;
+                if (!connectionString) {
+                    return {
+                        content: [{ type: "text", text: "This is a schema-enabled lakehouse and its SQL endpoint is not yet available. The endpoint may still be provisioning. Try again shortly or use sql_endpoint_execute_query once it's ready." }],
+                        isError: true,
+                    };
+                }
+                const server = connectionString.replace(/^.*?:\/\//, "").replace(/,.*$/, "").replace(/;.*$/, "");
+                const dbName = displayName ?? lakehouseId;
+                const result = await sqlClient.executeQuery(server, dbName, "SELECT TABLE_SCHEMA, TABLE_NAME, TABLE_TYPE FROM INFORMATION_SCHEMA.TABLES ORDER BY TABLE_SCHEMA, TABLE_NAME");
+                return {
+                    content: [{
+                            type: "text",
+                            text: `Schema-enabled lakehouse — tables retrieved via SQL endpoint:\n${JSON.stringify(result, null, 2)}`,
+                        }],
+                };
+            }
+            catch (fallbackError) {
+                return formatToolError(fallbackError);
+            }
         }
     });
-    server.tool("lakehouse_load_table", "Load data into a lakehouse table from a file path (long-running operation)", {
+    server.tool("lakehouse_load_table", "Load data into a lakehouse table from a file path (long-running operation). Not supported for schema-enabled lakehouses — use sql_endpoint_execute_query with COPY INTO or notebooks instead.", {
         workspaceId: z.string().describe("The workspace ID"),
         lakehouseId: z.string().describe("The lakehouse ID"),
         tableName: z.string().describe("Target table name"),
@@ -121,6 +155,12 @@ export function registerLakehouseTools(server, fabricClient, workspaceGuard) {
             return { content: [{ type: "text", text: "Table load initiated successfully" }] };
         }
         catch (error) {
+            if (isSchemaEnabledError(error)) {
+                return {
+                    content: [{ type: "text", text: "The table load API is not supported for schema-enabled lakehouses. Alternatives:\n• Use sql_endpoint_execute_query with a COPY INTO statement to load data via the SQL endpoint.\n• Use a notebook (notebook_run) with Spark to load data.\n• Recreate the lakehouse with enableSchemas=false if schemas are not needed." }],
+                    isError: true,
+                };
+            }
             return formatToolError(error);
         }
     });
@@ -222,5 +262,110 @@ export function registerLakehouseTools(server, fabricClient, workspaceGuard) {
             return formatToolError(error);
         }
     });
+    server.tool("lakehouse_get_definition", "Get the definition of a lakehouse (long-running). Writes definition files to the specified output directory.", {
+        workspaceId: z.string().describe("The workspace ID"),
+        lakehouseId: z.string().describe("The lakehouse ID"),
+        outputDirectoryPath: z.string().describe("Directory path where definition files will be written"),
+    }, async ({ workspaceId, lakehouseId, outputDirectoryPath }) => {
+        try {
+            const response = await fabricClient.post(`/workspaces/${workspaceId}/lakehouses/${lakehouseId}/getDefinition`);
+            let parts;
+            if (response.lro) {
+                await pollOperation(fabricClient, response.lro.operationId);
+                const result = await getOperationResult(fabricClient, response.lro.operationId);
+                if (result?.definition) {
+                    parts = result.definition.parts;
+                }
+            }
+            if (!parts && response.data?.definition) {
+                parts = response.data.definition.parts;
+            }
+            if (!parts) {
+                return { content: [{ type: "text", text: "No definition returned from Fabric API" }], isError: true };
+            }
+            const files = parts.map((part) => ({
+                path: part.path,
+                content: part.payloadType === "InlineBase64" ? decodeBase64(part.payload) : part.payload,
+            }));
+            const written = await writeFilesToDirectory(outputDirectoryPath, files);
+            return { content: [{ type: "text", text: `Lakehouse definition written to: ${outputDirectoryPath}\nFiles:\n${written.map((f) => `  ${f}`).join("\n")}` }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("lakehouse_update_definition", "Update a lakehouse's definition (long-running). Accepts definition parts inline or a directory path.", {
+        workspaceId: z.string().describe("The workspace ID"),
+        lakehouseId: z.string().describe("The lakehouse ID"),
+        parts: z.array(z.object({
+            path: z.string().describe("The definition part path"),
+            content: z.string().describe("The file content as a string"),
+        })).optional().describe("Array of definition parts to upload"),
+        partsDirectoryPath: z.string().optional().describe("Path to a directory containing definition files"),
+    }, async ({ workspaceId, lakehouseId, parts, partsDirectoryPath }) => {
+        try {
+            await workspaceGuard.assertWorkspaceAllowed(fabricClient, workspaceId);
+            const resolved = await resolveFilesOrDirectory(parts, partsDirectoryPath);
+            const body = {
+                definition: {
+                    parts: resolved.map((part) => ({
+                        path: part.path,
+                        payload: encodeBase64(part.content),
+                        payloadType: "InlineBase64",
+                    })),
+                },
+            };
+            const response = await fabricClient.post(`/workspaces/${workspaceId}/lakehouses/${lakehouseId}/updateDefinition`, body);
+            if (response.lro) {
+                const state = await pollOperation(fabricClient, response.lro.operationId);
+                return { content: [{ type: "text", text: JSON.stringify(state, null, 2) }] };
+            }
+            return { content: [{ type: "text", text: "Lakehouse definition updated successfully" }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("lakehouse_list_shortcuts", "List all OneLake shortcuts in a lakehouse", {
+        workspaceId: z.string().describe("The workspace ID"),
+        lakehouseId: z.string().describe("The lakehouse ID"),
+    }, async ({ workspaceId, lakehouseId }) => {
+        try {
+            const shortcuts = await paginateAll(fabricClient, `/workspaces/${workspaceId}/items/${lakehouseId}/shortcuts`);
+            return { content: [{ type: "text", text: JSON.stringify(shortcuts, null, 2) }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("lakehouse_get_shortcut", "Get details of a specific OneLake shortcut in a lakehouse", {
+        workspaceId: z.string().describe("The workspace ID"),
+        lakehouseId: z.string().describe("The lakehouse ID"),
+        shortcutPath: z.string().describe("The shortcut path (e.g., 'Tables' or 'Files/landingZone')"),
+        shortcutName: z.string().describe("The shortcut name"),
+    }, async ({ workspaceId, lakehouseId, shortcutPath, shortcutName }) => {
+        try {
+            const response = await fabricClient.get(`/workspaces/${workspaceId}/items/${lakehouseId}/shortcuts/${shortcutPath}/${shortcutName}`);
+            return { content: [{ type: "text", text: JSON.stringify(response.data, null, 2) }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
+    server.tool("lakehouse_delete_shortcut", "Delete a OneLake shortcut from a lakehouse", {
+        workspaceId: z.string().describe("The workspace ID"),
+        lakehouseId: z.string().describe("The lakehouse ID"),
+        shortcutPath: z.string().describe("The shortcut path (e.g., 'Tables' or 'Files/landingZone')"),
+        shortcutName: z.string().describe("The shortcut name"),
+    }, async ({ workspaceId, lakehouseId, shortcutPath, shortcutName }) => {
+        try {
+            await workspaceGuard.assertWorkspaceAllowed(fabricClient, workspaceId);
+            await fabricClient.delete(`/workspaces/${workspaceId}/items/${lakehouseId}/shortcuts/${shortcutPath}/${shortcutName}`);
+            return { content: [{ type: "text", text: `Shortcut ${shortcutName} at ${shortcutPath} deleted successfully` }] };
+        }
+        catch (error) {
+            return formatToolError(error);
+        }
+    });
 }
 //# sourceMappingURL=lakehouse.js.map