Andrew's git / gitweb.git / contrib/examples/builtin-fetch--tool.c
?
contrib / examples / builtin-fetch--tool.con commit Merge branch 'dj/fetch-all-tags' (0f80d89)
   1#include "builtin.h"
   2#include "cache.h"
   3#include "refs.h"
   4#include "commit.h"
   5#include "sigchain.h"
   6
   7static char *get_stdin(void)
   8{
   9        struct strbuf buf = STRBUF_INIT;
  10        if (strbuf_read(&buf, 0, 1024) < 0) {
  11                die_errno("error reading standard input");
  12        }
  13        return strbuf_detach(&buf, NULL);
  14}
  15
  16static void show_new(enum object_type type, unsigned char *sha1_new)
  17{
  18        fprintf(stderr, "  %s: %s\n", typename(type),
  19                find_unique_abbrev(sha1_new, DEFAULT_ABBREV));
  20}
  21
  22static int update_ref_env(const char *action,
  23                      const char *refname,
  24                      unsigned char *sha1,
  25                      unsigned char *oldval)
  26{
  27        char msg[1024];
  28        const char *rla = getenv("GIT_REFLOG_ACTION");
  29
  30        if (!rla)
  31                rla = "(reflog update)";
  32        if (snprintf(msg, sizeof(msg), "%s: %s", rla, action) >= sizeof(msg))
  33                warning("reflog message too long: %.*s...", 50, msg);
  34        return update_ref(msg, refname, sha1, oldval, 0, QUIET_ON_ERR);
  35}
  36
  37static int update_local_ref(const char *name,
  38                            const char *new_head,
  39                            const char *note,
  40                            int verbose, int force)
  41{
  42        unsigned char sha1_old[20], sha1_new[20];
  43        char oldh[41], newh[41];
  44        struct commit *current, *updated;
  45        enum object_type type;
  46
  47        if (get_sha1_hex(new_head, sha1_new))
  48                die("malformed object name %s", new_head);
  49
  50        type = sha1_object_info(sha1_new, NULL);
  51        if (type < 0)
  52                die("object %s not found", new_head);
  53
  54        if (!*name) {
  55                /* Not storing */
  56                if (verbose) {
  57                        fprintf(stderr, "* fetched %s\n", note);
  58                        show_new(type, sha1_new);
  59                }
  60                return 0;
  61        }
  62
  63        if (get_sha1(name, sha1_old)) {
  64                const char *msg;
  65        just_store:
  66                /* new ref */
  67                if (!strncmp(name, "refs/tags/", 10))
  68                        msg = "storing tag";
  69                else
  70                        msg = "storing head";
  71                fprintf(stderr, "* %s: storing %s\n",
  72                        name, note);
  73                show_new(type, sha1_new);
  74                return update_ref_env(msg, name, sha1_new, NULL);
  75        }
  76
  77        if (!hashcmp(sha1_old, sha1_new)) {
  78                if (verbose) {
  79                        fprintf(stderr, "* %s: same as %s\n", name, note);
  80                        show_new(type, sha1_new);
  81                }
  82                return 0;
  83        }
  84
  85        if (!strncmp(name, "refs/tags/", 10)) {
  86                fprintf(stderr, "* %s: updating with %s\n", name, note);
  87                show_new(type, sha1_new);
  88                return update_ref_env("updating tag", name, sha1_new, NULL);
  89        }
  90
  91        current = lookup_commit_reference(sha1_old);
  92        updated = lookup_commit_reference(sha1_new);
  93        if (!current || !updated)
  94                goto just_store;
  95
  96        strcpy(oldh, find_unique_abbrev(current->object.sha1, DEFAULT_ABBREV));
  97        strcpy(newh, find_unique_abbrev(sha1_new, DEFAULT_ABBREV));
  98
  99        if (in_merge_bases(current, updated)) {
 100                fprintf(stderr, "* %s: fast-forward to %s\n",
 101                        name, note);
 102                fprintf(stderr, "  old..new: %s..%s\n", oldh, newh);
 103                return update_ref_env("fast-forward", name, sha1_new, sha1_old);
 104        }
 105        if (!force) {
 106                fprintf(stderr,
 107                        "* %s: not updating to non-fast-forward %s\n",
 108                        name, note);
 109                fprintf(stderr,
 110                        "  old...new: %s...%s\n", oldh, newh);
 111                return 1;
 112        }
 113        fprintf(stderr,
 114                "* %s: forcing update to non-fast-forward %s\n",
 115                name, note);
 116        fprintf(stderr, "  old...new: %s...%s\n", oldh, newh);
 117        return update_ref_env("forced-update", name, sha1_new, sha1_old);
 118}
 119
 120static int append_fetch_head(FILE *fp,
 121                             const char *head, const char *remote,
 122                             const char *remote_name, const char *remote_nick,
 123                             const char *local_name, int not_for_merge,
 124                             int verbose, int force)
 125{
 126        struct commit *commit;
 127        int remote_len, i, note_len;
 128        unsigned char sha1[20];
 129        char note[1024];
 130        const char *what, *kind;
 131
 132        if (get_sha1(head, sha1))
 133                return error("Not a valid object name: %s", head);
 134        commit = lookup_commit_reference_gently(sha1, 1);
 135        if (!commit)
 136                not_for_merge = 1;
 137
 138        if (!strcmp(remote_name, "HEAD")) {
 139                kind = "";
 140                what = "";
 141        }
 142        else if (!strncmp(remote_name, "refs/heads/", 11)) {
 143                kind = "branch";
 144                what = remote_name + 11;
 145        }
 146        else if (!strncmp(remote_name, "refs/tags/", 10)) {
 147                kind = "tag";
 148                what = remote_name + 10;
 149        }
 150        else if (!strncmp(remote_name, "refs/remotes/", 13)) {
 151                kind = "remote-tracking branch";
 152                what = remote_name + 13;
 153        }
 154        else {
 155                kind = "";
 156                what = remote_name;
 157        }
 158
 159        remote_len = strlen(remote);
 160        for (i = remote_len - 1; remote[i] == '/' && 0 <= i; i--)
 161                ;
 162        remote_len = i + 1;
 163        if (4 < i && !strncmp(".git", remote + i - 3, 4))
 164                remote_len = i - 3;
 165
 166        note_len = 0;
 167        if (*what) {
 168                if (*kind)
 169                        note_len += sprintf(note + note_len, "%s ", kind);
 170                note_len += sprintf(note + note_len, "'%s' of ", what);
 171        }
 172        note_len += sprintf(note + note_len, "%.*s", remote_len, remote);
 173        fprintf(fp, "%s\t%s\t%s\n",
 174                sha1_to_hex(commit ? commit->object.sha1 : sha1),
 175                not_for_merge ? "not-for-merge" : "",
 176                note);
 177        return update_local_ref(local_name, head, note, verbose, force);
 178}
 179
 180static char *keep;
 181static void remove_keep(void)
 182{
 183        if (keep && *keep)
 184                unlink(keep);
 185}
 186
 187static void remove_keep_on_signal(int signo)
 188{
 189        remove_keep();
 190        sigchain_pop(signo);
 191        raise(signo);
 192}
 193
 194static char *find_local_name(const char *remote_name, const char *refs,
 195                             int *force_p, int *not_for_merge_p)
 196{
 197        const char *ref = refs;
 198        int len = strlen(remote_name);
 199
 200        while (ref) {
 201                const char *next;
 202                int single_force, not_for_merge;
 203
 204                while (*ref == '\n')
 205                        ref++;
 206                if (!*ref)
 207                        break;
 208                next = strchr(ref, '\n');
 209
 210                single_force = not_for_merge = 0;
 211                if (*ref == '+') {
 212                        single_force = 1;
 213                        ref++;
 214                }
 215                if (*ref == '.') {
 216                        not_for_merge = 1;
 217                        ref++;
 218                        if (*ref == '+') {
 219                                single_force = 1;
 220                                ref++;
 221                        }
 222                }
 223                if (!strncmp(remote_name, ref, len) && ref[len] == ':') {
 224                        const char *local_part = ref + len + 1;
 225                        int retlen;
 226
 227                        if (!next)
 228                                retlen = strlen(local_part);
 229                        else
 230                                retlen = next - local_part;
 231                        *force_p = single_force;
 232                        *not_for_merge_p = not_for_merge;
 233                        return xmemdupz(local_part, retlen);
 234                }
 235                ref = next;
 236        }
 237        return NULL;
 238}
 239
 240static int fetch_native_store(FILE *fp,
 241                              const char *remote,
 242                              const char *remote_nick,
 243                              const char *refs,
 244                              int verbose, int force)
 245{
 246        char buffer[1024];
 247        int err = 0;
 248
 249        sigchain_push_common(remove_keep_on_signal);
 250        atexit(remove_keep);
 251
 252        while (fgets(buffer, sizeof(buffer), stdin)) {
 253                int len;
 254                char *cp;
 255                char *local_name;
 256                int single_force, not_for_merge;
 257
 258                for (cp = buffer; *cp && !isspace(*cp); cp++)
 259                        ;
 260                if (*cp)
 261                        *cp++ = 0;
 262                len = strlen(cp);
 263                if (len && cp[len-1] == '\n')
 264                        cp[--len] = 0;
 265                if (!strcmp(buffer, "failed"))
 266                        die("Fetch failure: %s", remote);
 267                if (!strcmp(buffer, "pack"))
 268                        continue;
 269                if (!strcmp(buffer, "keep")) {
 270                        char *od = get_object_directory();
 271                        int len = strlen(od) + strlen(cp) + 50;
 272                        keep = xmalloc(len);
 273                        sprintf(keep, "%s/pack/pack-%s.keep", od, cp);
 274                        continue;
 275                }
 276
 277                local_name = find_local_name(cp, refs,
 278                                             &single_force, &not_for_merge);
 279                if (!local_name)
 280                        continue;
 281                err |= append_fetch_head(fp,
 282                                         buffer, remote, cp, remote_nick,
 283                                         local_name, not_for_merge,
 284                                         verbose, force || single_force);
 285        }
 286        return err;
 287}
 288
 289static int parse_reflist(const char *reflist)
 290{
 291        const char *ref;
 292
 293        printf("refs='");
 294        for (ref = reflist; ref; ) {
 295                const char *next;
 296                while (*ref && isspace(*ref))
 297                        ref++;
 298                if (!*ref)
 299                        break;
 300                for (next = ref; *next && !isspace(*next); next++)
 301                        ;
 302                printf("\n%.*s", (int)(next - ref), ref);
 303                ref = next;
 304        }
 305        printf("'\n");
 306
 307        printf("rref='");
 308        for (ref = reflist; ref; ) {
 309                const char *next, *colon;
 310                while (*ref && isspace(*ref))
 311                        ref++;
 312                if (!*ref)
 313                        break;
 314                for (next = ref; *next && !isspace(*next); next++)
 315                        ;
 316                if (*ref == '.')
 317                        ref++;
 318                if (*ref == '+')
 319                        ref++;
 320                colon = strchr(ref, ':');
 321                putchar('\n');
 322                printf("%.*s", (int)((colon ? colon : next) - ref), ref);
 323                ref = next;
 324        }
 325        printf("'\n");
 326        return 0;
 327}
 328
 329static int expand_refs_wildcard(const char *ls_remote_result, int numrefs,
 330                                const char **refs)
 331{
 332        int i, matchlen, replacelen;
 333        int found_one = 0;
 334        const char *remote = *refs++;
 335        numrefs--;
 336
 337        if (numrefs == 0) {
 338                fprintf(stderr, "Nothing specified for fetching with remote.%s.fetch\n",
 339                        remote);
 340                printf("empty\n");
 341        }
 342
 343        for (i = 0; i < numrefs; i++) {
 344                const char *ref = refs[i];
 345                const char *lref = ref;
 346                const char *colon;
 347                const char *tail;
 348                const char *ls;
 349                const char *next;
 350
 351                if (*lref == '+')
 352                        lref++;
 353                colon = strchr(lref, ':');
 354                tail = lref + strlen(lref);
 355                if (!(colon &&
 356                      2 < colon - lref &&
 357                      colon[-1] == '*' &&
 358                      colon[-2] == '/' &&
 359                      2 < tail - (colon + 1) &&
 360                      tail[-1] == '*' &&
 361                      tail[-2] == '/')) {
 362                        /* not a glob */
 363                        if (!found_one++)
 364                                printf("explicit\n");
 365                        printf("%s\n", ref);
 366                        continue;
 367                }
 368
 369                /* glob */
 370                if (!found_one++)
 371                        printf("glob\n");
 372
 373                /* lref to colon-2 is remote hierarchy name;
 374                 * colon+1 to tail-2 is local.
 375                 */
 376                matchlen = (colon-1) - lref;
 377                replacelen = (tail-1) - (colon+1);
 378                for (ls = ls_remote_result; ls; ls = next) {
 379                        const char *eol;
 380                        unsigned char sha1[20];
 381                        int namelen;
 382
 383                        while (*ls && isspace(*ls))
 384                                ls++;
 385                        next = strchr(ls, '\n');
 386                        eol = !next ? (ls + strlen(ls)) : next;
 387                        if (!memcmp("^{}", eol-3, 3))
 388                                continue;
 389                        if (eol - ls < 40)
 390                                continue;
 391                        if (get_sha1_hex(ls, sha1))
 392                                continue;
 393                        ls += 40;
 394                        while (ls < eol && isspace(*ls))
 395                                ls++;
 396                        /* ls to next (or eol) is the name.
 397                         * is it identical to lref to colon-2?
 398                         */
 399                        if ((eol - ls) <= matchlen ||
 400                            strncmp(ls, lref, matchlen))
 401                                continue;
 402
 403                        /* Yes, it is a match */
 404                        namelen = eol - ls;
 405                        if (lref != ref)
 406                                putchar('+');
 407                        printf("%.*s:%.*s%.*s\n",
 408                               namelen, ls,
 409                               replacelen, colon + 1,
 410                               namelen - matchlen, ls + matchlen);
 411                }
 412        }
 413        return 0;
 414}
 415
 416static int pick_rref(int sha1_only, const char *rref, const char *ls_remote_result)
 417{
 418        int err = 0;
 419        int lrr_count = lrr_count, i, pass;
 420        const char *cp;
 421        struct lrr {
 422                const char *line;
 423                const char *name;
 424                int namelen;
 425                int shown;
 426        } *lrr_list = lrr_list;
 427
 428        for (pass = 0; pass < 2; pass++) {
 429                /* pass 0 counts and allocates, pass 1 fills... */
 430                cp = ls_remote_result;
 431                i = 0;
 432                while (1) {
 433                        const char *np;
 434                        while (*cp && isspace(*cp))
 435                                cp++;
 436                        if (!*cp)
 437                                break;
 438                        np = strchrnul(cp, '\n');
 439                        if (pass) {
 440                                lrr_list[i].line = cp;
 441                                lrr_list[i].name = cp + 41;
 442                                lrr_list[i].namelen = np - (cp + 41);
 443                        }
 444                        i++;
 445                        cp = np;
 446                }
 447                if (!pass) {
 448                        lrr_count = i;
 449                        lrr_list = xcalloc(lrr_count, sizeof(*lrr_list));
 450                }
 451        }
 452
 453        while (1) {
 454                const char *next;
 455                int rreflen;
 456                int i;
 457
 458                while (*rref && isspace(*rref))
 459                        rref++;
 460                if (!*rref)
 461                        break;
 462                next = strchrnul(rref, '\n');
 463                rreflen = next - rref;
 464
 465                for (i = 0; i < lrr_count; i++) {
 466                        struct lrr *lrr = &(lrr_list[i]);
 467
 468                        if (rreflen == lrr->namelen &&
 469                            !memcmp(lrr->name, rref, rreflen)) {
 470                                if (!lrr->shown)
 471                                        printf("%.*s\n",
 472                                               sha1_only ? 40 : lrr->namelen + 41,
 473                                               lrr->line);
 474                                lrr->shown = 1;
 475                                break;
 476                        }
 477                }
 478                if (lrr_count <= i) {
 479                        error("pick-rref: %.*s not found", rreflen, rref);
 480                        err = 1;
 481                }
 482                rref = next;
 483        }
 484        free(lrr_list);
 485        return err;
 486}
 487
 488int cmd_fetch__tool(int argc, const char **argv, const char *prefix)
 489{
 490        int verbose = 0;
 491        int force = 0;
 492        int sopt = 0;
 493
 494        while (1 < argc) {
 495                const char *arg = argv[1];
 496                if (!strcmp("-v", arg))
 497                        verbose = 1;
 498                else if (!strcmp("-f", arg))
 499                        force = 1;
 500                else if (!strcmp("-s", arg))
 501                        sopt = 1;
 502                else
 503                        break;
 504                argc--;
 505                argv++;
 506        }
 507
 508        if (argc <= 1)
 509                return error("Missing subcommand");
 510
 511        if (!strcmp("append-fetch-head", argv[1])) {
 512                int result;
 513                FILE *fp;
 514                char *filename;
 515
 516                if (argc != 8)
 517                        return error("append-fetch-head takes 6 args");
 518                filename = git_path("FETCH_HEAD");
 519                fp = fopen(filename, "a");
 520                if (!fp)
 521                        return error("cannot open %s: %s", filename, strerror(errno));
 522                result = append_fetch_head(fp, argv[2], argv[3],
 523                                           argv[4], argv[5],
 524                                           argv[6], !!argv[7][0],
 525                                           verbose, force);
 526                fclose(fp);
 527                return result;
 528        }
 529        if (!strcmp("native-store", argv[1])) {
 530                int result;
 531                FILE *fp;
 532                char *filename;
 533
 534                if (argc != 5)
 535                        return error("fetch-native-store takes 3 args");
 536                filename = git_path("FETCH_HEAD");
 537                fp = fopen(filename, "a");
 538                if (!fp)
 539                        return error("cannot open %s: %s", filename, strerror(errno));
 540                result = fetch_native_store(fp, argv[2], argv[3], argv[4],
 541                                            verbose, force);
 542                fclose(fp);
 543                return result;
 544        }
 545        if (!strcmp("parse-reflist", argv[1])) {
 546                const char *reflist;
 547                if (argc != 3)
 548                        return error("parse-reflist takes 1 arg");
 549                reflist = argv[2];
 550                if (!strcmp(reflist, "-"))
 551                        reflist = get_stdin();
 552                return parse_reflist(reflist);
 553        }
 554        if (!strcmp("pick-rref", argv[1])) {
 555                const char *ls_remote_result;
 556                if (argc != 4)
 557                        return error("pick-rref takes 2 args");
 558                ls_remote_result = argv[3];
 559                if (!strcmp(ls_remote_result, "-"))
 560                        ls_remote_result = get_stdin();
 561                return pick_rref(sopt, argv[2], ls_remote_result);
 562        }
 563        if (!strcmp("expand-refs-wildcard", argv[1])) {
 564                const char *reflist;
 565                if (argc < 4)
 566                        return error("expand-refs-wildcard takes at least 2 args");
 567                reflist = argv[2];
 568                if (!strcmp(reflist, "-"))
 569                        reflist = get_stdin();
 570                return expand_refs_wildcard(reflist, argc - 3, argv + 3);
 571        }
 572
 573        return error("Unknown subcommand: %s", argv[1]);
 574}