@goplusvn/core 0.1.57 → 0.1.58

package/src/storage/upload-handler.ts

@@ -0,0 +1,100 @@
+import type { NextRequest } from "next/server";
+
+import { safeFileName } from "./keys";
+import { putFile } from "./storage-service";
+
+/**
+ * Route factory nhận tập tin tải lên.
+ *
+ * Trả về URL dạng `/api/files/<key>` cho MỌI chế độ kho (S3 hay đĩa local) —
+ * cố ý: URL đã lưu trong cột `attachments` không được đổi nghĩa khi admin bật
+ * hay tắt S3. Bản cũ ở app ghi thẳng vào `public/uploads` khi chưa có S3, tức
+ * chứng từ tài chính và CCCD nằm trong thư mục web đọc công khai không cần
+ * phiên; ở đây đĩa local là thư mục PRIVATE và vẫn phải đi qua proxy có quyền.
+ *
+ *   // src/app/api/upload/route.ts
+ *   export const POST = apiHandler(createUploadHandler())
+ */
+
+export interface UploadHandlerOptions {
+  /** Mặc định 25MB. */
+  maxBytes?: number;
+  /** Đuôi cho phép (chữ thường, không dấu chấm). */
+  allowedExtensions?: string[];
+  /** Tiền tố key trong kho. Mặc định `uploads`. */
+  prefix?: string;
+  /** Tiền tố URL trả về. Mặc định `/api/files`. */
+  urlPrefix?: string;
+  onError?: (error: unknown, req: NextRequest) => Response | Promise<Response>;
+}
+
+const DEFAULT_ALLOWED = [
+  "pdf",
+  "jpg",
+  "jpeg",
+  "png",
+  "gif",
+  "webp",
+  "doc",
+  "docx",
+  "xls",
+  "xlsx",
+  "csv",
+  "txt",
+];
+
+/** Thư mục theo ngày (YYYYMMDD) để một ngày gom về một chỗ. */
+function dateDir(now: Date): string {
+  const y = now.getFullYear();
+  const m = String(now.getMonth() + 1).padStart(2, "0");
+  const d = String(now.getDate()).padStart(2, "0");
+  return `${y}${m}${d}`;
+}
+
+export function createUploadHandler(options: UploadHandlerOptions = {}) {
+  const maxBytes = options.maxBytes ?? 25 * 1024 * 1024;
+  const allowed = new Set(options.allowedExtensions ?? DEFAULT_ALLOWED);
+  const prefix = (options.prefix ?? "uploads").replace(/^\/|\/$/g, "");
+  const urlPrefix = (options.urlPrefix ?? "/api/files").replace(/\/$/, "");
+
+  return async function handler(req: NextRequest): Promise<Response> {
+    try {
+      const formData = await req.formData();
+      const file = formData.get("file");
+      if (!file || typeof file === "string") {
+        return Response.json({ error: "No file uploaded" }, { status: 400 });
+      }
+
+      if (file.size > maxBytes) {
+        const mb = Math.round(maxBytes / (1024 * 1024));
+        return Response.json(
+          { error: `Tập tin quá lớn (tối đa ${mb}MB)` },
+          { status: 413 },
+        );
+      }
+
+      const safeName = safeFileName(file.name || "file");
+      const ext = (safeName.split(".").pop() || "").toLowerCase();
+      if (!allowed.has(ext)) {
+        return Response.json(
+          { error: "Định dạng tập tin không được hỗ trợ" },
+          { status: 415 },
+        );
+      }
+
+      // Chèn timestamp trước phần mở rộng — cùng tên tải lên hai lần không đè nhau.
+      const dot = safeName.lastIndexOf(".");
+      const base = dot > 0 ? safeName.slice(0, dot) : safeName;
+      const key = `${prefix}/${dateDir(new Date())}/${base}_${Date.now()}.${ext}`;
+
+      const buffer = Buffer.from(await file.arrayBuffer());
+      await putFile(key, buffer, file.type || "application/octet-stream");
+
+      return Response.json({ url: `${urlPrefix}/${key}`, key });
+    } catch (error) {
+      if (options.onError) return options.onError(error, req);
+      console.error("[storage] upload lỗi:", error);
+      return Response.json({ error: "Upload failed" }, { status: 500 });
+    }
+  };
+}