From d8b2df1479e8c73080be859c3b0d7b7d2b02bf0e Mon Sep 17 00:00:00 2001
From: "C. McEnroe" <june@causal.agency>
Date: Wed, 3 Mar 2021 17:37:13 -0500
Subject: Add -t flag to truncate stdout before writing paths

---
 archive.c | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

(limited to 'archive.c')

diff --git a/archive.c b/archive.c
index 1ef4f33..91c6118 100644
--- a/archive.c
+++ b/archive.c
@@ -82,6 +82,7 @@ static void createDirs(void) {
 
 int main(int argc, char *argv[]) {
 	int exitStatus = 0;
+	bool truncate = false;
 
 	const char *host = NULL;
 	const char *port = "imaps";
@@ -95,7 +96,7 @@ int main(int argc, char *argv[]) {
 
 	for (
 		int opt;
-		0 < (opt = getopt(argc, argv, "C:H:S:a:h:im:p:qs:t:u:vw:y:"));
+		0 < (opt = getopt(argc, argv, "C:H:S:T:a:h:im:p:qs:tu:vw:y:"));
 	) {
 		switch (opt) {
 			break; case 'C': {
@@ -104,6 +105,7 @@ int main(int argc, char *argv[]) {
 			}
 			break; case 'H': concatHead = optarg;
 			break; case 'S': search = optarg;
+			break; case 'T': baseTitle = optarg;
 			break; case 'a': algo = optarg;
 			break; case 'h': host = optarg;
 			break; case 'i': idle = true;
@@ -111,7 +113,7 @@ int main(int argc, char *argv[]) {
 			break; case 'p': port = optarg;
 			break; case 'q': quiet = true; exitStatus = EXIT_FAILURE;
 			break; case 's': baseSubscribe = optarg;
-			break; case 't': baseTitle = optarg;
+			break; case 't': truncate = true;
 			break; case 'u': baseURL = optarg;
 			break; case 'v': imapVerbose = true;
 			break; case 'w': passPath = optarg;
@@ -177,7 +179,13 @@ int main(int argc, char *argv[]) {
 	for (; resp = respOk(imapResp(&imap)), resp.tag != login; respFree(resp));
 	respFree(resp);
 
-examine:;
+examine:
+	if (truncate) {
+		int error = ftruncate(STDOUT_FILENO, 0);
+		if (error) warn("ftruncate");
+		rewind(stdout);
+	}
+
 	uint32_t uidNext = 0;
 	uint32_t uidValidity = 0;
 	enum Atom examine = atom("examine");
@@ -259,6 +267,7 @@ concat:;
 
 	concatThreads(threads, envelopes);
 	concatIndex(threads, envelopes);
+	fflush(stdout);
 	uidWrite("UIDNEXT", uidNext);
 
 	for (size_t i = 0; i < threads.len; ++i) {
-- 
cgit 1.4.1