-#include "cache.h"
+#include "object-store.h"
+#include "packfile.h"
+#include "progress.h"
+#include "prune-packed.h"
-static const char prune_packed_usage[] =
-"git-prune-packed [-n]";
+static struct progress *progress;
-static int dryrun;
-
-static void prune_dir(int i, DIR *dir, char *pathname, int len)
+static int prune_subdir(unsigned int nr, const char *path, void *data)
{
- struct dirent *de;
- char hex[40];
-
- 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;
- memcpy(pathname + len, de->d_name, 38);
- if (dryrun)
- printf("rm -f %s\n", pathname);
- else if (unlink(pathname) < 0)
- error("unable to unlink %s", pathname);
- }
- pathname[len] = 0;
- rmdir(pathname);
+ int *opts = data;
+ display_progress(progress, nr + 1);
+ if (!(*opts & PRUNE_PACKED_DRY_RUN))
+ rmdir(path);
+ return 0;
}
-static void prune_packed_objects(void)
+static int prune_object(const struct object_id *oid, const char *path,
+ void *data)
{
- int i;
- static char pathname[PATH_MAX];
- const char *dir = get_object_directory();
- int len = strlen(dir);
+ int *opts = data;
- if (len > PATH_MAX - 42)
- die("impossible object directory");
- memcpy(pathname, dir, len);
- if (len && pathname[len-1] != '/')
- pathname[len++] = '/';
- for (i = 0; i < 256; i++) {
- DIR *d;
+ if (!has_object_pack(oid))
+ return 0;
- sprintf(pathname + len, "%02x/", i);
- d = opendir(pathname);
- if (!d)
- continue;
- prune_dir(i, d, pathname, len + 3);
- closedir(d);
- }
+ if (*opts & PRUNE_PACKED_DRY_RUN)
+ printf("rm -f %s\n", path);
+ else
+ unlink_or_warn(path);
+ return 0;
}
-int main(int argc, char **argv)
+void prune_packed_objects(int opts)
{
- int i;
-
- setup_git_directory();
+ if (opts & PRUNE_PACKED_VERBOSE)
+ progress = start_delayed_progress(_("Removing duplicate objects"), 256);
- for (i = 1; i < argc; i++) {
- const char *arg = argv[i];
+ for_each_loose_file_in_objdir(get_object_directory(),
+ prune_object, NULL, prune_subdir, &opts);
- if (*arg == '-') {
- if (!strcmp(arg, "-n"))
- dryrun = 1;
- else
- usage(prune_packed_usage);
- continue;
- }
- /* Handle arguments here .. */
- usage(prune_packed_usage);
- }
- sync();
- prune_packed_objects();
- return 0;
+ /* Ensure we show 100% before finishing progress */
+ display_progress(progress, 256);
+ stop_progress(&progress);
}