Fixed index count, revwalks and references

John Haley · John Haley · commit ed1b9c8a3ec5 · 2014-11-19T10:52:29.000-07:00
diff --git a/generate/descriptor.json b/generate/descriptor.json
@@ -507,12 +507,18 @@
       "git_index_conflict_next": {
         "ignore": true
       },
+      "git_index_entrycount": {
+        "jsFunctionName": "size"
+      },
       "git_index_find": {
         "ignore": true
       },
       "git_index_free": {
         "ignore": true
       },
+      "git_index_get_byindex": {
+        "jsFunctionName": "entry"
+      },
       "git_index_new": {
         "ignore": true
       },
@@ -900,6 +906,19 @@
       "git_reference_iterator_new": {
         "ignore": true
       },
+      "git_reference_list": {
+        "args": {
+          "array": {
+            "isReturn": true,
+            "shouldAlloc": true,
+            "cppClassName": "Array",
+            "jsClassName": "Array",
+            "size": "count",
+            "key": "strings"
+          }
+        },
+        "isAsync": true
+      },
       "git_reference_next": {
         "ignore": true
       },
diff --git a/lib/nodegit.js b/lib/nodegit.js
@@ -52,6 +52,7 @@ require("./oid");
 require("./patch");
 require("./refs");
 require("./remote");
+require("./revwalk");
 require("./repository");
 require("./revwalk");
 require("./signature");
diff --git a/lib/repository.js b/lib/repository.js
@@ -85,6 +85,52 @@ Repository.prototype.getReference = function(name, callback) {
   }, callback);
 };
 
+Repository.getReferences = function(repo, type, refNamesOnly, callback) {
+  return Reference.list(repo).then(function(refList) {
+    var refFilterPromises = [];
+    var filteredRefs = [];
+
+    refList.forEach(function(refName) {
+      refFilterPromises.push(Reference.lookup(repo, refName)
+        .then(function(ref) {
+          if (type == Reference.Type.All || ref.type() == type) {
+            if (refNamesOnly) {
+              filteredRefs.push(refName);
+              return;
+            }
+
+            if (ref.type() == Reference.Type.Symbolic) {
+              return ref.resolve().then(function(resolvedRef) {
+                resolvedRef.repo = repo;
+
+                filteredRefs.push(resolvedRef);
+              });
+            }
+            else {
+              filteredRefs.push(ref);
+            }
+          }
+        })
+      );
+    });
+
+    return Promise.all(refFilterPromises).then(function() {
+      if (callback) {
+        callback(null, filteredRefs);
+      }
+      return filteredRefs;
+    }, callback);
+  });
+};
+
+Repository.prototype.getReferences = function(type, callback) {
+  return Repository.getReferences(this, type, false, callback);
+};
+
+Repository.prototype.getReferenceNames = function(type, callback) {
+  return Repository.getReferences(this, type, true, callback);
+};
+
 /**
  * Retrieve the commit identified by oid.
  *
