All files / server/utils/join index.js

100% Statements 27/27
100% Branches 14/14
100% Functions 3/3
100% Lines 26/26
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 5731x 31x 31x     31x 3460x     153x 53x           100x   100x 99x   1x       31x     351x     351x 372x 372x 3491x   3491x   35x 31x 31x 31x         3456x 3429x     34x     370x      
const load = require('loadware');
const assert = require('assert');
const reply = require('../../reply');
 
// Recursively resolve possible function returns
const processReturn = async (ctx, ret) => {
  if (!ret) return;
 
  // Use the returned reply instance
  if (ret.constructor.name === 'Reply') {
    return await ret.exec(ctx);
  }
 
  // TODO: make a check for only accepting the right types of return values
 
  // Create a whole new reply thing
  const fn = typeof ret === 'number' ? 'status' : 'send';
 
  if (ctx.res) {
    return await reply[fn](ret).exec(ctx);
  }
  return ret;
};
 
// Pass an array of modern middleware and return a single modern middleware
module.exports = (...middles) => {
 
  // Flattify all of the middleware
  const middle = load(middles);
 
  // Go through each of them
  return async ctx => {
    let globalReturn = false;
    for (const mid of middle) {
      try {
        // DO NOT MERGE; the else is relevant only for ctx.error
        if (ctx.error) {
          // See if this middleware can fix it
          if (mid.error) {
            assert(mid.error instanceof Function, 'Error handler should be a function');
            let ret = await mid.error(ctx);
            await processReturn(ctx, ret);
          }
        }
        // No error, call next middleware. Skips middleware if there's an error
        else {
          let ret = await mid(ctx);
          globalReturn = await processReturn(ctx, ret) || globalReturn;
        }
      } catch (err) {
        ctx.error = err;
      }
    }
    return globalReturn;
  };
};