tools/afcclient: Allow removing non-empty directories with -r

1 files changed, 84 insertions, 32 deletions

diff --git a/tools/afcclient.c b/tools/afcclient.c
index 3975b31..9bcd77b 100644
--- a/tools/afcclient.c
+++ b/tools/afcclient.c
@@ -114,6 +114,36 @@ static void print_usage(int argc, char **argv, int is_error)
 	);
 }
 
+#ifndef HAVE_READLINE
+#ifdef WIN32
+#define BS_CC '\b'
+#else
+#define BS_CC 0x7f
+#define getch getchar
+#endif
+static void get_input(char *buf, int maxlen)
+{
+	int len = 0;
+	int c;
+
+	while ((c = getch())) {
+		if ((c == '\r') || (c == '\n')) {
+			break;
+		}
+		if (isprint(c)) {
+			if (len < maxlen-1)
+				buf[len++] = c;
+		} else if (c == BS_CC) {
+			if (len > 0) {
+				fputs("\b \b", stdout);
+				len--;
+			}
+		}
+	}
+	buf[len] = 0;
+}
+#endif
+
 #define OPT_DOCUMENTS 1
 #define OPT_CONTAINER 2
 
@@ -576,15 +606,67 @@ static void handle_link(afc_client_t afc, int argc, char** argv)
 	}
 }
 
+static int ask_yesno(const char* prompt)
+{
+	int ret = 0;
+#ifdef HAVE_READLINE
+	char* result = readline(prompt);
+	if (result && result[0] == 'y') {
+		ret = 1;
+	}
+#else
+	char cmdbuf[2] = {0, };
+	printf("%s", prompt);
+	fflush(stdout);
+	get_input(cmdbuf, sizeof(cmdbuf));
+	if (cmdbuf[0] == 'y') {
+		ret = 1;
+	}
+#endif
+#ifdef HAVE_READLINE
+	free(result);
+#endif
+	return ret;
+}
+
 static void handle_remove(afc_client_t afc, int argc, char** argv)
 {
-	for (int i = 0; i < argc; i++) {
+	int recursive = 0;
+	int force = 0;
+	int i = 0;
+	for (i = 0; i < argc; i++) {
+		if (!strcmp(argv[i], "--")) {
+			i++;
+			break;
+		} else if (!strcmp(argv[i], "-r")) {
+			recursive = 1;
+		} else if (!strcmp(argv[i], "-f")) {
+			force = 1;
+		} else if (!strcmp(argv[i], "-rf") || !strcmp(argv[i], "-fr")) {
+			recursive = 1;
+			force = 1;
+		} else {
+			break;
+		}
+	}
+	if (recursive && !force) {
+		if (!ask_yesno("WARNING: This operation will remove all contents of the given path(s). Continue? [y/N] ")) {
+			printf("Aborted.\n");
+			return;
+		}
+	}
+	for ( ; i < argc; i++) {
 		char* abspath = get_absolute_path(argv[i]);
 		if (!abspath) {
 			printf("Error: Invalid argument '%s'\n", argv[i]);
 			continue;
 		}
-		afc_error_t err = afc_remove_path(afc, abspath);
+		afc_error_t err;
+		if (recursive) {
+			err = afc_remove_path_and_contents(afc, abspath);
+		} else {
+			err = afc_remove_path(afc, abspath);
+		}
 		if (err != AFC_E_SUCCESS) {
 			printf("Error: Failed to remove '%s': %s (%d)\n", argv[i], afc_strerror(err), err);
 		}
@@ -857,36 +939,6 @@ static void handle_cd(afc_client_t afc, int argc, char** argv)
 	curdir_len = strlen(curdir);
 }
 
-#ifndef HAVE_READLINE
-#ifdef WIN32
-#define BS_CC '\b'
-#else
-#define BS_CC 0x7f
-#define getch getchar
-#endif
-static void get_input(char *buf, int maxlen)
-{
-	int len = 0;
-	int c;
-
-	while ((c = getch())) {
-		if ((c == '\r') || (c == '\n')) {
-			break;
-		}
-		if (isprint(c)) {
-			if (len < maxlen-1)
-				buf[len++] = c;
-		} else if (c == BS_CC) {
-			if (len > 0) {
-				fputs("\b \b", stdout);
-				len--;
-			}
-		}
-	}
-	buf[len] = 0;
-}
-#endif
-
 static void parse_cmdline(int* p_argc, char*** p_argv, const char* cmdline)
 {
 	char **argv = NULL;