Git.pm: Implement options for the command interface
[git/builtin-gsoc.git] / builtin-push.c
blob66b94078229c09d8105fcd7f5800a2cba567c972
1 /*
2 * "git push"
3 */
4 #include "cache.h"
5 #include "refs.h"
6 #include "run-command.h"
7 #include "builtin.h"
9 #define MAX_URI (16)
11 static const char push_usage[] = "git push [--all] [--tags] [--force] <repository> [<refspec>...]";
13 static int all = 0, tags = 0, force = 0, thin = 1;
14 static const char *execute = NULL;
16 #define BUF_SIZE (2084)
17 static char buffer[BUF_SIZE];
19 static const char **refspec = NULL;
20 static int refspec_nr = 0;
22 static void add_refspec(const char *ref)
24 int nr = refspec_nr + 1;
25 refspec = xrealloc(refspec, nr * sizeof(char *));
26 refspec[nr-1] = ref;
27 refspec_nr = nr;
30 static int expand_one_ref(const char *ref, const unsigned char *sha1)
32 /* Ignore the "refs/" at the beginning of the refname */
33 ref += 5;
35 if (strncmp(ref, "tags/", 5))
36 return 0;
38 add_refspec(strdup(ref));
39 return 0;
42 static void expand_refspecs(void)
44 if (all) {
45 if (refspec_nr)
46 die("cannot mix '--all' and a refspec");
49 * No need to expand "--all" - we'll just use
50 * the "--all" flag to send-pack
52 return;
54 if (!tags)
55 return;
56 for_each_ref(expand_one_ref);
59 static void set_refspecs(const char **refs, int nr)
61 if (nr) {
62 size_t bytes = nr * sizeof(char *);
64 refspec = xrealloc(refspec, bytes);
65 memcpy(refspec, refs, bytes);
66 refspec_nr = nr;
68 expand_refspecs();
71 static int get_remotes_uri(const char *repo, const char *uri[MAX_URI])
73 int n = 0;
74 FILE *f = fopen(git_path("remotes/%s", repo), "r");
75 int has_explicit_refspec = refspec_nr || all || tags;
77 if (!f)
78 return -1;
79 while (fgets(buffer, BUF_SIZE, f)) {
80 int is_refspec;
81 char *s, *p;
83 if (!strncmp("URL: ", buffer, 5)) {
84 is_refspec = 0;
85 s = buffer + 5;
86 } else if (!strncmp("Push: ", buffer, 6)) {
87 is_refspec = 1;
88 s = buffer + 6;
89 } else
90 continue;
92 /* Remove whitespace at the head.. */
93 while (isspace(*s))
94 s++;
95 if (!*s)
96 continue;
98 /* ..and at the end */
99 p = s + strlen(s);
100 while (isspace(p[-1]))
101 *--p = 0;
103 if (!is_refspec) {
104 if (n < MAX_URI)
105 uri[n++] = strdup(s);
106 else
107 error("more than %d URL's specified, ignoreing the rest", MAX_URI);
109 else if (is_refspec && !has_explicit_refspec)
110 add_refspec(strdup(s));
112 fclose(f);
113 if (!n)
114 die("remote '%s' has no URL", repo);
115 return n;
118 static const char **config_uri;
119 static const char *config_repo;
120 static int config_repo_len;
121 static int config_current_uri;
122 static int config_get_refspecs;
124 static int get_remote_config(const char* key, const char* value)
126 if (!strncmp(key, "remote.", 7) &&
127 !strncmp(key + 7, config_repo, config_repo_len)) {
128 if (!strcmp(key + 7 + config_repo_len, ".url")) {
129 if (config_current_uri < MAX_URI)
130 config_uri[config_current_uri++] = strdup(value);
131 else
132 error("more than %d URL's specified, ignoring the rest", MAX_URI);
134 else if (config_get_refspecs &&
135 !strcmp(key + 7 + config_repo_len, ".push"))
136 add_refspec(strdup(value));
138 return 0;
141 static int get_config_remotes_uri(const char *repo, const char *uri[MAX_URI])
143 config_repo_len = strlen(repo);
144 config_repo = repo;
145 config_current_uri = 0;
146 config_uri = uri;
147 config_get_refspecs = !(refspec_nr || all || tags);
149 git_config(get_remote_config);
150 return config_current_uri;
153 static int get_branches_uri(const char *repo, const char *uri[MAX_URI])
155 const char *slash = strchr(repo, '/');
156 int n = slash ? slash - repo : 1000;
157 FILE *f = fopen(git_path("branches/%.*s", n, repo), "r");
158 char *s, *p;
159 int len;
161 if (!f)
162 return 0;
163 s = fgets(buffer, BUF_SIZE, f);
164 fclose(f);
165 if (!s)
166 return 0;
167 while (isspace(*s))
168 s++;
169 if (!*s)
170 return 0;
171 p = s + strlen(s);
172 while (isspace(p[-1]))
173 *--p = 0;
174 len = p - s;
175 if (slash)
176 len += strlen(slash);
177 p = xmalloc(len + 1);
178 strcpy(p, s);
179 if (slash)
180 strcat(p, slash);
181 uri[0] = p;
182 return 1;
186 * Read remotes and branches file, fill the push target URI
187 * list. If there is no command line refspecs, read Push: lines
188 * to set up the *refspec list as well.
189 * return the number of push target URIs
191 static int read_config(const char *repo, const char *uri[MAX_URI])
193 int n;
195 if (*repo != '/') {
196 n = get_remotes_uri(repo, uri);
197 if (n > 0)
198 return n;
200 n = get_config_remotes_uri(repo, uri);
201 if (n > 0)
202 return n;
204 n = get_branches_uri(repo, uri);
205 if (n > 0)
206 return n;
209 uri[0] = repo;
210 return 1;
213 static int do_push(const char *repo)
215 const char *uri[MAX_URI];
216 int i, n;
217 int common_argc;
218 const char **argv;
219 int argc;
221 n = read_config(repo, uri);
222 if (n <= 0)
223 die("bad repository '%s'", repo);
225 argv = xmalloc((refspec_nr + 10) * sizeof(char *));
226 argv[0] = "dummy-send-pack";
227 argc = 1;
228 if (all)
229 argv[argc++] = "--all";
230 if (force)
231 argv[argc++] = "--force";
232 if (execute)
233 argv[argc++] = execute;
234 common_argc = argc;
236 for (i = 0; i < n; i++) {
237 int error;
238 int dest_argc = common_argc;
239 int dest_refspec_nr = refspec_nr;
240 const char **dest_refspec = refspec;
241 const char *dest = uri[i];
242 const char *sender = "git-send-pack";
243 if (!strncmp(dest, "http://", 7) ||
244 !strncmp(dest, "https://", 8))
245 sender = "git-http-push";
246 else if (thin)
247 argv[dest_argc++] = "--thin";
248 argv[0] = sender;
249 argv[dest_argc++] = dest;
250 while (dest_refspec_nr--)
251 argv[dest_argc++] = *dest_refspec++;
252 argv[dest_argc] = NULL;
253 error = run_command_v(argc, argv);
254 if (!error)
255 continue;
256 switch (error) {
257 case -ERR_RUN_COMMAND_FORK:
258 die("unable to fork for %s", sender);
259 case -ERR_RUN_COMMAND_EXEC:
260 die("unable to exec %s", sender);
261 case -ERR_RUN_COMMAND_WAITPID:
262 case -ERR_RUN_COMMAND_WAITPID_WRONG_PID:
263 case -ERR_RUN_COMMAND_WAITPID_SIGNAL:
264 case -ERR_RUN_COMMAND_WAITPID_NOEXIT:
265 die("%s died with strange error", sender);
266 default:
267 return -error;
270 return 0;
273 int cmd_push(int argc, const char **argv, char **envp)
275 int i;
276 const char *repo = "origin"; // default repository
278 for (i = 1; i < argc; i++) {
279 const char *arg = argv[i];
281 if (arg[0] != '-') {
282 repo = arg;
283 i++;
284 break;
286 if (!strcmp(arg, "--all")) {
287 all = 1;
288 continue;
290 if (!strcmp(arg, "--tags")) {
291 tags = 1;
292 continue;
294 if (!strcmp(arg, "--force")) {
295 force = 1;
296 continue;
298 if (!strcmp(arg, "--thin")) {
299 thin = 1;
300 continue;
302 if (!strcmp(arg, "--no-thin")) {
303 thin = 0;
304 continue;
306 if (!strncmp(arg, "--exec=", 7)) {
307 execute = arg;
308 continue;
310 usage(push_usage);
312 set_refspecs(argv + i, argc - i);
313 return do_push(repo);