jdb/lib/server.js

// Generated by CoffeeScript 1.9.0
require('../');


/*
    JDB.Server
 */

JDB.Server = (function() {
  function Server() {
    var ego, k, self, v;
    self = {};
    ego = {
      opts: {
        interactive: false,
        dbPath: 'jdb.db',
        port: 8137,
        host: '127.0.0.1',
        compactDBFile: true,
        config_path: null
      },
      init: function() {
        ego.init_options();
        ego.jdb = new JDB.Jdb;
        return ego.jdb.init(ego.opts).done(function() {
          if (ego.opts.interactive) {
            return ego.init_interactive();
          } else {
            return ego.init_server();
          }
        });
      },
      init_options: function() {
        var cmder, conf, config, defaults;
        cmder = require('commander');
        cmder.usage('[options] [config.json or config.js]').option('-d, --dbPath <path>', 'Data base file path').option('-i, --interactive', 'Start with interactive mode').option('-p, --port <port>', 'Port to listen to. Default is ' + ego.opts.port, parseInt).option('--host <host>', "Host to listen to. Default is " + ego.opts.host + " only").option('-c, --compactDBFile <true>', 'Whether compact db file at start up or not', function(data) {
          return data === 'true';
        }).option('-v, --ver', 'Print JDB version').parse(process.argv);
        if (cmder.ver) {
          conf = require('../package');
          console.log('JDB v' + conf.version);
          process.exit();
        }
        if (cmder.args[0]) {
          ego.opts.config_path = cmder.args[0];
        }
        if (ego.opts.config_path) {
          config = require(ego.opts.config_path);
        }
        defaults = function(opts) {
          var k, v, _ref, _results;
          if (!opts) {
            return;
          }
          _ref = ego.opts;
          _results = [];
          for (k in _ref) {
            v = _ref[k];
            if (opts[k]) {
              _results.push(ego.opts[k] = opts[k]);
            } else {
              _results.push(void 0);
            }
          }
          return _results;
        };
        defaults(config);
        return defaults(cmder);
      },
      init_interactive: function() {
        global.save = function() {
          return ego.jdb.exec({
            data: global.doc,
            command: function(jdb, data) {
              jdb.doc = data;
              return jdb.save();
            }
          });
        };
        return ego.jdb.exec({
          command: function(jdb) {
            return jdb.send(jdb.doc);
          },
          callback: function(err, doc) {
            var cmd;
            global.doc = doc;
            process.argv = [];
            cmd = require('coffee-script/lib/coffee-script/command');
            return cmd.run();
          }
        });
      },
      init_server: function() {
        var http;
        http = require('http');
        ego.server = http.createServer(ego.init_routes);
        ego.server.listen(ego.opts.port, ego.opts.host);
        return ego.log("Listen: " + ego.opts.host + ":" + ego.opts.port);
      },
      init_routes: function(req, res) {
        var ht;
        ht = {
          req: req,
          res: res
        };
        switch (req.url) {
          case '/exec':
            return ego.exec(ht);
          case '/compactDBFile':
            return ego.compactDBFile(ht);
          default:
            return ego.not_found(ht);
        }
      },
      log: function(msg, level) {
        if (level == null) {
          level = 0;
        }
        return console.log(">>", msg);
      },
      send: function(ht, body, status, type) {
        var buf;
        if (body == null) {
          body = '';
        }
        if (status == null) {
          status = 200;
        }
        if (type == null) {
          type = 'application/json; charset=utf-8';
        }
        buf = new Buffer(body);
        ht.res.writeHead(status, {
          'Content-Type': type,
          'Content-Length': buf.length,
          'X-Powered-By': 'jdb'
        });
        return ht.res.end(buf);
      },
      exec: function(ht) {
        var body;
        body = '';
        ht.req.on('data', function(chunk) {
          return body += chunk;
        });
        return ht.req.on('end', function() {
          var cmd, command, e;
          try {
            cmd = JSON.parse(body);
          } catch (_error) {
            e = _error;
            ego.send(ht, "JSON syntax error: \n" + body, 500);
            return;
          }
          if (!cmd.command) {
            ego.send(ht, 'Empty command', 403);
            return;
          }
          try {
            command = eval("(" + cmd.command + ")");
          } catch (_error) {
            e = _error;
            ego.send(ht, 'Command syntax error: \n' + cmd.command, 500);
            return;
          }
          return ego.jdb.exec({
            data: cmd.data,
            command: command,
            callback: function(err, data) {
              if (err) {
                return ego.send(ht, JSON.stringify({
                  error: err.message
                }), 500);
              } else {
                return ego.send(ht, JSON.stringify(data || 'ok'));
              }
            }
          });
        });
      },
      compactDBFile: function(ht) {
        return ego.jdb.compactDBFile.then(function(err) {
          return ego.send(ht, 'OK');
        })["catch"](function(err) {
          return ego.send(ht, JSON.stringify({
            error: err.message
          }), 500);
        }).done();
      },
      not_found: function(ht) {
        return ego.send(ht, 'not found', 404);
      }
    };
    for (k in self) {
      v = self[k];
      this[k] = v;
    }
    self = this;
    for (k in ego) {
      v = ego[k];
      if (typeof v === 'function') {
        v.bind(self);
      }
    }
    ego.init();
    return self;
  }

  return Server;

})();