haiku/src/bin/trash.cpp

215 lines
5.0 KiB
C++
Raw Normal View History

/*
* [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 <app/Message.h>
#include <app/Messenger.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";
static const char *kTrackerSig = "application/x-vnd.Be-TRAK";
int usage(int ret)
{
printf("\nSend files to trash, or restore them.\nUsage:\n");
printf("trash [--restore|--empty|--list] file ...\n");
printf("\t--restore\trestore files (act as untrash)\n");
printf("\t--empty\t\tempty the Trash\n");
printf("\t--list\t\tlist what's already in 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];
BPath path(f);
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(path.Path(), 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(original_path, 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;
}
status_t show_trashed_file(const char *f)
{
status_t err;
char original_path[B_PATH_NAME_LENGTH];
BPath path(f);
BNode node(f);
err = node.ReadAttr(kAttrOriginalPath, B_STRING_TYPE, 0LL, original_path, B_PATH_NAME_LENGTH);
if (err < 0)
return 0;
//printf("%s\n\t[from] %s\n", f, original_path);
printf("%s\n\tas: %s\n", original_path, f);
return 0;
}
status_t foreach_in_trash(status_t (*iterator)(const char *))
{
status_t err;
dev_t dev;
char trash_dir[B_PATH_NAME_LENGTH];
for (dev = 0; ; ) {
if (next_dev(&dev) < B_OK)
break;
//for each in trash_dir
err = find_directory(B_TRASH_DIRECTORY, dev, false, trash_dir, B_PATH_NAME_LENGTH);
if (err)
continue; /* skip trashless volumes */
BDirectory trashDir(trash_dir);
err = trashDir.InitCheck();
if (err < 0)
return err;
entry_ref er;
while (trashDir.GetNextRef(&er) == B_OK) {
BPath path(&er);
if ((err = path.InitCheck()))
return err;
err = iterator(path.Path());
if (err)
return err;
}
}
return B_OK;
}
int main(int argc, char **argv)
{
int dountrash = 0;
int i = 1;
int err = 0;
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 (!strcmp(argv[1], "--restore")) {
dountrash = 1;
i++;
}
if (!dountrash && !strcmp(argv[1], "--empty")) {
/* XXX: clean that */
BMessage msg(B_DELETE_PROPERTY);
msg.AddSpecifier("Trash");
BMessenger msgr(kTrackerSig);
err = msgr.SendMessage(&msg);
if (err < 0) {
fprintf(stderr, "Emptying Trash: %s\n", strerror(err));
return 1;
}
return 0;
}
if (dountrash && !strcmp(argv[i], "--all")) {
/* restore all trashed files */
err = foreach_in_trash(untrash);
if (err) {
fprintf(stderr, "untrash: %s\n", strerror(err));
return 1;
}
return 0;
}
if (!strcmp(argv[i], "--list")) {
err = foreach_in_trash(show_trashed_file);
return 0;
}
/* restore files... */
if (dountrash) {
for (; 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;
}