@kiyasov/platform-hono 1.3.7 → 1.3.9

package/src/multer/storage/disk-storage.ts

@@ -0,0 +1,107 @@
+import { tmpdir } from 'os';
+import { createWriteStream } from 'fs';
+import { mkdir, unlink } from 'fs/promises';
+import { join } from 'path';
+
+import { StorageFile, Storage } from './storage';
+import { getUniqueFilename, pathExists } from '../fs';
+import { pump } from '../stream';
+import { HonoRequest } from 'hono';
+import { Readable } from 'node:stream';
+
+export interface DiskStorageFile extends StorageFile {
+  dest: string;
+  filename: string;
+  path: string;
+}
+
+type DiskStorageOptionHandler =
+  | ((file: File, req: HonoRequest) => Promise<string> | string)
+  | string;
+
+export interface DiskStorageOptions {
+  dest?: DiskStorageOptionHandler;
+  filename?: DiskStorageOptionHandler;
+  removeAfter?: boolean;
+}
+
+const excecuteStorageHandler = (
+  file: File,
+  req: HonoRequest,
+  obj?: DiskStorageOptionHandler,
+) => {
+  if (typeof obj === 'function') {
+    return obj(file, req);
+  }
+
+  if (obj != null) return obj;
+
+  return null;
+};
+
+const ENV_TESTS_STORAGE_TMP_PATH = process.env.__TESTS_TMP_PATH__;
+export class DiskStorage
+  implements Storage<DiskStorageFile, DiskStorageOptions>
+{
+  public readonly options?: DiskStorageOptions;
+
+  constructor(options?: DiskStorageOptions) {
+    this.options = options;
+
+    if (ENV_TESTS_STORAGE_TMP_PATH != null) {
+      this.options = { ...this.options, dest: ENV_TESTS_STORAGE_TMP_PATH };
+    }
+  }
+
+  public async handleFile(file: File, req: HonoRequest, fieldName: string) {
+    const filename = await this.getFilename(file, req, this.options?.filename);
+    const dest = await this.getFileDestination(file, req, this.options?.dest);
+
+    if (!(await pathExists(dest))) {
+      await mkdir(dest, { recursive: true });
+    }
+
+    const path = join(dest, filename);
+    const stream = createWriteStream(path);
+
+    const buffer = await file.arrayBuffer();
+    const readableStream = Readable.from(Buffer.from(buffer));
+
+    await pump(readableStream, stream);
+
+    return {
+      size: stream.bytesWritten,
+      dest,
+      filename,
+      originalFilename: file.name,
+      path,
+      mimetype: file.type,
+      encoding: 'utf-8',
+      fieldname: fieldName,
+    };
+  }
+
+  public async removeFile(file: DiskStorageFile, force?: boolean) {
+    if (!this.options?.removeAfter && !force) return;
+
+    await unlink(file.path);
+  }
+
+  protected async getFilename(
+    file: File,
+    req: HonoRequest,
+    obj?: DiskStorageOptionHandler,
+  ): Promise<string> {
+    return (
+      excecuteStorageHandler(file, req, obj) ?? getUniqueFilename(file.name)
+    );
+  }
+
+  protected async getFileDestination(
+    file: File,
+    req: HonoRequest,
+    obj?: DiskStorageOptionHandler,
+  ): Promise<string> {
+    return excecuteStorageHandler(file, req, obj) ?? tmpdir();
+  }
+}

package/src/multer/storage/index.ts

@@ -0,0 +1,3 @@
+export * from "./disk-storage";
+export * from "./memory-storage";
+export * from "./storage";

package/src/multer/storage/memory-storage.ts

@@ -0,0 +1,25 @@
+import { StorageFile, Storage } from './storage';
+import { HonoRequest } from 'hono';
+
+export interface MemoryStorageFile extends StorageFile {
+  buffer: Buffer;
+}
+
+export class MemoryStorage implements Storage<MemoryStorageFile> {
+  public async handleFile(file: File, _req: HonoRequest, fieldName: string) {
+    const buffer = await file.arrayBuffer().then(Buffer.from);
+
+    return {
+      buffer,
+      size: buffer.length,
+      encoding: 'utf-8',
+      mimetype: file.type,
+      fieldname: fieldName,
+      originalFilename: file.name,
+    };
+  }
+
+  public async removeFile(file: MemoryStorageFile) {
+    delete file.buffer;
+  }
+}

package/src/multer/storage/storage.ts

@@ -0,0 +1,15 @@
+import { HonoRequest } from 'hono';
+
+export interface StorageFile {
+  size: number;
+  fieldname: string;
+  encoding: string;
+  mimetype: string;
+  originalFilename: string;
+}
+
+export interface Storage<T extends StorageFile = StorageFile, K = any> {
+  handleFile: (file: File, req: HonoRequest, fieldName: string) => Promise<T>;
+  removeFile: (file: T, force?: boolean) => Promise<void> | void;
+  options?: K;
+}

package/src/multer/stream/index.ts

@@ -0,0 +1,4 @@
+import { promisify } from 'node:util';
+import { pipeline } from 'node:stream';
+
+export const pump = promisify(pipeline);