/**
 * Created by JetBrains WebStorm.
 * User: chishaxie
 * Date: 12-1-5
 * Time: 下午4:59
 * To change this template use File | Settings | File Templates.
 */
var conf = {
    Root : 'html', //文件的根路径
    IndexEnable : true, //开启目录功能？
    IndexFile : 'index.html', //目录欢迎文件
    DynamicExt : /^\.njs$/ig, //动态页面后缀（需要.）
    ServerName : 'httpNgin/nodeJS', //服务器名字
    FileCache : { //文件（内存）缓存
        MaxSingleSize : 1024*1024, //单个文件最大尺寸
        MaxTotalSize : 30*1024*1024 //整个文件Cache最大尺寸
    },
    Expires : { //浏览器缓存
        FileMatch : /gif|jpg|png|js|css|ico/ig, //匹配的文件格式
        MaxAge : 3600*24*365 //最大缓存时间
    },
    Compress : { //编码压缩
        FileMatch : /css|js|html|txt/ig //匹配的文件格式
    },
    MIME : {
        'css': 'text/css',
        'gif': 'image/gif',
        'html': 'text/html',
        'ico': 'image/x-icon',
        'jpg': 'image/jpeg',
        'js': 'text/javascript',
        'png': 'image/png',
        'rar' : 'application/x-rar-compressed',
        'txt': 'text/plain',
        'json': 'text/plain',
        'jar': 'application/java-archive'
    }
};

/* 计算长度（中文算2个长度） */
String.prototype.len = function(){
    return this.replace(/[^\x00-\xff]/g,'**').length;
};

/* 填充（长度，字符串，填充到左边？） */
String.prototype.pad = function(len,char,isLeft){
    var bArr = len.toString(2).split('');
    var ret = '';
    var step = char;
    for(var i=bArr.length-1;i>=0;i--){
        if(bArr[i] == '1') ret += step;
        step += step;
    }
    if(!isLeft) return this + ret;
    else return ret + this;
};

/* 获取Http时间（2012-12-21 19:30形式） */
Date.prototype.getHttpTime = function(){
    return this.getFullYear() + '-' + (this.getMonth()+1) + '-' + this.getDate() +  ' ' + this.getHours() + ':' + this.getMinutes() ;
};

/* 缓存类（maxSize 最大字节数） */
function Cache(maxSize){
    this.maxSize = maxSize; //最大尺寸
    this.curSize = 0; //当前尺寸
    this._bufs = {}; //缓存Map
    this._accessCount = 0; //访问计数器
    this._lastClearCount = 0; //上次清理的计数器
}

Cache.prototype.put = function(key,buf){
    buf.access = this._accessCount++;
    var obuf = this._bufs[key];
    if(obuf) this.curSize -= obuf.length;
    this._bufs[key] = buf;
    this.curSize += buf.length;
    while(this.curSize > this.maxSize){
        this._clear();
    }
};

Cache.prototype.get = function (key) {
    var buf = this._bufs[key];
    if (buf) buf.access = this._accessCount++;
    return buf;
};

Cache.prototype.del = function (key) {
    var buf = this._bufs[key];
    if (buf) {
        this.curSize -= buf.length;
        delete this._bufs[key];
    }
};

Cache.prototype._clear = function () {
    var clearCount = (this._lastClearCount + this._accessCount) / 2;
    for (var e in this._bufs) {
        var buf = this._bufs[e];
        if (buf.access <= clearCount) {
            this.curSize -= buf.length;
            delete this._bufs[e];
        }
    }
    this._lastClearCount = clearCount;
};

/* HTTP缓存类（mtime不可更改） */
function HttpCache(mtime,obuf,gbuf,dbuf){
    this.mtime = mtime; //修改时间
    this.obuf = obuf; //原始数据
    this.gbuf = gbuf; //gzip数据
    this.dbuf = dbuf; //deflate数据
    this.length = (obuf?obuf.length:0) + (dbuf?dbuf.length:0) + (gbuf?gbuf.length:0);
}

/*
HttpCache.prototype.setObuf = function(obuf){
    this.length += obuf.length - (this.obuf?this.obuf.length:0);
    this.obuf = obuf;
};
*/

HttpCache.prototype.setGbuf = function(gbuf){
    this.length += gbuf.length - (this.gbuf?this.gbuf.length:0);
    this.gbuf = gbuf;
};

HttpCache.prototype.setDbuf = function(dbuf){
    this.length += dbuf.length - (this.dbuf?this.dbuf.length:0);
    this.dbuf = dbuf;
};

var http = require('http'),
    url = require('url'),
    path = require('path'),
    fs = require('fs'),
    zlib = require('zlib');

/* 路径状态查询（自带文件名封装） */
fs.statWithFN = function(dirpath,filename,callback){
    fs.stat(path.join(dirpath,filename),function(err,stats){
        callback(err,stats,filename);
    });
};

var Com = {
    fileCache : new Cache(conf.FileCache.MaxTotalSize),
    ifModifiedSince : 'If-Modified-Since'.toLowerCase(),
    parseRange : function(str,size){ //范围解析（HTTP Range字段）
        if(str.indexOf(",") != -1) //不支持多段请求
            return;
        var strs = str.split("=");
        str = strs[1] || '';
        var range = str.split("-"),
            start = parseInt(range[0], 10),
            end = parseInt(range[1], 10);
        // Case: -100
        if(isNaN(start)) {
            start = size - end;
            end = size - 1;
        }
        // Case: 100-
        else if(isNaN(end)) {
            end = size - 1;
        }
        // Invalid
        if(isNaN(start) || isNaN(end) || start > end || end > size)
            return;
        return {start: start, end: end};
    },
    error : function(response,id,err){ //返回错误
        response.writeHeader(id, {'Content-Type': 'text/html'});
        var txt;
        switch(id){
            case 404:
                txt = '<h3>404: Not Found</h3>';
                break;
            case 403:
                txt = '<h3>403: Forbidden</h3>';
                break;
            case 416:
                txt = '<h3>416: Requested Range not satisfiable</h3>';
                break;
            case 500:
                txt = '<h3>500: Internal Server Error</h3>';
                break;
        }
        if(err) txt += err;
        response.end(txt);
    },
    cache : function(response,lastModified,ext){ //写客户端Cache
        response.setHeader('Last-Modified', lastModified);
        if(ext && ext.search(conf.Expires.FileMatch)!=-1){
            var expires = new Date();
            expires.setTime(expires.getTime() + conf.Expires.MaxAge * 1000);
            response.setHeader('Expires', expires.toUTCString());
            response.setHeader('Cache-Control', 'max-age=' + conf.Expires.MaxAge);
        }
    },
    compressHandle : function(request,response,raw,ext,contentType,statusCode){ //流压缩处理
        var stream = raw;
        var acceptEncoding = request.headers['accept-encoding'] || '';
        var matched = ext.match(conf.Compress.match);
        if (matched && acceptEncoding.match(/\bgzip\b/)) {
            response.setHeader('Content-Encoding', 'gzip');
            stream = raw.pipe(zlib.createGzip());
        } else if (matched && acceptEncoding.match(/\bdeflate\b/)) {
            response.setHeader('Content-Encoding', 'deflate');
            stream = raw.pipe(zlib.createDeflate());
        }
        response.setHeader('Content-Type', contentType);
        response.writeHead(statusCode);
        stream.pipe(response);
    },
    flush : function(request,response,cache,ext,contentType){ //Cache输出
        var acceptEncoding = request.headers['accept-encoding'] || "";
        var matched = ext.match(conf.Compress.FileMatch);
        if (matched && acceptEncoding.match(/\bgzip\b/)) {
            if(cache.gbuf){
                response.writeHead(200, {'Content-Encoding': 'gzip','Content-Type': contentType});
                response.end(cache.gbuf);
            }
            else{
                zlib.gzip(cache.obuf,function(err,buf){
                    if(err) Com.error(response,500,'<h4>Error : ' + err + '</h4>');
                    else{
                        response.writeHead(200, {'Content-Encoding': 'gzip','Content-Type': contentType});
                        response.end(buf);
                        cache.setGbuf(buf);
                    }
                });
            }
        } else if (matched && acceptEncoding.match(/\bdeflate\b/)) {
            if(cache.dbuf){
                response.writeHead(200, {'Content-Encoding': 'deflate','Content-Type': contentType});
                response.end(cache.dbuf);
            }
            else{
                zlib.deflate(cache.obuf,function(err,buf){
                    if(err) Com.error(response,500,'<h4>Error : ' + err + '</h4>');
                    else{
                        response.writeHead(200, {'Content-Encoding': 'deflate','Content-Type': contentType});
                        response.end(buf);
                        cache.setDbuf(buf);
                    }
                });
            }
        } else {
            response.writeHead(200,{'Content-Type': contentType});
            response.end(cache.obuf);
        }
    },
    pathHandle : function(request,response,realpath,httppath,dirmtime){
        fs.stat(realpath,function(err,stats){
            if(err){
                if(dirmtime){
                    var dirPath = path.dirname(realpath);
                    fs.readdir(dirPath,function(err,files){
                        if(err) Com.error(response,404);
                        else{
                            var httpP = httppath.replace(/\\/g,'/');
                            var txt = '<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"><html xmlns="http://www.w3.org/1999/xhtml"><head><meta http-equiv="Content-Type" content="text/html; charset=utf-8" /><title>Index of '+httpP+'</title></head><body><h1>Index of '+httpP+'</h1><hr ><pre>';
                            if(httpP!='/')
                                txt += '<a href="'+path.dirname(httppath).replace(/\\/g,'/')+'">../</a>\n';
                            if(files.length == 0){ //空目录
                                txt += '</pre><hr ><h3>Powered by '+conf.ServerName+'</h3></body></html>';
                                var cache = new HttpCache(dirmtime.getTime(),new Buffer(txt));
                                Com.cache(response,dirmtime.toUTCString(),'html');
                                Com.flush(request,response,cache,'html','text/html');
                                Com.fileCache.put(dirPath+'\\',cache);
                                return;
                            }
                            var fileI = 0;
                            var fileInfos = [];
                            var fsCallback = function(err,stats,filename){
                                if(!err) fileInfos.push([stats,filename]);
                                fileI++;
                                if(fileI == files.length){
                                    fileInfos.sort(function(a,b){
                                        if(a[0].isDirectory() == b[0].isDirectory())
                                            return a[1].localeCompare(b[1]);
                                        return b[0].isDirectory() - a[0].isDirectory();
                                    });
                                    for(var i=0;i<fileInfos.length;i++){
                                        if(fileInfos[i][0].isDirectory()) fileInfos[i][1] += '/';
                                        var sf = fileInfos[i][1];
                                        var st = fileInfos[i][0].mtime.getHttpTime();
                                        var ss = fileInfos[i][0].isDirectory()?'-':fileInfos[i][0].size.toString();
                                        txt += '<a href="'+path.join(httppath,sf).replace(/\\/g,'/')+'">'+sf+'</a>'+''.pad(50-sf.len(),' ')+st+''.pad(35-st.len()-ss.length,' ')+ss+'\n'
                                    }
                                    txt += '</pre><hr ><h3>Powered by '+conf.ServerName+'</h3></body></html>';
                                    var cache = new HttpCache(dirmtime.getTime(),new Buffer(txt));
                                    Com.cache(response,dirmtime.toUTCString(),'html');
                                    Com.flush(request,response,cache,'html','text/html');
                                    Com.fileCache.put(dirPath+'\\',cache);
                                }
                            };
                            for(var i=0;i<files.length;i++)
                                fs.statWithFN(dirPath,files[i],fsCallback);
                        }
                    });
                }
                else{
                    Com.fileCache.del(realpath);
                    Com.error(response,404);
                }
            }
            else{
                var lastModified = stats.mtime.toUTCString();
                //304 客户端有Cache，且木有改动
                if(request.headers[Com.ifModifiedSince] && lastModified == request.headers[Com.ifModifiedSince]){
                    response.writeHead(304);
                    response.end();
                    return;
                }
                var ext = path.extname(realpath);
                ext = ext?ext.slice(1):'unknown';
                ext = stats.isDirectory()?'html':ext;
                var contentType = conf.MIME[ext];
                var cache = Com.fileCache.get(realpath);
                //服务端有Cache，且木有改动
                if(cache && cache.mtime == stats.mtime.getTime()){
                    Com.cache(response,lastModified,ext);
                    Com.flush(request,response,cache,ext,contentType);
                    Com.fileCache.put(realpath,cache);
                    return;
                }
                if(stats.isDirectory()){
                    realpath = path.join(realpath,conf.IndexFile);
                    Com.pathHandle(request,response,realpath,httppath,conf.IndexEnable?stats.mtime:0);
                }
                else{
                    //不合法的MIME
                    if(!contentType){
                        Com.error(response,403);
                        return;
                    }
                    Com.cache(response,lastModified,ext);
                    //文件太大，服务端不Cache
                    if(stats.size > conf.FileCache.MaxSingleSize){
                        if(request.headers['range']){
                            var range = Com.parseRange(request.headers['range'], stats.size);
                            if(range){
                                response.setHeader('Content-Range', 'bytes ' + range.start + '-' + range.end + '/' + stats.size);
                                response.setHeader('Content-Length', (range.end - range.start + 1));
                                var raw = fs.createReadStream(realpath, {'start': range.start, 'end': range.end});
                                Com.compressHandle(request,response,raw,ext,contentType,206);
                            }
                            else
                                Com.error(response,416);
                        }
                        else{
                            var raw = fs.createReadStream(realpath);
                            Com.compressHandle(request,response,raw,ext,contentType,200);
                        }
                    }
                    else{
                        fs.readFile(realpath,function(err,data){
                            if(err) Com.error(response,500,'<h4>Error : ' + err + '</h4>');
                            else{
                                var buf = new HttpCache(stats.mtime.getTime(),data);
                                Com.flush(request,response,buf,ext,contentType);
                                Com.fileCache.put(realpath,buf);
                            }
                        });
                    }
                }
            }
        });
    }
};

/* 对外的接口 */
exports.createServer = function(port,dynamicCallBack){
    if(!port) port = 80;
    http.createServer(function(req,res){
        if(conf.ServerName) res.setHeader('Server',conf.ServerName);
        var httppath = '/';
        try{
            httppath = path.normalize(decodeURI(url.parse(req.url).pathname.replace(/\.\./g, '')));
        }
        catch(err){
            httppath = path.normalize(url.parse(req.url).pathname.replace(/\.\./g, ''));
        }
        var realpath = path.join(conf.Root, httppath );
        var ext = path.extname(realpath);
        if(ext.search(conf.DynamicExt) != -1){
            if(typeof dynamicCallBack === 'function')
                dynamicCallBack(req,res,httppath,realpath);
            else
                Com.error(res,500,"<h4>Error : Can't find the dynamic page callback function!</h4>");
        }
        else
            Com.pathHandle(req,res,realpath,httppath);
    }).listen(port);
};