thinkjs/lib/Lib/Core/Think.js

//自动加载进行识别的路径
var autoloadPaths = {};
/**
 * [exports description]
 * @type {Object}
 */
module.exports = {
    /**
     * [start description]
     * @return {[type]} [description]
     */
    start: function(){
        "use strict";
        //debug模式下不捕获异常
        // if (!APP_DEBUG) {
        //     this.register_exception();
        // };
        //加载文件
        this.loadFiles();
        //合并自动加载的路径
        this.mergeAutoloadPath();
        //thinkRequire的autoload
        registerAutoload(this.autoload);
        //debug模式
        if (APP_DEBUG) {
            this.debug();
        }
        //记录进程的id
        this.log_process_pid();

        thinkRequire('App').run();
    },
    /**
     * 注册异常处理
     * @return {[type]} [description]
     */
    register_exception: function(){
        "use strict";
        process.on('uncaughtException', function(err) {
            console.log(isError(err) ? err.stack : err);
        });
    },
    /**
     * 加载项目下对应的文件
     * @return {[type]} [description]
     */
    loadFiles: function(){
        "use strict";
        //加载系统默认配置
        C(require(THINK_PATH + '/Conf/config.js'));
        //加载用户配置
        var file = CONF_PATH + '/config.js';
        if (isFile(file)) {
            C(require(file));
        }
        //加载模式的配置文件
        if (APP_MODE) {
            var modeFiles = [
                THINK_PATH + "/Conf/mode.js",
                CONF_PATH + "/mode.js"
            ];
            modeFiles.forEach(function(file){
                if (isFile(file)) {
                    var conf = require(file);
                    if (conf[APP_MODE]) {
                        C(conf[APP_MODE]);
                    }
                }
            });
        }
        //自定义路由
        if (C('url_route_on') && isFile(CONF_PATH + "/route.js")) {
            C("url_route_rules", require(CONF_PATH + "/route.js"));
        }
        //加载行为配置
        if (C('app_tag_on')) {
            //加载系统行为配置
            C('sys_tag', require(THINK_PATH + "/Conf/tag.js"));
            //加载用户的行为配置
            var tagFile = CONF_PATH + "/tag.js";
            if (isFile(tagFile)) {
                C('tag', require(tagFile));
            }
        }
        //common文件
        if (isFile(COMMON_PATH + "/common.js")) {
            require(COMMON_PATH + "/common.js");
        }
        //别名文件
        if (isFile(COMMON_PATH + "/alias.js")) {
            aliasImport(require(COMMON_PATH + "/alias.js"));
        }
        this.loadExtConfig();
        this.loadExtFiles();
    },
    //加载自定义外部文件
    loadExtFiles: function(){
        "use strict";
        var files = C('load_ext_file');
        if (files) {
            if (isString(files)) {
                files = files.split(',');
            }
            files.forEach(function(file){
                file = COMMON_PATH + "/" + file + ".js";
                if (isFile(file)) {
                    require(file);
                }
            });
        }
    },
    //加载额外的配置
    loadExtConfig: function(){
        "use strict";
        var files = C('load_ext_config');
        if (files) {
            if (isString(files)) {
                files = files.split(",");
            }
            files.forEach(function(file){
                file = CONF_PATH + "/" + file + ".js";
                if (isFile(file)) {
                    C(require(file));
                }
            });
        }
    },
    //加载debug模式配置文件
    loadDebugFiles: function(){
        "use strict";
        //加载debug模式下的配置
        C(require(THINK_PATH + "/Conf/debug.js"));
        //debug下自定义状态的配置
        var status = C('app_status');
        if (status) {
            if (isFile(CONF_PATH + "/" + status + '.js')) {
                C(require(CONF_PATH + "/" + status + ".js"));
            }
        }else{
            if (isFile(CONF_PATH + '/debug.js')) {
                C(require(CONF_PATH + '/debug.js'));
            }
        }
        if (APP_MODE) {
            var modeFiles = [
                THINK_PATH + "/Conf/mode.js",
                CONF_PATH + "/mode.js"
            ];
            modeFiles.forEach(function(file){
                if (isFile(file)) {
                    var conf = require(file);
                    var key = APP_MODE + "_debug";
                    if (conf[key]) {
                        C(conf[key]);
                    }
                }
            });
        }
    },
    /**
     * debug模式下一些特殊处理
     * @return {[type]} [description]
     */
    debug: function(){
        "use strict";
        this.loadDebugFiles();
        //清除require的缓存
        if (C('clear_require_cache')) {
            //这些文件不清除缓存
            var retainFiles = C('debug_retain_files');
            var self = this;
            setInterval(function(){
                //清除已经绑定的自定义事件
                E(null);
                var fn = function(item){
                    if (file.indexOf(item) > -1) {
                        return true;
                    }
                };
                for(var file in require.cache){
                    var flag = retainFiles.some(fn);
                    if (!flag) {
                        delete require.cache[file];
                    }
                }
                self.loadFiles();
                self.loadDebugFiles();
            }, 100);
        }
    },
    /**
     * 记录当前进程的id
     * 记录在Runtime/Data/app.pid文件里
     * @return {[type]} [description]
     */
    log_process_pid: function(){
        "use strict";
        var isMaster = require('cluster').isMaster;
        if (C('log_process_pid') && isMaster) {
            mkdir(DATA_PATH);
            var pidFile = DATA_PATH + "/app.pid";
            var fs = require("fs");
            fs.writeFileSync(pidFile, process.pid);
            chmod(pidFile);
            //进程退出时删除该文件
            process.on('SIGTERM', function () {
                if (fs.existsSync(pidFile)) {
                    fs.unlinkSync(pidFile);
                }
                process.exit(0);
            });
        }
    },
    /**
     * 合并autoload的path
     * @return {[type]} [description]
     */
    mergeAutoloadPath: function(){
        "use strict";
        var file = "__CLASS__.js";
        var sysAutoloadPath = {
            "Behavior": [
                LIB_PATH + "/Behavior/" + file,
                THINK_LIB_PATH + "/Behavior/" + file
            ],
            "Model": [
                LIB_PATH + "/Model/" + file,
                THINK_EXTEND_PATH + "/Model/" + file
            ],
            "Controller": [
                LIB_PATH + "/Controller/" + file,
                THINK_EXTEND_PATH + "/Controller/" + file
            ],
            "Cache": [
                LIB_PATH + "/Driver/Cache/" + file,
                THINK_LIB_PATH + "/Driver/Cache/" + file
            ],
            "Db": [
                LIB_PATH + "/Driver/Db/" + file,
                THINK_LIB_PATH + "/Driver/Db/" + file
            ],
            "Template": [
                LIB_PATH + "/Driver/Template/" + file,
                THINK_LIB_PATH + "/Driver/Template/" + file
            ],
            "Socket": [
                LIB_PATH + "/Driver/Socket/" + file,
                THINK_LIB_PATH + "/Driver/Socket/" + file
            ],
            "Session": [
                LIB_PATH + "/Driver/Session/" + file,
                THINK_LIB_PATH + "/Driver/Session/" + file
            ]
        };
        var autoloadPath = C('autoload_path');
        for(var type in autoloadPath){
            var paths = autoloadPath[type];
            if (!isArray(paths)) {
                paths = [paths];
            }
            var override = false;
            if (isBoolean(paths[0])) {
                override = paths[0];
                paths.shift();
            }
            if (override) {
                sysAutoloadPath[type] = paths;  
            }else{
                paths.push.apply(paths, sysAutoloadPath[type]);
                sysAutoloadPath[type] = paths;
            }
        }
        autoloadPaths = sysAutoloadPath;
    },
    //thinkRequire的自动加载
    autoload: function(cls){
        "use strict";
        var filepath = '';
        var fn = function(item){
            item = item.replace(/__CLASS__/g, cls);
            if (isFile(item)) {
                filepath = item;
                return true;
            }
        };
        for(var name in autoloadPaths){
            var length = name.length;
            if (cls.substr(0 - length) === name) {
                var list = autoloadPaths[name];
                list.some(fn);
                if (filepath) {
                    return filepath;
                }
            }
        }
    }
};