Merge branch 'mh/refs-have-new'
[gitweb.git] / builtin / prune-packed.c
index 6879468c46a7baca69e5e28b0c6e38ff7099a6ff..7cf900ea0765e334f02df5600f0911d41563a027 100644 (file)
@@ -4,69 +4,47 @@
 #include "parse-options.h"
 
 static const char * const prune_packed_usage[] = {
-       N_("git prune-packed [-n|--dry-run] [-q|--quiet]"),
+       N_("git prune-packed [-n | --dry-run] [-q | --quiet]"),
        NULL
 };
 
 static struct progress *progress;
 
-static void prune_dir(int i, DIR *dir, struct strbuf *pathname, int opts)
+static int prune_subdir(int nr, const char *path, void *data)
 {
-       struct dirent *de;
-       char hex[40];
-       int top_len = pathname->len;
+       int *opts = data;
+       display_progress(progress, nr + 1);
+       if (!(*opts & PRUNE_PACKED_DRY_RUN))
+               rmdir(path);
+       return 0;
+}
+
+static int prune_object(const unsigned char *sha1, const char *path,
+                        void *data)
+{
+       int *opts = data;
 
-       sprintf(hex, "%02x", i);
-       while ((de = readdir(dir)) != NULL) {
-               unsigned char sha1[20];
-               if (strlen(de->d_name) != 38)
-                       continue;
-               memcpy(hex + 2, de->d_name, 38);
-               if (get_sha1_hex(hex, sha1))
-                       continue;
-               if (!has_sha1_pack(sha1))
-                       continue;
+       if (!has_sha1_pack(sha1))
+               return 0;
 
-               strbuf_add(pathname, de->d_name, 38);
-               if (opts & PRUNE_PACKED_DRY_RUN)
-                       printf("rm -f %s\n", pathname->buf);
-               else
-                       unlink_or_warn(pathname->buf);
-               display_progress(progress, i + 1);
-               strbuf_setlen(pathname, top_len);
-       }
+       if (*opts & PRUNE_PACKED_DRY_RUN)
+               printf("rm -f %s\n", path);
+       else
+               unlink_or_warn(path);
+       return 0;
 }
 
 void prune_packed_objects(int opts)
 {
-       int i;
-       const char *dir = get_object_directory();
-       struct strbuf pathname = STRBUF_INIT;
-       int top_len;
-
-       strbuf_addstr(&pathname, dir);
        if (opts & PRUNE_PACKED_VERBOSE)
                progress = start_progress_delay(_("Removing duplicate objects"),
                        256, 95, 2);
 
-       if (pathname.len && pathname.buf[pathname.len - 1] != '/')
-               strbuf_addch(&pathname, '/');
-
-       top_len = pathname.len;
-       for (i = 0; i < 256; i++) {
-               DIR *d;
+       for_each_loose_file_in_objdir(get_object_directory(),
+                                     prune_object, NULL, prune_subdir, &opts);
 
-               display_progress(progress, i + 1);
-               strbuf_setlen(&pathname, top_len);
-               strbuf_addf(&pathname, "%02x/", i);
-               d = opendir(pathname.buf);
-               if (!d)
-                       continue;
-               prune_dir(i, d, &pathname, opts);
-               closedir(d);
-               strbuf_setlen(&pathname, top_len + 2);
-               rmdir(pathname.buf);
-       }
+       /* Ensure we show 100% before finishing progress */
+       display_progress(progress, 256);
        stop_progress(&progress);
 }