Move window size handling to cmdedit.c. Move prompt setup to setup_prompt_string()

1 files changed, 33 insertions, 53 deletions

diff --git a/sh.c b/sh.c
index 1457116..dd1d3aa 100644
--- a/sh.c
+++ b/sh.c
@@ -184,10 +184,8 @@ static struct built_in_command bltins_forking[] = {
 	{NULL, NULL, NULL}
 };
 
-static char prompt[3];
 static char *cwd;
 static char *local_pending_command = NULL;
-static char *prompt_str = NULL;
 static struct jobset job_list = { NULL, NULL };
 static int argc;
 static char **argv;
@@ -212,19 +210,6 @@ static inline void debug_printf(const char *format, ...)
 static inline void debug_printf(const char *format, ...) { }
 #endif
 
-#ifdef BB_FEATURE_SH_COMMAND_EDITING
-static inline void win_changed(int junk)
-{
-	struct winsize win = { 0, 0, 0, 0 };
-	ioctl(0, TIOCGWINSZ, &win);
-	if (win.ws_col > 0) {
-		cmdedit_setwidth( win.ws_col - 1);
-	}
-}
-#else
-static inline void win_changed(int junk) {}
-#endif
-
 /*
 	Most builtins need access to the struct child_prog that has
 	their arguments, previously coded as cmd->progs[0].  That coding
@@ -743,26 +728,18 @@ static void restore_redirects(int squirrel[])
 	}
 }
 
-static int get_command(FILE * source, char *command)
+static char* setup_prompt_string(int state)
 {
 	char user[9],buf[255],*s;
-	
-	if (source == NULL) {
-		if (local_pending_command) {
-			/* a command specified (-c option): return it & mark it done */
-			strcpy(command, local_pending_command);
-			free(local_pending_command);
-			local_pending_command = NULL;
-			return 0;
-		}
-		return 1;
-	}
+	char prompt[3];
+	char prompt_str[BUFSIZ];
 
-	if (shell_context == 0) {
+	/* Set up the prompt */
+	if (state == 0) {
 		/* get User Name and setup prompt */
 		strcpy(prompt,( geteuid() != 0 ) ? "$ ":"# ");
 		my_getpwuid(user, geteuid());
-		
+
 		/* get HostName */
 		gethostname(buf, 255);
 		s = strchr(buf, '.');
@@ -772,11 +749,35 @@ static int get_command(FILE * source, char *command)
 	} else {
 		strcpy(prompt,"> ");
 	}
+
+	if (state == 0) {
+		snprintf(prompt_str, BUFSIZ-1, "[%s@%s %s]%s", user, buf, 
+				get_last_path_component(cwd), prompt);
+	} else {
+		sprintf(prompt_str, "%s", prompt);
+	}
+	return(strdup(prompt_str));  /* Must free this memory */
+}
+
+static int get_command(FILE * source, char *command)
+{
+	char *prompt_str;
+	
+	if (source == NULL) {
+		if (local_pending_command) {
+			/* a command specified (-c option): return it & mark it done */
+			strcpy(command, local_pending_command);
+			free(local_pending_command);
+			local_pending_command = NULL;
+			return 0;
+		}
+		return 1;
+	}
+
+	prompt_str = setup_prompt_string(shell_context);
 	
 	if (source == stdin) {
 #ifdef BB_FEATURE_SH_COMMAND_EDITING
-		int len;
-
 		/*
 		** enable command line editing only while a command line
 		** is actually being read; otherwise, we'll end up bequeathing
@@ -784,32 +785,12 @@ static int get_command(FILE * source, char *command)
 		** child processes (rob@sysgo.de)
 		*/
 		cmdedit_init();
-		signal(SIGWINCH, win_changed);
-		debug_printf( "in get_command() -- job_context=%d\n", shell_context);
-		fflush(stdout);
-		if (shell_context == 0) {
-			len=fprintf(stdout, "[%s@%s %s]%s", user, buf, 
-					get_last_path_component(cwd), prompt);
-		} else {
-			len=fprintf(stdout, "%s", prompt);
-		}
-		fflush(stdout);
-		prompt_str=(char*)xmalloc(sizeof(char)*(len+1));
-		if (shell_context == 0) {
-			sprintf(prompt_str, "[%s@%s %s]%s", user, buf, 
-					get_last_path_component(cwd), prompt);
-		} else {
-			sprintf(prompt_str, "%s", prompt);
-		}
 		cmdedit_read_input(prompt_str, command);
 		free( prompt_str);
 		cmdedit_terminate();
-		signal(SIGWINCH, SIG_DFL);
 		return 0;
 #else
-		fprintf(stdout, "[%s@%s %s]%s",user, buf, 
-				get_last_path_component(cwd), prompt);
-		fflush(stdout);
+		fprintf(stdout, "%s", prompt_str);
 #endif
 	}
 
@@ -1712,6 +1693,5 @@ int shell_main(int argc_l, char **argv_l)
 	atexit(free_memory);
 #endif
 
-	win_changed(0);
 	return (busy_loop(input));
 }