builtin/prune.c - jrn/git - Git at Google

 #include "cache.h"
 #include "commit.h"
 #include "diff.h"
 #include "revision.h"
 #include "builtin.h"
 #include "reachable.h"
 #include "parse-options.h"
 #include "progress.h"
 #include "dir.h"

 static const char * const prune_usage[] = {
 	N_("git prune [-n] [-v] [--expire <time>] [--] [<head>...]"),
 	NULL
 };
 static int show_only;
 static int verbose;
 static unsigned long expire;
 static int show_progress = -1;

 static int prune_tmp_file(const char *fullpath)
 {
 	struct stat st;
 	if (lstat(fullpath, &st))
 		return error("Could not stat '%s'", fullpath);
 	if (st.st_mtime > expire)
 		return 0;
 	if (show_only || verbose)
 		printf("Removing stale temporary file %s\n", fullpath);
 	if (!show_only)
 		unlink_or_warn(fullpath);
 	return 0;
 }

 static int prune_object(const unsigned char *sha1, const char *fullpath,
 			void *data)
 {
 	struct stat st;

 	/*
 	 * Do we know about this object?
 	 * It must have been reachable
 	 */
 	if (lookup_object(sha1))
 		return 0;

 	if (lstat(fullpath, &st)) {
 		/* report errors, but do not stop pruning */
 		error("Could not stat '%s'", fullpath);
 		return 0;
 	}
 	if (st.st_mtime > expire)
 		return 0;
 	if (show_only || verbose) {
 		enum object_type type = sha1_object_info(sha1, NULL);
 		printf("%s %s\n", sha1_to_hex(sha1),
 		       (type > 0) ? typename(type) : "unknown");
 	}
 	if (!show_only)
 		unlink_or_warn(fullpath);
 	return 0;
 }

 static int prune_cruft(const char *basename, const char *path, void *data)
 {
 	if (starts_with(basename, "tmp_obj_"))
 		prune_tmp_file(path);
 	else
 		fprintf(stderr, "bad sha1 file: %s\n", path);
 	return 0;
 }

 static int prune_subdir(int nr, const char *path, void *data)
 {
 	if (!show_only)
 		rmdir(path);
 	return 0;
 }

 static int prune_worktree(const char *id, struct strbuf *reason)
 {
 	struct stat st;
 	char *path;
 	int fd, len;

 	if (!is_directory(git_path("worktrees/%s", id))) {
 		strbuf_addf(reason, _("Removing worktrees/%s: not a valid directory"), id);
 		return 1;
 	}
 	if (file_exists(git_path("worktrees/%s/locked", id)))
 		return 0;
 	if (stat(git_path("worktrees/%s/gitdir", id), &st)) {
 		strbuf_addf(reason, _("Removing worktrees/%s: gitdir file does not exist"), id);
 		return 1;
 	}
 	fd = open(git_path("worktrees/%s/gitdir", id), O_RDONLY);
 	if (fd < 0) {
 		strbuf_addf(reason, _("Removing worktrees/%s: unable to read gitdir file (%s)"),
 			    id, strerror(errno));
 		return 1;
 	}
 	len = st.st_size;
 	path = xmalloc(len + 1);
 	read_in_full(fd, path, len);
 	close(fd);
 	while (len && (path[len - 1] == '\n' || path[len - 1] == '\r'))
 		len--;
 	if (!len) {
 		strbuf_addf(reason, _("Removing worktrees/%s: invalid gitdir file"), id);
 		free(path);
 		return 1;
 	}
 	path[len] = '\0';
 	if (!file_exists(path)) {
 		struct stat st_link;
 		free(path);
 		/*
 		 * the repo is moved manually and has not been
 		 * accessed since?
 		 */
 		if (!stat(git_path("worktrees/%s/link", id), &st_link) &&
 		    st_link.st_nlink > 1)
 			return 0;
 		if (st.st_mtime <= expire) {
 			strbuf_addf(reason, _("Removing worktrees/%s: gitdir file points to non-existent location"), id);
 			return 1;
 		} else {
 			return 0;
 		}
 	}
 	free(path);
 	return 0;
 }

 static void prune_worktrees(void)
 {
 	struct strbuf reason = STRBUF_INIT;
 	struct strbuf path = STRBUF_INIT;
 	DIR *dir = opendir(git_path("worktrees"));
 	struct dirent *d;
 	int ret;
 	if (!dir)
 		return;
 	while ((d = readdir(dir)) != NULL) {
 		if (!strcmp(d->d_name, ".") || !strcmp(d->d_name, ".."))
 			continue;
 		strbuf_reset(&reason);
 		if (!prune_worktree(d->d_name, &reason))
 			continue;
 		if (show_only || verbose)
 			printf("%s\n", reason.buf);
 		if (show_only)
 			continue;
 		strbuf_reset(&path);
 		strbuf_addstr(&path, git_path("worktrees/%s", d->d_name));
 		ret = remove_dir_recursively(&path, 0);
 		if (ret < 0 && errno == ENOTDIR)
 			ret = unlink(path.buf);
 		if (ret)
 			error(_("failed to remove: %s"), strerror(errno));
 	}
 	closedir(dir);
 	if (!show_only)
 		rmdir(git_path("worktrees"));
 	strbuf_release(&reason);
 	strbuf_release(&path);
 }

 /*
  * Write errors (particularly out of space) can result in
  * failed temporary packs (and more rarely indexes and other
  * files beginning with "tmp_") accumulating in the object
  * and the pack directories.
  */
 static void remove_temporary_files(const char *path)
 {
 	DIR *dir;
 	struct dirent *de;

 	dir = opendir(path);
 	if (!dir) {
 		fprintf(stderr, "Unable to open directory %s\n", path);
 		return;
 	}
 	while ((de = readdir(dir)) != NULL)
 		if (starts_with(de->d_name, "tmp_"))
 			prune_tmp_file(mkpath("%s/%s", path, de->d_name));
 	closedir(dir);
 }

 int cmd_prune(int argc, const char **argv, const char *prefix)
 {
 	struct rev_info revs;
 	struct progress *progress = NULL;
 	int do_prune_worktrees = 0;
 	const struct option options[] = {
 		OPT__DRY_RUN(&show_only, N_("do not remove, show only")),
 		OPT__VERBOSE(&verbose, N_("report pruned objects")),
 		OPT_BOOL(0, "progress", &show_progress, N_("show progress")),
 		OPT_BOOL(0, "worktrees", &do_prune_worktrees, N_("prune .git/worktrees")),
 		OPT_EXPIRY_DATE(0, "expire", &expire,
 				N_("expire objects older than <time>")),
 		OPT_END()
 	};
 	char *s;

 	expire = ULONG_MAX;
 	save_commit_buffer = 0;
 	check_replace_refs = 0;
 	ref_paranoia = 1;
 	init_revisions(&revs, prefix);

 	argc = parse_options(argc, argv, prefix, options, prune_usage, 0);

 	if (do_prune_worktrees) {
 		if (argc)
 			die(_("--worktrees does not take extra arguments"));
 		prune_worktrees();
 		return 0;
 	}

 	while (argc--) {
 		unsigned char sha1[20];
 		const char *name = *argv++;

 		if (!get_sha1(name, sha1)) {
 			struct object *object = parse_object_or_die(sha1, name);
 			add_pending_object(&revs, object, "");
 		}
 		else
 			die("unrecognized argument: %s", name);
 	}

 	if (show_progress == -1)
 		show_progress = isatty(2);
 	if (show_progress)
 		progress = start_progress_delay(_("Checking connectivity"), 0, 0, 2);

 	mark_reachable_objects(&revs, 1, expire, progress);
 	stop_progress(&progress);
 	for_each_loose_file_in_objdir(get_object_directory(), prune_object,
 				      prune_cruft, prune_subdir, NULL);

 	prune_packed_objects(show_only ? PRUNE_PACKED_DRY_RUN : 0);
 	remove_temporary_files(get_object_directory());
 	s = mkpathdup("%s/pack", get_object_directory());
 	remove_temporary_files(s);
 	free(s);

 	if (is_repository_shallow())
 		prune_shallow(show_only);

 	return 0;
 }
	#include "cache.h"
	#include "commit.h"
	#include "diff.h"
	#include "revision.h"
	#include "builtin.h"
	#include "reachable.h"
	#include "parse-options.h"
	#include "progress.h"
	#include "dir.h"

	static const char * const prune_usage[] = {
	N_("git prune [-n] [-v] [--expire <time>] [--] [<head>...]"),
	NULL
	};
	static int show_only;
	static int verbose;
	static unsigned long expire;
	static int show_progress = -1;

	static int prune_tmp_file(const char *fullpath)
	{
	struct stat st;
	if (lstat(fullpath, &st))
	return error("Could not stat '%s'", fullpath);
	if (st.st_mtime > expire)
	return 0;
	if (show_only \|\| verbose)
	printf("Removing stale temporary file %s\n", fullpath);
	if (!show_only)
	unlink_or_warn(fullpath);
	return 0;
	}

	static int prune_object(const unsigned char sha1, const char fullpath,
	void *data)
	{
	struct stat st;

	/*
	* Do we know about this object?
	* It must have been reachable
	*/
	if (lookup_object(sha1))
	return 0;

	if (lstat(fullpath, &st)) {
	/* report errors, but do not stop pruning */
	error("Could not stat '%s'", fullpath);
	return 0;
	}
	if (st.st_mtime > expire)
	return 0;
	if (show_only \|\| verbose) {
	enum object_type type = sha1_object_info(sha1, NULL);
	printf("%s %s\n", sha1_to_hex(sha1),
	(type > 0) ? typename(type) : "unknown");
	}
	if (!show_only)
	unlink_or_warn(fullpath);
	return 0;
	}

	static int prune_cruft(const char basename, const char path, void *data)
	{
	if (starts_with(basename, "tmp_obj_"))
	prune_tmp_file(path);
	else
	fprintf(stderr, "bad sha1 file: %s\n", path);
	return 0;
	}

	static int prune_subdir(int nr, const char path, void data)
	{
	if (!show_only)
	rmdir(path);
	return 0;
	}

	static int prune_worktree(const char id, struct strbuf reason)
	{
	struct stat st;
	char *path;
	int fd, len;

	if (!is_directory(git_path("worktrees/%s", id))) {
	strbuf_addf(reason, _("Removing worktrees/%s: not a valid directory"), id);
	return 1;
	}
	if (file_exists(git_path("worktrees/%s/locked", id)))
	return 0;
	if (stat(git_path("worktrees/%s/gitdir", id), &st)) {
	strbuf_addf(reason, _("Removing worktrees/%s: gitdir file does not exist"), id);
	return 1;
	}
	fd = open(git_path("worktrees/%s/gitdir", id), O_RDONLY);
	if (fd < 0) {
	strbuf_addf(reason, _("Removing worktrees/%s: unable to read gitdir file (%s)"),
	id, strerror(errno));
	return 1;
	}
	len = st.st_size;
	path = xmalloc(len + 1);
	read_in_full(fd, path, len);
	close(fd);
	while (len && (path[len - 1] == '\n' \|\| path[len - 1] == '\r'))
	len--;
	if (!len) {
	strbuf_addf(reason, _("Removing worktrees/%s: invalid gitdir file"), id);
	free(path);
	return 1;
	}
	path[len] = '\0';
	if (!file_exists(path)) {
	struct stat st_link;
	free(path);
	/*
	* the repo is moved manually and has not been
	* accessed since?
	*/
	if (!stat(git_path("worktrees/%s/link", id), &st_link) &&
	st_link.st_nlink > 1)
	return 0;
	if (st.st_mtime <= expire) {
	strbuf_addf(reason, _("Removing worktrees/%s: gitdir file points to non-existent location"), id);
	return 1;
	} else {
	return 0;
	}
	}
	free(path);
	return 0;
	}

	static void prune_worktrees(void)
	{
	struct strbuf reason = STRBUF_INIT;
	struct strbuf path = STRBUF_INIT;
	DIR *dir = opendir(git_path("worktrees"));
	struct dirent *d;
	int ret;
	if (!dir)
	return;
	while ((d = readdir(dir)) != NULL) {
	if (!strcmp(d->d_name, ".") \|\| !strcmp(d->d_name, ".."))
	continue;
	strbuf_reset(&reason);
	if (!prune_worktree(d->d_name, &reason))
	continue;
	if (show_only \|\| verbose)
	printf("%s\n", reason.buf);
	if (show_only)
	continue;
	strbuf_reset(&path);
	strbuf_addstr(&path, git_path("worktrees/%s", d->d_name));
	ret = remove_dir_recursively(&path, 0);
	if (ret < 0 && errno == ENOTDIR)
	ret = unlink(path.buf);
	if (ret)
	error(_("failed to remove: %s"), strerror(errno));
	}
	closedir(dir);
	if (!show_only)
	rmdir(git_path("worktrees"));
	strbuf_release(&reason);
	strbuf_release(&path);
	}

	/*
	* Write errors (particularly out of space) can result in
	* failed temporary packs (and more rarely indexes and other
	* files beginning with "tmp_") accumulating in the object
	* and the pack directories.
	*/
	static void remove_temporary_files(const char *path)
	{
	DIR *dir;
	struct dirent *de;

	dir = opendir(path);
	if (!dir) {
	fprintf(stderr, "Unable to open directory %s\n", path);
	return;
	}
	while ((de = readdir(dir)) != NULL)
	if (starts_with(de->d_name, "tmp_"))
	prune_tmp_file(mkpath("%s/%s", path, de->d_name));
	closedir(dir);
	}

	int cmd_prune(int argc, const char *argv, const char prefix)
	{
	struct rev_info revs;
	struct progress *progress = NULL;
	int do_prune_worktrees = 0;
	const struct option options[] = {
	OPT__DRY_RUN(&show_only, N_("do not remove, show only")),
	OPT__VERBOSE(&verbose, N_("report pruned objects")),
	OPT_BOOL(0, "progress", &show_progress, N_("show progress")),
	OPT_BOOL(0, "worktrees", &do_prune_worktrees, N_("prune .git/worktrees")),
	OPT_EXPIRY_DATE(0, "expire", &expire,
	N_("expire objects older than <time>")),
	OPT_END()
	};
	char *s;

	expire = ULONG_MAX;
	save_commit_buffer = 0;
	check_replace_refs = 0;
	ref_paranoia = 1;
	init_revisions(&revs, prefix);

	argc = parse_options(argc, argv, prefix, options, prune_usage, 0);

	if (do_prune_worktrees) {
	if (argc)
	die(_("--worktrees does not take extra arguments"));
	prune_worktrees();
	return 0;
	}

	while (argc--) {
	unsigned char sha1[20];
	const char name = argv++;

	if (!get_sha1(name, sha1)) {
	struct object *object = parse_object_or_die(sha1, name);
	add_pending_object(&revs, object, "");
	}
	else
	die("unrecognized argument: %s", name);
	}

	if (show_progress == -1)
	show_progress = isatty(2);
	if (show_progress)
	progress = start_progress_delay(_("Checking connectivity"), 0, 0, 2);

	mark_reachable_objects(&revs, 1, expire, progress);
	stop_progress(&progress);
	for_each_loose_file_in_objdir(get_object_directory(), prune_object,
	prune_cruft, prune_subdir, NULL);

	prune_packed_objects(show_only ? PRUNE_PACKED_DRY_RUN : 0);
	remove_temporary_files(get_object_directory());
	s = mkpathdup("%s/pack", get_object_directory());
	remove_temporary_files(s);
	free(s);

	if (is_repository_shallow())
	prune_shallow(show_only);

	return 0;
	}