server.js

/**
 * Created by Henrie on 12/1/2018.
 */

const config = require('./config.js'),
    unirest = require('unirest'),
    CookieJar = unirest.jar(),
    Pac = require('node-pac'),
    express = require('express'),
    errorHandler = require('express-error-handler'),
    handler = errorHandler({
        handlers: {
            '404': (err, req, res) => {
                trans(req.body, res, req.url);
            }
        }
    }),
    app = express(),
    fs = require('fs'),
    glob = require('glob'),
    bodyParser = require('body-parser'),
    multer = require('multer'),
    accessControlAllowHeaders = config.accessControlAllowHeaders ? ',' + config.accessControlAllowHeaders : '',
    storage = multer.diskStorage({
        destination (req, file, cb) {
            // 文件目錄
            cb(null, './upload')
        },
        filename (req, file, cb) {
            // 文件名稱
            let fileFormat = (file.originalname).split(".");
            cb(null, file.fieldname + '-' + Date.now() + "." + fileFormat[fileFormat.length - 1]);
        }
    }),
    uploader = multer({
        storage: storage
    }),
    log4js = require('log4js');

log4js.configure({
    appenders: {
        console: { type: 'console' }
    },
    categories: {
        http: { appenders: ['console'], level: 'info' },
        params: { appenders: ['console'], level: 'mark' },
        warner: { appenders: ['console'], level: 'warn' },
        default: { appenders: ['console'], level: 'info' }
    }
});

let connectLogger = log4js.getLogger('http'),
    logger = log4js.getLogger('params'),
    warner = log4js.getLogger('warner');
app.use(log4js.connectLogger(connectLogger, {
    level: 'auto',
    format (req, res, str) {
        // method, 請求接口, 狀態碼, 響應時間
        let arr = [req.method, req.url.split('?')[0], res.statusCode, res.responseTime + 'ms'];
        return arr.join(' ') + '\n';
    }
}));

// app.use(bodyParser.json());
app.use(bodyParser.urlencoded({ extended: true }));
app.all('*', (req, res, next) => {
    res.header("Access-Control-Allow-Origin", req.headers.origin); // 跨域
    res.header("Access-Control-Allow-Methods", "GET,HEAD,OPTIONS,POST,PUT"); // 跨域
    config.withCredentials && res.header("Access-Control-Allow-Credentials", "true"); // 跨域
    res.header("Access-Control-Allow-Headers", "Origin, X-Requested-With, Content-Type, Accept" + accessControlAllowHeaders);
    let pathname = req._parsedUrl.pathname;
    if ('OPTIONS' === req.method) {
        res.send(200);
    } else if (pathname !== config.uploadUrl && pathname !== config.downloadFileUrl && pathname !== config.downloadBase64Url) {
        let path = 'json/' + req.method.toLowerCase() + pathname + '.json';
        fsReader(req, res, path);
    } else {
        next();
    }
});

let server,
    fsReader = (req, res, filepath) => {
        fs.readFile(__dirname + '/' + filepath, 'utf8', (err, data) => {
            if (err) {
                trans(req.body, res, req.url);
            } else {
                // 出参在這裡
                res.send(data);
                res.end();
            }
        });
    },
    passProxy = (data, mockRes, Request) => {
        if (config.proxyUrl) {
            Request.proxy(config.proxyUrl);
            getRequest(data, mockRes, Request);
        } else {
            let pac = new Pac(config.pac);
            pac.FindWhistleProxyForURL(config.transHost, (err, res) => {
                let proxy = 'http://' + res.split('//')[1];
                Request.proxy(proxy);
                getRequest(data, mockRes, Request);
            });
        }
    },
    getRequest = (data, mockRes, Request) => {
        Request.send(data).end((res) => {
            let dis = res.headers['content-disposition'];
            if (dis && dis.search('attachment') > -1) {
                logger.mark(res.headers['content-disposition']);
            } else {
                logger.mark(res.body);
            }
            mockRes.set(res.headers);
            setCookies(res);
            mockRes.send(res.body);
            mockRes.end();
        });
    },
    setCookies = (response) => {
        let cookies = response.headers['set-cookie'],
            cookie;
        if (!cookies) return;
        for (let i = 0; i < cookies.length; i++) {
            cookie = cookies[i];
            CookieJar.add(cookie.split(';')[0], '/');
        }
    },
    trans = (data, mockRes, url) => {
        let Request = null,
            method = mockRes.req.method,
            headers = config.transHeaders,
            headersKeys = accessControlAllowHeaders.split(',');
        for (let i = 0; i < headersKeys.length; i++) {
            let key = headersKeys[i].trim();
            if (key) {
                headers[key] = headersKeys[key] || '';
            }
        }
        switch (method) {
            case 'GET':
                Request = unirest.get(config.transHost + config.transPath + url).headers(headers).jar(CookieJar);
                break;
            case 'POST':
                Request = unirest.post(config.transHost + config.transPath + url).headers(headers).jar(CookieJar);
                break;
        }
        warner.warn('mockserver沒有這個接口，轉發到' + config.transHost + config.transPath + url + '中');
        if (config.useProxy) {
            passProxy(data, mockRes, Request);
        } else {
            getRequest(data, mockRes, Request);
        }
    },
    uploadApi = () => { // 上傳接口
        app.post(config.uploadUrl, uploader.single(config.uploadName), (req, res) => {
            logger.info(req.body);
            let buf = fs.readFileSync(__dirname + '/' + req.file.path);
            res.write(JSON.stringify({
                resultCode: "000000",
                resultMsg: '成功',
                data: {
                    base64: buf.toString('base64'),
                    attachementNo: req.file.filename // 一般應為ID，不建數據庫就直接用名字吧
                }
            }));
            res.end();
        });
        console.log(config.uploadUrl);
    },
    downloadApi = () => {
        app.get(config.downloadFileUrl, (req, res) => {
            let params = JSON.parse(req.query.params),
                filepath = '/upload/' + params.attachementNo,
                fReadStream;
            logger.info(req.query);
            fs.readFile(__dirname + filepath, 'utf8', (err, data) => {
                if (err) {
                    console.log(err);
                    res.write(JSON.stringify({
                        resultCode: "999999",
                        resultMsg: '失敗'
                    }));
                    res.end();
                } else {
                    res.set({
                        "Content-type":"application/octet-stream",
                        "Content-Disposition":"attachment;filename=" + encodeURI(params.attachementNo)
                    });
                    fReadStream = fs.createReadStream(__dirname + filepath);
                    fReadStream.on('data', (chunk) => {
                        res.write(chunk, 'binary');
                    });
                    fReadStream.on('end', () => {
                        res.end();
                    });
                }
            });
        });
        app.get(config.downloadBase64Url, (req, res) => {
            let params = JSON.parse(req.query.params),
                buf,
                opt;
            logger.info(req.query);
            try {
                buf = fs.readFileSync(__dirname + '/upload/' + params.attachementNo);
                opt = {
                    resultCode: "000000",
                    resultMsg: '成功',
                    data: {
                        base64: buf.toString('base64')
                    }
                }
            } catch (e) {
                opt = {
                    resultCode: "999999",
                    resultMsg: e.code,
                    data: e
                }
            }
            res.write(JSON.stringify(opt));
            res.end();
        });
        console.log(config.downloadFileUrl);
        console.log(config.downloadBase64Url);
    };

console.log('======== 創建以下接口 ========');
uploadApi();
downloadApi();
console.log('========== 創建完畢 ==========\n');

app.use(errorHandler.httpError(404));
app.use(handler);

server = app.listen(config.port, () => {
    let port = server.address().port;
    console.log('致Root。服務器地址：127.0.0.1:' + port + '\n');
});