thinkjs/lib/Lib/Core/Session.class.js
2013-11-04 18:31:40 +08:00

68 lines
1.9 KiB
JavaScript

var crypto = require('crypto');
var uid = function(length){
var ratio = Math.log(64) / Math.log(256);
var numbytes = Math.ceil(length * ratio);
var str = crypto.randomBytes(numbytes).toString('base64').slice(0, length);
return str.replace(/\+/g, '_').replace(/\//g, '-');
};
var cookie_sign = function(val, secret){
secret = crypto.createHmac('sha256', secret).update(val).digest('base64');
secret = secret.replace(/\=+$/, '');
return val + '.' + secret;
};
var cookie_unsign = function(val, secret){
var str = val.slice(0, val.lastIndexOf('.'));
return cookie_sign(str, secret) === val ? str : false;
}
module.exports = Class(function(){
return {
key: "",
cookie: null,
init: function(http){
this.http = http;
this.cookie = this.http.req.session_cookie;
},
get: function(name){
var deferrd = when.defer();
process.nextTick(function(){
deferrd.resolve("");
})
return deferrd.promise;
},
set: function(name, value){
return true;
}
}
});
//解析cookie
module.exports.start = function(http){
if (http.req.session_cookie) {
return http.req.session_cookie;
};
var name = C('session_id');
var secret = C('session_cookie_sign');
var cookie = http.req.cookie[name];
if (cookie && secret) {
cookie = cookie_unsign(cookie, secret);
};
if (!cookie) {
cookie = uid(32);
if (secret) {
cookie = cookie_sign(cookie, secret);
};
http.res.setCookie(name, cookie, C('session_options'));
}
http.req.session_cookie = cookie;
var name = C('session_type') + "Session";
var cls = think_require(name)(http);
http.req.session = cls;
return cookie;
};
module.exports.cookie_sign = cookie_sign;
module.exports.cookie_unsign = cookie_unsign;