app.js 5.29 KB
/**
 * yohobuy app
 * @author: xuqi<qi.xu@yoho.cn>
 * @date: 2016/4/25
 */
'use strict';

// if (process.env.USE_APM === '1' && process.env.NODE_ENV === 'production') {
//     require('oneapm');
// }

const config = require('./config/common');

global.Promise = require('bluebird');

const yohoLib = require('yoho-node-lib');

const express = require('express');
const path = require('path');
const compression = require('compression');
const bodyParser = require('body-parser');
const cookieParser = require('cookie-parser');
const moment = require('moment');
const uuid = require('uuid');

const pkg = require('./package.json');
const hbsEvent = require('./config/hbsevent');
const _ = require('lodash');

const app = express();

// 向模板注入变量
app.locals.devEnv = app.get('env') === 'development';
app.locals.isProduction = app.get('env') === 'production';
app.locals.version = pkg.version;
app.locals.startTime = moment().format('YYYYMMDDHH');

// 全局注册library
yohoLib.global(config);

const logger = global.yoho.logger;

// zookeeper
if (config.zookeeperServer) {
    require('yoho-zookeeper')(config.zookeeperServer, 'wap', app.locals.wap = {}, global.yoho.cache);
}

// 访问域名层级
app.set('subdomain offset', 3);

app.set('etag', false);

app.enable('trust proxy');

app.disable('x-powered-by');

const safeStringify = require('fast-safe-stringify');

// 循环对象检测
app.set('json replacer', function(key, value) {
    if (!key) {
        let safeStr = safeStringify(value) || '{}';

        if (/\[Circular\]/.test(safeStr)) {
            logger.error('stringify error:', value);
        }

        return JSON.parse(safeStr);
    }

    return value;
});

// 请求限制中间件
if (!app.locals.devEnv) {
    app.use(require('./doraemon/middleware/limiter'));
}

// 指定libray目录
global.utils = path.resolve('./utils');

// 添加请求上下文
app.use(global.yoho.httpCtx());

// 访问域名层级
app.set('subdomain offset', 3);

app.use(global.yoho.hbs({
    extname: '.hbs',
    defaultLayout: 'layout',
    layoutsDir: path.join(__dirname, 'doraemon/views'),
    partialsDir: path.join(__dirname, 'doraemon/views/partial'),
    views: path.join(__dirname, 'doraemon/views'),
    helpers: _.assign(global.yoho.helpers, require('./utils/helpers')),
    cb: hbsEvent.cb
}));


app.use(bodyParser.json());
app.use(bodyParser.urlencoded({
    extended: false
}));
app.use(cookieParser());
app.use(compression());

require('./doraemon/middleware/yoho-session')(app);

// 添加请求上下文
app.use(global.yoho.httpCtx());

app.use((req, res, next) => {
    req.user = {}; // 全局的用户数据
    req.yoho = {}; // req和res绑定yoho对象,用于传递全局数据, 如req.yoho.channel等
    req.app.locals.wap = app.locals.wap; // zookeper对象赋值

    req.isApmReport = app.locals.wap.open && app.locals.wap.open.bughd; // 把错误上报的开关绑定到上下文,node-lib 库要使用

    // 独立的 UDID
    if (!req.cookies.udid) {
        let udid = uuid.v4();

        res.cookie('udid', udid, {
            domain: 'yohobuy.com',
            expires: new Date(Date.now() + 365 * 24 * 60 * 60 * 1000)
        });

        req.cookies.udid = udid;
    }
    next();
});

// dispatcher
try {
    const tdkUrl = require('./doraemon/middleware/redis-url');
    const user = require('./doraemon/middleware/user');
    const reqParamsFilter = require('./doraemon/middleware/req-params-filter');
    const urlRewrite = require('./doraemon/middleware/url-rewrite');
    const subDomain = require('./doraemon/middleware/sub-domain');
    const itemNameHandler = require('./doraemon/middleware/item-name-handler');
    const setYohoData = require('./doraemon/middleware/set-yoho-data');
    const errorHanlder = require('./doraemon/middleware/error-handler');
    const setPageInfo = require('./doraemon/middleware/set-pageinfo');
    const devtools = require('./doraemon/middleware/devtools');
    const layoutTools = require('./doraemon/middleware/layout-tools');
    const seo = require('./doraemon/middleware/seo');
    const pageCache = require('./doraemon/middleware/page-cache');
    const downloadBar = require('./doraemon/middleware/download-bar');
    const routeEncode = require('./doraemon/middleware/route-encode');

    // YOHO 前置中间件
    app.use(tdkUrl());
    app.use(reqParamsFilter());
    app.use(urlRewrite());
    app.use(subDomain());
    app.use(itemNameHandler);
    app.use(setYohoData());
    app.use(user());
    app.use(seo());
    app.use(setPageInfo());

    if (app.locals.devEnv) {
        app.use(devtools());
    }

    app.use(layoutTools());
    app.use(pageCache());
    app.use(routeEncode.md);
    app.use(downloadBar());

    require('./dispatch')(app);
    app.all('*', errorHanlder.notFound()); // 404

    // YOHO 后置中间件
    app.use(errorHanlder.serverError());
} catch (err) {
    logger.error(err);
}

// listener
app.listen(config.port, function() {
    logger.info('yohobuy start');
    hbsEvent.event.on('hbs-complete', () => {
        logger.info('hbs-all-complete');
        if (!process.env.NODE_ENV || process.env.NODE_ENV === 'dev') {
            let {devHost, port} = require('./config/devtools');

            require('request-promise')({
                url: `http://${devHost}:${port}/event?action=reload`
            }).then(() => {}, () =>{});
        }
    });
});