Started from strider-git

worker.js

@@ -0,0 +1,174 @@
+
+var path = require('path')
+  , fs = require('fs')
+  , spawn = require('child_process').spawn
+  , mkdirp = require('mkdirp')
+
+  , utils = require('./lib')
+
+function safespawn() {
+  var c
+  try {
+    c = spawn.apply(null, arguments)
+  } catch (e) {
+    throw new Error('Failed to start command: ' + JSON.stringify([].slice.call(arguments)))
+  }
+  c.on('error', function (err) {
+    // suppress node errors
+  })
+  return c
+}
+
+function httpCloneCmd(config, branch) {
+  var urls = utils.httpUrl(config)
+    , screen = 'git clone --recursive ' + urls[1] + ' .'
+    , args = ['clone', '--recursive', urls[0], '.']
+  if (branch) {
+    args = args.concat(['-b', branch])
+    screen += ' -b ' + branch
+  }
+  return {
+    command: 'git',
+    args: args,
+    screen: screen
+  }
+}
+
+function pull(dest, config, context, done) {
+  context.cmd({
+    cmd: 'git reset --hard',
+    cwd: dest
+  }, function (exitCode) {
+    utils.gitCmd('git pull', dest, config.auth, context, done)
+  })
+}
+
+function gitVersion(next) {
+  var child = safespawn('git', ['--version'])
+    , out = ''
+  child.stdout.setEncoding('utf8')
+  child.stderr.setEncoding('utf8')
+  child.stdout.on('data', function (data) {
+    out += data
+  })
+  child.stderr.on('data', function (data) {
+    out += data
+  })
+  child.on('close', function (code) {
+    if (code) return next(new Error('Failed to get git version: ' + out))
+    next(null, out)
+  })
+  child.on('error', function () {})
+}
+ 
+
+function clone(dest, config, ref, context, done) {
+  if (config.auth.type === 'ssh') {
+    var cmd = 'git clone --recursive ' + utils.sshUrl(config)[0] + ' .'
+    if (ref.branch) {
+      cmd += ' -b ' + ref.branch
+      // this /only/ gets the one branch; so only use if we won't be caching
+      if (!config.cache) cmd += ' --single-branch'
+    }
+    return utils.gitaneCmd(cmd, dest, config.auth.privkey, context, done)
+  }
+  context.cmd({
+    cmd: httpCloneCmd(config),
+    cwd: dest
+  }, done)
+}
+
+function badCode(name, code) {
+  var e = new Error(name + ' failed with code ' + code)
+  e.code = code
+  e.exitCode = code
+  return e
+}
+
+module.exports = {
+  init: function (dirs, config, job, done) {
+    return done(null, {
+      config: config,
+      fetch: function (context, done) {
+        module.exports.fetch(dirs.data, config, job, context, done)
+      }
+    })
+  },
+  fetch: fetch
+}
+
+function getMasterPrivKey(branches) {
+  for (var i=0; i<branches.length; i++) {
+    if (branches[i].name === 'master') {
+      return branches[i].privkey
+    }
+  }
+}
+
+function checkoutRef(dest, cmd, ref, done) {
+  return cmd({
+    cmd: 'git checkout -qf ' + utils.shellEscape(ref.id || ref.branch),
+    cwd: dest
+  }, function (exitCode) {
+    done(exitCode && badCode('Checkout', exitCode))
+  })
+}
+
+function fetch(dest, config, job, context, done) {
+  if (config.auth.type === 'ssh' && !config.auth.privkey) {
+    config.auth.privkey = getMasterPrivKey(job.project.branches)
+  }
+  var get = pull
+    , cloning = false
+    , pleaseClone = function () {
+        cloning = true
+        mkdirp(dest, function () {
+          clone(dest, config, job.ref, context, updateCache)
+        })
+      }
+  if (!config.cache) return pleaseClone()
+
+  context.cachier.get(dest, function (err) {
+    if (err) return pleaseClone()
+    // make sure .git exists
+    fs.exists(path.join(dest, '.git'), function (exists) {
+      if (exists) {
+        context.comment('restored code from cache')
+        return pull(dest, config, context, updateCache)
+      }
+      safespawn('rm', ['-rf', dest]).on('close', function (exitCode) {
+        pleaseClone()
+      })
+    })
+  })
+
+  function updateCache(exitCode) {
+    if (exitCode) return done(badCode('Git ' + (cloning ? 'clone' : 'pull'), exitCode))
+    if (!config.cache) return gotten()
+    context.comment('saved code to cache')
+    context.cachier.update(dest, gotten)
+  }
+
+  function gotten (err) {
+    if (err) return done(err)
+    // fetch the ref
+    if (job.ref.branch && !job.ref.fetch) {
+      return checkoutRef(dest, context.cmd, job.ref, done)
+    }
+    fetchRef(job.ref.fetch, dest, config.auth, context, done)
+  }
+}
+
+function fetchRef(what, dest, auth, context, done) {
+  utils.gitCmd('git fetch origin ' + utils.shellEscape(what), dest, auth, context, function (exitCode) {
+    if (exitCode) return done(badCode('Fetch ' + what, exitCode))
+    context.cmd({
+      cmd: 'git checkout -qf FETCH_HEAD',
+      cwd: dest
+    }, function (exitCode) {
+      done(exitCode && badCode('Checkout', exitCode))
+    })
+  })
+}
+
+