X-Git-Url: http://git.bytex64.net/?a=blobdiff_plain;f=tools%2Fblergtool.c;h=ab98a942d36744c31a15b631cbb8019e26dd7aaa;hb=fd07ab46f5f37057e9290441931d5de92d62761b;hp=cc78c7dae46e99ae37044f7ebe49b08b1ef60175;hpb=96ec261b36bdbb701e05f5ee1aab70dec44085f9;p=blerg.git

diff --git a/tools/blergtool.c b/tools/blergtool.c
index cc78c7d..ab98a94 100644
--- a/tools/blergtool.c
+++ b/tools/blergtool.c
@@ -1,3 +1,6 @@
+/* Blerg is (C) 2011 The Dominion of Awesome, and is distributed under a
+ * BSD-style license.  Please see the COPYING file for details.
+ */
 #include <stdio.h>
 #include <stdlib.h>
 #include "database.h"
@@ -47,14 +50,15 @@ int main(int argc, char *argv[]) {
 
 		char *data;
 		int size;
-		blerg_fetch(f, record, &data, &size);
+		if (blerg_fetch(f, record, &data, &size)) {
+			fwrite(data, 1, size, stdout);
+			free(data);
+		}
 		blerg_close(f);
-		fwrite(data, 1, size, stdout);
-		free(data);
 	} else if (strncmp(argv[1], "list", 4) == 0) {
 		char *tag = argv[2];
 		int count = 50;
-		struct tag *list = tag_list(tag, 0, &count, -1);
+		struct blergref *list = tag_list(tag, 0, &count, -1);
 		if (list == NULL) {
 			printf("No entries");
 		} else {
@@ -64,6 +68,26 @@ int main(int argc, char *argv[]) {
 			}
 			free(list);
 		}
+	} else if (strncmp(argv[1], "mute", 4) == 0) {
+		char *store = argv[2];
+		struct blerg *f = blerg_open(store);
+		if (!f) {
+			printf("Blerg open failed\n");
+			exit(1);
+		}
+
+		blerg_set_mute(f, 1);
+		blerg_close(f);
+	} else if (strncmp(argv[1], "unmute", 6) == 0) {
+		char *store = argv[2];
+		struct blerg *f = blerg_open(store);
+		if (!f) {
+			printf("Blerg open failed\n");
+			exit(1);
+		}
+
+		blerg_set_mute(f, 0);
+		blerg_close(f);
 	} else {
 		help();
 	}