All files / server/utils/join index.js

100% Statements 29/29
100% Branches 16/16
100% Functions 3/3
100% Lines 28/28
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 57 58 59 60 61 62 6333x 33x 33x     33x 3613x     163x 54x       109x 4x           105x   105x 104x     1x       33x     367x     367x 388x 388x 3649x   3649x   45x 37x 37x 37x         3604x 3576x     41x     386x      
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 ret.exec(ctx);
  }
 
  // Errors should be thrown to trickle down
  if (ret instanceof Error) {
    throw ret;
  }
 
  // 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;
  };
};