Andrew's git - gitweb.git/blob

   1#include <fcntl.h>
   2#include <unistd.h>
   3#include <string.h>
   4#include <stdlib.h>
   5#include "cache.h"
   6#include "commit.h"
   7#include <errno.h>
   8#include <stdio.h>
   9#include "rsh.h"
  10
  11static int tree = 0;
  12static int commits = 0;
  13static int all = 0;
  14
  15static int fd_in;
  16static int fd_out;
  17
  18static int fetch(unsigned char *sha1)
  19{
  20        if (has_sha1_file(sha1))
  21                return 0;
  22        write(fd_out, sha1, 20);
  23        return write_sha1_from_fd(sha1, fd_in);
  24}
  25
  26static int process_tree(unsigned char *sha1)
  27{
  28        struct tree *tree = lookup_tree(sha1);
  29        struct tree_entry_list *entries;
  30
  31        if (parse_tree(tree))
  32                return -1;
  33
  34        for (entries = tree->entries; entries; entries = entries->next) {
  35                /*
  36                  fprintf(stderr, "Tree %s ", sha1_to_hex(sha1));
  37                  fprintf(stderr, "needs %s\n", 
  38                  sha1_to_hex(entries->item.tree->object.sha1));
  39                */
  40                if (fetch(entries->item.tree->object.sha1)) {
  41                        return error("Missing item %s",
  42                                     sha1_to_hex(entries->item.tree->object.sha1));
  43                }
  44                if (entries->directory) {
  45                        if (process_tree(entries->item.tree->object.sha1))
  46                                return -1;
  47                }
  48        }
  49        return 0;
  50}
  51
  52static int process_commit(unsigned char *sha1)
  53{
  54        struct commit *obj = lookup_commit(sha1);
  55
  56        if (fetch(sha1)) {
  57                return error("Fetching %s", sha1_to_hex(sha1));
  58        }
  59
  60        if (parse_commit(obj))
  61                return -1;
  62
  63        if (tree) {
  64                if (fetch(obj->tree->object.sha1))
  65                        return -1;
  66                if (process_tree(obj->tree->object.sha1))
  67                        return -1;
  68                if (!all)
  69                        tree = 0;
  70        }
  71        if (commits) {
  72                struct commit_list *parents = obj->parents;
  73                for (; parents; parents = parents->next) {
  74                        if (has_sha1_file(parents->item->object.sha1))
  75                                continue;
  76                        if (fetch(parents->item->object.sha1)) {
  77                                /* The server might not have it, and
  78                                 * we don't mind. 
  79                                 */
  80                                error("Missing tree %s; continuing", 
  81                                      sha1_to_hex(parents->item->object.sha1));
  82                                continue;
  83                        }
  84                        if (process_commit(parents->item->object.sha1))
  85                                return -1;
  86                }
  87        }
  88        return 0;
  89}
  90
  91int main(int argc, char **argv)
  92{
  93        char *commit_id;
  94        char *url;
  95        int arg = 1;
  96        unsigned char sha1[20];
  97
  98        while (arg < argc && argv[arg][0] == '-') {
  99                if (argv[arg][1] == 't') {
 100                        tree = 1;
 101                } else if (argv[arg][1] == 'c') {
 102                        commits = 1;
 103                } else if (argv[arg][1] == 'a') {
 104                        all = 1;
 105                        tree = 1;
 106                        commits = 1;
 107                }
 108                arg++;
 109        }
 110        if (argc < arg + 2) {
 111                usage("rpull [-c] [-t] [-a] commit-id url");
 112                return 1;
 113        }
 114        commit_id = argv[arg];
 115        url = argv[arg + 1];
 116
 117        if (setup_connection(&fd_in, &fd_out, "rpush", url, arg, argv + 1))
 118                return 1;
 119
 120        get_sha1_hex(commit_id, sha1);
 121
 122        if (fetch(sha1))
 123                return 1;
 124        if (process_commit(sha1))
 125                return 1;
 126
 127        return 0;
 128}