userHome default rule

This commit is contained in:
想当当 2014-09-06 09:50:49 +08:00
parent 0343eb7aeb
commit 946c4ee755
3 changed files with 14 additions and 6 deletions

View File

@ -5,9 +5,10 @@ var exec = require('child_process').exec,
os = require("os"), os = require("os"),
color = require('colorful'), color = require('colorful'),
readline = require('readline'), readline = require('readline'),
util = require('./util'),
asyncTask = require("async-task-mgr"); asyncTask = require("async-task-mgr");
var certDir = path.join(getUserHome(),"/.anyproxy_certs/"), var certDir = path.join(util.getUserHome(),"/.anyproxy_certs/"),
cmdDir = path.join(__dirname,"..","./cert/"), cmdDir = path.join(__dirname,"..","./cert/"),
asyncTaskMgr = new asyncTask(); asyncTaskMgr = new asyncTask();
@ -37,10 +38,6 @@ function getCertificate(hostname,cb){
} }
} }
function getUserHome() {
return process.env.HOME || process.env.HOMEPATH || process.env.USERPROFILE;
}
function createCert(hostname,callback){ function createCert(hostname,callback){
console.log(hostname); console.log(hostname);
checkRootCA(); checkRootCA();

View File

@ -18,3 +18,8 @@ module.exports.merge = function(baseObj, extendObj){
return baseObj; return baseObj;
} }
module.exports.getUserHome = function(){
return process.env.HOME || process.env.HOMEPATH || process.env.USERPROFILE;
}

View File

@ -19,6 +19,7 @@ var http = require('http'),
util = require("./lib/util"), util = require("./lib/util"),
entities = require("entities"), entities = require("entities"),
express = require("express"), express = require("express"),
path = require("path"),
WebSocketServer= require('ws').Server; WebSocketServer= require('ws').Server;
GLOBAL.recorder = new Recorder(); GLOBAL.recorder = new Recorder();
@ -31,6 +32,11 @@ var T_TYPE_HTTP = 0,
DEFAULT_HOST = "localhost", DEFAULT_HOST = "localhost",
DEFAULT_TYPE = T_TYPE_HTTP; DEFAULT_TYPE = T_TYPE_HTTP;
var default_rule = require('./lib/rule_default');
if(fs.existsSync(path.join(util.getUserHome(),"/.anyproxy/rule_default.js"))){
default_rule = require(path.join(util.getUserHome(),"/.anyproxy/rule_default"));
}
//option //option
//option.type : 'http'(default) or 'https' //option.type : 'http'(default) or 'https'
//option.port : 8001(default) //option.port : 8001(default)
@ -43,7 +49,7 @@ function proxyServer(option){
proxyType = /https/i.test(option.type || DEFAULT_TYPE) ? T_TYPE_HTTPS : T_TYPE_HTTP , proxyType = /https/i.test(option.type || DEFAULT_TYPE) ? T_TYPE_HTTPS : T_TYPE_HTTP ,
proxyPort = option.port || DEFAULT_PORT, proxyPort = option.port || DEFAULT_PORT,
proxyHost = option.hostname || DEFAULT_HOST, proxyHost = option.hostname || DEFAULT_HOST,
proxyRules = option.rule || require('./lib/rule_default'); proxyRules = option.rule || default_rule;
requestHandler.setRules(proxyRules); //TODO : optimize calling for set rule requestHandler.setRules(proxyRules); //TODO : optimize calling for set rule
self.httpProxyServer = null; self.httpProxyServer = null;