From 3915b4307b7501828f69588cdf2068cd5da3e50e Mon Sep 17 00:00:00 2001
From: Joop Kiefte <ikojba@gmail.com>
Date: Wed, 2 Dec 2020 21:34:00 +0100
Subject: Add restart option

---
 main.go | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

(limited to 'main.go')

diff --git a/main.go b/main.go
index a8db9ac..5bd752f 100644
--- a/main.go
+++ b/main.go
@@ -54,6 +54,7 @@ func GetKey(t *term.Term) []byte {
 }
 
 func main() {
+restart:
 	var startlevel int
 	if len(os.Args) > 1 {
 		startlevel, _ = strconv.Atoi(os.Args[1])
@@ -189,7 +190,18 @@ func main() {
 
 		if topout {
 			ppos(4, 15, " GAME OVER ")
-			return
+			break
 		}
 	}
+	ppos(6, 15, " replay? [Y/n] ")
+	t.Restore() // return from raw mode
+	var yes string
+	fmt.Scanln(&yes)
+	if len(yes) < 1 {
+		goto restart
+	}
+	if yes[0] == 'n' || yes[0] == 'N' {
+		return
+	}
+	goto restart
 }
-- 
cgit v1.2.3-70-g09d2