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 5728x 28x 28x     28x 2905x     117x 40x           77x   77x 76x   1x       28x     300x     300x 320x 320x 2937x   2937x   32x 28x 28x 28x         2905x 2877x     29x     317x      
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;
  };
};