427 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			427 lines
		
	
	
		
			12 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
//app
 | 
						|
//external modules
 | 
						|
var express = require('express');
 | 
						|
var toobusy = require('toobusy-js');
 | 
						|
var ejs = require('ejs');
 | 
						|
var passport = require('passport');
 | 
						|
var methodOverride = require('method-override');
 | 
						|
var cookieParser = require('cookie-parser');
 | 
						|
var bodyParser = require('body-parser');
 | 
						|
var mongoose = require('mongoose');
 | 
						|
var compression = require('compression')
 | 
						|
var session = require('express-session');
 | 
						|
var MongoStore = require('connect-mongo')(session);
 | 
						|
var fs = require('fs');
 | 
						|
var shortid = require('shortid');
 | 
						|
var imgur = require('imgur');
 | 
						|
var formidable = require('formidable');
 | 
						|
var morgan = require('morgan');
 | 
						|
var passportSocketIo = require("passport.socketio");
 | 
						|
var helmet = require('helmet');
 | 
						|
 | 
						|
//core
 | 
						|
var config = require("./config.js");
 | 
						|
var logger = require("./lib/logger.js");
 | 
						|
var User = require("./lib/user.js");
 | 
						|
var Temp = require("./lib/temp.js");
 | 
						|
var auth = require("./lib/auth.js");
 | 
						|
var response = require("./lib/response.js");
 | 
						|
 | 
						|
//server setup
 | 
						|
if (config.usessl) {
 | 
						|
    var ca = (function () {
 | 
						|
        var i, len, results;
 | 
						|
        results = [];
 | 
						|
        for (i = 0, len = config.sslcapath.length; i < len; i++) {
 | 
						|
            results.push(fs.readFileSync(config.sslcapath[i], 'utf8'));
 | 
						|
        }
 | 
						|
        return results;
 | 
						|
    })();
 | 
						|
    var options = {
 | 
						|
        key: fs.readFileSync(config.sslkeypath, 'utf8'),
 | 
						|
        cert: fs.readFileSync(config.sslcertpath, 'utf8'),
 | 
						|
        ca: ca,
 | 
						|
        dhparam: fs.readFileSync(config.dhparampath, 'utf8'),
 | 
						|
        requestCert: false,
 | 
						|
        rejectUnauthorized: false
 | 
						|
    };
 | 
						|
    var app = express();
 | 
						|
    var server = require('https').createServer(options, app);
 | 
						|
} else {
 | 
						|
    var app = express();
 | 
						|
    var server = require('http').createServer(app);
 | 
						|
}
 | 
						|
 | 
						|
//logger
 | 
						|
app.use(morgan('combined', {
 | 
						|
    "stream": logger.stream
 | 
						|
}));
 | 
						|
 | 
						|
//socket io
 | 
						|
var io = require('socket.io')(server);
 | 
						|
 | 
						|
// connect to the mongodb
 | 
						|
mongoose.connect(process.env.MONGOLAB_URI || config.mongodbstring);
 | 
						|
 | 
						|
//others
 | 
						|
var db = require("./lib/db.js");
 | 
						|
var realtime = require("./lib/realtime.js");
 | 
						|
 | 
						|
//assign socket io to realtime
 | 
						|
realtime.io = io;
 | 
						|
 | 
						|
//methodOverride
 | 
						|
app.use(methodOverride('_method'));
 | 
						|
 | 
						|
// create application/json parser
 | 
						|
var jsonParser = bodyParser.json();
 | 
						|
 | 
						|
// create application/x-www-form-urlencoded parser
 | 
						|
var urlencodedParser = bodyParser.urlencoded({
 | 
						|
    extended: false
 | 
						|
});
 | 
						|
 | 
						|
//session store
 | 
						|
var sessionStore = new MongoStore({
 | 
						|
        mongooseConnection: mongoose.connection,
 | 
						|
        touchAfter: config.sessiontouch
 | 
						|
    },
 | 
						|
    function (err) {
 | 
						|
        logger.info(err);
 | 
						|
    });
 | 
						|
 | 
						|
//compression
 | 
						|
app.use(compression());
 | 
						|
 | 
						|
// use hsts to tell https users stick to this
 | 
						|
app.use(helmet.hsts({
 | 
						|
    maxAge: 31536000 * 1000, // 365 days
 | 
						|
    includeSubdomains: true,
 | 
						|
    preload: true
 | 
						|
}));
 | 
						|
 | 
						|
//session
 | 
						|
app.use(session({
 | 
						|
    name: config.sessionname,
 | 
						|
    secret: config.sessionsecret,
 | 
						|
    resave: false, //don't save session if unmodified
 | 
						|
    saveUninitialized: true, //always create session to ensure the origin
 | 
						|
    cookie: {
 | 
						|
        maxAge: new Date(Date.now() + config.sessionlife),
 | 
						|
        expires: new Date(Date.now() + config.sessionlife),
 | 
						|
    },
 | 
						|
    maxAge: new Date(Date.now() + config.sessionlife),
 | 
						|
    store: sessionStore
 | 
						|
}));
 | 
						|
 | 
						|
// session resumption
 | 
						|
var tlsSessionStore = {};
 | 
						|
server.on('newSession', function (id, data, cb) {
 | 
						|
    tlsSessionStore[id.toString('hex')] = data;
 | 
						|
    cb();
 | 
						|
});
 | 
						|
server.on('resumeSession', function (id, cb) {
 | 
						|
    cb(null, tlsSessionStore[id.toString('hex')] || null);
 | 
						|
});
 | 
						|
 | 
						|
//middleware which blocks requests when we're too busy
 | 
						|
app.use(function (req, res, next) {
 | 
						|
    if (toobusy()) {
 | 
						|
        response.errorServiceUnavailable(res);
 | 
						|
    } else {
 | 
						|
        next();
 | 
						|
    }
 | 
						|
});
 | 
						|
 | 
						|
//passport
 | 
						|
app.use(passport.initialize());
 | 
						|
app.use(passport.session());
 | 
						|
 | 
						|
//serialize and deserialize
 | 
						|
passport.serializeUser(function (user, done) {
 | 
						|
    //logger.info('serializeUser: ' + user._id);
 | 
						|
    done(null, user._id);
 | 
						|
});
 | 
						|
passport.deserializeUser(function (id, done) {
 | 
						|
    User.model.findById(id, function (err, user) {
 | 
						|
        //logger.info(user)
 | 
						|
        if (!err) done(null, user);
 | 
						|
        else done(err, null);
 | 
						|
    })
 | 
						|
});
 | 
						|
 | 
						|
//routes
 | 
						|
//static files
 | 
						|
app.use('/', express.static(__dirname + '/public'));
 | 
						|
app.use('/vendor/', express.static(__dirname + '/bower_components'));
 | 
						|
//template files
 | 
						|
app.set('views', __dirname + '/public');
 | 
						|
//set render engine
 | 
						|
app.engine('html', ejs.renderFile);
 | 
						|
//get index
 | 
						|
app.get("/", response.showIndex);
 | 
						|
//get 403 forbidden
 | 
						|
app.get("/403", function(req, res) {
 | 
						|
    response.errorForbidden(res);
 | 
						|
});
 | 
						|
//get 404 not found
 | 
						|
app.get("/404", function(req, res) {
 | 
						|
    response.errorNotFound(res);
 | 
						|
});
 | 
						|
//get status
 | 
						|
app.get("/status", function (req, res, next) {
 | 
						|
    realtime.getStatus(function (data) {
 | 
						|
        res.end(JSON.stringify(data));
 | 
						|
    });
 | 
						|
});
 | 
						|
//get status
 | 
						|
app.get("/temp", function (req, res) {
 | 
						|
    var host = req.get('host');
 | 
						|
    if (config.alloworigin.indexOf(host) == -1)
 | 
						|
        response.errorForbidden(res);
 | 
						|
    else {
 | 
						|
        var tempid = req.query.tempid;
 | 
						|
        if (!tempid)
 | 
						|
            response.errorForbidden(res);
 | 
						|
        else {
 | 
						|
            Temp.findTemp(tempid, function (err, temp) {
 | 
						|
                if (err || !temp)
 | 
						|
                    response.errorForbidden(res);
 | 
						|
                else {
 | 
						|
                    res.header("Access-Control-Allow-Origin", "*");
 | 
						|
                    res.send({
 | 
						|
                        temp: temp.data
 | 
						|
                    });
 | 
						|
                    temp.remove(function (err) {
 | 
						|
                        if (err)
 | 
						|
                            logger.error('remove temp failed: ' + err);
 | 
						|
                    });
 | 
						|
                }
 | 
						|
            });
 | 
						|
        }
 | 
						|
    }
 | 
						|
});
 | 
						|
//post status
 | 
						|
app.post("/temp", urlencodedParser, function (req, res) {
 | 
						|
    var host = req.get('host');
 | 
						|
    if (config.alloworigin.indexOf(host) == -1)
 | 
						|
        response.errorForbidden(res);
 | 
						|
    else {
 | 
						|
        var id = shortid.generate();
 | 
						|
        var data = req.body.data;
 | 
						|
        if (!id || !data)
 | 
						|
            response.errorForbidden(res);
 | 
						|
        else {
 | 
						|
            if (config.debug)
 | 
						|
                logger.info('SERVER received temp from [' + host + ']: ' + req.body.data);
 | 
						|
            Temp.newTemp(id, data, function (err, temp) {
 | 
						|
                if (!err && temp) {
 | 
						|
                    res.header("Access-Control-Allow-Origin", "*");
 | 
						|
                    res.send({
 | 
						|
                        status: 'ok',
 | 
						|
                        id: temp.id
 | 
						|
                    });
 | 
						|
                } else
 | 
						|
                    response.errorInternalError(res);
 | 
						|
            });
 | 
						|
        }
 | 
						|
    }
 | 
						|
});
 | 
						|
//facebook auth
 | 
						|
app.get('/auth/facebook',
 | 
						|
    passport.authenticate('facebook'),
 | 
						|
    function (req, res) {});
 | 
						|
//facebook auth callback
 | 
						|
app.get('/auth/facebook/callback',
 | 
						|
    passport.authenticate('facebook', {
 | 
						|
        failureRedirect: config.getserverurl()
 | 
						|
    }),
 | 
						|
    function (req, res) {
 | 
						|
        res.redirect(config.getserverurl());
 | 
						|
    });
 | 
						|
//twitter auth
 | 
						|
app.get('/auth/twitter',
 | 
						|
    passport.authenticate('twitter'),
 | 
						|
    function (req, res) {});
 | 
						|
//twitter auth callback
 | 
						|
app.get('/auth/twitter/callback',
 | 
						|
    passport.authenticate('twitter', {
 | 
						|
        failureRedirect: config.getserverurl()
 | 
						|
    }),
 | 
						|
    function (req, res) {
 | 
						|
        res.redirect(config.getserverurl());
 | 
						|
    });
 | 
						|
//github auth
 | 
						|
app.get('/auth/github',
 | 
						|
    passport.authenticate('github'),
 | 
						|
    function (req, res) {});
 | 
						|
//github auth callback
 | 
						|
app.get('/auth/github/callback',
 | 
						|
    passport.authenticate('github', {
 | 
						|
        failureRedirect: config.getserverurl()
 | 
						|
    }),
 | 
						|
    function (req, res) {
 | 
						|
        res.redirect(config.getserverurl());
 | 
						|
    });
 | 
						|
//github callback actions
 | 
						|
app.get('/auth/github/callback/:noteId/:action', response.githubActions);
 | 
						|
//dropbox auth
 | 
						|
app.get('/auth/dropbox',
 | 
						|
    passport.authenticate('dropbox-oauth2'),
 | 
						|
    function (req, res) {});
 | 
						|
//dropbox auth callback
 | 
						|
app.get('/auth/dropbox/callback',
 | 
						|
    passport.authenticate('dropbox-oauth2', {
 | 
						|
        failureRedirect: config.getserverurl()
 | 
						|
    }),
 | 
						|
    function (req, res) {
 | 
						|
        res.redirect(config.getserverurl());
 | 
						|
    });
 | 
						|
//logout
 | 
						|
app.get('/logout', function (req, res) {
 | 
						|
    if (config.debug && req.isAuthenticated())
 | 
						|
        logger.info('user logout: ' + req.user._id);
 | 
						|
    req.logout();
 | 
						|
    res.redirect(config.getserverurl());
 | 
						|
});
 | 
						|
//get history
 | 
						|
app.get('/history', function (req, res) {
 | 
						|
    if (req.isAuthenticated()) {
 | 
						|
        User.model.findById(req.user._id, function (err, user) {
 | 
						|
            if (err) {
 | 
						|
                logger.error('read history failed: ' + err);
 | 
						|
            } else {
 | 
						|
                var history = [];
 | 
						|
                if (user.history)
 | 
						|
                    history = JSON.parse(user.history);
 | 
						|
                res.send({
 | 
						|
                    history: history
 | 
						|
                });
 | 
						|
            }
 | 
						|
        });
 | 
						|
    } else {
 | 
						|
        response.errorForbidden(res);
 | 
						|
    }
 | 
						|
});
 | 
						|
//post history
 | 
						|
app.post('/history', urlencodedParser, function (req, res) {
 | 
						|
    if (req.isAuthenticated()) {
 | 
						|
        if (config.debug)
 | 
						|
            logger.info('SERVER received history from [' + req.user._id + ']: ' + req.body.history);
 | 
						|
        User.model.findById(req.user._id, function (err, user) {
 | 
						|
            if (err) {
 | 
						|
                logger.error('write history failed: ' + err);
 | 
						|
            } else {
 | 
						|
                user.history = req.body.history;
 | 
						|
                user.save(function (err) {
 | 
						|
                    if (err) {
 | 
						|
                        logger.error('write user history failed: ' + err);
 | 
						|
                    } else {
 | 
						|
                        if (config.debug)
 | 
						|
                            logger.info("write user history success: " + user._id);
 | 
						|
                    };
 | 
						|
                });
 | 
						|
            }
 | 
						|
        });
 | 
						|
        res.end();
 | 
						|
    } else {
 | 
						|
        response.errorForbidden(res);
 | 
						|
    }
 | 
						|
});
 | 
						|
//get me info
 | 
						|
app.get('/me', function (req, res) {
 | 
						|
    if (req.isAuthenticated()) {
 | 
						|
        User.model.findById(req.user._id, function (err, user) {
 | 
						|
            if (err) {
 | 
						|
                logger.error('read me failed: ' + err);
 | 
						|
            } else {
 | 
						|
                var profile = JSON.parse(user.profile);
 | 
						|
                res.send({
 | 
						|
                    status: 'ok',
 | 
						|
                    id: req.user._id,
 | 
						|
                    name: profile.displayName || profile.username
 | 
						|
                });
 | 
						|
            }
 | 
						|
        });
 | 
						|
    } else {
 | 
						|
        res.send({
 | 
						|
            status: 'forbidden'
 | 
						|
        });
 | 
						|
    }
 | 
						|
});
 | 
						|
//upload to imgur
 | 
						|
app.post('/uploadimage', function (req, res) {
 | 
						|
    var form = new formidable.IncomingForm();
 | 
						|
    form.parse(req, function (err, fields, files) {
 | 
						|
        if (err || !files.image || !files.image.path) {
 | 
						|
            response.errorForbidden(res);
 | 
						|
        } else {
 | 
						|
            if (config.debug)
 | 
						|
                logger.info('SERVER received uploadimage: ' + JSON.stringify(files.image));
 | 
						|
            imgur.setClientId(config.imgur.clientID);
 | 
						|
            try {
 | 
						|
                imgur.uploadFile(files.image.path)
 | 
						|
                    .then(function (json) {
 | 
						|
                        if (config.debug)
 | 
						|
                            logger.info('SERVER uploadimage success: ' + JSON.stringify(json));
 | 
						|
                        res.send({
 | 
						|
                            link: json.data.link
 | 
						|
                        });
 | 
						|
                    })
 | 
						|
                    .catch(function (err) {
 | 
						|
                        logger.error(err);
 | 
						|
                        res.send('upload image error');
 | 
						|
                    });
 | 
						|
            } catch (err) {
 | 
						|
                logger.error(err);
 | 
						|
                res.send('upload image error');
 | 
						|
            }
 | 
						|
        }
 | 
						|
    });
 | 
						|
});
 | 
						|
//get new note
 | 
						|
app.get("/new", response.newNote);
 | 
						|
//get features
 | 
						|
app.get("/features", response.showFeatures);
 | 
						|
//get publish note
 | 
						|
app.get("/s/:shortid", response.showPublishNote);
 | 
						|
//publish note actions
 | 
						|
app.get("/s/:shortid/:action", response.publishNoteActions);
 | 
						|
//get publish slide
 | 
						|
app.get("/p/:shortid", response.showPublishSlide);
 | 
						|
//get note by id
 | 
						|
app.get("/:noteId", response.showNote);
 | 
						|
//note actions
 | 
						|
app.get("/:noteId/:action", response.noteActions);
 | 
						|
 | 
						|
//socket.io secure
 | 
						|
io.use(realtime.secure);
 | 
						|
//socket.io auth
 | 
						|
io.use(passportSocketIo.authorize({
 | 
						|
    cookieParser: cookieParser,
 | 
						|
    key: config.sessionname,
 | 
						|
    secret: config.sessionsecret,
 | 
						|
    store: sessionStore,
 | 
						|
    success: realtime.onAuthorizeSuccess,
 | 
						|
    fail: realtime.onAuthorizeFail
 | 
						|
}));
 | 
						|
//socket.io heartbeat
 | 
						|
io.set('heartbeat interval', config.heartbeatinterval);
 | 
						|
io.set('heartbeat timeout', config.heartbeattimeout);
 | 
						|
//socket.io connection
 | 
						|
io.sockets.on('connection', realtime.connection);
 | 
						|
 | 
						|
//listen
 | 
						|
if (config.usessl) {
 | 
						|
    server.listen(config.sslport, function () {
 | 
						|
        logger.info('HTTPS Server listening at sslport %d', config.sslport);
 | 
						|
    });
 | 
						|
} else {
 | 
						|
    server.listen(config.port, function () {
 | 
						|
        logger.info('HTTP Server listening at port %d', config.port);
 | 
						|
    });
 | 
						|
}
 | 
						|
process.on('uncaughtException', function (err) {
 | 
						|
    logger.error(err);
 | 
						|
});
 |