githubmirror/ChatGPT

src/utils.ts
66 строк2 KB
lencx
chore: optim
24 дек 2022, 15:04
24 дек 2022, 15:0425ab2b0
import { readTextFile, writeTextFile, exists, createDir, BaseDirectory } from '@tauri-apps/api/fs';
import { homeDir, join, dirname } from '@tauri-apps/api/path';
import dayjs from 'dayjs';

export const CHAT_MODEL_JSON = 'chat.model.json';
export const CHAT_MODEL_CMD_JSON = 'chat.model.cmd.json';
export const CHAT_PROMPTS_CSV = 'chat.prompts.csv';
export const GITHUB_PROMPTS_CSV_URL = 'https://raw.githubusercontent.com/f/awesome-chatgpt-prompts/main/prompts.csv';
export const DISABLE_AUTO_COMPLETE = {
  autoCapitalize: 'off',
  autoComplete: 'off',
  spellCheck: false
};

export const chatRoot = async () => {
  return join(await homeDir(), '.chatgpt')
}

export const chatModelPath = async (): Promise<string> => {
  return join(await chatRoot(), CHAT_MODEL_JSON);
}

export const chatPromptsPath = async (): Promise<string> => {
  return join(await chatRoot(), CHAT_PROMPTS_CSV);
}

type readJSONOpts = { defaultVal?: Record<string, any>, isRoot?: boolean, isList?: boolean };
export const readJSON = async (path: string, opts: readJSONOpts = {}) => {
  const { defaultVal = {}, isRoot = false, isList = false } = opts;
  const root = await chatRoot();
  const file = await join(isRoot ? '' : root, path);

  if (!await exists(file)) {
    if (await dirname(file) !== root) {
      await createDir(await dirname(file), { recursive: true });
    }
    await writeTextFile(file, isList ? '[]' : JSON.stringify({
      name: 'ChatGPT',
      link: 'https://github.com/lencx/ChatGPT',
      ...defaultVal,
    }, null, 2))
  }

  try {
    return JSON.parse(await readTextFile(file));
  } catch(e) {
    return {};
  }
}

type writeJSONOpts = { dir?: string, isRoot?: boolean };
export const writeJSON = async (path: string, data: Record<string, any>, opts: writeJSONOpts = {}) => {
  const { isRoot = false } = opts;
  const root = await chatRoot();
  const file = await join(isRoot ? '' : root, path);

  if (isRoot && !await exists(await dirname(file))) {
    await createDir(await dirname(file), { recursive: true });
  }

  await writeTextFile(file, JSON.stringify(data, null, 2));
}

export const fmtDate = (date: any) => dayjs(date).format('YYYY-MM-DD HH:mm:ss');

export const genCmd = (act: string) => act.replace(/\s+|\/+/g, '_').replace(/[^\d\w]/g, '').toLocaleLowerCase();