@wzyjs/utils 0.3.8 → 0.3.10

package/dist/node.esm.js

@@ -9420,7 +9420,20 @@ import * as cheerio from "cheerio";
 import { default as default3 } from "axios";
 import { default as default4 } from "json5";
 import { default as default5 } from "consola";
-import { default as default6, DataStoreOptions } from "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
 import axios_ from "axios";
@@ -9460,12 +9473,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
 import * as url from "url";
 import * as path from "path";
 
-// src/common/ai/keling/axios.ts
+// src/common/ai/video/keling/axios.ts
 import axios_2 from "axios";
 var axios2 = axios_2.create({
   headers: {
@@ -9474,7 +9500,7 @@ var axios2 = axios_2.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", {
@@ -9647,7 +9673,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;
@@ -9674,6 +9700,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
+});
+import json52 from "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 json52.parse(match[2]);
+      }
+      return json52.parse(content);
+    }
+    return content;
+  } catch (error) {
+    console.error("AI completion error:", error);
+    throw error;
+  }
+};
 // src/common/base/string.ts
 var getChineseByStr = (str) => {
   if (!str) {
@@ -12426,7 +12494,6 @@ import isSameOrAfter from "dayjs/plugin/isSameOrAfter";
 import isSameOrBefore from "dayjs/plugin/isSameOrBefore";
 import utc from "dayjs/plugin/utc";
 import"dayjs/locale/zh-cn";
-import { Dayjs } from "dayjs";
 dayjs.extend(utc);
 dayjs.extend(isBetween);
 dayjs.extend(weekday);
@@ -12454,7 +12521,148 @@ var initChinaDayjs = () => {
 };
 var chinaDayjs = initChinaDayjs();
 var dayjs_default = dayjs;
-// 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
+});
+import OSS from "ali-oss";
+var oss = null;
+var getOssClient = () => {
+  if (!oss) {
+    oss = new OSS({
+      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
+});
+import { S3Client, PutObjectCommand, DeleteObjectCommand } from "@aws-sdk/client-s3";
+var s3Client = null;
+var getS3Client = () => {
+  if (!s3Client) {
+    s3Client = new 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 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 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
+});
+import cron from "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();
+  }
+  cron.schedule(cronExpression, () => {
+    fn();
+  }, {
+    timezone: "Asia/Shanghai"
+  });
+};
+// src/node/mail/index.ts
+var exports_mail = {};
+__export(exports_mail, {
+  send: () => send
+});
 import nodemailer from "nodemailer";
 var authMap = {
   163: {
@@ -12472,17 +12680,73 @@ var transporter = nodemailer.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
+});
 import fs from "fs-extra";
 import axios4 from "axios";
 import url2 from "url";
 import * as path2 from "path";
+import crypto2 from "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 fs.readFile(filePath, "utf8");
@@ -12529,84 +12793,76 @@ var downloadFile = async (httpUrl, outputPath) => {
     throw error;
   }
 };
-// src/node/oss.ts
-import OSS from "ali-oss";
-import axios5 from "axios";
-import fs2 from "fs-extra";
-import * as path3 from "path";
-var oss = new OSS({
-  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 axios5.get(file, { responseType: "arraybuffer" });
+        const response = await axios4.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 = fs2.readFileSync(file);
+        buffer = fs.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
-import cron from "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 = crypto2.createHash("md5").update(buffer).digest("hex");
+  const { finalFileName, ext } = generateUniqueFileName2(originalName);
+  if (!mimeType) {
+    mimeType = getMimeType(finalFileName);
   }
-  cron.schedule(cronExpression, () => {
-    fn();
-  }, {
-    timezone: "Asia/Shanghai"
-  });
+  return {
+    buffer,
+    finalFileName,
+    originalName,
+    hash,
+    mimeType,
+    size,
+    ext
+  };
 };
 var export__ = import_lodash.default;
 
@@ -12615,7 +12871,6 @@ export {
   watch,
   voidType as void,
   util,
-  uploadFile,
   unknownType as unknown,
   unionType as union,
   undefinedType as undefined,
@@ -12627,12 +12882,8 @@ export {
   strictObjectType as strictObject,
   shuffleArray,
   setErrorMap,
-  setCookie,
   setType as set,
-  sendMail,
-  schedule,
   retryPromise,
-  replaceContentInFile,
   replaceByVariables,
   replaceByRules,
   replaceAll,
@@ -12645,6 +12896,7 @@ export {
   pipelineType as pipeline,
   performDecimalOperation,
   ostring,
+  exports_oss as oss,
   optionsToEnum,
   optionalType as optional,
   onumber,
@@ -12660,6 +12912,7 @@ export {
   nanType as nan,
   mapType as map,
   makeIssue,
+  exports_mail as mail,
   literalType as literal,
   limitDecimals,
   levenshteinDistance,
@@ -12673,12 +12926,10 @@ export {
   isDirty,
   isAsync,
   isAborted,
-  interval,
   intersectionType as intersection,
   instanceOfType as instanceof,
   initChinaDayjs,
   imageToBase64,
-  getVideoStatus,
   getType,
   getTotal,
   getStrLength,
@@ -12694,16 +12945,15 @@ export {
   getCookie,
   getChineseByStr,
   getChinese,
-  generateVideo,
   generateUniqueFileName,
   generateAlphabetArray,
   functionType as function,
   findItem,
   filterParams,
+  exports_file as file,
   executePromise,
   enumType as enum,
   effectsType as effect,
-  downloadFile,
   discriminatedUnionType as discriminatedUnion,
   delay,
   en_default as defaultErrorMap,
@@ -12711,11 +12961,11 @@ export {
   datetimeRegex,
   dateType as date,
   custom,
+  exports_cron as cron,
   default5 as consola,
   coerce,
   chinaDayjs,
   cheerio,
-  chat,
   calcJsText,
   booleanType as boolean,
   bigIntType as bigint,
@@ -12723,6 +12973,7 @@ export {
   arrayType as array,
   anyType as any,
   amount,
+  exports_ai as ai,
   addIssueToContext,
   export__ as _,
   ZodVoid,
@@ -12776,9 +13027,6 @@ export {
   INVALID,
   Enum,
   EMPTY_PATH,
-  Dayjs,
-  DataStoreOptions,
-  default6 as DataStore,
   DIRTY,
   BRAND
 };