Merge pull request #831 from schu/tree-walk-order tree: fix ordering for git_tree_walk
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63
diff --git a/src/tree.c b/src/tree.c
index 31a581c..9d793cb 100644
--- a/src/tree.c
+++ b/src/tree.c
@@ -759,11 +759,12 @@ int git_tree_entry_bypath(
return error;
}
-static int tree_walk_post(
+static int tree_walk(
git_tree *tree,
git_treewalk_cb callback,
git_buf *path,
- void *payload)
+ void *payload,
+ bool preorder)
{
int error = 0;
unsigned int i;
@@ -771,8 +772,8 @@ static int tree_walk_post(
for (i = 0; i < tree->entries.length; ++i) {
git_tree_entry *entry = tree->entries.contents[i];
- if (callback(path->ptr, entry, payload) < 0)
- continue;
+ if (preorder && callback(path->ptr, entry, payload) < 0)
+ return -1;
if (git_tree_entry__is_tree(entry)) {
git_tree *subtree;
@@ -789,12 +790,15 @@ static int tree_walk_post(
if (git_buf_oom(path))
return -1;
- if (tree_walk_post(subtree, callback, path, payload) < 0)
+ if (tree_walk(subtree, callback, path, payload, preorder) < 0)
return -1;
git_buf_truncate(path, path_len);
git_tree_free(subtree);
}
+
+ if (!preorder && callback(path->ptr, entry, payload) < 0)
+ return -1;
}
return 0;
@@ -807,12 +811,12 @@ int git_tree_walk(git_tree *tree, git_treewalk_cb callback, int mode, void *payl
switch (mode) {
case GIT_TREEWALK_POST:
- error = tree_walk_post(tree, callback, &root_path, payload);
+ error = tree_walk(tree, callback, &root_path, payload, false);
break;
case GIT_TREEWALK_PRE:
- tree_error("Preorder tree walking is still not implemented");
- return -1;
+ error = tree_walk(tree, callback, &root_path, payload, true);
+ break;
default:
giterr_set(GITERR_INVALID, "Invalid walking mode for tree walk");