sequencer: unify label lookup

The arguments to the `reset` and `merge` commands may be a label created
with a `label` command or an arbitrary commit name. The `merge` command
uses the lookup_label() function to lookup its arguments but `reset` has
a slightly different version of that function in do_reset(). Reduce this
code duplication by calling lookup_label() from do_reset() as well.

This change improves the behavior of `reset` when the argument is a
tree.  Previously `reset` would accept a tree only for the rebase to
fail with

       update_ref failed for ref 'HEAD': cannot update ref 'HEAD': trying to write non-commit object da5497437fd67ca928333aab79c4b4b55036ea66 to branch 'HEAD'

Using lookup_label() means do_reset() will now error out straight away
if its argument is not a commit.

Signed-off-by: Phillip Wood <phillip.wood@dunelm.org.uk>
Signed-off-by: Taylor Blau <me@ttaylorr.com>
This commit is contained in:
Phillip Wood 2022-11-10 16:43:40 +00:00 committed by Taylor Blau
parent 319605f8f0
commit 82766b2961
2 changed files with 33 additions and 24 deletions

View File

@ -3694,6 +3694,26 @@ static const char *reflog_message(struct replay_opts *opts,
return buf.buf; return buf.buf;
} }
static struct commit *lookup_label(const char *label, int len,
struct strbuf *buf)
{
struct commit *commit;
strbuf_reset(buf);
strbuf_addf(buf, "refs/rewritten/%.*s", len, label);
commit = lookup_commit_reference_by_name(buf->buf);
if (!commit) {
/* fall back to non-rewritten ref or commit */
strbuf_splice(buf, 0, strlen("refs/rewritten/"), "", 0);
commit = lookup_commit_reference_by_name(buf->buf);
}
if (!commit)
error(_("could not resolve '%s'"), buf->buf);
return commit;
}
static int do_reset(struct repository *r, static int do_reset(struct repository *r,
const char *name, int len, const char *name, int len,
struct replay_opts *opts) struct replay_opts *opts)
@ -3725,6 +3745,7 @@ static int do_reset(struct repository *r,
oidcpy(&oid, &opts->squash_onto); oidcpy(&oid, &opts->squash_onto);
} else { } else {
int i; int i;
struct commit *commit;
/* Determine the length of the label */ /* Determine the length of the label */
for (i = 0; i < len; i++) for (i = 0; i < len; i++)
@ -3732,12 +3753,12 @@ static int do_reset(struct repository *r,
break; break;
len = i; len = i;
strbuf_addf(&ref_name, "refs/rewritten/%.*s", len, name); commit = lookup_label(name, len, &ref_name);
if (get_oid(ref_name.buf, &oid) && if (!commit) {
get_oid(ref_name.buf + strlen("refs/rewritten/"), &oid)) { ret = -1;
ret = error(_("could not read '%s'"), ref_name.buf);
goto cleanup; goto cleanup;
} }
oid = commit->object.oid;
} }
setup_unpack_trees_porcelain(&unpack_tree_opts, "reset"); setup_unpack_trees_porcelain(&unpack_tree_opts, "reset");
@ -3784,26 +3805,6 @@ cleanup:
return ret; return ret;
} }
static struct commit *lookup_label(const char *label, int len,
struct strbuf *buf)
{
struct commit *commit;
strbuf_reset(buf);
strbuf_addf(buf, "refs/rewritten/%.*s", len, label);
commit = lookup_commit_reference_by_name(buf->buf);
if (!commit) {
/* fall back to non-rewritten ref or commit */
strbuf_splice(buf, 0, strlen("refs/rewritten/"), "", 0);
commit = lookup_commit_reference_by_name(buf->buf);
}
if (!commit)
error(_("could not resolve '%s'"), buf->buf);
return commit;
}
static int do_merge(struct repository *r, static int do_merge(struct repository *r,
struct commit *commit, struct commit *commit,
const char *arg, int arg_len, const char *arg, int arg_len,

View File

@ -138,6 +138,14 @@ test_expect_success '`reset` refuses to overwrite untracked files' '
git rebase --abort git rebase --abort
' '
test_expect_success '`reset` rejects trees' '
test_when_finished "test_might_fail git rebase --abort" &&
test_must_fail env GIT_SEQUENCE_EDITOR="echo reset A^{tree} >" \
git rebase -i B C >out 2>err &&
grep "object .* is a tree" err &&
test_must_be_empty out
'
test_expect_success 'failed `merge -C` writes patch (may be rescheduled, too)' ' test_expect_success 'failed `merge -C` writes patch (may be rescheduled, too)' '
test_when_finished "test_might_fail git rebase --abort" && test_when_finished "test_might_fail git rebase --abort" &&
git checkout -b conflicting-merge A && git checkout -b conflicting-merge A &&