@microsoft/teamsfx-core 3.0.10-alpha.5aa2874bb.0 → 3.0.10-alpha.6bfb69e98.0

package/build/core/FxCore.declarativeAgent.js

@@ -0,0 +1,210 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT license.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FxCoreDeclarativeAgentPart = void 0;
+const tslib_1 = require("tslib");
+const hooks_1 = require("@feathersjs/hooks");
+const teamsfx_api_1 = require("@microsoft/teamsfx-api");
+const axios_1 = tslib_1.__importDefault(require("axios"));
+const fs_extra_1 = tslib_1.__importDefault(require("fs-extra"));
+const path = tslib_1.__importStar(require("path"));
+const globalVars_1 = require("../common/globalVars");
+const localizeUtils_1 = require("../common/localizeUtils");
+const actionInjector_1 = require("../component/configManager/actionInjector");
+const constants_1 = require("../component/constants");
+const questionMW_1 = require("../component/middleware/questionMW");
+const pathUtils_1 = require("../component/utils/pathUtils");
+const constants_2 = require("../question/constants");
+const concurrentLocker_1 = require("./middleware/concurrentLocker");
+const errorHandler_1 = require("./middleware/errorHandler");
+class FxCoreDeclarativeAgentPart {
+    async updateActionWithMCP(inputs) {
+        var _a, _b, _c, _d;
+        const context = (0, globalVars_1.createContext)();
+        const projectPath = inputs.projectPath;
+        if (!projectPath) {
+            throw new Error("projectPath is undefined"); // should never happen
+        }
+        const aiPluginFilePath = inputs[constants_2.QuestionNames.PluginManifestFilePath];
+        if (!(await fs_extra_1.default.pathExists(aiPluginFilePath))) {
+            const error = new teamsfx_api_1.SystemError("MCPForDAPluginManifestNotFound", "PluginManifestNotFound", (0, localizeUtils_1.getDefaultString)("core.MCPForDA.pluginManifestNotFound", aiPluginFilePath), (0, localizeUtils_1.getLocalizedString)("core.MCPForDA.pluginManifestNotFound", aiPluginFilePath));
+            return (0, teamsfx_api_1.err)(error);
+        }
+        const aiPluginFilePathRelative = path.basename(aiPluginFilePath);
+        const mcpServerUrl = inputs[constants_2.QuestionNames.MCPForDAServerUrl];
+        const serverName = inputs[constants_2.QuestionNames.MCPForDAServerName];
+        const mcpAuth = inputs[constants_2.QuestionNames.MCPForDAAuth];
+        const authType = inputs[constants_2.QuestionNames.MCPForDAAuthType];
+        let oauthAuthorizationUrl = undefined;
+        let oauthTokenUrl = undefined;
+        let oauthRefreshUrl = undefined;
+        let registrationId = undefined;
+        if (mcpAuth === "OAuthPluginVault") {
+            try {
+                registrationId = `MCP_DA_AUTH_ID_${serverName.toUpperCase()}`;
+                if (authType === "oauth") {
+                    let wellKnownMetadataUrl = inputs[constants_2.QuestionNames.MCPForDAAuthWellKnownUrl];
+                    if (!wellKnownMetadataUrl) {
+                        const mcpAuthMetadataUrl = inputs[constants_2.QuestionNames.MCPForDAAuthMetadataUrl];
+                        if (!mcpAuthMetadataUrl) {
+                            throw new Error((0, localizeUtils_1.getLocalizedString)("core.MCPForDA.mcpAuthMetadataUrlNotFound"));
+                        }
+                        const response = await axios_1.default.get(mcpAuthMetadataUrl);
+                        if (response.status === 200 &&
+                            response.data &&
+                            response.data.authorization_servers &&
+                            response.data.authorization_servers.length > 0) {
+                            const mcpServerMetadataUrl = (_a = response.data.authorization_servers) === null || _a === void 0 ? void 0 : _a[0];
+                            // Transform the URL to the proper OAuth authorization server metadata endpoint
+                            // According to RFC 8414, the well-known endpoint should be constructed as:
+                            // https://{domain}/.well-known/oauth-authorization-server{path}
+                            const serverUrl = new URL(mcpServerMetadataUrl);
+                            wellKnownMetadataUrl = `${serverUrl.protocol}//${serverUrl.host}/.well-known/oauth-authorization-server${serverUrl.pathname}`;
+                        }
+                        else {
+                            throw new Error((0, localizeUtils_1.getLocalizedString)("core.MCPForDA.mcpServerMetadataUrlNotFound"));
+                        }
+                    }
+                    const metadataResponse = await axios_1.default.get(wellKnownMetadataUrl);
+                    if (metadataResponse.status === 200) {
+                        oauthAuthorizationUrl = metadataResponse.data.authorization_endpoint;
+                        oauthTokenUrl = metadataResponse.data.token_endpoint;
+                        oauthRefreshUrl = metadataResponse.data.refresh_endpoint;
+                    }
+                    if (!oauthAuthorizationUrl || !oauthTokenUrl) {
+                        throw new Error((0, localizeUtils_1.getLocalizedString)("core.MCPForDA.authUrlNotFound"));
+                    }
+                }
+            }
+            catch (error) {
+                void context.userInteraction.showMessage("error", (0, localizeUtils_1.getLocalizedString)("core.MCPForDA.mcpAuthMetadataMissingError", error.message), false);
+            }
+        }
+        // 2. Read ai-plugin.json
+        const aiPluginContent = await fs_extra_1.default.readJSON(aiPluginFilePath);
+        // For dynamic fetch tools, keep the functions empty and add runtime info
+        // TODO: support dynamic fetch tools in the future
+        const mcpToolsDetail = inputs[constants_2.QuestionNames.MCPForDAAvailableTools];
+        const mcpToolsSelected = inputs[constants_2.QuestionNames.MCPForDAPreFetchTools];
+        if (!mcpToolsDetail || !mcpToolsSelected) {
+            const error = new teamsfx_api_1.UserError("MCPForDAPreFetchToolsNotFound", "PreFetchToolsNotFound", (0, localizeUtils_1.getDefaultString)("core.MCPForDA.preFetchToolsNotFound"), (0, localizeUtils_1.getLocalizedString)("core.MCPForDA.preFetchToolsNotFound"));
+            return (0, teamsfx_api_1.err)(error);
+        }
+        const toolsSelectedPrevious = [];
+        aiPluginContent.runtimes
+            .filter((runtime) => (runtime.type === "RemoteMCPServer" && runtime.spec.url === mcpServerUrl) ||
+            runtime.type === "LocalPlugin")
+            .forEach((runtime) => {
+            toolsSelectedPrevious.push(...runtime.run_for_functions);
+        });
+        aiPluginContent.functions = aiPluginContent.functions.filter((func) => !toolsSelectedPrevious.includes(func.name));
+        aiPluginContent.functions = [
+            ...aiPluginContent.functions,
+            ...mcpToolsDetail
+                .filter((tool) => mcpToolsSelected.includes(tool.name))
+                .map((tool) => {
+                if (inputs[constants_2.QuestionNames.MCPLocalServerIdentifier] != null) {
+                    return {
+                        name: tool.name,
+                        description: tool.description,
+                        parameters: {
+                            type: tool.inputSchema.type || "object",
+                            properties: tool.inputSchema.properties,
+                            required: tool.inputSchema.required || [],
+                        },
+                    };
+                }
+                else {
+                    return {
+                        name: tool.name,
+                        description: tool.description,
+                    };
+                }
+            }),
+        ];
+        const matchedRuntime = aiPluginContent.runtimes.find((runtime) => runtime.type === "RemoteMCPServer" && runtime.spec.url === mcpServerUrl);
+        aiPluginContent.runtimes = aiPluginContent.runtimes.filter((runtime) => (runtime.type !== "RemoteMCPServer" && runtime.type !== "LocalPlugin") ||
+            runtime.spec.url !== mcpServerUrl);
+        if (inputs[constants_2.QuestionNames.MCPLocalServerIdentifier] != null) {
+            aiPluginContent.runtimes.push({
+                type: "LocalPlugin",
+                spec: {
+                    local_endpoint: `${constants_1.LocalMcpPrefix}${inputs[constants_2.QuestionNames.MCPLocalServerIdentifier]}`,
+                },
+                run_for_functions: mcpToolsSelected,
+            });
+        }
+        else {
+            let mcpFile = (_b = matchedRuntime === null || matchedRuntime === void 0 ? void 0 : matchedRuntime.spec.mcp_tool_description) === null || _b === void 0 ? void 0 : _b.file;
+            if (!mcpFile) {
+                mcpFile = "mcp-tools.json";
+                let suffix = 1;
+                while (await fs_extra_1.default.pathExists(path.join(path.dirname(aiPluginFilePath), mcpFile))) {
+                    mcpFile = `mcp-tools-${suffix}.json`;
+                    suffix += 1;
+                }
+            }
+            await fs_extra_1.default.writeJSON(path.join(path.dirname(aiPluginFilePath), mcpFile), {
+                tools: [
+                    ...mcpToolsDetail
+                        .filter((tool) => mcpToolsSelected.includes(tool.name))
+                        .map((tool) => {
+                        return Object.assign(Object.assign({}, tool), { title: tool.name
+                                .replace(/_/g, " ")
+                                .replace(/^./, (str) => str.toUpperCase()) });
+                    }),
+                ],
+            }, { spaces: 4 });
+            aiPluginContent.runtimes.push({
+                type: "RemoteMCPServer",
+                spec: {
+                    url: mcpServerUrl,
+                    mcp_tool_description: {
+                        file: mcpFile,
+                    },
+                },
+                run_for_functions: mcpToolsSelected,
+                auth: mcpAuth === "OAuthPluginVault" && !!registrationId
+                    ? {
+                        type: "OAuthPluginVault",
+                        reference_id: `$\{\{${registrationId}\}\}`,
+                    }
+                    : {
+                        type: "None",
+                    },
+            });
+        }
+        if (mcpAuth === "OAuthPluginVault" && !!registrationId) {
+            // insert oauth info in teamsapp.yaml
+            await actionInjector_1.ActionInjector.injectCreateOAuthActionForMCP(pathUtils_1.pathUtils.getYmlFilePath(projectPath), authType, serverName, registrationId, mcpServerUrl, oauthAuthorizationUrl, oauthTokenUrl, oauthRefreshUrl);
+        }
+        void context.userInteraction
+            .showMessage("info", (0, localizeUtils_1.getLocalizedString)("core.MCPForDA.updatePluginManifest", aiPluginFilePathRelative), false, "Provision")
+            .then((result) => {
+            if (result.isOk() && result.value === "Provision") {
+                void this.provisionResources(inputs);
+            }
+        });
+        await fs_extra_1.default.writeJSON(aiPluginFilePath, aiPluginContent, { spaces: 4 });
+        void ((_d = (_c = context.userInteraction).openFile) === null || _d === void 0 ? void 0 : _d.call(_c, aiPluginFilePath));
+        return (0, teamsfx_api_1.ok)(undefined);
+    }
+    // This method will be implemented by FxCore
+    provisionResources(_inputs) {
+        throw new Error("Method not implemented.");
+    }
+}
+tslib_1.__decorate([
+    (0, hooks_1.hooks)([
+        (0, globalVars_1.ErrorContextMW)({ component: "FxCore", stage: teamsfx_api_1.Stage.installApp }),
+        errorHandler_1.ErrorHandlerMW,
+        (0, questionMW_1.QuestionMW)("updateActionWithMCP"),
+        concurrentLocker_1.ConcurrentLockerMW,
+    ]),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], FxCoreDeclarativeAgentPart.prototype, "updateActionWithMCP", null);
+exports.FxCoreDeclarativeAgentPart = FxCoreDeclarativeAgentPart;
+//# sourceMappingURL=FxCore.declarativeAgent.js.map

package/build/core/FxCore.declarativeAgent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"FxCore.declarativeAgent.js","sourceRoot":"","sources":["../../src/core/FxCore.declarativeAgent.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;;AAElC,6CAA0C;AAC1C,wDASgC;AAChC,0DAA0B;AAC1B,gEAA0B;AAC1B,mDAA6B;AAC7B,qDAAqE;AACrE,2DAA+E;AAC/E,8EAA2E;AAC3E,sDAAwD;AACxD,mEAAgE;AAChE,4DAAyD;AACzD,qDAAsD;AACtD,oEAAmE;AACnE,4DAA2D;AAE3D,MAAa,0BAA0B;IAOrC,KAAK,CAAC,mBAAmB,CAAC,MAAc;;QACtC,MAAM,OAAO,GAAG,IAAA,0BAAa,GAAE,CAAC;QAChC,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACvC,IAAI,CAAC,WAAW,EAAE;YAChB,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC,CAAC,sBAAsB;SACpE;QACD,MAAM,gBAAgB,GAAG,MAAM,CAAC,yBAAa,CAAC,sBAAsB,CAAW,CAAC;QAChF,IAAI,CAAC,CAAC,MAAM,kBAAE,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,EAAE;YAC5C,MAAM,KAAK,GAAG,IAAI,yBAAW,CAC3B,gCAAgC,EAChC,wBAAwB,EACxB,IAAA,gCAAgB,EAAC,sCAAsC,EAAE,gBAAgB,CAAC,EAC1E,IAAA,kCAAkB,EAAC,sCAAsC,EAAE,gBAAgB,CAAC,CAC7E,CAAC;YACF,OAAO,IAAA,iBAAG,EAAC,KAAK,CAAC,CAAC;SACnB;QACD,MAAM,wBAAwB,GAAG,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QAEjE,MAAM,YAAY,GAAG,MAAM,CAAC,yBAAa,CAAC,iBAAiB,CAAC,CAAC;QAC7D,MAAM,UAAU,GAAG,MAAM,CAAC,yBAAa,CAAC,kBAAkB,CAAW,CAAC;QACtE,MAAM,OAAO,GAAG,MAAM,CAAC,yBAAa,CAAC,YAAY,CAAC,CAAC;QACnD,MAAM,QAAQ,GAAG,MAAM,CAAC,yBAAa,CAAC,gBAAgB,CAAC,CAAC;QAExD,IAAI,qBAAqB,GAAuB,SAAS,CAAC;QAC1D,IAAI,aAAa,GAAuB,SAAS,CAAC;QAClD,IAAI,eAAe,GAAuB,SAAS,CAAC;QACpD,IAAI,cAAc,GAAuB,SAAS,CAAC;QAEnD,IAAI,OAAO,KAAK,kBAAkB,EAAE;YAClC,IAAI;gBACF,cAAc,GAAG,kBAAkB,UAAU,CAAC,WAAW,EAAE,EAAE,CAAC;gBAC9D,IAAI,QAAQ,KAAK,OAAO,EAAE;oBACxB,IAAI,oBAAoB,GAAG,MAAM,CAAC,yBAAa,CAAC,wBAAwB,CAAC,CAAC;oBAC1E,IAAI,CAAC,oBAAoB,EAAE;wBACzB,MAAM,kBAAkB,GAAG,MAAM,CAAC,yBAAa,CAAC,uBAAuB,CAAC,CAAC;wBACzE,IAAI,CAAC,kBAAkB,EAAE;4BACvB,MAAM,IAAI,KAAK,CAAC,IAAA,kCAAkB,EAAC,0CAA0C,CAAC,CAAC,CAAC;yBACjF;wBAED,MAAM,QAAQ,GAAG,MAAM,eAAK,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;wBACrD,IACE,QAAQ,CAAC,MAAM,KAAK,GAAG;4BACvB,QAAQ,CAAC,IAAI;4BACb,QAAQ,CAAC,IAAI,CAAC,qBAAqB;4BACnC,QAAQ,CAAC,IAAI,CAAC,qBAAqB,CAAC,MAAM,GAAG,CAAC,EAC9C;4BACA,MAAM,oBAAoB,GAAG,MAAA,QAAQ,CAAC,IAAI,CAAC,qBAAqB,0CAAG,CAAC,CAAC,CAAC;4BACtE,+EAA+E;4BAC/E,2EAA2E;4BAC3E,gEAAgE;4BAChE,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,oBAAoB,CAAC,CAAC;4BAChD,oBAAoB,GAAG,GAAG,SAAS,CAAC,QAAQ,KAAK,SAAS,CAAC,IAAI,0CAA0C,SAAS,CAAC,QAAQ,EAAE,CAAC;yBAC/H;6BAAM;4BACL,MAAM,IAAI,KAAK,CAAC,IAAA,kCAAkB,EAAC,4CAA4C,CAAC,CAAC,CAAC;yBACnF;qBACF;oBACD,MAAM,gBAAgB,GAAG,MAAM,eAAK,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;oBAC/D,IAAI,gBAAgB,CAAC,MAAM,KAAK,GAAG,EAAE;wBACnC,qBAAqB,GAAG,gBAAgB,CAAC,IAAI,CAAC,sBAAsB,CAAC;wBACrE,aAAa,GAAG,gBAAgB,CAAC,IAAI,CAAC,cAAc,CAAC;wBACrD,eAAe,GAAG,gBAAgB,CAAC,IAAI,CAAC,gBAAgB,CAAC;qBAC1D;oBACD,IAAI,CAAC,qBAAqB,IAAI,CAAC,aAAa,EAAE;wBAC5C,MAAM,IAAI,KAAK,CAAC,IAAA,kCAAkB,EAAC,+BAA+B,CAAC,CAAC,CAAC;qBACtE;iBACF;aACF;YAAC,OAAO,KAAU,EAAE;gBACnB,KAAK,OAAO,CAAC,eAAe,CAAC,WAAW,CACtC,OAAO,EACP,IAAA,kCAAkB,EAAC,2CAA2C,EAAE,KAAK,CAAC,OAAO,CAAC,EAC9E,KAAK,CACN,CAAC;aACH;SACF;QAED,yBAAyB;QACzB,MAAM,eAAe,GAAG,MAAM,kBAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QAE5D,yEAAyE;QACzE,kDAAkD;QAClD,MAAM,cAAc,GAAG,MAAM,CAAC,yBAAa,CAAC,sBAAsB,CAAC,CAAC;QACpE,MAAM,gBAAgB,GAAG,MAAM,CAAC,yBAAa,CAAC,qBAAqB,CAAC,CAAC;QACrE,IAAI,CAAC,cAAc,IAAI,CAAC,gBAAgB,EAAE;YACxC,MAAM,KAAK,GAAG,IAAI,uBAAS,CACzB,+BAA+B,EAC/B,uBAAuB,EACvB,IAAA,gCAAgB,EAAC,qCAAqC,CAAC,EACvD,IAAA,kCAAkB,EAAC,qCAAqC,CAAC,CAC1D,CAAC;YACF,OAAO,IAAA,iBAAG,EAAC,KAAK,CAAC,CAAC;SACnB;QAED,MAAM,qBAAqB,GAAa,EAAE,CAAC;QAC3C,eAAe,CAAC,QAAQ;aACrB,MAAM,CACL,CAAC,OAAY,EAAE,EAAE,CACf,CAAC,OAAO,CAAC,IAAI,KAAK,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC,GAAG,KAAK,YAAY,CAAC;YACzE,OAAO,CAAC,IAAI,KAAK,aAAa,CACjC;aACA,OAAO,CAAC,CAAC,OAAY,EAAE,EAAE;YACxB,qBAAqB,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QACL,eAAe,CAAC,SAAS,GAAG,eAAe,CAAC,SAAS,CAAC,MAAM,CAC1D,CAAC,IAAS,EAAE,EAAE,CAAC,CAAC,qBAAqB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAC1D,CAAC;QACF,eAAe,CAAC,SAAS,GAAG;YAC1B,GAAG,eAAe,CAAC,SAAS;YAC5B,GAAG,cAAc;iBACd,MAAM,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,gBAAgB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;iBAC3D,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE;gBACjB,IAAI,MAAM,CAAC,yBAAa,CAAC,wBAAwB,CAAC,IAAI,IAAI,EAAE;oBAC1D,OAAO;wBACL,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,WAAW,EAAE,IAAI,CAAC,WAAW;wBAC7B,UAAU,EAAE;4BACV,IAAI,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI,IAAI,QAAQ;4BACvC,UAAU,EAAE,IAAI,CAAC,WAAW,CAAC,UAAU;4BACvC,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,IAAI,EAAE;yBAC1C;qBACF,CAAC;iBACH;qBAAM;oBACL,OAAO;wBACL,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,WAAW,EAAE,IAAI,CAAC,WAAW;qBAC9B,CAAC;iBACH;YACH,CAAC,CAAC;SACL,CAAC;QAEF,MAAM,cAAc,GAAG,eAAe,CAAC,QAAQ,CAAC,IAAI,CAClD,CAAC,OAAY,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC,GAAG,KAAK,YAAY,CAC1F,CAAC;QAEF,eAAe,CAAC,QAAQ,GAAG,eAAe,CAAC,QAAQ,CAAC,MAAM,CACxD,CAAC,OAAY,EAAE,EAAE,CACf,CAAC,OAAO,CAAC,IAAI,KAAK,iBAAiB,IAAI,OAAO,CAAC,IAAI,KAAK,aAAa,CAAC;YACtE,OAAO,CAAC,IAAI,CAAC,GAAG,KAAK,YAAY,CACpC,CAAC;QAEF,IAAI,MAAM,CAAC,yBAAa,CAAC,wBAAwB,CAAC,IAAI,IAAI,EAAE;YACzD,eAAe,CAAC,QAAkB,CAAC,IAAI,CAAC;gBACvC,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE;oBACJ,cAAc,EAAE,GAAG,0BAAc,GAC/B,MAAM,CAAC,yBAAa,CAAC,wBAAwB,CAC/C,EAAE;iBACH;gBACD,iBAAiB,EAAE,gBAAgB;aACpC,CAAC,CAAC;SACJ;aAAM;YACL,IAAI,OAAO,GAAG,MAAA,cAAc,aAAd,cAAc,uBAAd,cAAc,CAAE,IAAI,CAAC,oBAAoB,0CAAE,IAAI,CAAC;YAC9D,IAAI,CAAC,OAAO,EAAE;gBACZ,OAAO,GAAG,gBAAgB,CAAC;gBAC3B,IAAI,MAAM,GAAG,CAAC,CAAC;gBACf,OAAO,MAAM,kBAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,OAAO,CAAC,CAAC,EAAE;oBAC9E,OAAO,GAAG,aAAa,MAAM,OAAO,CAAC;oBACrC,MAAM,IAAI,CAAC,CAAC;iBACb;aACF;YACD,MAAM,kBAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,OAAO,CAAC,EAClD;gBACE,KAAK,EAAE;oBACL,GAAG,cAAc;yBACd,MAAM,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,gBAAgB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;yBAC3D,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE;wBACjB,uCACK,IAAI,KACP,KAAK,EAAG,IAAI,CAAC,IAAe;iCACzB,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC;iCAClB,OAAO,CAAC,IAAI,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,IAC5C;oBACJ,CAAC,CAAC;iBACL;aACF,EACD,EAAE,MAAM,EAAE,CAAC,EAAE,CACd,CAAC;YACD,eAAe,CAAC,QAAkB,CAAC,IAAI,CAAC;gBACvC,IAAI,EAAE,iBAAiB;gBACvB,IAAI,EAAE;oBACJ,GAAG,EAAE,YAAY;oBACjB,oBAAoB,EAAE;wBACpB,IAAI,EAAE,OAAO;qBACd;iBACF;gBACD,iBAAiB,EAAE,gBAAgB;gBACnC,IAAI,EACF,OAAO,KAAK,kBAAkB,IAAI,CAAC,CAAC,cAAc;oBAChD,CAAC,CAAC;wBACE,IAAI,EAAE,kBAAkB;wBACxB,YAAY,EAAE,QAAQ,cAAc,MAAM;qBAC3C;oBACH,CAAC,CAAC;wBACE,IAAI,EAAE,MAAM;qBACb;aACR,CAAC,CAAC;SACJ;QAED,IAAI,OAAO,KAAK,kBAAkB,IAAI,CAAC,CAAC,cAAc,EAAE;YACtD,qCAAqC;YACrC,MAAM,+BAAc,CAAC,6BAA6B,CAChD,qBAAS,CAAC,cAAc,CAAC,WAAW,CAAW,EAC/C,QAAQ,EACR,UAAU,EACV,cAAc,EACd,YAAY,EACZ,qBAAqB,EACrB,aAAa,EACb,eAAe,CAChB,CAAC;SACH;QACD,KAAK,OAAO,CAAC,eAAe;aACzB,WAAW,CACV,MAAM,EACN,IAAA,kCAAkB,EAAC,oCAAoC,EAAE,wBAAwB,CAAC,EAClF,KAAK,EACL,WAAW,CACZ;aACA,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YACf,IAAI,MAAM,CAAC,IAAI,EAAE,IAAI,MAAM,CAAC,KAAK,KAAK,WAAW,EAAE;gBACjD,KAAK,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;aACtC;QACH,CAAC,CAAC,CAAC;QACL,MAAM,kBAAE,CAAC,SAAS,CAAC,gBAAgB,EAAE,eAAe,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC;QACrE,KAAK,CAAA,MAAA,MAAA,OAAO,CAAC,eAAe,EAAC,QAAQ,mDAAG,gBAAgB,CAAC,CAAA,CAAC;QAC1D,OAAO,IAAA,gBAAE,EAAC,SAAS,CAAC,CAAC;IACvB,CAAC;IAED,4CAA4C;IAC5C,kBAAkB,CAAC,OAAe;QAChC,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC7C,CAAC;CACF;AAxOC;IANC,IAAA,aAAK,EAAC;QACL,IAAA,2BAAc,EAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,KAAK,EAAE,mBAAK,CAAC,UAAU,EAAE,CAAC;QAChE,6BAAc;QACd,IAAA,uBAAU,EAAC,qBAAqB,CAAC;QACjC,qCAAkB;KACnB,CAAC;;;;qEAmOD;AAzOH,gEA+OC"}

package/build/core/FxCore.js

@@ -7,7 +7,6 @@ const tslib_1 = require("tslib");
 const hooks_1 = require("@feathersjs/hooks");
 const m365_spec_parser_1 = require("@microsoft/m365-spec-parser");
 const teamsfx_api_1 = require("@microsoft/teamsfx-api");
-const axios_1 = tslib_1.__importDefault(require("axios"));
 const fs_extra_1 = tslib_1.__importDefault(require("fs-extra"));
 const jsonschema = tslib_1.__importStar(require("jsonschema"));
 const os = tslib_1.__importStar(require("os"));
@@ -27,7 +26,6 @@ const templates_config_json_1 = tslib_1.__importDefault(require("../common/templ
 const tools_1 = require("../common/tools");
 const utils_1 = require("../common/utils");
 const versionMetadata_1 = require("../common/versionMetadata");
-const actionInjector_1 = require("../component/configManager/actionInjector");
 const constant_1 = require("../component/configManager/constant");
 const parser_1 = require("../component/configManager/parser");
 const constants_2 = require("../component/constants");
@@ -73,6 +71,7 @@ const callback_1 = require("./callback");
 const collaborator_1 = require("./collaborator");
 const crypto_1 = require("./crypto");
 const environmentName_1 = require("./environmentName");
+const generateConfigFiles_1 = require("./generateConfigFiles");
 const concurrentLocker_1 = require("./middleware/concurrentLocker");
 const contextInjector_1 = require("./middleware/contextInjector");
 const errorHandler_1 = require("./middleware/errorHandler");
@@ -82,8 +81,10 @@ const retry_1 = require("./middleware/retry");
 const v3MigrationUtils_1 = require("./middleware/utils/v3MigrationUtils");
 const share_2 = require("./share");
 const telemetry_2 = require("./telemetry");
-class FxCore {
+const FxCore_declarativeAgent_1 = require("./FxCore.declarativeAgent");
+class FxCore extends FxCore_declarativeAgent_1.FxCoreDeclarativeAgentPart {
     constructor(tools) {
+        super();
         (0, globalVars_1.setTools)(tools);
     }
     /**
@@ -1856,165 +1857,6 @@ class FxCore {
         }
         return (0, teamsfx_api_1.ok)(undefined);
     }
-    async updateActionWithMCP(inputs) {
-        var _a, _b, _c, _d;
-        const context = (0, globalVars_1.createContext)();
-        const projectPath = inputs.projectPath;
-        if (!projectPath) {
-            throw new Error("projectPath is undefined"); // should never happen
-        }
-        const aiPluginFilePath = inputs[constants_3.QuestionNames.PluginManifestFilePath];
-        if (!(await fs_extra_1.default.pathExists(aiPluginFilePath))) {
-            const error = new teamsfx_api_1.SystemError("MCPForDAPluginManifestNotFound", "PluginManifestNotFound", (0, localizeUtils_1.getDefaultString)("core.MCPForDA.pluginManifestNotFound", aiPluginFilePath), (0, localizeUtils_1.getLocalizedString)("core.MCPForDA.pluginManifestNotFound", aiPluginFilePath));
-            return (0, teamsfx_api_1.err)(error);
-        }
-        const aiPluginFilePathRelative = path.basename(aiPluginFilePath);
-        const mcpServerUrl = inputs[constants_3.QuestionNames.MCPForDAServerUrl];
-        const serverName = inputs[constants_3.QuestionNames.MCPForDAServerName];
-        const mcpAuth = inputs[constants_3.QuestionNames.MCPForDAAuth];
-        const authType = inputs[constants_3.QuestionNames.MCPForDAAuthType];
-        let oauthAuthorizationUrl = undefined;
-        let oauthTokenUrl = undefined;
-        let oauthRefreshUrl = undefined;
-        let registrationId = undefined;
-        if (mcpAuth === "OAuthPluginVault") {
-            try {
-                registrationId = `MCP_DA_AUTH_ID_${serverName.toUpperCase()}`;
-                if (authType === "oauth") {
-                    let wellKnownMetadataUrl = inputs[constants_3.QuestionNames.MCPForDAAuthWellKnownUrl];
-                    if (!wellKnownMetadataUrl) {
-                        const mcpAuthMetadataUrl = inputs[constants_3.QuestionNames.MCPForDAAuthMetadataUrl];
-                        if (!mcpAuthMetadataUrl) {
-                            throw new Error((0, localizeUtils_1.getLocalizedString)("core.MCPForDA.mcpAuthMetadataUrlNotFound"));
-                        }
-                        const response = await axios_1.default.get(mcpAuthMetadataUrl);
-                        if (response.status === 200 &&
-                            response.data &&
-                            response.data.authorization_servers &&
-                            response.data.authorization_servers.length > 0) {
-                            const mcpServerMetadataUrl = (_a = response.data.authorization_servers) === null || _a === void 0 ? void 0 : _a[0];
-                            // Transform the URL to the proper OAuth authorization server metadata endpoint
-                            // According to RFC 8414, the well-known endpoint should be constructed as:
-                            // https://{domain}/.well-known/oauth-authorization-server{path}
-                            const serverUrl = new URL(mcpServerMetadataUrl);
-                            wellKnownMetadataUrl = `${serverUrl.protocol}//${serverUrl.host}/.well-known/oauth-authorization-server${serverUrl.pathname}`;
-                        }
-                        else {
-                            throw new Error((0, localizeUtils_1.getLocalizedString)("core.MCPForDA.mcpServerMetadataUrlNotFound"));
-                        }
-                    }
-                    const metadataResponse = await axios_1.default.get(wellKnownMetadataUrl);
-                    if (metadataResponse.status === 200) {
-                        oauthAuthorizationUrl = metadataResponse.data.authorization_endpoint;
-                        oauthTokenUrl = metadataResponse.data.token_endpoint;
-                        oauthRefreshUrl = metadataResponse.data.refresh_endpoint;
-                    }
-                    if (!oauthAuthorizationUrl || !oauthTokenUrl) {
-                        throw new Error((0, localizeUtils_1.getLocalizedString)("core.MCPForDA.authUrlNotFound"));
-                    }
-                }
-            }
-            catch (error) {
-                void context.userInteraction.showMessage("error", (0, localizeUtils_1.getLocalizedString)("core.MCPForDA.mcpAuthMetadataMissingError", error.message), false);
-            }
-        }
-        // 2. Read ai-plugin.json
-        const aiPluginContent = await fs_extra_1.default.readJSON(aiPluginFilePath);
-        // For dynamic fetch tools, keep the functions empty and add runtime info
-        // TODO: support dynamic fetch tools in the future
-        const mcpToolsDetail = inputs[constants_3.QuestionNames.MCPForDAAvailableTools];
-        const mcpToolsSelected = inputs[constants_3.QuestionNames.MCPForDAPreFetchTools];
-        if (!mcpToolsDetail || !mcpToolsSelected) {
-            const error = new teamsfx_api_1.UserError("MCPForDAPreFetchToolsNotFound", "PreFetchToolsNotFound", (0, localizeUtils_1.getDefaultString)("core.MCPForDA.preFetchToolsNotFound"), (0, localizeUtils_1.getLocalizedString)("core.MCPForDA.preFetchToolsNotFound"));
-            return (0, teamsfx_api_1.err)(error);
-        }
-        const toolsSelectedPrevious = [];
-        aiPluginContent.runtimes
-            .filter((runtime) => (runtime.type === "RemoteMCPServer" &&
-            runtime.spec.url === mcpServerUrl &&
-            !runtime.spec["enable_dynamic_discovery"]) ||
-            runtime.type === "LocalPlugin")
-            .forEach((runtime) => {
-            toolsSelectedPrevious.push(...runtime.run_for_functions);
-        });
-        aiPluginContent.functions = aiPluginContent.functions.filter((func) => !toolsSelectedPrevious.includes(func.name));
-        aiPluginContent.functions = [
-            ...aiPluginContent.functions,
-            ...mcpToolsDetail
-                .filter((tool) => mcpToolsSelected.includes(tool.name))
-                .map((tool) => {
-                return {
-                    name: tool.name,
-                    description: tool.description,
-                };
-            }),
-        ];
-        const matchedRuntime = aiPluginContent.runtimes.find((runtime) => runtime.type === "RemoteMCPServer" && runtime.spec.url === mcpServerUrl);
-        aiPluginContent.runtimes = aiPluginContent.runtimes.filter((runtime) => (runtime.type !== "RemoteMCPServer" && runtime.type !== "LocalPlugin") ||
-            runtime.spec.url !== mcpServerUrl ||
-            runtime.spec["enable_dynamic_discovery"] === true);
-        if (inputs[constants_3.QuestionNames.MCPLocalServerIdentifier] != null) {
-            aiPluginContent.runtimes.push({
-                type: "LocalPlugin",
-                spec: {
-                    local_endpoint: `${constants_2.LocalMcpPrefix}${inputs[constants_3.QuestionNames.MCPLocalServerIdentifier]}`,
-                },
-                run_for_functions: mcpToolsSelected,
-            });
-        }
-        else {
-            let mcpFile = (_b = matchedRuntime === null || matchedRuntime === void 0 ? void 0 : matchedRuntime.spec.mcp_tool_description) === null || _b === void 0 ? void 0 : _b.file;
-            if (!mcpFile) {
-                mcpFile = "mcp-tools.json";
-                let suffix = 1;
-                while (await fs_extra_1.default.pathExists(path.join(path.dirname(aiPluginFilePath), mcpFile))) {
-                    mcpFile = `mcp-tools-${suffix}.json`;
-                    suffix += 1;
-                }
-            }
-            await fs_extra_1.default.writeJSON(path.join(path.dirname(aiPluginFilePath), mcpFile), {
-                tools: [
-                    ...mcpToolsDetail
-                        .filter((tool) => mcpToolsSelected.includes(tool.name))
-                        .map((tool) => {
-                        return Object.assign(Object.assign({}, tool), { title: tool.name
-                                .replace(/_/g, " ")
-                                .replace(/^./, (str) => str.toUpperCase()) });
-                    }),
-                ],
-            }, { spaces: 4 });
-            aiPluginContent.runtimes.push({
-                type: "RemoteMCPServer",
-                spec: {
-                    url: mcpServerUrl,
-                    mcp_tool_description: {
-                        file: mcpFile,
-                    },
-                },
-                run_for_functions: mcpToolsSelected,
-                auth: mcpAuth === "OAuthPluginVault" && !!registrationId
-                    ? {
-                        type: "OAuthPluginVault",
-                        reference_id: `$\{\{${registrationId}\}\}`,
-                    }
-                    : undefined,
-            });
-        }
-        if (mcpAuth === "OAuthPluginVault" && !!registrationId) {
-            // insert oauth info in teamsapp.yaml
-            await actionInjector_1.ActionInjector.injectCreateOAuthActionForMCP(pathUtils_1.pathUtils.getYmlFilePath(projectPath), authType, serverName, registrationId, mcpServerUrl, oauthAuthorizationUrl, oauthTokenUrl, oauthRefreshUrl);
-        }
-        void context.userInteraction
-            .showMessage("info", (0, localizeUtils_1.getLocalizedString)("core.MCPForDA.updatePluginManifest", aiPluginFilePathRelative), false, "Provision")
-            .then((result) => {
-            if (result.isOk() && result.value === "Provision") {
-                void this.provisionResources(inputs);
-            }
-        });
-        await fs_extra_1.default.writeJSON(aiPluginFilePath, aiPluginContent, { spaces: 4 });
-        void ((_d = (_c = context.userInteraction).openFile) === null || _d === void 0 ? void 0 : _d.call(_c, aiPluginFilePath));
-        return (0, teamsfx_api_1.ok)(undefined);
-    }
     /**
      * dynamic template metadata download
      */
@@ -2070,6 +1912,12 @@ class FxCore {
             return (0, teamsfx_api_1.err)(systemErr);
         }
     }
+    /**
+     * dynamic template metadata download
+     */
+    async generateConfigFiles(inputs) {
+        return await (0, generateConfigFiles_1.generateConfigFiles)(inputs);
+    }
     async updateAuthActionInYaml(authName, authScheme, projectPath, apSpecPath, pluginManifestPath, forceToAddNew = true) {
         if (authName && authScheme) {
             const authInjectRes = await (0, helper_2.injectAuthAction)(projectPath, authName, authScheme, apSpecPath, forceToAddNew);
@@ -2892,17 +2740,6 @@ tslib_1.__decorate([
     tslib_1.__metadata("design:paramtypes", [Object]),
     tslib_1.__metadata("design:returntype", Promise)
 ], FxCore.prototype, "installAppToChannel", null);
-tslib_1.__decorate([
-    (0, hooks_1.hooks)([
-        (0, globalVars_1.ErrorContextMW)({ component: "FxCore", stage: teamsfx_api_1.Stage.installApp }),
-        errorHandler_1.ErrorHandlerMW,
-        (0, questionMW_1.QuestionMW)("updateActionWithMCP"),
-        concurrentLocker_1.ConcurrentLockerMW,
-    ]),
-    tslib_1.__metadata("design:type", Function),
-    tslib_1.__metadata("design:paramtypes", [Object]),
-    tslib_1.__metadata("design:returntype", Promise)
-], FxCore.prototype, "updateActionWithMCP", null);
 tslib_1.__decorate([
     (0, hooks_1.hooks)([
         (0, globalVars_1.ErrorContextMW)({ component: "FxCore", stage: "fetchOnlineTemplateMetadata" }),
@@ -2912,5 +2749,11 @@ tslib_1.__decorate([
     tslib_1.__metadata("design:paramtypes", []),
     tslib_1.__metadata("design:returntype", Promise)
 ], FxCore.prototype, "fetchOnlineTemplateMetadata", null);
+tslib_1.__decorate([
+    (0, hooks_1.hooks)([(0, globalVars_1.ErrorContextMW)({ component: "FxCore", stage: "generateConfigFiles" }), errorHandler_1.ErrorHandlerMW]),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", [Object]),
+    tslib_1.__metadata("design:returntype", Promise)
+], FxCore.prototype, "generateConfigFiles", null);
 exports.FxCore = FxCore;
 //# sourceMappingURL=FxCore.js.map