Trends Hub

by baranwang
Verified
TypeScript
165
Reddit Discord
Overview InspectNew Schema Related Servers Reviews Score
Need Help?View Source Code Report Issue
mcp-trends-hub
scripts
scripts/readme.ts
import { promises as fs } from 'node:fs';
import { Client } from '@modelcontextprotocol/sdk/client/index.js';
import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
import type { RsbuildPlugin } from '@rsbuild/core';
import { logger } from '@rslib/core';
import { version } from '../package.json';

interface Options {
  readmePath: string;
}

const getToolsContent = async () => {
  const transport = new StdioClientTransport({
    command: 'node',
    args: ['.'],
  });

  const client = new Client({
    name: 'test-client',
    version: '0.0.0',
  });

  await client.connect(transport);

  const resp = await client.listTools();

  await client.close();

  let result = '| 工具名称 | 描述 |\n| --- | --- |\n';
  for (const tool of resp.tools) {
    result += `| ${tool.name} | ${tool.description} |\n`;
  }

  return result;
};

const getUsageJSONContent = () => {
  const mcpServerConfig = {
    mcpServers: {
      'trends-hub': {
        command: 'npx',
        args: ['-y', `mcp-trends-hub@${version ?? 'latest'}`],
      },
    },
  };
  let result = '```json\n';
  result += JSON.stringify(mcpServerConfig, null, 2);
  result += '\n```';
  return result;
};

const getUsageBashContent = () => {
  let result = '```bash\n';
  result += `npx -y mcp-trends-hub@${version ?? 'latest'}`;
  result += '\n```';
  return result;
};

const createContentUpdater = (initialContent: string) => {
  let content = initialContent;

  return {
    update: async (markName: string, contentGenerator: () => string | Promise<string>): Promise<void> => {
      const markStart = `<!-- ${markName}-start -->`;
      const markEnd = `<!-- ${markName}-end -->`;
      const newContent = await contentGenerator();
      const regex = new RegExp(`(${markStart}\\n)([\\s\\S]*?)(\\n${markEnd})`, 'g');
      content = content.replace(regex, `$1${newContent}\n$3`);
    },
    getContent: () => content,
  };
};

export const readmePlugin = ({ readmePath }: Options): RsbuildPlugin => {
  return {
    name: 'readme-plugin',

    setup: (api) => {
      api.onAfterBuild(async ({ isWatch }) => {
        if (isWatch) {
          return;
        }

        try {
          logger.ready('更新 README.md 文件...');

          const readmeContent = await fs.readFile(readmePath, 'utf-8');

          const contentUpdater = createContentUpdater(readmeContent);

          await contentUpdater.update('tools', getToolsContent);
          await contentUpdater.update('usage-json', getUsageJSONContent);
          await contentUpdater.update('usage-bash', getUsageBashContent);
          await fs.writeFile(readmePath, contentUpdater.getContent());

          logger.success('README.md 更新成功');
        } catch (error) {
          logger.error('更新 README.md 失败:', error);
        }
      });
    },
  };
};