Initial commit.

api/app.js

@@ -0,0 +1,154 @@
+import spdy from 'spdy';
+import koa from 'koa';
+import route from 'koa-route';
+import websockify from 'koa-websocket';
+import logger from 'koa-logger';
+import compress from 'koa-compress';
+import bodyparser from 'koa-bodyparser';
+import cors from 'kcors';
+import co from 'co';
+import redis from 'redis-url';
+import coRedis from 'co-redis';
+import raven from 'raven';
+import auth from './lib/auth';
+import mongoConnect from '../config/mongo';
+import * as user from './routes/user';
+import * as file from './routes/file';
+import debugname from 'debug';
+const debug = debugname('hostr-api');
+
+if (process.env.SENTRY_DSN) {
+  const ravenClient = new raven.Client(process.env.SENTRY_DSN);
+  ravenClient.patchGlobal();
+}
+
+const app = websockify(koa());
+
+const redisUrl = process.env.REDIS_URL || process.env.REDISTOGO_URL || 'redis://localhost:6379';
+
+app.use(logger());
+
+app.use(cors({
+  origin: '*',
+  credentials: true
+}));
+
+app.use(function* (next){
+  this.set('Server', 'Nintendo 64');
+  if(this.req.headers['x-forwarded-proto'] === 'http'){
+    return this.redirect('https://' + this.req.headers.host + this.req.url);
+  }
+  yield next;
+});
+
+const redisConn = redis.connect(redisUrl);
+let coRedisConn = {};
+
+co(function*() {
+  coRedisConn = coRedis(redisConn);
+  coRedisConn.on('error', function (err) {
+    debug('Redis error ' + err);
+  });
+}).catch(function(err) {
+  console.error(err);
+});
+
+let mongoConnecting = false;
+const mongoDeferred = {};
+mongoDeferred.promise = new Promise(function(resolve, reject) {
+  mongoDeferred.resolve = resolve;
+  mongoDeferred.reject = reject;
+});
+
+function* getMongo() {
+  if (!mongoConnecting) {
+    mongoConnecting = true;
+    const db = yield mongoConnect();
+    mongoDeferred.resolve(db);
+    return db;
+  } else {
+    return mongoDeferred.promise;
+  }
+}
+
+function* setupConnections(next){
+  this.db = yield getMongo();
+  this.redis = coRedisConn;
+  yield next;
+}
+app.ws.use(setupConnections);
+app.use(setupConnections);
+
+app.use(route.get('/', function* (){
+  this.status = 200;
+  this.body = '';
+}));
+
+app.use(function* (next){
+  try {
+    yield next;
+    if (this.response.status === 404 && !this.response.body) {
+      this.throw(404);
+    }
+  } catch (err) {
+    if (err.status === 401) {
+      this.set('WWW-Authenticate', 'Basic');
+      this.status = 401;
+      this.body = err.message;
+    } else if(err.status === 404) {
+      this.status = 404;
+      this.body = {
+        error: {
+          message: 'File not found',
+          code: 604
+        }
+      };
+    } else {
+      if (!err.status) {
+        debug(err);
+        throw err;
+      } else {
+        this.status = err.status;
+        this.body = err.message;
+      }
+    }
+  }
+  this.type = 'application/json';
+});
+
+app.use(compress());
+app.use(bodyparser());
+
+app.ws.use(route.all('/file/:id', file.events));
+app.ws.use(route.all('/user', user.events));
+
+app.use(route.get('/file/:id', file.get));
+
+// Run auth middleware before all other endpoints
+app.use(auth);
+
+app.use(route.get('/user', user.get));
+app.use(route.get('/user/token', user.token));
+app.use(route.get('/user/transaction', user.transaction));
+app.use(route.post('/user/settings', user.settings));
+app.use(route.get('/file', file.list));
+app.use(route.post('/file', file.post));
+app.use(route.put('/file/:id', file.put));
+app.use(route.delete('/file/:id', file.del));
+
+if (!module.parent) {
+  if (process.env.LOCALHOST_KEY) {
+    spdy.createServer({
+      key: process.env.LOCALHOST_KEY,
+      cert: process.env.LOCALHOST_CRT
+    }, app.callback()).listen(4042, function() {
+      debug('Koa SPDY server listening on port ' + (process.env.PORT || 4042));
+    });
+  } else {
+    app.listen(process.env.PORT || 4042, function() {
+      debug('Koa HTTP server listening on port ' + (process.env.PORT || 4042));
+    });
+  }
+}
+
+export default app;