gitweb: tree view: eliminate redundant "blob"
[git] / builtin-prune.c
1 #include "cache.h"
2 #include "refs.h"
3 #include "tag.h"
4 #include "commit.h"
5 #include "tree.h"
6 #include "blob.h"
7 #include "tree-walk.h"
8 #include "diff.h"
9 #include "revision.h"
10 #include "builtin.h"
11 #include "cache-tree.h"
12
13 static const char prune_usage[] = "git-prune [-n]";
14 static int show_only;
15 static struct rev_info revs;
16
17 static int prune_object(char *path, const char *filename, const unsigned char *sha1)
18 {
19         if (show_only) {
20                 printf("would prune %s/%s\n", path, filename);
21                 return 0;
22         }
23         unlink(mkpath("%s/%s", path, filename));
24         rmdir(path);
25         return 0;
26 }
27
28 static int prune_dir(int i, char *path)
29 {
30         DIR *dir = opendir(path);
31         struct dirent *de;
32
33         if (!dir)
34                 return 0;
35
36         while ((de = readdir(dir)) != NULL) {
37                 char name[100];
38                 unsigned char sha1[20];
39                 int len = strlen(de->d_name);
40
41                 switch (len) {
42                 case 2:
43                         if (de->d_name[1] != '.')
44                                 break;
45                 case 1:
46                         if (de->d_name[0] != '.')
47                                 break;
48                         continue;
49                 case 38:
50                         sprintf(name, "%02x", i);
51                         memcpy(name+2, de->d_name, len+1);
52                         if (get_sha1_hex(name, sha1) < 0)
53                                 break;
54
55                         /*
56                          * Do we know about this object?
57                          * It must have been reachable
58                          */
59                         if (lookup_object(sha1))
60                                 continue;
61
62                         prune_object(path, de->d_name, sha1);
63                         continue;
64                 }
65                 fprintf(stderr, "bad sha1 file: %s/%s\n", path, de->d_name);
66         }
67         closedir(dir);
68         return 0;
69 }
70
71 static void prune_object_dir(const char *path)
72 {
73         int i;
74         for (i = 0; i < 256; i++) {
75                 static char dir[4096];
76                 sprintf(dir, "%s/%02x", path, i);
77                 prune_dir(i, dir);
78         }
79 }
80
81 static void process_blob(struct blob *blob,
82                          struct object_array *p,
83                          struct name_path *path,
84                          const char *name)
85 {
86         struct object *obj = &blob->object;
87
88         if (obj->flags & SEEN)
89                 return;
90         obj->flags |= SEEN;
91         /* Nothing to do, really .. The blob lookup was the important part */
92 }
93
94 static void process_tree(struct tree *tree,
95                          struct object_array *p,
96                          struct name_path *path,
97                          const char *name)
98 {
99         struct object *obj = &tree->object;
100         struct tree_desc desc;
101         struct name_entry entry;
102         struct name_path me;
103
104         if (obj->flags & SEEN)
105                 return;
106         obj->flags |= SEEN;
107         if (parse_tree(tree) < 0)
108                 die("bad tree object %s", sha1_to_hex(obj->sha1));
109         name = xstrdup(name);
110         add_object(obj, p, path, name);
111         me.up = path;
112         me.elem = name;
113         me.elem_len = strlen(name);
114
115         desc.buf = tree->buffer;
116         desc.size = tree->size;
117
118         while (tree_entry(&desc, &entry)) {
119                 if (S_ISDIR(entry.mode))
120                         process_tree(lookup_tree(entry.sha1), p, &me, entry.path);
121                 else
122                         process_blob(lookup_blob(entry.sha1), p, &me, entry.path);
123         }
124         free(tree->buffer);
125         tree->buffer = NULL;
126 }
127
128 static void process_tag(struct tag *tag, struct object_array *p, const char *name)
129 {
130         struct object *obj = &tag->object;
131         struct name_path me;
132
133         if (obj->flags & SEEN)
134                 return;
135         obj->flags |= SEEN;
136
137         me.up = NULL;
138         me.elem = "tag:/";
139         me.elem_len = 5;
140
141         if (parse_tag(tag) < 0)
142                 die("bad tag object %s", sha1_to_hex(obj->sha1));
143         add_object(tag->tagged, p, NULL, name);
144 }
145
146 static void walk_commit_list(struct rev_info *revs)
147 {
148         int i;
149         struct commit *commit;
150         struct object_array objects = { 0, 0, NULL };
151
152         /* Walk all commits, process their trees */
153         while ((commit = get_revision(revs)) != NULL)
154                 process_tree(commit->tree, &objects, NULL, "");
155
156         /* Then walk all the pending objects, recursively processing them too */
157         for (i = 0; i < revs->pending.nr; i++) {
158                 struct object_array_entry *pending = revs->pending.objects + i;
159                 struct object *obj = pending->item;
160                 const char *name = pending->name;
161                 if (obj->type == OBJ_TAG) {
162                         process_tag((struct tag *) obj, &objects, name);
163                         continue;
164                 }
165                 if (obj->type == OBJ_TREE) {
166                         process_tree((struct tree *)obj, &objects, NULL, name);
167                         continue;
168                 }
169                 if (obj->type == OBJ_BLOB) {
170                         process_blob((struct blob *)obj, &objects, NULL, name);
171                         continue;
172                 }
173                 die("unknown pending object %s (%s)", sha1_to_hex(obj->sha1), name);
174         }
175 }
176
177 static int add_one_ref(const char *path, const unsigned char *sha1)
178 {
179         struct object *object = parse_object(sha1);
180         if (!object)
181                 die("bad object ref: %s:%s", path, sha1_to_hex(sha1));
182         add_pending_object(&revs, object, "");
183         return 0;
184 }
185
186 static void add_one_tree(const unsigned char *sha1)
187 {
188         struct tree *tree = lookup_tree(sha1);
189         add_pending_object(&revs, &tree->object, "");
190 }
191
192 static void add_cache_tree(struct cache_tree *it)
193 {
194         int i;
195
196         if (it->entry_count >= 0)
197                 add_one_tree(it->sha1);
198         for (i = 0; i < it->subtree_nr; i++)
199                 add_cache_tree(it->down[i]->cache_tree);
200 }
201
202 static void add_cache_refs(void)
203 {
204         int i;
205
206         read_cache();
207         for (i = 0; i < active_nr; i++) {
208                 lookup_blob(active_cache[i]->sha1);
209                 /*
210                  * We could add the blobs to the pending list, but quite
211                  * frankly, we don't care. Once we've looked them up, and
212                  * added them as objects, we've really done everything
213                  * there is to do for a blob
214                  */
215         }
216         if (active_cache_tree)
217                 add_cache_tree(active_cache_tree);
218 }
219
220 int cmd_prune(int argc, const char **argv, const char *prefix)
221 {
222         int i;
223
224         for (i = 1; i < argc; i++) {
225                 const char *arg = argv[i];
226                 if (!strcmp(arg, "-n")) {
227                         show_only = 1;
228                         continue;
229                 }
230                 usage(prune_usage);
231         }
232
233         /*
234          * Set up revision parsing, and mark us as being interested
235          * in all object types, not just commits.
236          */
237         init_revisions(&revs, prefix);
238         revs.tag_objects = 1;
239         revs.blob_objects = 1;
240         revs.tree_objects = 1;
241
242         /* Add all external refs */
243         for_each_ref(add_one_ref);
244
245         /* Add all refs from the index file */
246         add_cache_refs();
247
248         /*
249          * Set up the revision walk - this will move all commits
250          * from the pending list to the commit walking list.
251          */
252         prepare_revision_walk(&revs);
253
254         walk_commit_list(&revs);
255
256         prune_object_dir(get_object_directory());
257
258         return 0;
259 }