393 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			393 lines
		
	
	
		
			11 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");
 | |
| 
 | |
| //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,
 | |
|         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");
 | |
| 
 | |
| //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());
 | |
| 
 | |
| //session
 | |
| app.use(session({
 | |
|     name: config.sessionname,
 | |
|     secret: config.sessionsecret,
 | |
|     resave: false, //don't save session if unmodified
 | |
|     saveUninitialized: false, //don't create session until something stored
 | |
|     cookie: {
 | |
|         maxAge: new Date(Date.now() + config.sessionlife),
 | |
|         expires: new Date(Date.now() + config.sessionlife),
 | |
|     },
 | |
|     maxAge: new Date(Date.now() + config.sessionlife),
 | |
|     store: sessionStore
 | |
| }));
 | |
| 
 | |
| //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'));
 | |
| //template files
 | |
| app.set('views', __dirname + '/public');
 | |
| //set render engine
 | |
| app.engine('html', ejs.renderFile);
 | |
| //get index
 | |
| app.get("/", function (req, res, next) {
 | |
|     res.render("index.html");
 | |
| });
 | |
| //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: '/'
 | |
|     }),
 | |
|     function (req, res) {
 | |
|         res.redirect('/');
 | |
|     });
 | |
| //twitter auth
 | |
| app.get('/auth/twitter',
 | |
|     passport.authenticate('twitter'),
 | |
|     function (req, res) {});
 | |
| //twitter auth callback
 | |
| app.get('/auth/twitter/callback',
 | |
|     passport.authenticate('twitter', {
 | |
|         failureRedirect: '/'
 | |
|     }),
 | |
|     function (req, res) {
 | |
|         res.redirect('/');
 | |
|     });
 | |
| //github auth
 | |
| app.get('/auth/github',
 | |
|     passport.authenticate('github'),
 | |
|     function (req, res) {});
 | |
| //github auth callback
 | |
| app.get('/auth/github/callback',
 | |
|     passport.authenticate('github', {
 | |
|         failureRedirect: '/'
 | |
|     }),
 | |
|     function (req, res) {
 | |
|         res.redirect('/');
 | |
|     });
 | |
| //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: '/'
 | |
|     }),
 | |
|     function (req, res) {
 | |
|         res.redirect('/');
 | |
|     });
 | |
| //logout
 | |
| app.get('/logout', function (req, res) {
 | |
|     if (config.debug && req.session.passport.user)
 | |
|         logger.info('user logout: ' + req.session.passport.user);
 | |
|     req.logout();
 | |
|     res.redirect('/');
 | |
| });
 | |
| //get history
 | |
| app.get('/history', function (req, res) {
 | |
|     if (req.isAuthenticated()) {
 | |
|         User.model.findById(req.session.passport.user, 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.session.passport.user + ']: ' + req.body.history);
 | |
|         User.model.findById(req.session.passport.user, 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.session.passport.user, function (err, user) {
 | |
|             if (err) {
 | |
|                 logger.error('read me failed: ' + err);
 | |
|             } else {
 | |
|                 var profile = JSON.parse(user.profile);
 | |
|                 res.send({
 | |
|                     status: 'ok',
 | |
|                     id: req.session.passport.user,
 | |
|                     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 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);
 | |
| }); |