builtin / replace.con commit replace: factor object resolution out of replace_object (479bd75)
   1/*
   2 * Builtin "git replace"
   3 *
   4 * Copyright (c) 2008 Christian Couder <chriscool@tuxfamily.org>
   5 *
   6 * Based on builtin/tag.c by Kristian Høgsberg <krh@redhat.com>
   7 * and Carlos Rica <jasampler@gmail.com> that was itself based on
   8 * git-tag.sh and mktag.c by Linus Torvalds.
   9 */
  10
  11#include "cache.h"
  12#include "builtin.h"
  13#include "refs.h"
  14#include "parse-options.h"
  15
  16static const char * const git_replace_usage[] = {
  17        N_("git replace [-f] <object> <replacement>"),
  18        N_("git replace -d <object>..."),
  19        N_("git replace [--format=<format>] [-l [<pattern>]]"),
  20        NULL
  21};
  22
  23enum replace_format {
  24      REPLACE_FORMAT_SHORT,
  25      REPLACE_FORMAT_MEDIUM,
  26      REPLACE_FORMAT_LONG
  27};
  28
  29struct show_data {
  30        const char *pattern;
  31        enum replace_format format;
  32};
  33
  34static int show_reference(const char *refname, const unsigned char *sha1,
  35                          int flag, void *cb_data)
  36{
  37        struct show_data *data = cb_data;
  38
  39        if (!wildmatch(data->pattern, refname, 0, NULL)) {
  40                if (data->format == REPLACE_FORMAT_SHORT)
  41                        printf("%s\n", refname);
  42                else if (data->format == REPLACE_FORMAT_MEDIUM)
  43                        printf("%s -> %s\n", refname, sha1_to_hex(sha1));
  44                else { /* data->format == REPLACE_FORMAT_LONG */
  45                        unsigned char object[20];
  46                        enum object_type obj_type, repl_type;
  47
  48                        if (get_sha1(refname, object))
  49                                return error("Failed to resolve '%s' as a valid ref.", refname);
  50
  51                        obj_type = sha1_object_info(object, NULL);
  52                        repl_type = sha1_object_info(sha1, NULL);
  53
  54                        printf("%s (%s) -> %s (%s)\n", refname, typename(obj_type),
  55                               sha1_to_hex(sha1), typename(repl_type));
  56                }
  57        }
  58
  59        return 0;
  60}
  61
  62static int list_replace_refs(const char *pattern, const char *format)
  63{
  64        struct show_data data;
  65
  66        if (pattern == NULL)
  67                pattern = "*";
  68        data.pattern = pattern;
  69
  70        if (format == NULL || *format == '\0' || !strcmp(format, "short"))
  71                data.format = REPLACE_FORMAT_SHORT;
  72        else if (!strcmp(format, "medium"))
  73                data.format = REPLACE_FORMAT_MEDIUM;
  74        else if (!strcmp(format, "long"))
  75                data.format = REPLACE_FORMAT_LONG;
  76        else
  77                die("invalid replace format '%s'\n"
  78                    "valid formats are 'short', 'medium' and 'long'\n",
  79                    format);
  80
  81        for_each_replace_ref(show_reference, (void *) &data);
  82
  83        return 0;
  84}
  85
  86typedef int (*each_replace_name_fn)(const char *name, const char *ref,
  87                                    const unsigned char *sha1);
  88
  89static int for_each_replace_name(const char **argv, each_replace_name_fn fn)
  90{
  91        const char **p, *full_hex;
  92        char ref[PATH_MAX];
  93        int had_error = 0;
  94        unsigned char sha1[20];
  95
  96        for (p = argv; *p; p++) {
  97                if (get_sha1(*p, sha1)) {
  98                        error("Failed to resolve '%s' as a valid ref.", *p);
  99                        had_error = 1;
 100                        continue;
 101                }
 102                full_hex = sha1_to_hex(sha1);
 103                snprintf(ref, sizeof(ref), "refs/replace/%s", full_hex);
 104                /* read_ref() may reuse the buffer */
 105                full_hex = ref + strlen("refs/replace/");
 106                if (read_ref(ref, sha1)) {
 107                        error("replace ref '%s' not found.", full_hex);
 108                        had_error = 1;
 109                        continue;
 110                }
 111                if (fn(full_hex, ref, sha1))
 112                        had_error = 1;
 113        }
 114        return had_error;
 115}
 116
 117static int delete_replace_ref(const char *name, const char *ref,
 118                              const unsigned char *sha1)
 119{
 120        if (delete_ref(ref, sha1, 0))
 121                return 1;
 122        printf("Deleted replace ref '%s'\n", name);
 123        return 0;
 124}
 125
 126static int replace_object_sha1(const char *object_ref,
 127                               unsigned char object[20],
 128                               const char *replace_ref,
 129                               unsigned char repl[20],
 130                               int force)
 131{
 132        unsigned char prev[20];
 133        enum object_type obj_type, repl_type;
 134        char ref[PATH_MAX];
 135        struct ref_lock *lock;
 136
 137        if (snprintf(ref, sizeof(ref),
 138                     "refs/replace/%s",
 139                     sha1_to_hex(object)) > sizeof(ref) - 1)
 140                die("replace ref name too long: %.*s...", 50, ref);
 141        if (check_refname_format(ref, 0))
 142                die("'%s' is not a valid ref name.", ref);
 143
 144        obj_type = sha1_object_info(object, NULL);
 145        repl_type = sha1_object_info(repl, NULL);
 146        if (!force && obj_type != repl_type)
 147                die("Objects must be of the same type.\n"
 148                    "'%s' points to a replaced object of type '%s'\n"
 149                    "while '%s' points to a replacement object of type '%s'.",
 150                    object_ref, typename(obj_type),
 151                    replace_ref, typename(repl_type));
 152
 153        if (read_ref(ref, prev))
 154                hashclr(prev);
 155        else if (!force)
 156                die("replace ref '%s' already exists", ref);
 157
 158        lock = lock_any_ref_for_update(ref, prev, 0, NULL);
 159        if (!lock)
 160                die("%s: cannot lock the ref", ref);
 161        if (write_ref_sha1(lock, repl, NULL) < 0)
 162                die("%s: cannot update the ref", ref);
 163
 164        return 0;
 165}
 166
 167static int replace_object(const char *object_ref, const char *replace_ref, int force)
 168{
 169        unsigned char object[20], repl[20];
 170
 171        if (get_sha1(object_ref, object))
 172                die("Failed to resolve '%s' as a valid ref.", object_ref);
 173        if (get_sha1(replace_ref, repl))
 174                die("Failed to resolve '%s' as a valid ref.", replace_ref);
 175
 176        return replace_object_sha1(object_ref, object, replace_ref, repl, force);
 177}
 178
 179int cmd_replace(int argc, const char **argv, const char *prefix)
 180{
 181        int force = 0;
 182        const char *format = NULL;
 183        enum {
 184                MODE_UNSPECIFIED = 0,
 185                MODE_LIST,
 186                MODE_DELETE,
 187                MODE_REPLACE
 188        } cmdmode = MODE_UNSPECIFIED;
 189        struct option options[] = {
 190                OPT_CMDMODE('l', "list", &cmdmode, N_("list replace refs"), MODE_LIST),
 191                OPT_CMDMODE('d', "delete", &cmdmode, N_("delete replace refs"), MODE_DELETE),
 192                OPT_BOOL('f', "force", &force, N_("replace the ref if it exists")),
 193                OPT_STRING(0, "format", &format, N_("format"), N_("use this format")),
 194                OPT_END()
 195        };
 196
 197        check_replace_refs = 0;
 198
 199        argc = parse_options(argc, argv, prefix, options, git_replace_usage, 0);
 200
 201        if (!cmdmode)
 202                cmdmode = argc ? MODE_REPLACE : MODE_LIST;
 203
 204        if (format && cmdmode != MODE_LIST)
 205                usage_msg_opt("--format cannot be used when not listing",
 206                              git_replace_usage, options);
 207
 208        if (force && cmdmode != MODE_REPLACE)
 209                usage_msg_opt("-f only makes sense when writing a replacement",
 210                              git_replace_usage, options);
 211
 212        switch (cmdmode) {
 213        case MODE_DELETE:
 214                if (argc < 1)
 215                        usage_msg_opt("-d needs at least one argument",
 216                                      git_replace_usage, options);
 217                return for_each_replace_name(argv, delete_replace_ref);
 218
 219        case MODE_REPLACE:
 220                if (argc != 2)
 221                        usage_msg_opt("bad number of arguments",
 222                                      git_replace_usage, options);
 223                return replace_object(argv[0], argv[1], force);
 224
 225        case MODE_LIST:
 226                if (argc > 1)
 227                        usage_msg_opt("only one pattern can be given with -l",
 228                                      git_replace_usage, options);
 229                return list_replace_refs(argv[0], format);
 230
 231        default:
 232                die("BUG: invalid cmdmode %d", (int)cmdmode);
 233        }
 234}