ssr.js 8.94 KB
const fs = require('fs');
const path = require('path');
const url = require('url');
const _ = require('lodash');
const os = require('os');
const md5 = require('yoho-md5');
const pkg = require('../../package.json');
const routes = require('../../config/ssr-routes');
const redis = require('../../utils/redis');
const routeEncode = require('../../utils/route-encode');
const {createBundleRenderer} = require('vue-server-renderer');
const Handlebars = require('handlebars');
const logger = global.yoho.logger;
const config = global.yoho.config;

const isDev = process.env.NODE_ENV === 'development' || !process.env.NODE_ENV;

let renderer;
let serverBundle;
let degradeHtml;

const hbs = fs.readFileSync(path.join(__dirname, '../views/index.hbs'), 'utf-8');

const template = Handlebars.compile(hbs);

if (!isDev) {

  degradeHtml = fs.readFileSync(path.join(__dirname, '../../degrade.html'), 'utf-8');

  serverBundle = require('../../manifest.server.json');
  const clientManifest = require('../../manifest.json');

  renderer = createBundleRenderer(serverBundle, {
    runInNewContext: false,
    clientManifest,
    inject: false
  });
}

const REG_SCRIPT = /src="([^"]+)"/g;

const asyncLoadScripts = (renderScripts) => {
  let match;
  const scripts = [];

  while ((match = REG_SCRIPT.exec(renderScripts))) {
    scripts.push({
      src: match[1],
      index: scripts.length
    });
  }

  return scripts;
};

const getUserProfile = async(req, res, next) => {
  if (req.user && req.user.uid) {
    if (!req.session.headIco) {
      const apiCtx = req.ctx(global.yoho.BaseModel);

      result = await apiCtx.get({
        url: '',
        data: {
          method: 'app.passport.profile',
          uid: {
            toString: () => {
              return req.user.uid;
            },
            sessionKey: req.user.sessionKey,
            appSessionType: req.user.appSessionType
          }
        }
      });

      req.session.headIco = _.get(result, 'data.head_ico', '');
    }

    res.cookie('_head_ico', req.session.headIco);
  }

  next();
}

const versionToNumber = (str, len = 3) => {
  let verNum = 0;
  const ver = str.split('.').map(i => Number(i));

  for (let i = 0; i < len; i++) {
    verNum += (ver[i] || 0) * Math.pow(100, len - i - 1);
  }

  return verNum;
}

const getImmersedStatus = (req) => {
  let status = false;

  if (req.yoho.isYohoApp && req.yoho.isiOS) {
    let appVersion = req.cookies.app_version || req.query.app_version;

    if (appVersion && versionToNumber(appVersion) >= versionToNumber('6.9.2')) {
      status = true;
    }
  }

  return status;
}

const getContext = (req) => {
  return {
    url: req.url,
    title: req.query.share_title || '',
    user: req.user,
    env: {
      isApp: req.yoho.isApp,
      isiOS: req.yoho.isiOS,
      isAndroid: req.yoho.isAndroid,
      isYohoApp: req.yoho.isYohoApp,
      clientIp: req.yoho.clientIp,
    },
    ua: req.get('user-agent'),
    hostname: os.hostname(),
    route: `[${req.method}]${_.get(req, 'route.path', '')}`, // 请求路由
    udid: _.get(req, 'cookies.udid', 'yoho'),
    path: `[${req.method}]${routeEncode.getRouter(req)}`
  };
};

const handlerError = (err = {}, req, res, next) => {
  if (err.code === 404) {
    return res.redirect('/grass/error/404');
  } else if (err.code === 500) {
    return res.redirect('/grass/error/500');
  }
  return next(err);
};

const getCacheKey = (req, cacheKey = '') => {
  const isYohoProtocol = _.get(req.app.locals.wap, `webapp.${config.appName}-yoho-protocol`, false);
  const urlObj = url.parse(req.url);
  const isIos = req.yoho.isiOS;
  const yohoProtocol = (req.get('User-Agent').indexOf('yoho-protocol') >= 0 && isYohoProtocol) ? 'yoho-protocol' : '';

  return md5(cacheKey
    .replace('$url', urlObj.pathname)
    .replace('$params', urlObj.query || '')
    .replace('$yoho-protocol', yohoProtocol) + (isIos ? 'ios' : 'android'));
};

const render = (route) => {
  return async(req, res, next) => {
    try {
      res.setHeader('X-YOHO-Version', pkg.version);
      const isDegrade = _.get(req.app.locals.wap, `webapp.${config.appName}-degrade`, false);
      const isYohoProtocol = _.get(req.app.locals.wap, `webapp.${config.appName}-yoho-protocol`, false);

      if (isDegrade) {
        return res.send(degradeHtml);
      }
      const ck = route.cacheKey ? getCacheKey(req, route.cacheKey) : void 0;

      if (config.useCache && route.cache && ck) {
        const html = await redis.getAsync(ck);

        res.set({
          'Cache-Control': 'max-age=' + route.cacheTime || 60
        });
        if (html) {
          logger.debug(`cached ${req.url}`);
          res.setHeader('X-YOHO-Cached', 'HIT');
          return res.send(html);
        }
        res.setHeader('X-YOHO-Cached', 'MISS');
      } else {
        res.set({
          'Cache-Control': 'no-cache',
          Pragma: 'no-cache',
          Expires: (new Date(1900, 0, 1, 0, 0, 0, 0)).toUTCString()
        });
      }
      let context = getContext(req);

      renderer.renderToString(context, (err, html) => {
        if (err) {
          return handlerError(err, req, res, next);
        }
        let styles = context.renderStyles();
        let scripts = context.renderScripts();
        let resources = context.renderResourceHints();
        const states = context.renderState();
        let asyncScripts;
        let zk = {};

        if (process.env.NODE_ENV === 'production') {
          zk.webperf = _.get(req.app.locals.wap, 'open.webperf', false);
          zk.asyncJs = _.get(req.app.locals.wap, 'webapp.ios-async-js', true);
        }

        if (req.yoho.isiOS && zk.asyncJs) {
          asyncScripts = asyncLoadScripts(scripts);
        }
        if (req.get('User-Agent').indexOf('yoho-protocol') >= 0 &&
          route.cacheKey &&
          route.cacheKey.indexOf('$yoho-protocol') >= 0 &&
          isYohoProtocol) {
          styles = styles.replace(/"\/\//g, '"yoho-protocol://');
          resources = resources.replace(/<link rel="preload" href="[^"]+" as="style">/g, '');
        }

        const result = template({
          html,
          styles,
          scripts,
          asyncScripts,
          resources,
          states,
          zk,
          routeHash: routeEncode.getRouter(req),
          needImmersed: getImmersedStatus(req)
        });

        if (config.useCache && route.cache && ck) {
          redis.setex(ck, route.cacheTime || 60, result);
        }
        return res.send(result);
      });
    } catch (error) {
      return next(error);
    }
  };
};
const devRender = (route) => {
  return async(req, res, next) => {
    try {
      res.setHeader('X-YOHO-Version', pkg.version);
      const ck = route.cacheKey ? getCacheKey(req, route.cacheKey) : void 0;

      const isDegrade = _.get(req.app.locals.wap, `webapp.${config.appName}-degrade`, false);

      if (isDegrade) {
        return require('request-promise')({
          url: 'http://m.yohobuy.com:6005/degrade.html'
        }).pipe(res);
      }
      if (config.useCache && route.cache && ck) {
        const html = await redis.getAsync(ck);

        res.set({
          'Cache-Control': 'max-age=' + route.cacheTime || 60
        });
        if (html) {
          logger.debug(`cached ${req.url}`);
          res.setHeader('X-YOHO-Cached', 'HIT');
          return res.send(html);
        }
        res.setHeader('X-YOHO-Cached', 'MISS');
      } else {
        res.set({
          'Cache-Control': 'no-cache',
          Pragma: 'no-cache',
          Expires: (new Date(1900, 0, 1, 0, 0, 0, 0)).toUTCString()
        });
      }
      let context = getContext(req);

      process.send({action: 'ssr_request', context});
      let event = msg => {
        process.removeListener('message', event);
        if (msg.action === 'ssr_request') {
          if (msg.err) {
            let err = msg.err;

            try {
              err = JSON.parse(msg.err);
            } catch (error) {} // eslint-disable-line
            return handlerError(err, req, res, next);
          }
          let {styles, scripts, resources, states, html} = msg;

          if (req.get('User-Agent').indexOf('yoho-protocol') >= 0 &&
            route.cacheKey &&
            route.cacheKey.indexOf('$yoho-protocol') >= 0) {
            styles = styles.replace(/"\/\//g, '"yoho-protocol://');
          }
          const result = template({
            html,
            styles,
            scripts,
            resources,
            states,
            routeHash: routeEncode.getRouter(req),
            needImmersed: getImmersedStatus(req)
          });

          if (config.useCache && route.cache && ck) {
            redis.setex(ck, route.cacheTime || 60, result);
          }
          return res.end(result);
        }
      };

      process.on('message', event);
    } catch (error) {
      return next(error);
    }
  };
};

const router = require('express').Router(); //eslint-disable-line

_.each(routes, r => {
  if (!r.disable) {
    router.get(r.route, getUserProfile, isDev ? devRender(r) : render(r));
  }
});

exports.ssrRender = isDev ? devRender({}) : render({});

exports.routers = router;