Merge branch 'tr/protect-low-3-fds'
[git] / builtin / clean.c
1 /*
2  * "git clean" builtin command
3  *
4  * Copyright (C) 2007 Shawn Bohrer
5  *
6  * Based on git-clean.sh by Pavel Roskin
7  */
8
9 #include "builtin.h"
10 #include "cache.h"
11 #include "dir.h"
12 #include "parse-options.h"
13 #include "refs.h"
14 #include "string-list.h"
15 #include "quote.h"
16
17 static int force = -1; /* unset */
18
19 static const char *const builtin_clean_usage[] = {
20         N_("git clean [-d] [-f] [-n] [-q] [-e <pattern>] [-x | -X] [--] <paths>..."),
21         NULL
22 };
23
24 static const char *msg_remove = N_("Removing %s\n");
25 static const char *msg_would_remove = N_("Would remove %s\n");
26 static const char *msg_skip_git_dir = N_("Skipping repository %s\n");
27 static const char *msg_would_skip_git_dir = N_("Would skip repository %s\n");
28 static const char *msg_warn_remove_failed = N_("failed to remove %s");
29
30 static int git_clean_config(const char *var, const char *value, void *cb)
31 {
32         if (!strcmp(var, "clean.requireforce"))
33                 force = !git_config_bool(var, value);
34         return git_default_config(var, value, cb);
35 }
36
37 static int exclude_cb(const struct option *opt, const char *arg, int unset)
38 {
39         struct string_list *exclude_list = opt->value;
40         string_list_append(exclude_list, arg);
41         return 0;
42 }
43
44 static int remove_dirs(struct strbuf *path, const char *prefix, int force_flag,
45                 int dry_run, int quiet, int *dir_gone)
46 {
47         DIR *dir;
48         struct strbuf quoted = STRBUF_INIT;
49         struct dirent *e;
50         int res = 0, ret = 0, gone = 1, original_len = path->len, len, i;
51         unsigned char submodule_head[20];
52         struct string_list dels = STRING_LIST_INIT_DUP;
53
54         *dir_gone = 1;
55
56         if ((force_flag & REMOVE_DIR_KEEP_NESTED_GIT) &&
57                         !resolve_gitlink_ref(path->buf, "HEAD", submodule_head)) {
58                 if (!quiet) {
59                         quote_path_relative(path->buf, strlen(path->buf), &quoted, prefix);
60                         printf(dry_run ?  _(msg_would_skip_git_dir) : _(msg_skip_git_dir),
61                                         quoted.buf);
62                 }
63
64                 *dir_gone = 0;
65                 return 0;
66         }
67
68         dir = opendir(path->buf);
69         if (!dir) {
70                 /* an empty dir could be removed even if it is unreadble */
71                 res = dry_run ? 0 : rmdir(path->buf);
72                 if (res) {
73                         quote_path_relative(path->buf, strlen(path->buf), &quoted, prefix);
74                         warning(_(msg_warn_remove_failed), quoted.buf);
75                         *dir_gone = 0;
76                 }
77                 return res;
78         }
79
80         if (path->buf[original_len - 1] != '/')
81                 strbuf_addch(path, '/');
82
83         len = path->len;
84         while ((e = readdir(dir)) != NULL) {
85                 struct stat st;
86                 if (is_dot_or_dotdot(e->d_name))
87                         continue;
88
89                 strbuf_setlen(path, len);
90                 strbuf_addstr(path, e->d_name);
91                 if (lstat(path->buf, &st))
92                         ; /* fall thru */
93                 else if (S_ISDIR(st.st_mode)) {
94                         if (remove_dirs(path, prefix, force_flag, dry_run, quiet, &gone))
95                                 ret = 1;
96                         if (gone) {
97                                 quote_path_relative(path->buf, strlen(path->buf), &quoted, prefix);
98                                 string_list_append(&dels, quoted.buf);
99                         } else
100                                 *dir_gone = 0;
101                         continue;
102                 } else {
103                         res = dry_run ? 0 : unlink(path->buf);
104                         if (!res) {
105                                 quote_path_relative(path->buf, strlen(path->buf), &quoted, prefix);
106                                 string_list_append(&dels, quoted.buf);
107                         } else {
108                                 quote_path_relative(path->buf, strlen(path->buf), &quoted, prefix);
109                                 warning(_(msg_warn_remove_failed), quoted.buf);
110                                 *dir_gone = 0;
111                                 ret = 1;
112                         }
113                         continue;
114                 }
115
116                 /* path too long, stat fails, or non-directory still exists */
117                 *dir_gone = 0;
118                 ret = 1;
119                 break;
120         }
121         closedir(dir);
122
123         strbuf_setlen(path, original_len);
124
125         if (*dir_gone) {
126                 res = dry_run ? 0 : rmdir(path->buf);
127                 if (!res)
128                         *dir_gone = 1;
129                 else {
130                         quote_path_relative(path->buf, strlen(path->buf), &quoted, prefix);
131                         warning(_(msg_warn_remove_failed), quoted.buf);
132                         *dir_gone = 0;
133                         ret = 1;
134                 }
135         }
136
137         if (!*dir_gone && !quiet) {
138                 for (i = 0; i < dels.nr; i++)
139                         printf(dry_run ?  _(msg_would_remove) : _(msg_remove), dels.items[i].string);
140         }
141         string_list_clear(&dels, 0);
142         return ret;
143 }
144
145 int cmd_clean(int argc, const char **argv, const char *prefix)
146 {
147         int i, res;
148         int dry_run = 0, remove_directories = 0, quiet = 0, ignored = 0;
149         int ignored_only = 0, config_set = 0, errors = 0, gone = 1;
150         int rm_flags = REMOVE_DIR_KEEP_NESTED_GIT;
151         struct strbuf directory = STRBUF_INIT;
152         struct dir_struct dir;
153         static const char **pathspec;
154         struct strbuf buf = STRBUF_INIT;
155         struct string_list exclude_list = STRING_LIST_INIT_NODUP;
156         struct exclude_list *el;
157         const char *qname;
158         char *seen = NULL;
159         struct option options[] = {
160                 OPT__QUIET(&quiet, N_("do not print names of files removed")),
161                 OPT__DRY_RUN(&dry_run, N_("dry run")),
162                 OPT__FORCE(&force, N_("force")),
163                 OPT_BOOLEAN('d', NULL, &remove_directories,
164                                 N_("remove whole directories")),
165                 { OPTION_CALLBACK, 'e', "exclude", &exclude_list, N_("pattern"),
166                   N_("add <pattern> to ignore rules"), PARSE_OPT_NONEG, exclude_cb },
167                 OPT_BOOLEAN('x', NULL, &ignored, N_("remove ignored files, too")),
168                 OPT_BOOLEAN('X', NULL, &ignored_only,
169                                 N_("remove only ignored files")),
170                 OPT_END()
171         };
172
173         git_config(git_clean_config, NULL);
174         if (force < 0)
175                 force = 0;
176         else
177                 config_set = 1;
178
179         argc = parse_options(argc, argv, prefix, options, builtin_clean_usage,
180                              0);
181
182         memset(&dir, 0, sizeof(dir));
183         if (ignored_only)
184                 dir.flags |= DIR_SHOW_IGNORED;
185
186         if (ignored && ignored_only)
187                 die(_("-x and -X cannot be used together"));
188
189         if (!dry_run && !force) {
190                 if (config_set)
191                         die(_("clean.requireForce set to true and neither -n nor -f given; "
192                                   "refusing to clean"));
193                 else
194                         die(_("clean.requireForce defaults to true and neither -n nor -f given; "
195                                   "refusing to clean"));
196         }
197
198         if (force > 1)
199                 rm_flags = 0;
200
201         dir.flags |= DIR_SHOW_OTHER_DIRECTORIES;
202
203         if (read_cache() < 0)
204                 die(_("index file corrupt"));
205
206         if (!ignored)
207                 setup_standard_excludes(&dir);
208
209         el = add_exclude_list(&dir, EXC_CMDL, "--exclude option");
210         for (i = 0; i < exclude_list.nr; i++)
211                 add_exclude(exclude_list.items[i].string, "", 0, el, -(i+1));
212
213         pathspec = get_pathspec(prefix, argv);
214
215         fill_directory(&dir, pathspec);
216
217         if (pathspec)
218                 seen = xmalloc(argc > 0 ? argc : 1);
219
220         for (i = 0; i < dir.nr; i++) {
221                 struct dir_entry *ent = dir.entries[i];
222                 int len, pos;
223                 int matches = 0;
224                 struct cache_entry *ce;
225                 struct stat st;
226
227                 /*
228                  * Remove the '/' at the end that directory
229                  * walking adds for directory entries.
230                  */
231                 len = ent->len;
232                 if (len && ent->name[len-1] == '/')
233                         len--;
234                 pos = cache_name_pos(ent->name, len);
235                 if (0 <= pos)
236                         continue;       /* exact match */
237                 pos = -pos - 1;
238                 if (pos < active_nr) {
239                         ce = active_cache[pos];
240                         if (ce_namelen(ce) == len &&
241                             !memcmp(ce->name, ent->name, len))
242                                 continue; /* Yup, this one exists unmerged */
243                 }
244
245                 /*
246                  * we might have removed this as part of earlier
247                  * recursive directory removal, so lstat() here could
248                  * fail with ENOENT.
249                  */
250                 if (lstat(ent->name, &st))
251                         continue;
252
253                 if (pathspec) {
254                         memset(seen, 0, argc > 0 ? argc : 1);
255                         matches = match_pathspec(pathspec, ent->name, len,
256                                                  0, seen);
257                 }
258
259                 if (S_ISDIR(st.st_mode)) {
260                         strbuf_addstr(&directory, ent->name);
261                         if (remove_directories || (matches == MATCHED_EXACTLY)) {
262                                 if (remove_dirs(&directory, prefix, rm_flags, dry_run, quiet, &gone))
263                                         errors++;
264                                 if (gone && !quiet) {
265                                         qname = quote_path_relative(directory.buf, directory.len, &buf, prefix);
266                                         printf(dry_run ? _(msg_would_remove) : _(msg_remove), qname);
267                                 }
268                         }
269                         strbuf_reset(&directory);
270                 } else {
271                         if (pathspec && !matches)
272                                 continue;
273                         res = dry_run ? 0 : unlink(ent->name);
274                         if (res) {
275                                 qname = quote_path_relative(ent->name, -1, &buf, prefix);
276                                 warning(_(msg_warn_remove_failed), qname);
277                                 errors++;
278                         } else if (!quiet) {
279                                 qname = quote_path_relative(ent->name, -1, &buf, prefix);
280                                 printf(dry_run ? _(msg_would_remove) : _(msg_remove), qname);
281                         }
282                 }
283         }
284         free(seen);
285
286         strbuf_release(&directory);
287         string_list_clear(&exclude_list, 0);
288         return (errors != 0);
289 }