builtin-fetch--tool.c

   1 #include "cache.h"
   2 #include "refs.h"
   3 #include "commit.h"
   4
   5 static void show_new(char *type, unsigned char *sha1_new)
   6 {
   7         fprintf(stderr, "  %s: %s\n", type,
   8                 find_unique_abbrev(sha1_new, DEFAULT_ABBREV));
   9 }
  10
  11 static int update_ref(const char *action,
  12                       const char *refname,
  13                       unsigned char *sha1,
  14                       unsigned char *oldval)
  15 {
  16         int len;
  17         char msg[1024];
  18         char *rla = getenv("GIT_REFLOG_ACTION");
  19         static struct ref_lock *lock;
  20
  21         if (!rla)
  22                 rla = "(reflog update)";
  23         len = snprintf(msg, sizeof(msg), "%s: %s", rla, action);
  24         if (sizeof(msg) <= len)
  25                 die("insanely long action");
  26         lock = lock_any_ref_for_update(refname, oldval);
  27         if (!lock)
  28                 return 1;
  29         if (write_ref_sha1(lock, sha1, msg) < 0)
  30                 return 1;
  31         return 0;
  32 }
  33
  34 static int update_local_ref(const char *name,
  35                             const char *new_head,
  36                             const char *note,
  37                             int verbose, int force)
  38 {
  39         char type[20];
  40         unsigned char sha1_old[20], sha1_new[20];
  41         char oldh[41], newh[41];
  42         struct commit *current, *updated;
  43
  44         if (get_sha1_hex(new_head, sha1_new))
  45                 die("malformed object name %s", new_head);
  46         if (sha1_object_info(sha1_new, type, NULL))
  47                 die("object %s not found", new_head);
  48
  49         if (!*name) {
  50                 /* Not storing */
  51                 if (verbose) {
  52                         fprintf(stderr, "* fetched %s\n", note);
  53                         show_new(type, sha1_new);
  54                 }
  55                 return 0;
  56         }
  57
  58         if (get_sha1(name, sha1_old)) {
  59                 char *msg;
  60         just_store:
  61                 /* new ref */
  62                 if (!strncmp(name, "refs/tags/", 10))
  63                         msg = "storing tag";
  64                 else
  65                         msg = "storing head";
  66                 fprintf(stderr, "* %s: storing %s\n",
  67                         name, note);
  68                 show_new(type, sha1_new);
  69                 return update_ref(msg, name, sha1_new, NULL);
  70         }
  71
  72         if (!hashcmp(sha1_old, sha1_new)) {
  73                 if (verbose) {
  74                         fprintf(stderr, "* %s: same as %s\n", name, note);
  75                         show_new(type, sha1_new);
  76                 }
  77                 return 0;
  78         }
  79
  80         if (!strncmp(name, "refs/tags/", 10)) {
  81                 fprintf(stderr, "* %s: updating with %s\n", name, note);
  82                 show_new(type, sha1_new);
  83                 return update_ref("updating tag", name, sha1_new, NULL);
  84         }
  85
  86         current = lookup_commit_reference(sha1_old);
  87         updated = lookup_commit_reference(sha1_new);
  88         if (!current || !updated)
  89                 goto just_store;
  90
  91         strcpy(oldh, find_unique_abbrev(current->object.sha1, DEFAULT_ABBREV));
  92         strcpy(newh, find_unique_abbrev(sha1_new, DEFAULT_ABBREV));
  93
  94         if (in_merge_bases(current, &updated, 1)) {
  95                 fprintf(stderr, "* %s: fast forward to %s\n",
  96                         name, note);
  97                 fprintf(stderr, "  old..new: %s..%s\n", oldh, newh);
  98                 return update_ref("fast forward", name, sha1_new, sha1_old);
  99         }
 100         if (!force) {
 101                 fprintf(stderr,
 102                         "* %s: not updating to non-fast forward %s\n",
 103                         name, note);
 104                 fprintf(stderr,
 105                         "  old...new: %s...%s\n", oldh, newh);
 106                 return 1;
 107         }
 108         fprintf(stderr,
 109                 "* %s: forcing update to non-fast forward %s\n",
 110                 name, note);
 111         fprintf(stderr, "  old...new: %s...%s\n", oldh, newh);
 112         return update_ref("forced-update", name, sha1_new, sha1_old);
 113 }
 114
 115 static int append_fetch_head(FILE *fp,
 116                              const char *head, const char *remote,
 117                              const char *remote_name, const char *remote_nick,
 118                              const char *local_name, int not_for_merge,
 119                              int verbose, int force)
 120 {
 121         struct commit *commit;
 122         int remote_len, i, note_len;
 123         unsigned char sha1[20];
 124         char note[1024];
 125         const char *what, *kind;
 126
 127         if (get_sha1(head, sha1))
 128                 return error("Not a valid object name: %s", head);
 129         commit = lookup_commit_reference(sha1);
 130         if (!commit)
 131                 not_for_merge = 1;
 132
 133         if (!strcmp(remote_name, "HEAD")) {
 134                 kind = "";
 135                 what = "";
 136         }
 137         else if (!strncmp(remote_name, "refs/heads/", 11)) {
 138                 kind = "branch";
 139                 what = remote_name + 11;
 140         }
 141         else if (!strncmp(remote_name, "refs/tags/", 10)) {
 142                 kind = "tag";
 143                 what = remote_name + 10;
 144         }
 145         else if (!strncmp(remote_name, "refs/remotes/", 13)) {
 146                 kind = "remote branch";
 147                 what = remote_name + 13;
 148         }
 149         else {
 150                 kind = "";
 151                 what = remote_name;
 152         }
 153
 154         remote_len = strlen(remote);
 155         for (i = remote_len - 1; remote[i] == '/' && 0 <= i; i--)
 156                 ;
 157         remote_len = i + 1;
 158         if (4 < i && !strncmp(".git", remote + i - 3, 4))
 159                 remote_len = i - 3;
 160         note_len = sprintf(note, "%s\t%s\t",
 161                            sha1_to_hex(commit ? commit->object.sha1 : sha1),
 162                            not_for_merge ? "not-for-merge" : "");
 163         if (*what) {
 164                 if (*kind)
 165                         note_len += sprintf(note + note_len, "%s ", kind);
 166                 note_len += sprintf(note + note_len, "'%s' of ", what);
 167         }
 168         note_len += sprintf(note + note_len, "%.*s", remote_len, remote);
 169         fprintf(fp, "%s\n", note);
 170         return update_local_ref(local_name, head, note, verbose, force);
 171 }
 172
 173 int cmd_fetch__tool(int argc, const char **argv, const char *prefix)
 174 {
 175         int verbose = 0;
 176         int force = 0;
 177
 178         while (1 < argc) {
 179                 const char *arg = argv[1];
 180                 if (!strcmp("-v", arg))
 181                         verbose = 1;
 182                 else if (!strcmp("-f", arg))
 183                         force = 1;
 184                 else
 185                         break;
 186                 argc--;
 187                 argv++;
 188         }
 189
 190         if (argc <= 1)
 191                 return error("Missing subcommand");
 192
 193         if (!strcmp("append-fetch-head", argv[1])) {
 194                 int result;
 195                 FILE *fp;
 196
 197                 if (argc != 8)
 198                         return error("append-fetch-head takes 6 args");
 199                 fp = fopen(git_path("FETCH_HEAD"), "a");
 200                 result = append_fetch_head(fp, argv[2], argv[3],
 201                                            argv[4], argv[5],
 202                                            argv[6], !!argv[7][0],
 203                                            verbose, force);
 204                 fclose(fp);
 205                 return result;
 206         }
 207         if (!strcmp("update-local-ref", argv[1])) {
 208                 if (argc != 5)
 209                         return error("update-local-ref takes 3 args");
 210                 return update_local_ref(argv[2], argv[3], argv[4],
 211                                         verbose, force);
 212         }
 213         return error("Unknown subcommand: %s", argv[1]);
 214 }