Add getStatus convenience method to repository

orderedlist · orderedlist · commit 01fbc0c99e0f · 2015-02-03T17:00:39.000-05:00
diff --git a/lib/repository.js b/lib/repository.js
@@ -8,6 +8,8 @@ var Commit = require("./commit");
 var Remote = require("./remote");
 var Promise = require("nodegit-promise");
 var normalizeOptions = require("./util/normalize_options");
+var Status = require("./status");
+var StatusFile = require("./status_file");
 
 var TreeBuilder = NodeGit.Treebuilder;
 var Repository = NodeGit.Repository;
@@ -648,5 +650,28 @@ Repository.initExt = function(repo_path, opts) {
   return initExt(repo_path, opts);
 };
 
+/**
+ * Get the status of a repo to it's working directory
+ *
+ * @param {obj} opts
+ * @return {Object} Promise object.
+ */
+Repository.prototype.getStatus = function(opts) {
+  var statuses = [];
+  var statusCallback = function(path, status) {
+    statuses.push(new StatusFile(path, status));
+  };
+
+  if (!opts) {
+    opts = {
+      flags: Status.OPT.INCLUDE_UNTRACKED +
+             Status.OPT.RECURSE_UNTRACKED_DIRS
+    };
+  }
+
+  return Status.foreachExt(this, opts, statusCallback).then(function() {
+    return statuses;
+  });
+};
 
 module.exports = Repository;
diff --git a/test/tests/repository.js b/test/tests/repository.js
@@ -75,4 +75,21 @@ describe("Repository", function() {
 
     assert(sig instanceof Signature);
   });
+
+  it("gets statuses with StatusFile", function() {
+    var fileName = "my-new-file-that-shouldnt-exist";
+    var fileContent = "new file";
+    var repo = this.repository;
+    var filePath = path.join(repo.workdir(), fileName);
+
+    return fse.writeFile(filePath, fileContent)
+      .then(function() {
+        return repo.getStatus().then(function(statuses) {
+          assert.equal(statuses.length, 1);
+          assert.equal(statuses[0].path(), fileName);
+          assert.ok(statuses[0].isNew());
+          return fse.unlink(filePath);
+        });
+      });
+  });
 });
