Merge branch 'au/complete-restore-s'
[git] / http-fetch.c
1 #include "cache.h"
2 #include "config.h"
3 #include "exec-cmd.h"
4 #include "http.h"
5 #include "walker.h"
6
7 static const char http_fetch_usage[] = "git http-fetch "
8 "[-c] [-t] [-a] [-v] [--recover] [-w ref] [--stdin | --packfile=hash | commit-id] url";
9
10 static int fetch_using_walker(const char *raw_url, int get_verbosely,
11                               int get_recover, int commits, char **commit_id,
12                               const char **write_ref, int commits_on_stdin)
13 {
14         char *url = NULL;
15         struct walker *walker;
16         int rc;
17
18         str_end_url_with_slash(raw_url, &url);
19
20         http_init(NULL, url, 0);
21
22         walker = get_http_walker(url);
23         walker->get_verbosely = get_verbosely;
24         walker->get_recover = get_recover;
25         walker->get_progress = 0;
26
27         rc = walker_fetch(walker, commits, commit_id, write_ref, url);
28
29         if (commits_on_stdin)
30                 walker_targets_free(commits, commit_id, write_ref);
31
32         if (walker->corrupt_object_found) {
33                 fprintf(stderr,
34 "Some loose object were found to be corrupt, but they might be just\n"
35 "a false '404 Not Found' error message sent with incorrect HTTP\n"
36 "status code.  Suggest running 'git fsck'.\n");
37         }
38
39         walker_free(walker);
40         http_cleanup();
41         free(url);
42
43         return rc;
44 }
45
46 static void fetch_single_packfile(struct object_id *packfile_hash,
47                                   const char *url) {
48         struct http_pack_request *preq;
49         struct slot_results results;
50         int ret;
51
52         http_init(NULL, url, 0);
53
54         preq = new_direct_http_pack_request(packfile_hash->hash, xstrdup(url));
55         if (preq == NULL)
56                 die("couldn't create http pack request");
57         preq->slot->results = &results;
58         preq->generate_keep = 1;
59
60         if (start_active_slot(preq->slot)) {
61                 run_active_slot(preq->slot);
62                 if (results.curl_result != CURLE_OK) {
63                         die("Unable to get pack file %s\n%s", preq->url,
64                             curl_errorstr);
65                 }
66         } else {
67                 die("Unable to start request");
68         }
69
70         if ((ret = finish_http_pack_request(preq)))
71                 die("finish_http_pack_request gave result %d", ret);
72
73         release_http_pack_request(preq);
74         http_cleanup();
75 }
76
77 int cmd_main(int argc, const char **argv)
78 {
79         int commits_on_stdin = 0;
80         int commits;
81         const char **write_ref = NULL;
82         char **commit_id;
83         int arg = 1;
84         int get_verbosely = 0;
85         int get_recover = 0;
86         int packfile = 0;
87         int nongit;
88         struct object_id packfile_hash;
89
90         setup_git_directory_gently(&nongit);
91
92         while (arg < argc && argv[arg][0] == '-') {
93                 const char *p;
94
95                 if (argv[arg][1] == 't') {
96                 } else if (argv[arg][1] == 'c') {
97                 } else if (argv[arg][1] == 'a') {
98                 } else if (argv[arg][1] == 'v') {
99                         get_verbosely = 1;
100                 } else if (argv[arg][1] == 'w') {
101                         write_ref = &argv[arg + 1];
102                         arg++;
103                 } else if (argv[arg][1] == 'h') {
104                         usage(http_fetch_usage);
105                 } else if (!strcmp(argv[arg], "--recover")) {
106                         get_recover = 1;
107                 } else if (!strcmp(argv[arg], "--stdin")) {
108                         commits_on_stdin = 1;
109                 } else if (skip_prefix(argv[arg], "--packfile=", &p)) {
110                         const char *end;
111
112                         packfile = 1;
113                         if (parse_oid_hex(p, &packfile_hash, &end) || *end)
114                                 die(_("argument to --packfile must be a valid hash (got '%s')"), p);
115                 }
116                 arg++;
117         }
118         if (argc != arg + 2 - (commits_on_stdin || packfile))
119                 usage(http_fetch_usage);
120
121         if (nongit)
122                 die(_("not a git repository"));
123
124         git_config(git_default_config, NULL);
125
126         if (packfile) {
127                 fetch_single_packfile(&packfile_hash, argv[arg]);
128                 return 0;
129         }
130
131         if (commits_on_stdin) {
132                 commits = walker_targets_stdin(&commit_id, &write_ref);
133         } else {
134                 commit_id = (char **) &argv[arg++];
135                 commits = 1;
136         }
137         return fetch_using_walker(argv[arg], get_verbosely, get_recover,
138                                   commits, commit_id, write_ref,
139                                   commits_on_stdin);
140 }