[un]trash command, to trash (eh) files, and get them back. Could be useful with alias rm=trash :)

git-svn-id: file:///srv/svn/repos/haiku/trunk/current@10488 a95241bf-73f2-0310-859d-f6bbb57e9c96
2004-12-19 00:17:36 +00:00 · 2004-12-19 00:17:36 +00:00 · b03a7f8292
commit b03a7f8292
parent f0cb8f7257
1 changed files with 151 additions and 0 deletions
--- a/src/apps/bin/trash.cpp
+++ b/src/apps/bin/trash.cpp
@ -0,0 +1,151 @@
+/*
+ * [un]trash command for Haiku
+ * Copyright (c) 2004, Francois Revol - revol@free.fr
+ * provided under the MIT licence
+ */
+
+#include <stdio.h>
+#include <string.h>
+#include <unistd.h>
+#include <kernel/fs_attr.h>
+#include <kernel/fs_info.h>
+#include <storage/Directory.h>
+#include <storage/Entry.h>
+#include <storage/FindDirectory.h>
+#include <storage/Node.h>
+#include <storage/Path.h>
+#include <support/TypeConstants.h>
+
+static const char *kAttrOriginalPath = "_trk/original_path";
+
+int usage(int ret)
+{
+	printf("\nSend files to trash, or restore them.\nUsage:\n");
+	printf("trash [--empty] file ...\n");
+	printf("\t--empty\tempty the Trash\n");
+	printf("untrash [--all] [file ...]\n");
+	//printf("restore [--all] [file ...]\n");
+	return ret;
+}
+
+status_t untrash(const char *f)
+{
+	status_t err;
+	char original_path[B_PATH_NAME_LENGTH];
+	BNode node(f);
+	err = node.InitCheck();
+	if (err)
+		return err;
+	err = node.ReadAttr(kAttrOriginalPath, B_STRING_TYPE, 0LL, original_path, B_PATH_NAME_LENGTH);
+	if (err < 0)
+		return err;
+	err = rename(f, original_path);
+	if (err < 0)
+		return err;
+	node.RemoveAttr(kAttrOriginalPath);
+	return 0;
+}
+
+status_t trash(const char *f)
+{
+	status_t err;
+	attr_info ai;
+	dev_t dev = -1;
+	int nr;
+	const char *original_path;
+	char trash_dir[B_PATH_NAME_LENGTH];
+	char trashed_file[B_PATH_NAME_LENGTH];
+	dev = dev_for_path(f);
+	err = find_directory(B_TRASH_DIRECTORY, dev, false, trash_dir, B_PATH_NAME_LENGTH);
+	if (err < 0)
+		return err;
+	BNode node(f);
+	err = node.InitCheck();
+	if (err < 0)
+		return err;
+	err = node.GetAttrInfo(kAttrOriginalPath, &ai);
+	if (err == B_OK)
+		return EALREADY;
+	if (!strncmp(f, trash_dir, strlen(trash_dir)))
+		return EALREADY;
+	entry_ref er;
+	err = get_ref_for_path(f, &er);
+	BPath orgPath(&er);
+	err = orgPath.InitCheck();
+	if (err < 0)
+		return err;
+	original_path = orgPath.Path();
+	BDirectory trashDir(trash_dir);
+	err = trashDir.InitCheck();
+	if (err < 0)
+		return err;
+	for (nr = 0; ; nr++) {
+		if (nr > INT_MAX - 1)
+			return B_ERROR;
+		if (nr)
+			snprintf(trashed_file, B_PATH_NAME_LENGTH-1, "%s/%s %d", trash_dir, er.name, nr);
+		else
+			snprintf(trashed_file, B_PATH_NAME_LENGTH-1, "%s/%s", trash_dir, er.name);
+		if (!trashDir.Contains(trashed_file))
+			break;
+	}
+	err = rename(f, trashed_file);
+	if (err < 0)
+		return err;
+	
+	err = node.WriteAttr(kAttrOriginalPath, B_STRING_TYPE, 0LL, original_path, strlen(original_path)+1);
+	if (err < 0)
+		return err;
+	return 0;
+}
+
+int main(int argc, char **argv)
+{
+	int dountrash = 0;
+	int i;
+	int err = 0;
+	int fd;
+	char trash_dir[B_PATH_NAME_LENGTH];
+	if (strstr(argv[0], "untrash") || strstr(argv[0], "restore"))
+		dountrash = 1;
+	if (argc < 2)
+		return usage(1);
+	if (!strcmp(argv[1], "--help"))
+		return usage(0);
+	if (!dountrash && !strcmp(argv[1], "--empty")) {
+		/* XXX: clean that */
+		if (system("hey Tracker delete Trash"))
+			return 1;
+		return 0;
+	}
+	if (dountrash && !strcmp(argv[1], "--all")) {
+		/* restore all trashed files */
+		//for each in all volumes
+		
+			//for each in trash_dir
+			
+		return 0;
+	}
+	/* restore files... */
+	if (dountrash) {
+		for (i = 1; i < argc; i++) {
+			err = untrash(argv[i]);
+			if (err) {
+				fprintf(stderr, "%s: %s\n", argv[i], strerror(err));
+				return 1;
+			}
+		}
+		return err;
+	}
+	/* or trash them */
+	for (i = 1; i < argc; i++) {
+		err = trash(argv[i]);
+		if (err) {
+			fprintf(stderr, "%s: %s\n", argv[i], strerror(err));
+			return 1;
+		}
+	}
+	
+	return err;
+
+}