wt-status.c: make wt_status_check_rebase() work on any worktree

This is a preparation step for find_shared_symref() to detect if any
worktree is being rebased.

Signed-off-by: Nguyễn Thái Ngọc Duy <pclouds@gmail.com>
Signed-off-by: Junio C Hamano <gitster@pobox.com>
This commit is contained in:
Nguyễn Thái Ngọc Duy 2016-04-22 20:01:31 +07:00 committed by Junio C Hamano
parent bcd522a149
commit 81eff27b0f
2 changed files with 24 additions and 14 deletions

View File

@ -15,6 +15,7 @@
#include "column.h" #include "column.h"
#include "strbuf.h" #include "strbuf.h"
#include "utf8.h" #include "utf8.h"
#include "worktree.h"
static const char cut_line[] = static const char cut_line[] =
"------------------------ >8 ------------------------\n"; "------------------------ >8 ------------------------\n";
@ -1262,13 +1263,13 @@ static void show_bisect_in_progress(struct wt_status *s,
/* /*
* Extract branch information from rebase/bisect * Extract branch information from rebase/bisect
*/ */
static char *read_and_strip_branch(const char *path) static char *get_branch(const struct worktree *wt, const char *path)
{ {
struct strbuf sb = STRBUF_INIT; struct strbuf sb = STRBUF_INIT;
unsigned char sha1[20]; unsigned char sha1[20];
const char *branch_name; const char *branch_name;
if (strbuf_read_file(&sb, git_path("%s", path), 0) <= 0) if (strbuf_read_file(&sb, worktree_git_path(wt, "%s", path), 0) <= 0)
goto got_nothing; goto got_nothing;
while (sb.len && sb.buf[sb.len - 1] == '\n') while (sb.len && sb.buf[sb.len - 1] == '\n')
@ -1295,6 +1296,11 @@ got_nothing:
return NULL; return NULL;
} }
static char *read_and_strip_branch(const char *path)
{
return get_branch(NULL, path);
}
struct grab_1st_switch_cbdata { struct grab_1st_switch_cbdata {
struct strbuf buf; struct strbuf buf;
unsigned char nsha1[20]; unsigned char nsha1[20];
@ -1360,27 +1366,28 @@ static void wt_status_get_detached_from(struct wt_status_state *state)
strbuf_release(&cb.buf); strbuf_release(&cb.buf);
} }
int wt_status_check_rebase(struct wt_status_state *state) int wt_status_check_rebase(const struct worktree *wt,
struct wt_status_state *state)
{ {
struct stat st; struct stat st;
if (!stat(git_path("rebase-apply"), &st)) { if (!stat(worktree_git_path(wt, "rebase-apply"), &st)) {
if (!stat(git_path("rebase-apply/applying"), &st)) { if (!stat(worktree_git_path(wt, "rebase-apply/applying"), &st)) {
state->am_in_progress = 1; state->am_in_progress = 1;
if (!stat(git_path("rebase-apply/patch"), &st) && !st.st_size) if (!stat(worktree_git_path(wt, "rebase-apply/patch"), &st) && !st.st_size)
state->am_empty_patch = 1; state->am_empty_patch = 1;
} else { } else {
state->rebase_in_progress = 1; state->rebase_in_progress = 1;
state->branch = read_and_strip_branch("rebase-apply/head-name"); state->branch = get_branch(wt, "rebase-apply/head-name");
state->onto = read_and_strip_branch("rebase-apply/onto"); state->onto = get_branch(wt, "rebase-apply/onto");
} }
} else if (!stat(git_path("rebase-merge"), &st)) { } else if (!stat(worktree_git_path(wt, "rebase-merge"), &st)) {
if (!stat(git_path("rebase-merge/interactive"), &st)) if (!stat(worktree_git_path(wt, "rebase-merge/interactive"), &st))
state->rebase_interactive_in_progress = 1; state->rebase_interactive_in_progress = 1;
else else
state->rebase_in_progress = 1; state->rebase_in_progress = 1;
state->branch = read_and_strip_branch("rebase-merge/head-name"); state->branch = get_branch(wt, "rebase-merge/head-name");
state->onto = read_and_strip_branch("rebase-merge/onto"); state->onto = get_branch(wt, "rebase-merge/onto");
} else } else
return 0; return 0;
return 1; return 1;
@ -1394,7 +1401,7 @@ void wt_status_get_state(struct wt_status_state *state,
if (!stat(git_path_merge_head(), &st)) { if (!stat(git_path_merge_head(), &st)) {
state->merge_in_progress = 1; state->merge_in_progress = 1;
} else if (wt_status_check_rebase(state)) { } else if (wt_status_check_rebase(NULL, state)) {
; /* all set */ ; /* all set */
} else if (!stat(git_path_cherry_pick_head(), &st) && } else if (!stat(git_path_cherry_pick_head(), &st) &&
!get_sha1("CHERRY_PICK_HEAD", sha1)) { !get_sha1("CHERRY_PICK_HEAD", sha1)) {

View File

@ -6,6 +6,8 @@
#include "color.h" #include "color.h"
#include "pathspec.h" #include "pathspec.h"
struct worktree;
enum color_wt_status { enum color_wt_status {
WT_STATUS_HEADER = 0, WT_STATUS_HEADER = 0,
WT_STATUS_UPDATED, WT_STATUS_UPDATED,
@ -100,7 +102,8 @@ void wt_status_prepare(struct wt_status *s);
void wt_status_print(struct wt_status *s); void wt_status_print(struct wt_status *s);
void wt_status_collect(struct wt_status *s); void wt_status_collect(struct wt_status *s);
void wt_status_get_state(struct wt_status_state *state, int get_detached_from); void wt_status_get_state(struct wt_status_state *state, int get_detached_from);
int wt_status_check_rebase(struct wt_status_state *state); int wt_status_check_rebase(const struct worktree *wt,
struct wt_status_state *state);
void wt_shortstatus_print(struct wt_status *s); void wt_shortstatus_print(struct wt_status *s);
void wt_porcelain_print(struct wt_status *s); void wt_porcelain_print(struct wt_status *s);