From 05d02740e0adbb36cc461323647f0c1e7f493156 Mon Sep 17 00:00:00 2001
From: lassulus <lass@aidsballs.de>
Date: Thu, 8 Oct 2015 19:18:39 +0200
Subject: adopt from krebscode/painload

---
 index.js | 106 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 106 insertions(+)
 create mode 100644 index.js

(limited to 'index.js')

diff --git a/index.js b/index.js
new file mode 100644
index 0000000..e9b551c
--- /dev/null
+++ b/index.js
@@ -0,0 +1,106 @@
+// configuration
+var httpPort = process.env.PORT;
+var redisPrefix = process.env.REDIS_KEY_PREFIX;
+
+
+// automatic configuration
+if (!httpPort) {
+  httpPort = 1337;
+}
+if (!redisPrefix) {
+  redisPrefix = 'go:';
+}
+
+
+// load libraries
+var http = require('http');
+var url = require('url');
+var formidable = require('formidable');
+var redis = require('redis');
+
+
+// instantiate components
+var redisClient = redis.createClient();
+var httpServer = http.createServer(listener);
+
+
+// setup compoments
+redisClient.on('error', function (err) {
+  console.log('redis made a bubu:', err.message);
+  process.exit(23);
+});
+httpServer.listen(httpPort, function () {
+  console.log('http server listening on port', httpPort);
+});
+
+
+// http handler
+function listener (req, res) {
+  if (req.method === 'POST' && req.url === '/') {
+    return create(req, res);
+  } else if (req.method === 'GET') {
+    return retrieve(req, res);
+  } else {
+    return methodNotAllowed(req, res);
+  }
+}
+
+function create (req, res) {
+  redisClient.incr(redisPrefix + 'index', function (err, reply) {
+    if (err) {
+      return internalError(err, req, res);
+    }
+
+    var form = new formidable.IncomingForm();
+
+    form.parse(req, function(err, fields, files) {
+      if (err) {
+        return internalError(err, req, res);
+      }
+
+      var uri = fields.uri;
+      // TODO check uri(?)
+      var shortPath = '/' + reply;
+      var shortUri = 'http://' + req.headers.host + shortPath;
+      var key = redisPrefix + shortPath;
+
+      redisClient.set(key, uri, function (error) {
+        if (error) {
+          return internalError(err, req, res);
+        }
+
+        res.writeHead(200, { 'content-type': 'text/plain' });
+        return res.end(shortUri + '\r\n');
+      });
+    });
+  });
+}
+
+function retrieve (req, res) {
+  var key = redisPrefix + url.parse(req.url).path;
+  redisClient.get(key, function (error, reply) {
+    if (error) {
+      return internalError(err, req, res);
+    }
+    
+    if (typeof reply !== 'string') {
+      res.writeHead(404, { 'content-type': 'text/plain' });
+      return res.end('not found\r\n');
+    }
+
+    res.writeHead(302, {
+      'content-type': 'text/plain',
+      'location': reply,
+    });
+    return res.end();
+  });
+}
+
+function methodNotAllowed (req, res) {
+  res.writeHead(405, { 'content-type': 'text/plain' });
+  return res.end('method not allowed\r\n');
+}
+function internalError (error, req, res) {
+  res.writeHead(500, { 'content-type': 'text/plain' });
+  return res.end(error.message + '\r\n');
+}
-- 
cgit v1.2.3