Moved bin/ directory out of apps/.

git-svn-id: file:///srv/svn/repos/haiku/haiku/trunk@11909 a95241bf-73f2-0310-859d-f6bbb57e9c96

src/bin/query.cpp

@@ -0,0 +1,151 @@
+// query.cpp
+//
+// A shell utility for somewhat emulating the Tracker's "Find By Formula"
+// functionality.
+//
+// by Ficus Kirkpatrick ([email protected])
+//
+// Modified by Jerome Duval on November 03, 2003
+//
+
+
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <unistd.h>
+
+#include <storage/Path.h>
+#include <storage/Query.h>
+#include <storage/Entry.h>
+#include <storage/Volume.h>
+#include <storage/VolumeRoster.h>
+#include <support/SupportDefs.h>
+#include <support/String.h>
+
+
+// Option variables.
+bool o_all_volumes = false;       // Query all volumes?
+bool o_escaping = true;       // Escape metacharacters?
+
+extern const char *__progname;
+
+
+void
+usage(void)
+{
+	printf("usage: %s [ -e ] [ -a || -v <path-to-volume> ] expression\n"
+		"  -e\t\tdon't escape meta-characters\n"
+		"  -a\t\tperform the query on all volumes\n"
+		"  -v <file>\tperform the query on just one volume; <file> can be any\n"
+		"\t\tfile on that volume. Defaults to the current volume.\n"
+		"  Hint: '%s name=foo' will find files named \"foo\"\n",
+		__progname, __progname);
+	exit(0);
+}
+
+
+void
+perform_query(BVolume &volume, const char *predicate)
+{
+	BQuery query;
+
+	// Set up the volume and predicate for the query.
+	query.SetVolume(&volume);
+	query.SetPredicate(predicate);
+
+	status_t status = query.Fetch();
+	if (status == B_BAD_VALUE) {
+		// the "name=" part may be omitted in our arguments
+		BString string = "name=";
+		string << predicate;
+
+		query.SetPredicate(string.String());
+		status = query.Fetch();
+	}
+	if (status != B_OK) {
+		printf("query: bad query expression\n");
+		return;
+	}
+
+	BEntry entry;
+	BPath path;
+	while (query.GetNextEntry(&entry) == B_OK) {
+		if (entry.GetPath(&path) < B_OK) {
+			fprintf(stderr, "%s: could not get path for entry\n", __progname);
+			continue;
+		}
+
+		printf("%s\n", o_escaping ? 
+			BString().CharacterEscape(path.Path(), " ()?*&\"'[]^\\~|;!<>*$", '\\').String()
+			: path.Path());
+	}
+}
+
+
+int
+main(int32 argc, char **argv)
+{
+	// Make sure we have the minimum number of arguments.
+	if (argc < 2)
+		usage();	
+
+	// Which volume do we make the query on?
+	// Default to the current volume.
+	char volumePath[B_FILE_NAME_LENGTH];
+	strcpy(volumePath, ".");
+
+	// Parse command-line arguments.
+	int opt;
+	while ((opt = getopt(argc, argv, "eav:")) != -1) {
+		switch(opt) {
+			case 'a':
+				o_all_volumes = true;
+				break;
+			case 'e':
+				o_escaping = false;
+				break;
+			case 'v':
+				strncpy(volumePath, optarg, B_FILE_NAME_LENGTH);
+				break;
+
+			default:
+				usage();
+				break;
+		}
+	}
+
+	BVolume volume;
+
+	if (!o_all_volumes) {
+		// Find the volume that the query should be performed on,
+		// and set the query to it.
+		BEntry entry(volumePath);
+		if (entry.InitCheck() != B_OK) {
+			fprintf(stderr, "%s: \"%s\" is not a valid file\n", __progname, volumePath);
+			exit(1);
+		}
+
+		status_t status = entry.GetVolume(&volume);
+		if (status != B_OK) {
+			fprintf(stderr, "%s: could not get volume: %s\n", __progname, strerror(status));
+			exit(1);
+		}
+
+		if (!volume.KnowsQuery())
+			fprintf(stderr, "%s: volume containing %s is not query-enabled\n", __progname, volumePath);
+		else
+			perform_query(volume, argv[optind]);
+	} else {	
+		// Okay, we want to query all the disks -- so iterate over
+		// them, one by one, running the query.
+		BVolumeRoster volumeRoster;
+		while (volumeRoster.GetNextVolume(&volume) == B_OK) {
+			// We don't print errors here -- this will catch /pipe and
+			// other filesystems we don't care about.
+			if (volume.KnowsQuery())
+				perform_query(volume, argv[optind]);
+		}
+	}
+
+	return 0;
+}