pass 3st stage

1 files changed, 9 insertions, 7 deletions

diff --git a/cmd/myshell/main.go b/cmd/myshell/main.go
index 65ed355..f9ca0ee 100644
--- a/cmd/myshell/main.go
+++ b/cmd/myshell/main.go
@@ -8,13 +8,15 @@ import (
 )
 
 func main() {
-	fmt.Fprint(os.Stdout, "$ ")
+	stdin := bufio.NewReader(os.Stdin)
+	for {
+		fmt.Fprint(os.Stdout, "$ ")
 
-	// Wait for user input
-	in, err := bufio.NewReader(os.Stdin).ReadString('\n')
-	if err != nil {
-		fmt.Println(err.Error())
-	}
+		in, err := stdin.ReadString('\n')
+		if err != nil {
+			fmt.Println(err.Error())
+		}
 
-	fmt.Printf("%s: command not found\n", strings.TrimRight(in, "\n"))
+		fmt.Printf("%s: command not found\n", strings.TrimRight(in, "\n"))
+	}
 }