Merge branch 'jh/notes'
[git] / builtin / prune-packed.c
1 #include "builtin.h"
2 #include "cache.h"
3 #include "progress.h"
4 #include "parse-options.h"
5
6 static const char * const prune_packed_usage[] = {
7         "git prune-packed [-n|--dry-run] [-q|--quiet]",
8         NULL
9 };
10
11 #define DRY_RUN 01
12 #define VERBOSE 02
13
14 static struct progress *progress;
15
16 static void prune_dir(int i, DIR *dir, char *pathname, int len, int opts)
17 {
18         struct dirent *de;
19         char hex[40];
20
21         sprintf(hex, "%02x", i);
22         while ((de = readdir(dir)) != NULL) {
23                 unsigned char sha1[20];
24                 if (strlen(de->d_name) != 38)
25                         continue;
26                 memcpy(hex+2, de->d_name, 38);
27                 if (get_sha1_hex(hex, sha1))
28                         continue;
29                 if (!has_sha1_pack(sha1))
30                         continue;
31                 memcpy(pathname + len, de->d_name, 38);
32                 if (opts & DRY_RUN)
33                         printf("rm -f %s\n", pathname);
34                 else
35                         unlink_or_warn(pathname);
36                 display_progress(progress, i + 1);
37         }
38         pathname[len] = 0;
39         rmdir(pathname);
40 }
41
42 void prune_packed_objects(int opts)
43 {
44         int i;
45         static char pathname[PATH_MAX];
46         const char *dir = get_object_directory();
47         int len = strlen(dir);
48
49         if (opts == VERBOSE)
50                 progress = start_progress_delay("Removing duplicate objects",
51                         256, 95, 2);
52
53         if (len > PATH_MAX - 42)
54                 die("impossible object directory");
55         memcpy(pathname, dir, len);
56         if (len && pathname[len-1] != '/')
57                 pathname[len++] = '/';
58         for (i = 0; i < 256; i++) {
59                 DIR *d;
60
61                 display_progress(progress, i + 1);
62                 sprintf(pathname + len, "%02x/", i);
63                 d = opendir(pathname);
64                 if (!d)
65                         continue;
66                 prune_dir(i, d, pathname, len + 3, opts);
67                 closedir(d);
68         }
69         stop_progress(&progress);
70 }
71
72 int cmd_prune_packed(int argc, const char **argv, const char *prefix)
73 {
74         int opts = isatty(2) ? VERBOSE : 0;
75         const struct option prune_packed_options[] = {
76                 OPT_BIT('n', "dry-run", &opts, "dry run", DRY_RUN),
77                 OPT_NEGBIT('q', "quiet", &opts, "be quiet", VERBOSE),
78                 OPT_END()
79         };
80
81         argc = parse_options(argc, argv, prefix, prune_packed_options,
82                              prune_packed_usage, 0);
83
84         prune_packed_objects(opts);
85         return 0;
86 }