feat: 变量

src/controllers/index.ts

@@ -10,8 +10,8 @@ const combinedRouter = new Router();
 // 组合根路由
 combinedRouter.use(RootController.routes()).use(RootController.allowedMethods());
 
-// // 组合图像相关接口路由
-// combinedRouter.use(ImageController.routes()).use(ImageController.allowedMethods());
+// 组合图像相关接口路由
+combinedRouter.use(ImageController.routes()).use(ImageController.allowedMethods());
 
 // // 组合静态资源接口路由
 // combinedRouter.use(StaticResourceController.routes()).use(StaticResourceController.allowedMethods());

src/middleware/error-handler.ts

@@ -0,0 +1,11 @@
+import { Context } from 'koa';
+
+// 统一的错误处理中间件函数
+export const handleControllerError = async (ctx: Context, error: any) => {
+  console.error('Controller Error:', error);
+
+  ctx.status = error.statusCode || 500;
+  ctx.body = {
+    error: error.message,
+  };
+};

src/middleware/index.ts

@@ -0,0 +1 @@
+export * from './error-handler';

src/service/article-service.ts

@@ -0,0 +1,10 @@
+import axios from 'axios';
+import fs from 'fs';
+import htmlDocx from 'html-docx-js';
+import path from 'path';
+
+import { delay } from '@/utils';
+
+const uploadDir = path.join(__dirname, '../../uploads');
+
+export const processArticleServe = async (data: { title: string; content: string; config: { output: string; hfApiKey: string } }) => {};

src/service/image-service.ts

@@ -0,0 +1,40 @@
+import path from 'path';
+import sharp from 'sharp';
+
+import { createUploadDir, getFileStream } from '../utils/file-utils';
+
+const uploadDir = path.join(__dirname, '../uploads');
+
+// 处理接收二进制数据并转换为JPG图像的业务逻辑
+export const processImageServe = async (data: any) => {
+  if (!Buffer.isBuffer(data)) {
+    throw new Error('Expected application/octet-stream data');
+  }
+
+  const outputFilename = `image-${Date.now()}.jpg`;
+  const outputPath = path.join(uploadDir, outputFilename);
+
+  // 使用 sharp 将 Buffer 转换为 JPG 并保存到本地文件系统
+  await sharp(data).toFormat('jpg').toFile(outputPath);
+
+  // 返回图像的文件路径
+  return {
+    message: 'Buffer data received and converted to JPG',
+    imagePath: outputFilename,
+  };
+};
+
+// 获取生成的图像文件的业务逻辑
+export const getGeneratedFile = async (filename: string, type: string) => {
+  const uploadDir = createUploadDir();
+  const fileStream = getFileStream(filename, uploadDir);
+
+  if (fileStream) {
+    return {
+      type,
+      stream: fileStream,
+    };
+  } else {
+    throw new Error('File not found');
+  }
+};

src/service/index.ts

@@ -0,0 +1,2 @@
+export * from './article-service';
+export * from './image-service';

src/utils/common.ts

@@ -0,0 +1,7 @@
+import util from 'util';
+const setTimeoutPromise = util.promisify(setTimeout);
+
+export async function delay(time: number) {
+  await setTimeoutPromise(time);
+  console.log(`延迟${time}毫秒后执行`);
+}

src/utils/file-utils.ts

@@ -0,0 +1,72 @@
+import fs from 'fs';
+import os from 'os';
+import path from 'path';
+import sharp from 'sharp';
+
+// 创建上传文件夹，如果不存在
+export const createUploadDir = (): string => {
+  const uploadDir = path.join(__dirname, '../../uploads');
+  if (!fs.existsSync(uploadDir)) {
+    fs.mkdirSync(uploadDir);
+  }
+  return uploadDir;
+};
+
+// 使用 sharp 将 Buffer 转换为 JPG 并保存到本地文件系统
+export const bufferToJpg = async (data: Buffer, outputFilename: string, uploadDir: string): Promise<string> => {
+  const outputPath = path.join(uploadDir, outputFilename);
+  await sharp(data).toFormat('jpg').toFile(outputPath);
+  return outputFilename;
+};
+
+// 检查文件是否存在并返回文件流（如果存在）
+export const getFileStream = (filename: string, uploadDir: string) => {
+  const filePath = path.join(uploadDir, filename);
+  if (fs.existsSync(filePath)) {
+    return fs.createReadStream(filePath);
+  }
+  return null;
+};
+
+export function bufferToBase64ImageSrc(buffer: Buffer, mimeType: string): string {
+  // 将 Buffer 转换为 base64 字符串
+  const base64 = buffer.toString('base64');
+
+  // 构建 base64 图片源
+  return `data:${mimeType};base64,${base64}`;
+}
+
+export async function downloadDocx(url: string, fileName: string) {
+  try {
+    // 发起 GET 请求下载文件
+    const response = await fetch(url);
+    if (!response.ok) {
+      throw new Error(`Failed to fetch ${url}: ${response.statusText}`);
+    }
+    // 确保响应头中的 Content-Type 为 DOCX 文件
+    const contentType = response.headers.get('content-type');
+    if (contentType !== 'application/vnd.openxmlformats-officedocument.wordprocessingml.document') {
+      throw new Error(`Unexpected content type: ${contentType}`);
+    }
+
+    // 将响应体转换为 Buffer
+    const blob = await response.blob();
+    // 将 Blob 转换为 ArrayBuffer
+    const arrayBuffer = await blob.arrayBuffer();
+    // 将 ArrayBuffer 转换为 Buffer
+    const buffer = Buffer.from(arrayBuffer);
+    // 获取用户的桌面路径
+    const desktopPath = path.join(os.homedir(), 'Desktop');
+    if (!fs.existsSync(desktopPath)) {
+      throw new Error('Desktop path does not exist. Please ensure you have a Desktop directory.');
+    }
+    const filePath = path.join(desktopPath, fileName);
+
+    // 将 Buffer 写入到文件
+    fs.writeFileSync(filePath, buffer);
+
+    console.log(`DOCX 文件已保存到 ${filePath}`);
+  } catch (error) {
+    console.error('下载文件时出错:', error);
+  }
+}

src/utils/index.ts

@@ -0,0 +1,2 @@
+export * from './common';
+export * from './file-utils';