gogogo/lib/Service.js

// Generated by CoffeeScript 1.3.3
(function() {
  var Cron, PREFIX, Service, spawn,
    __bind = function(fn, me){ return function(){ return fn.apply(me, arguments); }; };

  spawn = require('child_process').spawn;

  Cron = require("./Cron");

  PREFIX = "ggg";

  Service = (function() {

    Service.prototype.log = function(msg) {
      return this.parent.log(this.host, msg);
    };

    Service.prototype.runCommand = function(commands, cb) {
      if (this.isLocal) {
        return this.localCommand("bash", ["-c", commands], cb);
      } else {
        return this.sshCommand(commands, cb);
      }
    };

    Service.prototype.sshCommand = function(commands, cb) {
      return this.localCommand('ssh', ["-o StrictHostKeyChecking=no", this.host, commands], function(err) {
        if (err != null) {
          return cb(new Error("SSH Command Failed"));
        }
        return cb();
      });
    };

    Service.prototype.localCommand = function(command, args, cb) {
      var proc,
        _this = this;
      proc = spawn(command, args, {
        env: process.env
      });
      proc.stdout.on('data', function(data) {
        return _this.log(data.toString().replace(/\n$/, ""));
      });
      proc.stderr.on('data', function(data) {
        return _this.log(data.toString().replace(/\n$/, ""));
      });
      proc.on('exit', function(code) {
        if (code) {
          return cb(new Error("Command Failed"));
        }
        return cb();
      });
      return proc;
    };

    function Service(name, repoName, config, parent, isLocal) {
      this.name = name;
      this.repoName = repoName;
      this.config = config;
      this.parent = parent;
      this.isLocal = isLocal != null ? isLocal : false;
      this.sshCommand = __bind(this.sshCommand, this);

      this.runCommand = __bind(this.runCommand, this);

      this.id = this.repoName + "_" + this.name;
      this.repoDir = "$HOME/" + PREFIX + "/" + this.id;
      this.historyFile = "$HOME/" + PREFIX + "/" + this.id + "-history.txt";
      this.serverUser = this.config.getUser();
      this.hookFile = "" + this.repoDir + "/.git/hooks/post-receive";
      this.logFile = "" + this.repoDir + "/ggg.log";
      this.upstartFile = "/etc/init/" + this.id + ".conf";
      if (!this.config.getStart()) {
        this.noUpstart = true;
      }
      if (this.isLocal) {
        this.host = "localhost";
        this.repoDir = "" + process.env.HOME + "/" + PREFIX + "/" + this.id;
        this.repoUrl = this.repoDir;
      } else {
        this.host = this.config.getHost();
        this.repoUrl = "ssh://" + this.host + "/~/" + PREFIX + "/" + this.id;
      }
    }

    Service.prototype.create = function(cb) {
      var createRemoteScript, cron, cronConfig, cronScript, hookScript, upstartScript;
      this.log(" - id: " + this.id);
      this.log(" - repo: " + this.repoDir);
      this.log(" - remote: " + this.repoUrl);
      this.log(" - start: " + (this.config.getStart()));
      this.log(" - install: " + (this.config.getInstall()));
      upstartScript = "";
      if (!this.noUpstart) {
        upstartScript = this.makeUpstartScript();
      }
      hookScript = this.makeHookScript();
      cronConfig = this.config.getCron();
      cronScript = '';
      if (cronConfig) {
        cron = new Cron(cronConfig, this.id, this.repoDir, this.serverUser);
        cronScript = cron.buildCron();
      }
      createRemoteScript = this.makeCreateScript(upstartScript, hookScript, cronScript);
      return this.runCommand(createRemoteScript, function(err) {
        if (err != null) {
          return cb(err);
        }
        return cb();
      });
    };

    Service.prototype.makeUpstartScript = function() {
      return "description '" + this.id + "'\nstart on [2345]\nstop on [!2345]\nlimit nofile 10000 15000\nrespawn\nrespawn limit 5 5 \nexec su " + this.serverUser + " -c 'cd " + this.repoDir + " && " + (this.config.getStart()) + "' >> " + this.logFile + " 2>&1";
    };

    Service.prototype.makeHookScript = function() {
      return "read oldrev newrev refname\necho 'GOGOGO checking out:'\necho \\$newrev\necho \\`date\\` - \\$newrev >> " + this.historyFile + "\ncd " + this.repoDir + "/.git\nGIT_WORK_TREE=" + this.repoDir + " git reset --hard \\$newrev || exit 1;";
    };

    Service.prototype.makeCreateScript = function(upstart, hook, cronInstallScript) {
      var upstartInstall;
      upstartInstall = "";
      if (upstart) {
        upstartInstall = "echo \"" + upstart + "\" | sudo tee " + this.upstartFile;
      }
      return "echo '\nCREATING...'\nmkdir -p " + this.repoDir + "\ncd " + this.repoDir + "\necho \"Locating git\"\nwhich git \nif (( $? )); then\n    echo \"Could not locate git\"\n    exit 1\nfi\ngit init\ngit config receive.denyCurrentBranch ignore\n\n" + upstartInstall + "\n\necho \"" + hook + "\" > " + this.hookFile + "\nchmod +x " + this.hookFile + "\necho \"[√] created\"\n" + cronInstallScript;
    };

    Service.prototype.deploy = function(branch, cb) {
      var _this = this;
      this.log(" - name: " + this.name);
      this.log(" - server: " + this.host);
      this.log(" - branch: " + branch);
      return this.create(function(err) {
        if (err != null) {
          return cb(err);
        }
        _this.log("\nPUSHING");
        return _this.localCommand("git", ["push", _this.repoUrl, branch, "-f"], function(err) {
          var installCommand;
          if (err != null) {
            return cb(err);
          }
          _this.log("[√] pushed");
          installCommand = _this.makeInstallCommand() + "\n" + _this.makeRestartCommand();
          return _this.runCommand(installCommand, function(err) {
            var command, kill;
            if (err != null) {
              return cb(err);
            }
            command = _this.serverLogs(10, function() {});
            kill = function() {
              command.kill();
              return cb();
            };
            return setTimeout(kill, 2000);
          });
        });
      });
    };

    Service.prototype.makeInstallCommand = function() {
      return "echo '\nINSTALLING'\ncd " + this.repoDir + "\n" + (this.config.getInstall()) + " || exit 1;\necho '[√] installed'";
    };

    Service.prototype.makeRestartCommand = function() {
      if (this.noUpstart) {
        return "";
      }
      return "echo '\nRESTARTING'\nsudo stop " + this.id + "\nsudo start " + this.id + "\necho '[√] restarted'";
    };

    Service.prototype.restart = function(cb) {
      if (this.noUpstart) {
        return this.log("nothing to restart");
      }
      this.log("RESTARTING");
      return this.runCommand(this.makeRestartCommand(), cb);
    };

    Service.prototype.stop = function(cb) {
      if (this.noUpstart) {
        return this.log("nothing to stop");
      }
      this.log("STOPPING");
      return this.runCommand("sudo stop " + this.id + ";", cb);
    };

    Service.prototype.start = function(cb) {
      if (this.noUpstart) {
        return this.log("nothing to start");
      }
      this.log("STARTING");
      return this.runCommand("sudo start " + this.id + ";", cb);
    };

    Service.prototype.serverLogs = function(lines, cb) {
      this.log("Tailing " + this.logFile + "... Control-C to exit");
      this.log("-------------------------------------------------------------");
      return this.runCommand("tail -n " + lines + " -f " + this.logFile, cb);
    };

    Service.prototype.getHistory = function(revisions, cb) {
      this.log("Retrieving last " + revisions + " deploys, most recent first!");
      this.log("-------------------------------------------------------------");
      return this.runCommand("tail -n " + revisions + " " + this.historyFile + " | tac", cb);
    };

    return Service;

  })();

  module.exports = Service;

}).call(this);