@wzyjs/utils 0.3.8 → 0.3.10

package/dist/node.cjs.js

@@ -5463,7 +5463,6 @@ __export(exports_node, {
   watch: () => watch,
   void: () => voidType,
   util: () => util,
-  uploadFile: () => uploadFile,
   unknown: () => unknownType,
   union: () => unionType,
   undefined: () => undefinedType,
@@ -5475,12 +5474,8 @@ __export(exports_node, {
   strictObject: () => strictObjectType,
   shuffleArray: () => shuffleArray,
   setErrorMap: () => setErrorMap,
-  setCookie: () => setCookie,
   set: () => setType,
-  sendMail: () => sendMail,
-  schedule: () => schedule,
   retryPromise: () => retryPromise,
-  replaceContentInFile: () => replaceContentInFile,
   replaceByVariables: () => replaceByVariables,
   replaceByRules: () => replaceByRules,
   replaceAll: () => replaceAll,
@@ -5493,6 +5488,7 @@ __export(exports_node, {
   pipeline: () => pipelineType,
   performDecimalOperation: () => performDecimalOperation,
   ostring: () => ostring,
+  oss: () => exports_oss,
   optionsToEnum: () => optionsToEnum,
   optional: () => optionalType,
   onumber: () => onumber,
@@ -5508,25 +5504,24 @@ __export(exports_node, {
   nan: () => nanType,
   map: () => mapType,
   makeIssue: () => makeIssue,
+  mail: () => exports_mail,
   literal: () => literalType,
   limitDecimals: () => limitDecimals,
   levenshteinDistance: () => levenshteinDistance,
   lazy: () => lazyType,
   late: () => late,
   jsonParse: () => jsonParse,
-  json5: () => import_json52.default,
+  json5: () => import_json53.default,
   isValidNumber: () => isValidNumber,
   isValid: () => isValid,
   isJson: () => isJson,
   isDirty: () => isDirty,
   isAsync: () => isAsync,
   isAborted: () => isAborted,
-  interval: () => interval,
   intersection: () => intersectionType,
   instanceof: () => instanceOfType,
   initChinaDayjs: () => initChinaDayjs,
   imageToBase64: () => imageToBase64,
-  getVideoStatus: () => getVideoStatus,
   getType: () => getType,
   getTotal: () => getTotal,
   getStrLength: () => getStrLength,
@@ -5542,16 +5537,15 @@ __export(exports_node, {
   getCookie: () => getCookie,
   getChineseByStr: () => getChineseByStr,
   getChinese: () => getChinese,
-  generateVideo: () => generateVideo,
   generateUniqueFileName: () => generateUniqueFileName,
   generateAlphabetArray: () => generateAlphabetArray,
   function: () => functionType,
   findItem: () => findItem,
   filterParams: () => filterParams,
+  file: () => exports_file,
   executePromise: () => executePromise,
   enum: () => enumType,
   effect: () => effectsType,
-  downloadFile: () => downloadFile,
   discriminatedUnion: () => discriminatedUnionType,
   delay: () => delay,
   defaultErrorMap: () => en_default,
@@ -5559,18 +5553,19 @@ __export(exports_node, {
   datetimeRegex: () => datetimeRegex,
   date: () => dateType,
   custom: () => custom,
+  cron: () => exports_cron,
   consola: () => import_consola.default,
   coerce: () => coerce,
   chinaDayjs: () => chinaDayjs,
   cheerio: () => cheerio,
-  chat: () => chat,
   calcJsText: () => calcJsText,
   boolean: () => booleanType,
   bigint: () => bigIntType,
-  axios: () => import_axios9.default,
+  axios: () => import_axios8.default,
   array: () => arrayType,
   any: () => anyType,
   amount: () => amount,
+  ai: () => exports_ai,
   addIssueToContext: () => addIssueToContext,
   _: () => import_lodash.default,
   ZodVoid: () => ZodVoid,
@@ -5624,9 +5619,6 @@ __export(exports_node, {
   INVALID: () => INVALID,
   Enum: () => Enum,
   EMPTY_PATH: () => EMPTY_PATH,
-  Dayjs: () => import_dayjs2.Dayjs,
-  DataStoreOptions: () => import_nedb.DataStoreOptions,
-  DataStore: () => import_nedb.default,
   DIRTY: () => DIRTY,
   BRAND: () => BRAND
 });
@@ -9608,10 +9600,23 @@ var NEVER = INVALID;
 // src/node.ts
 var cheerio = __toESM(require("cheerio"));
 var import_lodash = __toESM(require_lodash());
-var import_axios9 = __toESM(require("axios"));
-var import_json52 = __toESM(require("json5"));
+var import_axios8 = __toESM(require("axios"));
+var import_json53 = __toESM(require("json5"));
 var import_consola = __toESM(require("consola"));
-var import_nedb = __toESM(require("nedb"));
+
+// src/common/ai/index.ts
+var exports_ai = {};
+__export(exports_ai, {
+  video: () => exports_video,
+  text: () => exports_text,
+  ai302: () => exports_302
+});
+
+// src/common/ai/302/index.ts
+var exports_302 = {};
+__export(exports_302, {
+  chat: () => chat
+});
 
 // src/common/ai/302/axios.ts
 var import_axios = __toESM(require("axios"));
@@ -9651,12 +9656,25 @@ var chat = async (message, model = "gpt-4.1", isParse = true) => {
     throw new Error(error?.response?.statusText || error?.message || "未知原因");
   }
 };
+// src/common/ai/video/index.ts
+var exports_video = {};
+__export(exports_video, {
+  keling: () => exports_keling
+});
 
-// src/common/ai/keling/utils.ts
+// src/common/ai/video/keling/index.ts
+var exports_keling = {};
+__export(exports_keling, {
+  setCookie: () => setCookie,
+  getVideoStatus: () => getVideoStatus,
+  generateVideo: () => generateVideo
+});
+
+// src/common/ai/video/keling/utils.ts
 var url = __toESM(require("url"));
 var path = __toESM(require("path"));
 
-// src/common/ai/keling/axios.ts
+// src/common/ai/video/keling/axios.ts
 var import_axios3 = __toESM(require("axios"));
 var axios2 = import_axios3.default.create({
   headers: {
@@ -9665,7 +9683,7 @@ var axios2 = import_axios3.default.create({
   }
 });
 
-// src/common/ai/keling/utils.ts
+// src/common/ai/video/keling/utils.ts
 var getToken = async (filename) => {
   try {
     const response = await axios2.get("https://klingai.kuaishou.com/api/upload/issue/token", {
@@ -9838,7 +9856,7 @@ var setCookie = (newCookieValue) => {
   axios2.defaults.headers.Cookie = newCookieValue;
 };
 
-// src/common/ai/keling/index.ts
+// src/common/ai/video/keling/index.ts
 var getVideoStatus = async (taskId) => {
   if (!taskId) {
     return;
@@ -9865,6 +9883,48 @@ var generateVideo = async (imageUrl, prompt) => {
   console.log(666, "任务 id", taskId);
   return taskId;
 };
+// src/common/ai/text/index.ts
+var exports_text = {};
+__export(exports_text, {
+  complete: () => complete
+});
+var import_json52 = __toESM(require("json5"));
+var complete = async (apiUrl, model, token, messages, json = true) => {
+  try {
+    const response = await fetch(apiUrl, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        Authorization: `Bearer ${token}`
+      },
+      body: JSON.stringify({
+        model,
+        messages,
+        stream: false
+      })
+    });
+    if (!response.ok) {
+      const errorText = await response.text();
+      throw new Error(`API request failed: ${response.status} ${response.statusText} - ${errorText}`);
+    }
+    const data = await response.json();
+    const content = data?.choices?.[0]?.message?.content || data?.output || "";
+    if (!content) {
+      throw new Error("No content in response");
+    }
+    if (json) {
+      const match = /```(json)?(.*)```/s.exec(content);
+      if (match && match[2]) {
+        return import_json52.default.parse(match[2]);
+      }
+      return import_json52.default.parse(content);
+    }
+    return content;
+  } catch (error) {
+    console.error("AI completion error:", error);
+    throw error;
+  }
+};
 // src/common/base/string.ts
 var getChineseByStr = (str) => {
   if (!str) {
@@ -12617,7 +12677,6 @@ var import_isSameOrAfter = __toESM(require("dayjs/plugin/isSameOrAfter"));
 var import_isSameOrBefore = __toESM(require("dayjs/plugin/isSameOrBefore"));
 var import_utc = __toESM(require("dayjs/plugin/utc"));
 var import_zh_cn = require("dayjs/locale/zh-cn");
-var import_dayjs2 = require("dayjs");
 import_dayjs.default.extend(import_utc.default);
 import_dayjs.default.extend(import_isBetween.default);
 import_dayjs.default.extend(import_weekday.default);
@@ -12645,7 +12704,148 @@ var initChinaDayjs = () => {
 };
 var chinaDayjs = initChinaDayjs();
 var dayjs_default = import_dayjs.default;
-// src/node/mail.ts
+// src/node/oss/index.ts
+var exports_oss = {};
+__export(exports_oss, {
+  cf: () => exports_cloudflare,
+  ali: () => exports_ali
+});
+
+// src/node/oss/ali.ts
+var exports_ali = {};
+__export(exports_ali, {
+  uploadFileToOss: () => uploadFileToOss
+});
+var import_ali_oss = __toESM(require("ali-oss"));
+var oss = null;
+var getOssClient = () => {
+  if (!oss) {
+    oss = new import_ali_oss.default({
+      region: process.env.ALI_OSS_REGION,
+      accessKeyId: process.env.ALI_OSS_ACCESS_KEY_ID,
+      accessKeySecret: process.env.ALI_OSS_ACCESS_KEY_SECRET,
+      bucket: process.env.ALI_OSS_BUCKET
+    });
+  }
+  return oss;
+};
+var uploadFileToOss = async (params) => {
+  const { buffer, fileName } = params;
+  if (!buffer) {
+    throw new Error("Missing required parameters: buffer");
+  }
+  if (!fileName) {
+    throw new Error("Missing required parameters: fileName");
+  }
+  const client = getOssClient();
+  try {
+    const { url: url2 } = await client.put(decodeURIComponent(fileName), buffer, {
+      headers: {
+        "Cache-Control": "public, max-age=31536000, immutable"
+      }
+    });
+    return {
+      url: url2,
+      objectKey: fileName
+    };
+  } catch (error) {
+    throw new Error(`上传到 OSS 失败: ${error}`);
+  }
+};
+// src/node/oss/cloudflare.ts
+var exports_cloudflare = {};
+__export(exports_cloudflare, {
+  uploadFile: () => uploadFile,
+  deleteFile: () => deleteFile
+});
+var import_client_s3 = require("@aws-sdk/client-s3");
+var s3Client = null;
+var getS3Client = () => {
+  if (!s3Client) {
+    s3Client = new import_client_s3.S3Client({
+      region: "auto",
+      endpoint: `https://${process.env.CLOUDFLARE_ACCOUNT_ID}.r2.cloudflarestorage.com`,
+      credentials: {
+        accessKeyId: process.env.CLOUDFLARE_ACCESS_KEY_ID,
+        secretAccessKey: process.env.CLOUDFLARE_SECRET_ACCESS_KEY
+      }
+    });
+  }
+  return s3Client;
+};
+async function uploadFile(params) {
+  const { buffer, mimeType, fileName } = params;
+  if (!buffer) {
+    throw new Error("Missing required parameters: buffer");
+  }
+  if (!fileName) {
+    throw new Error("Missing required parameters: fileName");
+  }
+  if (!mimeType) {
+    throw new Error("Missing required parameters: mimeType");
+  }
+  const client = getS3Client();
+  try {
+    await client.send(new import_client_s3.PutObjectCommand({
+      Bucket: process.env.CLOUDFLARE_BUCKET,
+      Key: fileName,
+      Body: buffer,
+      ContentType: mimeType,
+      ContentDisposition: "inline",
+      CacheControl: "public, max-age=31536000, immutable"
+    }));
+    const url2 = `${process.env.CLOUDFLARE_PUBLIC_URL}/${fileName}`;
+    return {
+      url: url2,
+      objectKey: fileName
+    };
+  } catch (error) {
+    throw new Error(`上传到 Cloudflare 失败: ${error}`);
+  }
+}
+async function deleteFile(objectKey) {
+  const client = getS3Client();
+  try {
+    await client.send(new import_client_s3.DeleteObjectCommand({
+      Bucket: process.env.CLOUDFLARE_BUCKET,
+      Key: objectKey
+    }));
+  } catch (error) {
+    throw new Error(`从 Cloudflare 删除失败: ${error}`);
+  }
+}
+// src/node/cron/index.ts
+var exports_cron = {};
+__export(exports_cron, {
+  schedule: () => schedule,
+  interval: () => interval
+});
+var import_node_cron = __toESM(require("node-cron"));
+var interval = (intervalMinutes, immediately, fn) => {
+  console.log(`定时任务已启动，将每${intervalMinutes}分钟执行一次`);
+  if (immediately) {
+    fn();
+  }
+  setInterval(() => {
+    fn();
+  }, intervalMinutes * 60 * 1000);
+};
+var schedule = (cronExpression, immediately, fn) => {
+  console.log(`定时任务已启动，cron表达式: ${cronExpression}`);
+  if (immediately) {
+    fn();
+  }
+  import_node_cron.default.schedule(cronExpression, () => {
+    fn();
+  }, {
+    timezone: "Asia/Shanghai"
+  });
+};
+// src/node/mail/index.ts
+var exports_mail = {};
+__export(exports_mail, {
+  send: () => send
+});
 var import_nodemailer = __toESM(require("nodemailer"));
 var authMap = {
   163: {
@@ -12663,17 +12863,73 @@ var transporter = import_nodemailer.default.createTransport({
   secure: true,
   auth: authMap.gmail
 });
-var sendMail = (config2) => {
+var send = (config2) => {
   return transporter.sendMail({
     from: authMap.gmail.user,
     ...config2
   });
 };
-// src/node/file.ts
+// src/node/file/index.ts
+var exports_file = {};
+__export(exports_file, {
+  replaceContentInFile: () => replaceContentInFile,
+  processFile: () => processFile,
+  getMimeType: () => getMimeType,
+  downloadFile: () => downloadFile
+});
 var import_fs_extra = __toESM(require("fs-extra"));
 var import_axios7 = __toESM(require("axios"));
 var import_url = __toESM(require("url"));
 var path2 = __toESM(require("path"));
+var import_crypto = __toESM(require("crypto"));
+var getMimeType = (fileName) => {
+  const ext = fileName.split(".").pop()?.toLowerCase();
+  switch (ext) {
+    case "png":
+      return "image/png";
+    case "jpg":
+    case "jpeg":
+      return "image/jpeg";
+    case "gif":
+      return "image/gif";
+    case "webp":
+      return "image/webp";
+    case "svg":
+      return "image/svg+xml";
+    case "mp4":
+      return "video/mp4";
+    case "mp3":
+      return "audio/mpeg";
+    case "pdf":
+      return "application/pdf";
+    case "doc":
+      return "application/msword";
+    case "docx":
+      return "application/vnd.openxmlformats-officedocument.wordprocessingml.document";
+    case "xls":
+      return "application/vnd.ms-excel";
+    case "xlsx":
+      return "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet";
+    case "ppt":
+      return "application/vnd.ms-powerpoint";
+    case "pptx":
+      return "application/vnd.openxmlformats-officedocument.presentationml.presentation";
+    case "txt":
+      return "text/plain";
+    case "html":
+      return "text/html";
+    case "css":
+      return "text/css";
+    case "js":
+      return "application/javascript";
+    case "json":
+      return "application/json";
+    case "xml":
+      return "application/xml";
+    default:
+      return "application/octet-stream";
+  }
+};
 var replaceContentInFile = async (filePath, targetContent, replacement) => {
   try {
     const data = await import_fs_extra.default.readFile(filePath, "utf8");
@@ -12720,82 +12976,74 @@ var downloadFile = async (httpUrl, outputPath) => {
     throw error;
   }
 };
-// src/node/oss.ts
-var import_ali_oss = __toESM(require("ali-oss"));
-var import_axios8 = __toESM(require("axios"));
-var import_fs_extra2 = __toESM(require("fs-extra"));
-var path3 = __toESM(require("path"));
-var oss = new import_ali_oss.default({
-  region: process.env.VITE_OSS_REGION,
-  accessKeyId: process.env.VITE_OSS_ACCESS_KEY_ID,
-  accessKeySecret: process.env.VITE_OSS_ACCESS_KEY_SECRET,
-  bucket: process.env.VITE_OSS_BUCKET
-});
-async function uploadFile(file, filename) {
+var generateUniqueFileName2 = (originalName = "") => {
+  const ext = path2.extname(originalName);
+  const nameWithoutExt = path2.basename(originalName, ext);
+  const finalFileName = `${nameWithoutExt ? `${nameWithoutExt}_` : ""}${Date.now()}_${Math.random().toString(36).slice(2)}${ext}`;
+  return {
+    finalFileName,
+    ext: ext.toLowerCase()
+  };
+};
+var processFile = async (params) => {
+  const { file, type } = params;
   let buffer;
-  let finalFilename;
-  if (typeof file === "string") {
-    if (/^https?:\/\//.test(file)) {
+  let mimeType;
+  let originalName;
+  switch (type) {
+    case "url":
+      if (typeof file !== "string") {
+        throw new Error("当类型为 url 时，file 必须是字符串");
+      }
       try {
-        const response = await import_axios8.default.get(file, { responseType: "arraybuffer" });
+        const response = await import_axios7.default.get(file, { responseType: "arraybuffer" });
         buffer = Buffer.from(response.data);
       } catch (error) {
         throw new Error(`无法从 URL 下载文件: ${error}`);
       }
-      const originalName = path3.basename(new URL(file).pathname);
-      const ext = path3.extname(originalName);
-      const nameWithoutExt = path3.basename(originalName, ext);
-      finalFilename = `${nameWithoutExt}_${Date.now()}${ext}`;
-    } else {
+      originalName = path2.basename(new URL(file).pathname);
+      break;
+    case "path":
+      if (typeof file !== "string") {
+        throw new Error("当类型为 path 时，file 必须是字符串");
+      }
       try {
-        buffer = import_fs_extra2.default.readFileSync(file);
+        buffer = import_fs_extra.default.readFileSync(file);
+        originalName = path2.basename(file);
       } catch (error) {
         throw new Error(`无法读取文件路径: ${error}`);
       }
-      const originalName = path3.basename(file);
-      const ext = path3.extname(originalName);
-      const nameWithoutExt = path3.basename(originalName, ext);
-      finalFilename = `${nameWithoutExt}_${Date.now()}${ext}`;
-    }
-  } else if (Buffer.isBuffer(file)) {
-    if (!filename) {
-      throw new Error("当文件为 Buffer 时，必须提供文件名");
-    }
-    buffer = file;
-    finalFilename = filename;
-  } else {
-    throw new Error("不支持的文件类型");
-  }
-  try {
-    const result = await oss.put(decodeURIComponent(finalFilename), buffer, {
-      headers: {
-        "Cache-Control": "public, max-age=31536000, immutable"
+      break;
+    case "file":
+      if (!(file instanceof File)) {
+        throw new Error("当类型为 file 时，file 必须是 File");
       }
-    });
-    return result.url;
-  } catch (error) {
-    throw new Error(`上传到 OSS 失败: ${error}`);
-  }
-}
-// src/node/cron.ts
-var import_node_cron = __toESM(require("node-cron"));
-var interval = (intervalMinutes, immediately, fn) => {
-  console.log(`定时任务已启动，将每${intervalMinutes}分钟执行一次`);
-  if (immediately) {
-    fn();
+      buffer = Buffer.from(await file.arrayBuffer());
+      mimeType = file.type;
+      originalName = file.name || "";
+      break;
+    case "buffer":
+      if (!Buffer.isBuffer(file)) {
+        throw new Error("当类型为 buffer 时，file 必须是 Buffer");
+      }
+      buffer = file;
+      break;
+    default:
+      throw new Error("不支持的文件类型");
   }
-  setInterval(() => {
-    fn();
-  }, intervalMinutes * 60 * 1000);
-};
-var schedule = (cronExpression, immediately, fn) => {
-  console.log(`定时任务已启动，cron表达式: ${cronExpression}`);
-  if (immediately) {
-    fn();
+  const size = buffer.length;
+  const hash = import_crypto.default.createHash("md5").update(buffer).digest("hex");
+  const { finalFileName, ext } = generateUniqueFileName2(originalName);
+  if (!mimeType) {
+    mimeType = getMimeType(finalFileName);
   }
-  import_node_cron.default.schedule(cronExpression, () => {
-    fn();
-  }, {
-    timezone: "Asia/Shanghai"
-  });
+  return {
+    buffer,
+    finalFileName,
+    originalName,
+    hash,
+    mimeType,
+    size,
+    ext
+  };
 };

package/dist/node.d.ts

@@ -4,9 +4,8 @@ export { default as _ } from 'lodash';
 export { default as axios } from 'axios';
 export { default as json5 } from 'json5';
 export { default as consola } from 'consola';
-export { default as DataStore, DataStoreOptions } from 'nedb';
 export * from './common';
-export * from './node/mail';
-export * from './node/file';
-export * from './node/oss';
-export * from './node/cron';
+export * as oss from './node/oss';
+export * as cron from './node/cron';
+export * as mail from './node/mail';
+export * as file from './node/file';