millas 0.2.15 → 0.2.16

package/src/http/middleware/UploadMiddleware.js

@@ -0,0 +1,287 @@
+'use strict';
+
+/**
+ * UploadMiddleware
+ *
+ * Built-in multipart/form-data handler for Millas.
+ * Zero config required — just declare file() fields in your shape or
+ * pass options to the constructor and the framework takes care of the rest.
+ *
+ * Powered by multer under the hood (lazy-required, so it only needs to be
+ * installed if you actually handle uploads — multer is listed as an optional
+ * peer dependency).
+ *
+ * ── Auto-injection (recommended) ─────────────────────────────────────────────
+ *
+ *   When a route shape has encoding: 'multipart' or any file() field in its
+ *   "in" schema, the Router injects UploadMiddleware automatically.
+ *   You never touch it.
+ *
+ *   Route.post('/media/upload', MediaController, 'upload')
+ *     .shape({
+ *       label: 'Upload media',
+ *       encoding: 'multipart',
+ *       in: {
+ *         file: file().required().maxSize('50mb').mimeType(['image/jpeg','image/png','video/mp4']),
+ *       },
+ *     });
+ *
+ *   async upload({ body, file, user }) {
+ *     // file is a multer file object: { buffer, mimetype, size, originalname }
+ *     // body has the rest of the form fields
+ *   }
+ *
+ * ── Manual usage ──────────────────────────────────────────────────────────────
+ *
+ *   const { UploadMiddleware } = require('millas/core/http');
+ *
+ *   // Single named field
+ *   Route.post('/avatar', ['auth', new UploadMiddleware({ field: 'avatar' })], UserController, 'updateAvatar');
+ *
+ *   // Multiple named fields
+ *   Route.post('/listing', ['auth', new UploadMiddleware({ fields: [{ name: 'photo', maxCount: 5 }] })], ListingController, 'store');
+ *
+ *   // Any field name (multer .any())
+ *   Route.post('/upload', [new UploadMiddleware({ any: true })], UploadController, 'store');
+ *
+ * ── Options ───────────────────────────────────────────────────────────────────
+ *
+ *   {
+ *     field:    string           // single field name (default: 'file')
+ *     fields:   [{name, maxCount}]  // multiple named fields
+ *     any:      boolean          // accept any field name
+ *     maxSize:  number|string    // max file size bytes or '10mb' (default: 50mb)
+ *     storage:  'memory'|'disk'  // multer storage (default: 'memory')
+ *     dest:     string           // disk storage destination (only when storage: 'disk')
+ *     filter:   Function         // custom multer fileFilter(req, file, cb)
+ *   }
+ *
+ * ── Handler context ───────────────────────────────────────────────────────────
+ *
+ *   After this middleware runs, RequestContext exposes:
+ *
+ *     ctx.file   — single uploaded file (multer file object)
+ *     ctx.files  — multiple uploaded files (object or array)
+ *     ctx.body   — non-file form fields (strings)
+ *
+ *   Multer file object shape:
+ *     {
+ *       fieldname:    'file',
+ *       originalname: 'photo.jpg',
+ *       mimetype:     'image/jpeg',
+ *       size:         204800,
+ *       buffer:       <Buffer ...>    // memory storage only
+ *       path:         '/tmp/...'      // disk storage only
+ *     }
+ */
+class UploadMiddleware {
+  /**
+   * @param {object} options
+   * @param {string}              [options.field='file']   single field name
+   * @param {Array<{name,maxCount}>} [options.fields]      multiple named fields
+   * @param {boolean}             [options.any=false]      accept any field
+   * @param {number|string}       [options.maxSize]        max file size
+   * @param {'memory'|'disk'}     [options.storage='memory']
+   * @param {string}              [options.dest]           disk dest (storage:'disk')
+   * @param {Function}            [options.filter]         multer fileFilter
+   */
+  constructor(options = {}) {
+    this._options = {
+      field:   options.field   || 'file',
+      fields:  options.fields  || null,
+      any:     options.any     || false,
+      maxSize: options.maxSize !== undefined ? _parseSize(options.maxSize) : 50 * 1024 * 1024,
+      storage: options.storage || 'memory',
+      dest:    options.dest    || null,
+      filter:  options.filter  || null,
+    };
+    this._multerInstance = null;
+  }
+
+  // ── Millas Middleware protocol ─────────────────────────────────────────────
+
+  /**
+   * Build and cache the multer middleware on first request, then delegate.
+   * Runs as a standard Millas middleware — ctx + next.
+   *
+   * Because multer is Express middleware internally, we reach into ctx.req.raw
+   * to get the underlying Express req/res and call multer directly.
+   */
+  async handle(ctx, next) {
+    const multerMw = this._getMulterMiddleware();
+    const expressReq = ctx.req.raw;
+
+    await new Promise((resolve, reject) => {
+      // multer needs a real Express res object; we synthesize a minimal one
+      // that satisfies multer's internal usage (it only calls res.end / statusCode
+      // on hard errors, which we convert to thrown exceptions via reject).
+      const fakeRes = _buildFakeRes(reject);
+      multerMw(expressReq, fakeRes, (err) => {
+        if (err) return reject(err);
+        resolve();
+      });
+    });
+
+    // Wrap raw multer file objects in UploadedFile instances so handlers
+    // receive rich, manipulable file objects instead of plain multer blobs.
+    const UploadedFile = require('../UploadedFile');
+    const { file, files } = UploadedFile.wrapRequest(expressReq);
+
+    // Sync into RequestContext (which was constructed before multer ran).
+    ctx.file  = file;
+    ctx.files = files;
+
+    // Also mirror back onto the raw express req so Router._buildShapeMiddleware
+    // can pick them up when it merges file inputs for validation.
+    expressReq._millaFile  = file;
+    expressReq._millaFiles = files;
+
+    // req.body is already a live reference — multer mutated it in-place,
+    // so non-file form fields are automatically visible via ctx.body.
+
+    return next();
+  }
+
+  // ── Static factory helpers ──────────────────────────────────────────────────
+
+  /**
+   * Build an UploadMiddleware from a shape definition.
+   * Called internally by the Router when auto-injecting.
+   *
+   * @param {import('../Shape').ShapeDefinition} shape
+   * @returns {UploadMiddleware}
+   */
+  static fromShape(shape) {
+    const opts = { storage: 'memory' };
+
+    // Collect file field names from the shape's "in" validators
+    const fileFields = _fileFieldsFromShape(shape);
+
+    if (fileFields.length === 1) {
+      opts.field = fileFields[0].name;
+    } else if (fileFields.length > 1) {
+      opts.fields = fileFields; // [{name, maxCount}]
+    }
+
+    // Honour explicit maxSize from first file validator found
+    const firstFileValidator = _firstFileValidator(shape);
+    if (firstFileValidator?._maxSizeBytes) {
+      opts.maxSize = firstFileValidator._maxSizeBytes;
+    }
+
+    // Build a combined mimeType filter from all file validators
+    const allMimes = _allMimeTypes(shape);
+    if (allMimes.length) {
+      opts.filter = (req, file, cb) => {
+        cb(null, allMimes.includes(file.mimetype));
+      };
+    }
+
+    return new UploadMiddleware(opts);
+  }
+
+  // ── Internal ───────────────────────────────────────────────────────────────
+
+  _getMulterMiddleware() {
+    if (this._multerInstance) return this._multerInstance;
+
+    let multer;
+    try {
+      multer = require('multer');
+    } catch {
+      throw new Error(
+        '[Millas UploadMiddleware] multer is not installed.\n' +
+        'Run:  npm install multer\n' +
+        'multer is required for multipart/file upload support.'
+      );
+    }
+
+    const { field, fields, any, maxSize, storage, dest, filter } = this._options;
+
+    // Storage engine
+    const storageEngine = storage === 'disk' && dest
+      ? multer.diskStorage({ destination: dest, filename: (_req, file, cb) => cb(null, `${Date.now()}_${file.originalname}`) })
+      : multer.memoryStorage();
+
+    const multerConfig = { storage: storageEngine, limits: { fileSize: maxSize } };
+    if (filter) multerConfig.fileFilter = filter;
+
+    const upload = multer(multerConfig);
+
+    // Pick the right multer handler based on options
+    if (any) {
+      this._multerInstance = upload.any();
+    } else if (fields) {
+      this._multerInstance = upload.fields(fields);
+    } else {
+      this._multerInstance = upload.single(field);
+    }
+
+    return this._multerInstance;
+  }
+}
+
+// ── Helpers ───────────────────────────────────────────────────────────────────
+
+/**
+ * Parse a human-readable size string to bytes.
+ * '5mb' → 5242880, '500kb' → 512000, 1024 → 1024
+ */
+function _parseSize(size) {
+  if (typeof size === 'number') return size;
+  const s = String(size).trim().toLowerCase();
+  const n = parseFloat(s);
+  if (s.endsWith('gb')) return n * 1024 * 1024 * 1024;
+  if (s.endsWith('mb')) return n * 1024 * 1024;
+  if (s.endsWith('kb')) return n * 1024;
+  return n;
+}
+
+/**
+ * Scan a shape's "in" schema for FileValidator instances.
+ * Returns [{name, maxCount: 1}] for each file field found.
+ */
+function _fileFieldsFromShape(shape) {
+  if (!shape?.in) return [];
+  const fields = [];
+  for (const [name, validator] of Object.entries(shape.in)) {
+    if (validator?._type === 'file') {
+      fields.push({ name, maxCount: 1 });
+    }
+  }
+  return fields;
+}
+
+function _firstFileValidator(shape) {
+  if (!shape?.in) return null;
+  for (const validator of Object.values(shape.in)) {
+    if (validator?._type === 'file') return validator;
+  }
+  return null;
+}
+
+function _allMimeTypes(shape) {
+  if (!shape?.in) return [];
+  const mimes = [];
+  for (const validator of Object.values(shape.in)) {
+    if (validator?._type === 'file' && Array.isArray(validator._mimeTypes)) {
+      mimes.push(...validator._mimeTypes);
+    }
+  }
+  return [...new Set(mimes)];
+}
+
+/**
+ * Build a minimal fake Express response object for multer.
+ * Multer only interacts with res in error cases — we map those to rejections.
+ */
+function _buildFakeRes(reject) {
+  return {
+    statusCode: 200,
+    end: (msg) => reject(new Error(msg || 'Upload error')),
+    setHeader: () => {},
+    getHeader: () => null,
+  };
+}
+
+module.exports = UploadMiddleware;

package/src/router/Router.js

@@ -73,6 +73,15 @@ class Router {
     // validation middleware that runs BEFORE the handler.
     // On failure → 422 immediately, handler never runs.
     // On success → ctx.body is replaced with coerced, validated output.
+    //
+    // When the shape declares encoding:'multipart' or contains any file()
+    // field, UploadMiddleware is injected first (before validation) so that
+    // multer parses the multipart body and populates req.file / req.files /
+    // req.body before the validation step reads them.
+    const uploadMiddlewares = route.shape && this._shapeNeedsUpload(route.shape)
+      ? this._buildUploadMiddleware(route.shape)
+      : [];
+
     const shapeMiddlewares = route.shape
       ? this._buildShapeMiddleware(route.shape)
       : [];
@@ -87,29 +96,98 @@ class Router {
 
     this._adapter.mountRoute(route.verb, route.path, [
       ...middlewareHandlers,
+      ...uploadMiddlewares,
       ...shapeMiddlewares,
       terminalHandler,
     ]);
   }
 
+  /**
+   * Returns true if the shape requires multipart parsing —
+   * either because encoding is explicitly set to 'multipart', or because
+   * the "in" schema contains at least one file() validator.
+   *
+   * @param {import('../http/Shape').ShapeDefinition} shape
+   * @returns {boolean}
+   */
+  _shapeNeedsUpload(shape) {
+    if (shape.encoding === 'multipart') return true;
+    if (!shape.in) return false;
+    return Object.values(shape.in).some(v => v?._type === 'file');
+  }
+
+  /**
+   * Build an Express middleware array that runs UploadMiddleware before
+   * the shape validation step. The UploadMiddleware is configured from
+   * the shape (field names, maxSize, mimeTypes) so developers don't have
+   * to configure it separately.
+   *
+   * @param {import('../http/Shape').ShapeDefinition} shape
+   * @returns {Function[]}
+   */
+  _buildUploadMiddleware(shape) {
+    const UploadMiddleware = require('../http/middleware/UploadMiddleware');
+    const instance = UploadMiddleware.fromShape(shape);
+    // Wrap it through the adapter so it becomes an Express (req,res,next) fn
+    return [this._adapter.wrapMiddleware(instance, this._container)];
+  }
+
   /**
    * Build Express middleware functions from a shape definition.
    * Returns an array of 0, 1, or 2 middleware functions
    * (one for body/in, one for query) depending on what the shape declares.
    *
+   * For multipart routes, file fields from req.file / req.files are merged
+   * into the validation input so file() validators in the schema run correctly.
+   *
    * @param {import('../http/Shape').ShapeDefinition} shape
    * @returns {Function[]}
    */
   _buildShapeMiddleware(shape) {
     const { Validator } = require('../validation/Validator');
     const middlewares   = [];
+    const isMultipart   = this._shapeNeedsUpload(shape);
 
     // ── Body / in validation ───────────────────────────────────────────────
     if (shape.in && Object.keys(shape.in).length) {
       middlewares.push(async (req, res, next) => {
         try {
-          const rawBody = req.body || {};
-          const clean   = await Validator.validate(rawBody, shape.in);
+          // For multipart requests, merge uploaded files into the validation
+          // input so file() validators in the shape's "in" schema can run.
+          // req.file  → single file (multer .single())
+          // req.files → multiple files (multer .fields() or .any())
+          let rawBody = req.body || {};
+          if (isMultipart) {
+            const fileInputs = {};
+            // Prefer the already-wrapped UploadedFile instances written by
+            // UploadMiddleware so FileValidator receives the same UploadedFile
+            // objects that the handler will destructure.
+            if (req._millaFile) {
+              fileInputs[req._millaFile.fieldName] = req._millaFile;
+            }
+            if (req._millaFiles) {
+              for (const [fieldname, f] of Object.entries(req._millaFiles)) {
+                fileInputs[fieldname] = f;
+              }
+            }
+            // Fallback to raw multer objects if UploadMiddleware hasn't run
+            // (e.g. manual upload middleware setup without UploadMiddleware)
+            if (!Object.keys(fileInputs).length) {
+              if (req.file) fileInputs[req.file.fieldname] = req.file;
+              if (req.files) {
+                if (Array.isArray(req.files)) {
+                  for (const f of req.files) fileInputs[f.fieldname] = f;
+                } else {
+                  for (const [fieldname, arr] of Object.entries(req.files)) {
+                    fileInputs[fieldname] = arr.length === 1 ? arr[0] : arr;
+                  }
+                }
+              }
+            }
+            rawBody = { ...rawBody, ...fileInputs };
+          }
+
+          const clean = await Validator.validate(rawBody, shape.in);
           // Replace req.body with the coerced, validated subset so the
           // handler's { body } destructure gets clean data automatically.
           req.body = clean;

package/src/storage/Storage.js

@@ -129,6 +129,10 @@ class Storage {
     const driverName = diskConf.driver || name || 'local';
 
     switch (driverName) {
+      case 's3': {
+        const S3Driver = require('./drivers/S3Driver');
+        return new S3Driver(diskConf);
+      }
       case 'local':
       default: {
         const LocalDriver = require('./drivers/LocalDriver');