egg-jianghu/app/middleware/downloadUserInfo.js

egg-jianghu

'use strict';

const range = require('koa-range');
const send = require('koa-send');
const userInfoUtil = require('./middlewareUtil/userInfoUtil');

module.exports = options => {
  return async (ctx, next) => {

    const { jianghuKnex, logger, db, config } = ctx.app;
    const { appId, appType, jiangHuConfig = {} } = config;

    // 由于 userInfoUtil 针对的是 post 请求，所以需要构造一个结构一致的 body
    const mockBody = {
      appData: {
        authToken: ctx.cookies.get(`${appId}_authToken`, {
          httpOnly: false,
          signed: false,
        }),
      },
    };

    // 捕获 userInfo: { user, userGroupRoleList, allowPageList, userAppList } 到 ctx.userInfo
    ctx.userInfo = await userInfoUtil.getUserInfo({ config, body: mockBody, jianghuKnex, db, logger, appType });

    if (ctx.userInfo && ctx.userInfo.user && Object.keys(ctx.userInfo.user).length) {
      // 兼容 /upload 开头的配置
      let maxage = jiangHuConfig.uploadFileMaxAge || 0;
      if (ctx.path.includes('/upload/etag')) {
        maxage = null;
      }
      if (ctx.path.startsWith(`/${config.appId}/upload/`)) {
        await send(ctx, decodeURI(ctx.path.replace(`/${config.appId}/upload/`, '')), {
          root: config.baseDir + '/upload',
          maxage,
        });
      } else {
        await send(ctx, decodeURI(ctx.path.replace('/upload/', '')), {
          root: config.baseDir + '/upload',
          maxage,
        });
      }
    } else {
      ctx.redirect(ctx.app.config.loginPage || '/');
    }
    return range(ctx, next);
  };
};