for_each_worktree iterates through each worktree and invokes a callback
function. The main worktree (if not bare) is always encountered first,
followed by worktrees created by `git worktree add`.
If the callback function returns a non-zero value, iteration stops, and
the callback's return value is returned from the for_each_worktree
function.
Signed-off-by: Michael Rappazzo <rappa...@gmail.com>
---
builtin/worktree.c | 83 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 83 insertions(+)
diff --git a/builtin/worktree.c b/builtin/worktree.c
index 430b51e..7b3cb96 100644
--- a/builtin/worktree.c
+++ b/builtin/worktree.c
@@ -26,6 +26,14 @@ static int show_only;
static int verbose;
static unsigned long expire;
+/*
+ * The signature for the callback function for the for_each_worktree()
+ * function below. The memory pointed to by the callback arguments
+ * is only guaranteed to be valid for the duration of a single
+ * callback invocation.
+ */
+typedef int each_worktree_fn(const char *path, const char *git_dir, void
*cb_data);
+
static int prune_worktree(const char *id, struct strbuf *reason)
{
struct stat st;
@@ -359,6 +367,81 @@ static int add(int ac, const char **av, const char *prefix)
return add_worktree(path, branch, &opts);
}
+/*
+ * Iterate through each worktree and invoke the callback function. If
+ * the callback function returns non-zero, the iteration stops, and
+ * this function returns that return value
+ */
+static int for_each_worktree(each_worktree_fn fn, void *cb_data)
+{
+ struct strbuf worktree_path = STRBUF_INIT;
+ struct strbuf worktree_git = STRBUF_INIT;
+ const char *common_dir;
+ int main_is_bare = 0;
+ int ret = 0;
+
+ common_dir = get_git_common_dir();
+ if (!strcmp(common_dir, get_git_dir())) {
+ /* simple case - this is the main repo */
+ main_is_bare = is_bare_repository();
+ if (!main_is_bare) {
+ strbuf_addstr(&worktree_path, get_git_work_tree());
+ } else {
+ strbuf_addstr(&worktree_path, common_dir);
+ }
+ } else {
+ strbuf_addstr(&worktree_path, common_dir);
+ /* If the common_dir DOES NOT end with '/.git', then it is bare
*/
+ main_is_bare = !strbuf_strip_suffix(&worktree_path, "/.git");
+ }
+ strbuf_addstr(&worktree_git, worktree_path.buf);
+
+ if (!main_is_bare) {
+ strbuf_addstr(&worktree_git, "/.git");
+
+ ret = fn(worktree_path.buf, worktree_git.buf, cb_data);
+ if (ret)
+ goto done;
+ }
+ strbuf_addstr(&worktree_git, "/worktrees");
+
+ if (is_directory(worktree_git.buf)) {
+ DIR *dir = opendir(worktree_git.buf);
+ if (dir) {
+ struct stat st;
+ struct dirent *d;
+ size_t base_path_len = worktree_git.len;
+
+ while ((d = readdir(dir)) != NULL) {
+ if (!strcmp(d->d_name, ".") ||
!strcmp(d->d_name, ".."))
+ continue;
+
+ strbuf_setlen(&worktree_git, base_path_len);
+ strbuf_addf(&worktree_git, "/%s/gitdir",
d->d_name);
+
+ if (stat(worktree_git.buf, &st)) {
+ fprintf(stderr, "Unable to read
worktree git dir: %s\n", worktree_git.buf);
+ continue;
+ }
+
+ strbuf_reset(&worktree_path);
+ strbuf_read_file(&worktree_path,
worktree_git.buf, st.st_size);
+ strbuf_strip_suffix(&worktree_path, "/.git\n");
+
+ strbuf_strip_suffix(&worktree_git, "/gitdir");
+ ret = fn(worktree_path.buf, worktree_git.buf,
cb_data);
+ if (ret)
+ break;
+ }
+ }
+ closedir(dir);
+ }
+done:
+ strbuf_release(&worktree_git);
+ strbuf_release(&worktree_path);
+ return ret;
+}
+
int cmd_worktree(int ac, const char **av, const char *prefix)
{
struct option options[] = {
--
2.5.0
--
To unsubscribe from this list: send the line "unsubscribe git" in
the body of a message to majord...@vger.kernel.org
More majordomo info at http://vger.kernel.org/majordomo-info.html
