Commit | Line | Data |
---|---|---|
33db5f4d LT |
1 | /* |
2 | * Check-out files from the "current cache directory" | |
3 | * | |
4 | * Copyright (C) 2005 Linus Torvalds | |
5 | * | |
33db5f4d | 6 | */ |
baffc0e7 | 7 | #include "builtin.h" |
33db5f4d | 8 | #include "cache.h" |
9debe63d | 9 | #include "quote.h" |
bad68ec9 | 10 | #include "cache-tree.h" |
a6c7db1b | 11 | #include "parse-options.h" |
33db5f4d | 12 | |
de84f99c | 13 | #define CHECKOUT_ALL 4 |
de84f99c | 14 | static int line_termination = '\n'; |
3bd348ae | 15 | static int checkout_stage; /* default to checkout stage0 */ |
de84f99c | 16 | static int to_tempfile; |
af2a651d | 17 | static char topath[4][TEMPORARY_FILENAME_LENGTH + 1]; |
c3e9a653 | 18 | |
344c52ae | 19 | static struct checkout state; |
33db5f4d | 20 | |
e414156a | 21 | static void write_tempfile_record(const char *name, int prefix_length) |
de84f99c SP |
22 | { |
23 | int i; | |
24 | ||
25 | if (CHECKOUT_ALL == checkout_stage) { | |
26 | for (i = 1; i < 4; i++) { | |
27 | if (i > 1) | |
28 | putchar(' '); | |
29 | if (topath[i][0]) | |
30 | fputs(topath[i], stdout); | |
31 | else | |
32 | putchar('.'); | |
33 | } | |
34 | } else | |
35 | fputs(topath[checkout_stage], stdout); | |
36 | ||
37 | putchar('\t'); | |
663af342 | 38 | write_name_quoted(name + prefix_length, stdout, line_termination); |
de84f99c SP |
39 | |
40 | for (i = 0; i < 4; i++) { | |
41 | topath[i][0] = 0; | |
42 | } | |
43 | } | |
44 | ||
e414156a | 45 | static int checkout_file(const char *name, int prefix_length) |
33db5f4d | 46 | { |
3bd348ae JH |
47 | int namelen = strlen(name); |
48 | int pos = cache_name_pos(name, namelen); | |
49 | int has_same_name = 0; | |
de84f99c SP |
50 | int did_checkout = 0; |
51 | int errs = 0; | |
3bd348ae JH |
52 | |
53 | if (pos < 0) | |
54 | pos = -pos - 1; | |
55 | ||
56 | while (pos < active_nr) { | |
57 | struct cache_entry *ce = active_cache[pos]; | |
f4f9adae | 58 | if (ce_namelen(ce) != namelen || |
3bd348ae JH |
59 | memcmp(ce->name, name, namelen)) |
60 | break; | |
61 | has_same_name = 1; | |
3bd348ae | 62 | pos++; |
de84f99c SP |
63 | if (ce_stage(ce) != checkout_stage |
64 | && (CHECKOUT_ALL != checkout_stage || !ce_stage(ce))) | |
65 | continue; | |
66 | did_checkout = 1; | |
67 | if (checkout_entry(ce, &state, | |
68 | to_tempfile ? topath[ce_stage(ce)] : NULL) < 0) | |
69 | errs++; | |
70 | } | |
71 | ||
72 | if (did_checkout) { | |
73 | if (to_tempfile) | |
e414156a | 74 | write_tempfile_record(name, prefix_length); |
de84f99c | 75 | return errs > 0 ? -1 : 0; |
33db5f4d | 76 | } |
3bd348ae JH |
77 | |
78 | if (!state.quiet) { | |
05207a28 | 79 | fprintf(stderr, "git checkout-index: %s ", name); |
3bd348ae JH |
80 | if (!has_same_name) |
81 | fprintf(stderr, "is not in the cache"); | |
82 | else if (checkout_stage) | |
83 | fprintf(stderr, "does not exist at stage %d", | |
84 | checkout_stage); | |
85 | else | |
86 | fprintf(stderr, "is unmerged"); | |
87 | fputc('\n', stderr); | |
88 | } | |
89 | return -1; | |
33db5f4d LT |
90 | } |
91 | ||
f7f0fbfc | 92 | static void checkout_all(const char *prefix, int prefix_length) |
33db5f4d | 93 | { |
4b12dae6 | 94 | int i, errs = 0; |
4b25d091 | 95 | struct cache_entry *last_ce = NULL; |
33db5f4d LT |
96 | |
97 | for (i = 0; i < active_nr ; i++) { | |
98 | struct cache_entry *ce = active_cache[i]; | |
de84f99c SP |
99 | if (ce_stage(ce) != checkout_stage |
100 | && (CHECKOUT_ALL != checkout_stage || !ce_stage(ce))) | |
d9f98eeb | 101 | continue; |
c3e9a653 | 102 | if (prefix && *prefix && |
3bd348ae JH |
103 | (ce_namelen(ce) <= prefix_length || |
104 | memcmp(prefix, ce->name, prefix_length))) | |
c3e9a653 | 105 | continue; |
de84f99c SP |
106 | if (last_ce && to_tempfile) { |
107 | if (ce_namelen(last_ce) != ce_namelen(ce) | |
108 | || memcmp(last_ce->name, ce->name, ce_namelen(ce))) | |
e414156a | 109 | write_tempfile_record(last_ce->name, prefix_length); |
de84f99c SP |
110 | } |
111 | if (checkout_entry(ce, &state, | |
112 | to_tempfile ? topath[ce_stage(ce)] : NULL) < 0) | |
4b12dae6 | 113 | errs++; |
de84f99c | 114 | last_ce = ce; |
33db5f4d | 115 | } |
de84f99c | 116 | if (last_ce && to_tempfile) |
e414156a | 117 | write_tempfile_record(last_ce->name, prefix_length); |
4b12dae6 JH |
118 | if (errs) |
119 | /* we have already done our error reporting. | |
120 | * exit with the same code as die(). | |
121 | */ | |
122 | exit(128); | |
33db5f4d LT |
123 | } |
124 | ||
a6c7db1b | 125 | static const char * const builtin_checkout_index_usage[] = { |
0ed21718 | 126 | N_("git checkout-index [options] [--] [<file>...]"), |
a6c7db1b MV |
127 | NULL |
128 | }; | |
d46ad9c9 | 129 | |
021b6e45 | 130 | static struct lock_file lock_file; |
31f584c2 | 131 | |
a6c7db1b MV |
132 | static int option_parse_u(const struct option *opt, |
133 | const char *arg, int unset) | |
134 | { | |
135 | int *newfd = opt->value; | |
136 | ||
137 | state.refresh_cache = 1; | |
138 | if (*newfd < 0) | |
139 | *newfd = hold_locked_index(&lock_file, 1); | |
140 | return 0; | |
141 | } | |
142 | ||
143 | static int option_parse_z(const struct option *opt, | |
144 | const char *arg, int unset) | |
145 | { | |
146 | if (unset) | |
147 | line_termination = '\n'; | |
148 | else | |
149 | line_termination = 0; | |
150 | return 0; | |
151 | } | |
152 | ||
153 | static int option_parse_prefix(const struct option *opt, | |
154 | const char *arg, int unset) | |
155 | { | |
156 | state.base_dir = arg; | |
157 | state.base_dir_len = strlen(arg); | |
158 | return 0; | |
159 | } | |
160 | ||
161 | static int option_parse_stage(const struct option *opt, | |
162 | const char *arg, int unset) | |
163 | { | |
164 | if (!strcmp(arg, "all")) { | |
165 | to_tempfile = 1; | |
166 | checkout_stage = CHECKOUT_ALL; | |
167 | } else { | |
168 | int ch = arg[0]; | |
169 | if ('1' <= ch && ch <= '3') | |
170 | checkout_stage = arg[0] - '0'; | |
171 | else | |
172 | die("stage should be between 1 and 3 or all"); | |
173 | } | |
174 | return 0; | |
175 | } | |
176 | ||
e414156a | 177 | int cmd_checkout_index(int argc, const char **argv, const char *prefix) |
33db5f4d | 178 | { |
a65a686f | 179 | int i; |
415e96c8 | 180 | int newfd = -1; |
a65a686f | 181 | int all = 0; |
9debe63d | 182 | int read_from_stdin = 0; |
e414156a | 183 | int prefix_length; |
a6c7db1b MV |
184 | int force = 0, quiet = 0, not_new = 0; |
185 | struct option builtin_checkout_index_options[] = { | |
d5d09d47 | 186 | OPT_BOOL('a', "all", &all, |
f63cf8c9 NTND |
187 | N_("check out all files in the index")), |
188 | OPT__FORCE(&force, N_("force overwrite of existing files")), | |
8c839683 | 189 | OPT__QUIET(&quiet, |
0ed21718 | 190 | N_("no warning for existing files and files not in index")), |
5d4d1440 | 191 | OPT_BOOL('n', "no-create", ¬_new, |
0ed21718 | 192 | N_("don't checkout new files")), |
a6c7db1b | 193 | { OPTION_CALLBACK, 'u', "index", &newfd, NULL, |
0ed21718 | 194 | N_("update stat information in the index file"), |
a6c7db1b MV |
195 | PARSE_OPT_NOARG, option_parse_u }, |
196 | { OPTION_CALLBACK, 'z', NULL, NULL, NULL, | |
0ed21718 | 197 | N_("paths are separated with NUL character"), |
a6c7db1b | 198 | PARSE_OPT_NOARG, option_parse_z }, |
d5d09d47 | 199 | OPT_BOOL(0, "stdin", &read_from_stdin, |
0ed21718 | 200 | N_("read list of paths from the standard input")), |
d5d09d47 | 201 | OPT_BOOL(0, "temp", &to_tempfile, |
0ed21718 NTND |
202 | N_("write the content to temporary files")), |
203 | OPT_CALLBACK(0, "prefix", NULL, N_("string"), | |
204 | N_("when creating files, prepend <string>"), | |
a6c7db1b MV |
205 | option_parse_prefix), |
206 | OPT_CALLBACK(0, "stage", NULL, NULL, | |
0ed21718 | 207 | N_("copy out the files from named stage"), |
a6c7db1b MV |
208 | option_parse_stage), |
209 | OPT_END() | |
210 | }; | |
33db5f4d | 211 | |
cf9d52e4 NTND |
212 | if (argc == 2 && !strcmp(argv[1], "-h")) |
213 | usage_with_options(builtin_checkout_index_usage, | |
214 | builtin_checkout_index_options); | |
ef90d6d4 | 215 | git_config(git_default_config, NULL); |
e414156a | 216 | state.base_dir = ""; |
c3e9a653 JH |
217 | prefix_length = prefix ? strlen(prefix) : 0; |
218 | ||
33db5f4d | 219 | if (read_cache() < 0) { |
2de381f9 | 220 | die("invalid cache"); |
33db5f4d LT |
221 | } |
222 | ||
37782920 | 223 | argc = parse_options(argc, argv, prefix, builtin_checkout_index_options, |
a6c7db1b MV |
224 | builtin_checkout_index_usage, 0); |
225 | state.force = force; | |
226 | state.quiet = quiet; | |
227 | state.not_new = not_new; | |
a65a686f | 228 | |
de84f99c | 229 | if (state.base_dir_len || to_tempfile) { |
a65a686f LT |
230 | /* when --prefix is specified we do not |
231 | * want to update cache. | |
232 | */ | |
233 | if (state.refresh_cache) { | |
021b6e45 | 234 | rollback_lock_file(&lock_file); |
4ed7cd3a | 235 | newfd = -1; |
a65a686f LT |
236 | } |
237 | state.refresh_cache = 0; | |
238 | } | |
239 | ||
240 | /* Check out named files first */ | |
a6c7db1b | 241 | for (i = 0; i < argc; i++) { |
a65a686f | 242 | const char *arg = argv[i]; |
dc46da22 | 243 | const char *p; |
a65a686f LT |
244 | |
245 | if (all) | |
7e44c935 | 246 | die("git checkout-index: don't mix '--all' and explicit filenames"); |
9debe63d | 247 | if (read_from_stdin) |
7e44c935 | 248 | die("git checkout-index: don't mix '--stdin' and explicit filenames"); |
dc46da22 | 249 | p = prefix_path(prefix, prefix_length, arg); |
e414156a | 250 | checkout_file(p, prefix_length); |
be65e7d9 | 251 | if (p < arg || p > arg + strlen(arg)) |
4b25d091 | 252 | free((char *)p); |
33db5f4d | 253 | } |
415e96c8 | 254 | |
9debe63d | 255 | if (read_from_stdin) { |
f285a2d7 | 256 | struct strbuf buf = STRBUF_INIT, nbuf = STRBUF_INIT; |
7fb1011e | 257 | |
9debe63d | 258 | if (all) |
7e44c935 | 259 | die("git checkout-index: don't mix '--all' and '--stdin'"); |
7fb1011e | 260 | |
7fb1011e | 261 | while (strbuf_getline(&buf, stdin, line_termination) != EOF) { |
dc46da22 | 262 | const char *p; |
7fb1011e PH |
263 | if (line_termination && buf.buf[0] == '"') { |
264 | strbuf_reset(&nbuf); | |
265 | if (unquote_c_style(&nbuf, buf.buf, NULL)) | |
266 | die("line is badly quoted"); | |
267 | strbuf_swap(&buf, &nbuf); | |
268 | } | |
269 | p = prefix_path(prefix, prefix_length, buf.buf); | |
e414156a | 270 | checkout_file(p, prefix_length); |
7fb1011e | 271 | if (p < buf.buf || p > buf.buf + buf.len) |
dc46da22 | 272 | free((char *)p); |
9debe63d | 273 | } |
7fb1011e | 274 | strbuf_release(&nbuf); |
e6c019d0 | 275 | strbuf_release(&buf); |
9debe63d SP |
276 | } |
277 | ||
a65a686f | 278 | if (all) |
e414156a | 279 | checkout_all(prefix, prefix_length); |
a65a686f | 280 | |
415e96c8 JH |
281 | if (0 <= newfd && |
282 | (write_cache(newfd, active_cache, active_nr) || | |
4ed7cd3a | 283 | commit_locked_index(&lock_file))) |
021b6e45 | 284 | die("Unable to write new index file"); |
33db5f4d LT |
285 | return 0; |
286 | } |