lobe-chat

1
import { kebabCase } from 'lodash';
2
import { readFileSync, writeFileSync } from 'node:fs';
3
import { resolve } from 'node:path';
4

5
import { AGENT_CN_URL, AGENT_EN_URL, PLUGIN_CN_URL, PLUGIN_EN_URL } from './const';
6

7
const fetchIndex = async (url: string) => {
8
  const res = await fetch(url);
9
  return await res.json();
10
};
11

12
export const fetchAgentIndex = async (lang: string) => {
13
  const isCN = lang === 'zh-CN';
14
  const url = isCN ? AGENT_CN_URL : AGENT_EN_URL;
15
  const data = await fetchIndex(url);
16
  return data.agents;
17
};
18

19
export const fetchPluginIndex = async (lang: string) => {
20
  const isCN = lang === 'zh-CN';
21
  const url = isCN ? PLUGIN_CN_URL : PLUGIN_EN_URL;
22
  const data = await fetchIndex(url);
23
  return data.plugins;
24
};
25

26
export const genLink = (title: string, url: string) => `[${title}](${url})`;
27

28
export const genTags = (tags: string[]) =>
29
  tags
30
    .filter(Boolean)
31
    .map((tag) => `\`${kebabCase(tag)}\``)
32
    .join(' ');
33

34
const getReadmePath = (lang: string) => {
35
  const isCN = lang === 'zh-CN';
36
  return resolve(__dirname, '../../', isCN ? `./README.zh-CN.md` : `./README.md`);
37
};
38

39
export const readReadme = (lang: string): string => {
40
  return readFileSync(getReadmePath(lang), 'utf8');
41
};
42

43
export const writeReadme = (content: string, lang: string) => {
44
  writeFileSync(getReadmePath(lang), content, 'utf8');
45
};
46

47
export const updateReadme = (split: string, md: string, content: string): string => {
48
  const mds = md.split(split);
49
  mds[1] = [' ', content, ' '].join('\n\n');
50

51
  return mds.join(split);
52
};
53