X-Git-Url: https://git.tokkee.org/?p=sysdb.git;a=blobdiff_plain;f=src%2Ftools%2Fsysdb%2Finput.c;h=2a2d1707101404169febb67d0ac643aaa98ec649;hp=52036028d2905d35555705ec026e2bc30f96c55e;hb=eeecbb1672352c4dc15af2ca345f2bb5cc5a5f66;hpb=a7596ce606438569d2d7799ec7d68831f79897da

diff --git a/src/tools/sysdb/input.c b/src/tools/sysdb/input.c
index 5203602..2a2d170 100644
--- a/src/tools/sysdb/input.c
+++ b/src/tools/sysdb/input.c
@@ -159,8 +159,8 @@ input_readline(void)
 		return (ssize_t)(sdb_strbuf_len(sysdb_input->input) - len);
 	}
 
-	if (sysdb_input->query_len)
-		prompt = "sysdb-> ";
+	if (sysdb_input->have_input)
+		prompt = "     -> ";
 	if (sdb_client_eof(sysdb_input->client))
 		prompt = "!-> ";
 
@@ -203,12 +203,12 @@ input_readline(void)
 		/* some response / error message from the server pending */
 		/* XXX: clear current line */
 		printf("\n");
-		sdb_command_print_reply(sysdb_input->client);
+		sdb_command_print_reply(sysdb_input);
 
 		if (sdb_client_eof(sysdb_input->client)) {
 			rl_callback_handler_remove();
 			/* XXX */
-			printf("Remote side closed the connection.\n");
+			sdb_log(SDB_LOG_ERR, "Remote side closed the connection.");
 			/* return EOF -> restart scanner */
 			return 0;
 		}
@@ -237,6 +237,24 @@ sdb_input_init(sdb_input_t *input)
 	return 0;
 } /* sdb_input_init */
 
+void
+sdb_input_reset(sdb_input_t *input)
+{
+	sdb_input_t reset = SDB_INPUT_INIT;
+
+	if (! input)
+		return;
+
+	if (input->client)
+		sdb_client_destroy(input->client);
+	if (input->user)
+		free(input->user);
+	if (input->input)
+		sdb_strbuf_destroy(input->input);
+
+	*input = reset;
+} /* sdb_input_reset */
+
 int
 sdb_input_mainloop(void)
 {
@@ -275,24 +293,45 @@ sdb_input_readline(char *buf, size_t *n_chars, size_t max_chars)
 int
 sdb_input_exec_query(void)
 {
-	char *query = sdb_command_exec(sysdb_input);
+	char *query = NULL;
+
+	HIST_ENTRY *hist;
+	const char *prev = NULL;
 
-	HIST_ENTRY *current_hist;
-	const char *hist_line = NULL;
+	if (! sysdb_input->have_input) {
+		/* empty line */
+		if (sdb_client_eof(sysdb_input->client))
+			sdb_input_reconnect();
+		return 0;
+	}
 
+	query = sdb_command_exec(sysdb_input);
 	if (! query)
 		return -1;
 
-	current_hist = current_history();
-	if (current_hist)
-		hist_line = current_hist->line;
+	hist = history_get(history_length);
+	if (hist)
+		prev = hist->line;
 
 	if (*query != ' ')
-		if ((! hist_line) || strcmp(hist_line, query))
+		if ((! prev) || strcmp(prev, query))
 			add_history(query);
 	free(query);
 	return 0;
 } /* sdb_input_exec_query */
 
+int
+sdb_input_reconnect(void)
+{
+	sdb_client_close(sysdb_input->client);
+	if (sdb_client_connect(sysdb_input->client, sysdb_input->user)) {
+		sdb_log(SDB_LOG_ERR, "Failed to reconnect to SysDBd");
+		return -1;
+	}
+	sdb_log(SDB_LOG_INFO, "Successfully reconnected to SysDBd");
+	sdb_command_print_server_version(sysdb_input);
+	return 0;
+} /* sdb_input_reconnect */
+
 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */