think-trace/lib/index.js

Error trace for thinkjs 3.x

const statuses = require('statuses');
const helper = require('think-helper');
const sourceMapSupport = require('source-map-support');
const Tracer = require('./tracer');

module.exports = function(opts, app) {
  const tracer = new Tracer(opts);

  // source map support for compiled file
  if (opts && opts.sourceMap !== false) {
    sourceMapSupport.install();
  }

  let errorCallback;
  if (opts && helper.isFunction(opts.error)) {
    errorCallback = opts.error;
  } else {
    errorCallback = console.error.bind(console);
  }

  if (app) {
    app.think.beforeStartServer(() => tracer.getTemplateContent());
  }

  return (ctx, next) => {
    const beforeTrace = app ? Promise.resolve() : tracer.getTemplateContent();

    return beforeTrace.then(next).then(() => {
      if (ctx.res.statusCode !== 404) {
        return true;
      }

      return ctx.throw(404, `url \`${ctx.path}\` not found.`);
    }).catch(err => {
      if (errorCallback(err, ctx) === false) {
        return Promise.resolve();
      }

      // default to 500
      if (typeof err.status !== 'number' || !statuses[err.status]) {
        err.status = 500;
      }

      // set status to forbidden reset status 200 during set body
      ctx.status = err.status;

      return tracer.run(ctx, err);
    });
  };
};