@cnbcool/mcp-server/dist/tools/repoTools.js

CNB MCP Server. A comprehensive MCP server that provides seamless integration to the CNB's API(https://cnb.cool), offering a wide range of tools for repository management, pipelines operations and collaboration features

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.default = registerRepoTools;
const zod_1 = require("zod");
const toolNames_js_1 = require("../constants/toolNames.js");
const toolDescriptions_js_1 = require("../constants/toolDescriptions.js");
const repository_js_1 = require("../api/repository.js");
const user_js_1 = require("../api/user.js");
const formatToolResult_js_1 = require("../helpers/formatToolResult.js");
const formatRepoUrl_js_1 = require("../helpers/formatRepoUrl.js");
const checkRepoVisibility_js_1 = require("../helpers/checkRepoVisibility.js");
function registerRepoTools(server, client) {
    server.tool(toolNames_js_1.ToolNames.LIST_REPOSITORIES, toolDescriptions_js_1.toolDescriptions[toolNames_js_1.ToolNames.LIST_REPOSITORIES], {
        remote_url: zod_1.z.string().describe('远程仓库URL，需要先执行`git remote get-url origin`命令获取，获取不到传空字符串'),
        page: zod_1.z.number().default(1).describe('第几页,从1开始,默认值是1'),
        page_size: zod_1.z.number().default(10).describe('每页多少条数据,默认值为10'),
        search: zod_1.z.preprocess((val) => (val === null ? undefined : val), zod_1.z.string().optional()).describe('查询关键字'),
        filter_type: zod_1.z
            .preprocess((val) => (val === null ? undefined : val), zod_1.z.enum(['private', 'public', 'secret']).optional())
            .describe('仓库类型,为空表示所有仓库类型,默认值为空'),
        role: zod_1.z
            .preprocess((val) => (val === null ? undefined : val), zod_1.z.enum(['Reporter', 'Developer', 'Master', 'Owner']).optional())
            .describe('最小仓库权限,当用户未指定角色时,需要主动传入Reporter'),
        order_by: zod_1.z
            .preprocess((val) => (val === null ? undefined : val), zod_1.z.enum(['created_at', 'last_updated_at', 'stars']).optional())
            .describe('排序类型,默认值是last_updated_at'),
        desc: zod_1.z
            .preprocess((val) => (val === null ? undefined : val), zod_1.z.boolean().optional())
            .describe('是否开启倒叙排序，默认值是false')
    }, async ({ remote_url, page, page_size, search, filter_type, role, order_by, desc }) => {
        const isPublic = await (0, checkRepoVisibility_js_1.isRepoPublic)(client, remote_url);
        if (isPublic) {
            filter_type = 'public';
        }
        try {
            const repos = await (0, repository_js_1.listRepositories)(client, { page, page_size, search, filter_type, role, order_by, desc });
            return (0, formatToolResult_js_1.formatTextToolResult)(JSON.stringify(repos, null, 2), toolNames_js_1.ToolNames.LIST_REPOSITORIES);
        }
        catch (error) {
            return (0, formatToolResult_js_1.formatToolError)(error, toolNames_js_1.ToolNames.LIST_REPOSITORIES);
        }
    });
    server.tool(toolNames_js_1.ToolNames.LIST_GROUP_REPOSITORIES, toolDescriptions_js_1.toolDescriptions[toolNames_js_1.ToolNames.LIST_GROUP_REPOSITORIES], {
        group: zod_1.z.string().describe('组织名称'),
        page: zod_1.z.number().default(1).describe('第几页,从1开始,默认值是1'),
        page_size: zod_1.z.number().default(10).describe('每页多少条数据,默认值为10'),
        search: zod_1.z.preprocess((val) => (val === null ? undefined : val), zod_1.z.string().optional()).describe('仓库关键字'),
        filter_type: zod_1.z
            .preprocess((val) => (val === null ? undefined : val), zod_1.z.enum(['private', 'public', 'secret']).optional())
            .describe('仓库类型'),
        descendant: zod_1.z
            .preprocess((val) => (val === null ? undefined : val), zod_1.z.enum(['all', 'sub', 'grand']).optional())
            .describe('查全部、直接属于当前组织的仓库、子组织的仓库'),
        order_by: zod_1.z
            .preprocess((val) => (val === null ? undefined : val), zod_1.z.enum(['created_at', 'last_updated_at', 'stars', 'slug_path']).optional())
            .describe('排序类型'),
        desc: zod_1.z
            .preprocess((val) => (val === null ? undefined : val), zod_1.z.boolean().optional())
            .describe('是否开启倒叙排序，默认值是false')
    }, async ({ group, page, page_size, search, filter_type, descendant, order_by, desc }) => {
        try {
            const repos = await (0, repository_js_1.listGroupRepositories)(client, group, {
                page,
                page_size,
                search,
                filter_type,
                descendant,
                order_by,
                desc
            });
            return (0, formatToolResult_js_1.formatTextToolResult)(JSON.stringify(repos, null, 2), toolNames_js_1.ToolNames.LIST_GROUP_REPOSITORIES);
        }
        catch (error) {
            return (0, formatToolResult_js_1.formatToolError)(error, toolNames_js_1.ToolNames.LIST_GROUP_REPOSITORIES);
        }
    });
    server.tool(toolNames_js_1.ToolNames.GET_REPOSITORY, toolDescriptions_js_1.toolDescriptions[toolNames_js_1.ToolNames.GET_REPOSITORY], {
        repo: zod_1.z.string().describe('仓库路径')
    }, async ({ repo }) => {
        try {
            const repoInfo = await (0, repository_js_1.getRepository)(client, repo);
            return (0, formatToolResult_js_1.formatTextToolResult)(JSON.stringify(repoInfo, null, 2), toolNames_js_1.ToolNames.GET_REPOSITORY);
        }
        catch (error) {
            return (0, formatToolResult_js_1.formatToolError)(error, toolNames_js_1.ToolNames.GET_REPOSITORY);
        }
    });
    server.tool(toolNames_js_1.ToolNames.CREATE_REPOSITORY, toolDescriptions_js_1.toolDescriptions[toolNames_js_1.ToolNames.CREATE_REPOSITORY], {
        group: zod_1.z.preprocess((val) => (val === null ? undefined : val), zod_1.z.string().optional()).describe('仓库所属分组'),
        name: zod_1.z.string().describe('仓库名称'),
        description: zod_1.z.preprocess((val) => (val === null ? undefined : val), zod_1.z.string().optional()).describe('仓库描述'),
        license: zod_1.z.preprocess((val) => (val === null ? undefined : val), zod_1.z.string().optional()).describe('仓库许可'),
        visibility: zod_1.z
            .preprocess((val) => (val === null ? undefined : val), zod_1.z.enum(['public', 'private', 'secret']).default('public'))
            .describe('仓库可见性')
    }, async ({ group, name, description, license, visibility }) => {
        let repoGroup = group;
        if (!repoGroup) {
            const { username = '' } = await (0, user_js_1.getUser)(client);
            repoGroup = username;
        }
        try {
            const data = await (0, repository_js_1.createRepository)(client, repoGroup, { name, description, license, visibility });
            return (0, formatToolResult_js_1.formatTextToolResult)(JSON.stringify(data, null, 2), toolNames_js_1.ToolNames.CREATE_REPOSITORY);
        }
        catch (error) {
            return (0, formatToolResult_js_1.formatToolError)(error, toolNames_js_1.ToolNames.CREATE_REPOSITORY);
        }
    });
    server.tool(toolNames_js_1.ToolNames.GET_CURRENT_REPOSITORY, toolDescriptions_js_1.toolDescriptions[toolNames_js_1.ToolNames.GET_CURRENT_REPOSITORY], {
        remote_url: zod_1.z.string().describe('远程仓库URL, 需要先执行`git remote get-url origin`命令获取')
    }, async ({ remote_url }) => {
        try {
            const repoPath = (0, formatRepoUrl_js_1.getRepoPath)(remote_url);
            if (!repoPath) {
                return (0, formatToolResult_js_1.formatToolError)(`无法从远程仓库URL解析出仓库路径: ${remote_url}`, toolNames_js_1.ToolNames.GET_CURRENT_REPOSITORY);
            }
            // 获取仓库信息
            const data = await (0, repository_js_1.getRepository)(client, repoPath);
            return (0, formatToolResult_js_1.formatTextToolResult)(JSON.stringify(data, null, 2), toolNames_js_1.ToolNames.GET_CURRENT_REPOSITORY);
        }
        catch (error) {
            return (0, formatToolResult_js_1.formatToolError)(error, toolNames_js_1.ToolNames.GET_CURRENT_REPOSITORY);
        }
    });
}