@zzclub/z-cli 0.4.5

package/src/utils/common.js

@@ -0,0 +1,200 @@
+import fs from "node:fs";
+import ora from "ora";
+import { dirname } from "node:path";
+import { fileURLToPath } from "node:url";
+import path from "node:path";
+import chalk from "chalk";
+import os from "node:os";
+import latestVersion from 'latest-version';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+export const defaultConfig = {
+  "translate": {
+    "sourceDirName": "zh-CN",
+    "targetDirName": "en-US",
+    "ignoreFiles": ["node_modules"],
+    "account": {
+      "appId": "",
+      "key": ""
+    }
+  },
+  "editableConfig": ["translate"]
+}
+
+export async function getLocalConfig() {
+  let spinner = ora();
+  const homeDir = os.homedir();
+  const configDir = path.join(homeDir, '.zzclub-z-cli');
+  const configPath = path.join(configDir, 'config.json');
+  let config = {}
+  if (fs.existsSync(configPath)) {
+    try {
+      const content = fs.readFileSync(configPath, 'utf8');
+      config = JSON.parse(content);
+    } catch (err) {
+      spinner.fail('读取配置文件失败:' + JSON.stringify(err));
+      process.exit(1);
+    }
+  } else {
+    spinner.warn('未找到配置文件，正在初始化默认配置');
+
+   
+
+    config = setLocalConfig(defaultConfig, spinner);
+  }
+
+  return config;
+}
+
+/**
+ * 写入文件
+ * @param filePath 文件路径
+ * @param fileContent 文件内容
+ * @param onFinally 完成时回调
+ */
+export function writeFileContent(filePath, fileContent, onFinally = () => {}) {
+  let spinner = ora();
+  fs.writeFile(filePath, fileContent, "utf-8", (error) => {
+    if (!error) {
+      //   spinner.succeed(`已写入${chalk.yellow(filePath)}`);
+      onFinally && onFinally(spinner, true);
+    } else {
+      console.log(error);
+      //   spinner.fail(`写入${chalk.red(filePath)}文件失败, 请重试`);
+      onFinally && onFinally(spinner, false);
+    }
+  });
+}
+
+export const readJsonFile = async (filePath) => {
+  let jsonData;
+  try {
+    let jsonStr = fs.readFileSync(filePath);
+    jsonData = JSON.parse(jsonStr.toString());
+  } catch (err) {
+    jsonData = {};
+  }
+  return jsonData;
+};
+
+export const setHighLightStr = (
+  sourceText,
+  hightlightText,
+  chalkFn = chalk.red
+) => {
+  let index = sourceText.indexOf(hightlightText);
+  if (index === -1) return sourceText;
+  return sourceText.replaceAll(hightlightText, chalkFn(hightlightText));
+};
+
+// 
+export async function setLocalConfig(newConfig = {}, spinner) {
+  const homeDir = os.homedir();
+  const configDir = path.join(homeDir, '.zzclub-z-cli');
+  const configPath = path.join(configDir, 'config.json');
+  
+  // 确保配置目录存在
+  if (!fs.existsSync(configDir)) {
+    fs.mkdirSync(configDir, { recursive: true });
+  }
+
+  // 读取已存在的配置
+  let config = {};
+  if (fs.existsSync(configPath)) {
+    try {
+      const content = fs.readFileSync(configPath, 'utf8');
+      config = JSON.parse(content);
+    } catch (err) {
+      spinner && spinner.fail('读取配置文件失败')
+      process.exit(1);
+    }
+  }
+
+  config = {
+    ...config,
+    ...newConfig,
+    translate: {
+      ...config.translate,
+      ...newConfig.translate,
+      account: {
+        ...(config.translate?.account || {}),
+        ...newConfig.translate.account,
+      }
+    }
+  };
+
+  // 保存配置
+  fs.writeFileSync(configPath, JSON.stringify(config, null, 2));
+  spinner && spinner.succeed(`配置文件已更新：${chalk.yellow(configPath)}`)
+  return config;
+}
+
+
+// 获取 package.json
+export function getPackageJson() {
+  try {
+    const rootDir = path.resolve(__dirname, '../../');
+    const packagePath = path.join(rootDir, 'package.json');
+    const content = fs.readFileSync(packagePath, 'utf8');
+    return JSON.parse(content);
+  } catch (err) {
+    console.error('读取 package.json 失败:', err);
+    return null;
+  }
+}
+
+
+// 检查版本更新
+// ... 其他代码保持不变 ...
+
+// 检查版本更新
+export async function checkUpdate() {
+  // 获取缓存目录
+  const homeDir = os.homedir();
+  const cacheDir = path.join(homeDir, '.zzclub-z-cli');
+  const cacheFile = path.join(cacheDir, 'version-cache.json');
+
+  // 检查是否需要更新
+  const needCheck = () => {
+    try {
+      if (!fs.existsSync(cacheFile)) return true;
+      const cache = JSON.parse(fs.readFileSync(cacheFile, 'utf8'));
+      // 24小时内不重复检查
+      return Date.now() - cache.lastCheck > 24 * 60 * 60 * 1000;
+    } catch (err) {
+      return true;
+    }
+  };
+
+ 
+    if (!needCheck()) return;
+
+    try {
+      const pkg = getPackageJson();
+      if (!pkg) return;
+      let spinner = ora();
+      spinner.start('正在检测版本更新...')
+      const latest = await latestVersion('@zzclub/z-cli');
+      if (latest !== pkg.version) {
+        spinner.warn(chalk.yellow(`\n提示: 发现新版本 ${chalk.green(latest)}，当前版本 ${chalk.red(pkg.version)}\n`))
+      } else {
+        spinner.succeed('当前为最新版本！')
+      }
+
+      // 更新缓存
+      if (!fs.existsSync(cacheDir)) {
+        fs.mkdirSync(cacheDir, { recursive: true });
+      }
+      fs.writeFileSync(cacheFile, JSON.stringify({
+        lastCheck: Date.now(),
+        version: latest
+      }));
+
+      spinner.stop();
+      
+    } catch (err) {
+      // 静默处理错误
+    }
+}

package/src/utils/file.js

@@ -0,0 +1,61 @@
+import fs from "node:fs";
+// 导出一个异步函数，用于读取JSON文件
+export const readJsonFile = async (filePath) => {
+  let jsonData;
+  try {
+    // 使用fs.readFileSync读取文件内容
+    let jsonStr = fs.readFileSync(filePath);
+    // 将文件内容转换为JSON对象
+    jsonData = JSON.parse(jsonStr);
+  } catch (err) {
+    // 如果读取文件失败，则将jsonData初始化为空对象
+    jsonData = {};
+  }
+  // 返回JSON对象
+  return jsonData;
+};
+
+// 用于获取格式化的文件大小
+export const getFormatedFileSize = (byte) => {
+  if (byte === 0) return "0 Bytes";
+  const sizes = ["B", "KB", "MB", "GB", "TB"];
+  const i = parseInt(Math.floor(Math.log(byte) / Math.log(1024)), 10);
+  if (i === 0) return byte + " " + sizes[i];
+  return (byte / Math.pow(1024, i)).toFixed(1) + " " + sizes[i];
+};
+
+// 检测文件是否存在 且 是文件类型
+export const checkFileExist = (filePath) => {
+  try {
+    return fs.existsSync(filePath) && fs.statSync(filePath).isFile();
+  } catch (err) {
+    return false;
+  }
+};
+
+// 获取文件信息, 没返回说明文件不存在, 或不是一个文件
+export const getFileInfo = (filePath) => {
+  let fileInfo = {};
+  if (checkFileExist(filePath)) {
+    fileInfo.fileName = filePath.split("/").pop();
+    fileInfo.formatFileSize = getFormatedFileSize(fs.statSync(filePath).size);
+    fileInfo.rawFileSize = fs.statSync(filePath).size;
+    fileInfo.fileType = filePath.split(".").pop();
+  }
+  return fileInfo;
+};
+
+// 替换文件内的内容
+// replaceMaps: [ {text: '1', newText: '2'}, {text: '45', newText: '112'},]
+export const replaceFileContent = async (file, replaceMaps) => {
+  try {
+    let fileContent = fs.readFileSync(file, "utf8");
+    replaceMaps.forEach((item) => {
+      if (item.text && item.newText)
+        fileContent = fileContent.replaceAll(item.text, item.newText);
+    });
+    return fileContent;
+  } catch (err) {
+    return null;
+  }
+};

package/src/utils/picgo.js

@@ -0,0 +1,149 @@
+import axios from "axios";
+import ora from "ora";
+import fs from "node:fs";
+import { getFileInfo } from "./file.js";
+import sharp from "sharp";
+import chalk from "chalk";
+// 检查文件是否超过最大值, 或不满足最小值
+export const checkFileSize = (filePath, { max, min }) => {
+  const fileInfo = getFileInfo(filePath);
+  if (!fileInfo.fileName) {
+    return {
+      ok: false,
+      msg: "文件不存在",
+      fileInfo: {},
+    };
+  }
+  const { rawFileSize: fileSize, fileName } = fileInfo;
+  const kbValue = fileSize / 1024;
+  if (max && kbValue > max) {
+    return {
+      ok: false,
+      msg: `文件[${fileName}](${kbValue.toFixed(1)}kb)大小超过${max}kb`,
+      fileInfo,
+    };
+  }
+  if (min && kbValue < min) {
+    return {
+      ok: false,
+      msg: `文件[${fileName}](${kbValue.toFixed(1)}kb)大小低于${min}kb`,
+      fileInfo,
+    };
+  }
+
+  return {
+    ok: true,
+    msg: "ok",
+    fileInfo,
+  };
+};
+
+// 检测所有文件, 并过滤出可以被上传的文件
+export const checkFileSizeAndFilter = (filePaths, { max, min }) => {
+  const okFiles = [],
+    errFiles = [];
+  if (filePaths && filePaths.length) {
+    filePaths.forEach((filePath) => {
+      let checkInfo = checkFileSize(filePath, { max, min });
+      // 是一个文件, 并且可以被sharp处理, 则上传
+      if (checkInfo.ok) {
+        if (sharp(filePath)[checkInfo.fileInfo.fileType]) {
+          okFiles.push(filePath);
+        }
+      } else {
+        errFiles.push({
+          errFile: filePath,
+          fileInfo: checkInfo.fileInfo,
+          errMsg: checkInfo.msg,
+        });
+      }
+    });
+  }
+  return {
+    okFiles,
+    errFiles,
+  };
+};
+
+// 请求picgo
+export const requestPicGo = async (files) => {
+  let spinner = ora();
+  if (!files || !files.length) {
+    spinner.fail(`上传失败: 未指定文件`);
+    process.exit(1);
+  }
+
+  spinner.start("正在上传");
+  let picgoRes = await axios
+    .post("http://127.0.0.1:36677/upload", {
+      list: files,
+    })
+    .catch((err) => {
+      spinner.fail("上传失败!请启动PicGo后重试!");
+      process.exit(1);
+    });
+  // 成功
+  if (picgoRes.data && picgoRes.data.success) {
+    spinner.succeed(`上传成功 ${chalk.green(picgoRes.data.result.length)} 个!`);
+    // 返回上传后的url列表
+    return picgoRes.data.result;
+  } else {
+    spinner.fail(`上传失败: ${picgoRes.data.message}`);
+    process.exit(1);
+  }
+};
+
+// 上传文件
+export const uploadFileByPicGo = async (file, option) => {
+  const spinner = ora();
+  let filePth = path.resolve(process.cwd(), file);
+  let checkResult = checkFileSize(filePth, { max: option.max });
+  if (checkResult.ok) {
+    let fileUrls = await requestPicGo([filePth]);
+    spinner.succeed(`上传地址: \n ${fileUrls.join("\n")}`);
+  } else {
+    spinner.fail(`上传失败:  ${checkResult.msg}`);
+  }
+};
+
+// 批量上传文件
+export const batchUploadByPicGo = async (files, option) => {
+  const spinner = ora();
+  const { okFiles, errFiles } = checkFileSizeAndFilter(files, {
+    max: option.max,
+  });
+  if (!okFiles.length) {
+    spinner.fail(
+      `该目录下没有${
+        option.condition ? "名称含有[" + option.condition + "]的" : ""
+      }图片文件(<=${option.max}kb)可以上传`
+    );
+    process.exit(1);
+  }
+  const fileUrls = await requestPicGo(okFiles);
+  // 如果要替换, 重新组装map
+  if (option.replace) {
+    option.replaceMaps = option.replaceMaps.map((item) => {
+      // obsidian里的文件名带空格, 所以需要编码一下
+      // 因为上传后的url里空格被编码了
+      let uploadedUrl = fileUrls.find(
+        (url) => url.indexOf(encodeURIComponent(item.newText)) !== -1
+      );
+      return {
+        text: item.text,
+        newText: uploadedUrl ? `![](${uploadedUrl})` : "",
+      };
+    });
+  }
+  spinner.succeed(`上传地址: \n${fileUrls.join("\n")}`);
+  let realErrFiles = errFiles.filter(
+    (item) => item.fileInfo && item.fileInfo.fileName
+  );
+  if (realErrFiles.length) {
+    spinner.fail(
+      `未上传 ${chalk.red(realErrFiles.length)} 个: \n${realErrFiles
+        .map((item) => item.errMsg)
+        .join("\n")}`
+    );
+  }
+};