Add output maze size and time

game.go

@@ -54,7 +54,14 @@ const (
 	DOWN
 )
 
-func (g *Game) Loop() error {
+type Result int
+
+const (
+	GOALED Result = iota
+	STOPPED
+)
+
+func (g *Game) Loop() (Result, error) {
 	e := make(chan Event)
 	go input(g.screen, e)
 
@@ -65,10 +72,10 @@ func (g *Game) Loop() error {
 		case ev := <-e:
 			switch ev {
 			case EXIT:
-				return nil
+				return STOPPED, nil
 			case RIGHT:
 				if g.maze.CheckGoal() { // gaol
-					return nil
+					return GOALED, nil
 				}
 				if g.maze.CheckMaze(RIGHT) {
 					g.maze.MoveCurrent(RIGHT)
@@ -89,7 +96,7 @@ func (g *Game) Loop() error {
 		}
 	}
 
-	return nil
+	return STOPPED, nil
 }
 
 func input(s tcell.Screen, e chan<- Event) {

gomaze.go

@@ -23,10 +23,10 @@ func initScreen() (tcell.Screen, error) {
 	return s, nil
 }
 
-func startGame(width, height int, seed bool, format bool) error {
+func startGame(width, height int, seed bool, format bool) (Result, int, int, error) {
 	s, err := initScreen()
 	if err != nil {
-		return err
+		return STOPPED, 0, 0, err
 	}
 	defer s.Fini()
 
@@ -38,7 +38,8 @@ func startGame(width, height int, seed bool, format bool) error {
 		maze:   m,
 	}
 
-	return game.Loop()
+	res, err := game.Loop()
+	return res, w / 2, h, err
 }
 
 func main() {
@@ -95,7 +96,20 @@ func main() {
 		wi := c.Bool("format")
 
 		if sc {
-			return startGame(tw, th, se, wi)
+			start := time.Now()
+			res, w, h, err := startGame(tw, th, se, wi)
+			if err != nil {
+				return err
+			}
+			end := time.Now()
+
+			if res == GOALED {
+				fmt.Println("Congrats!")
+				fmt.Printf("[Maze size] Width: %d / Height: %d\n", w, h)
+				fmt.Printf("[Your time] %s\n", end.Sub(start))
+			}
+
+			return nil
 		} else {
 			m := NewMaze(tw, th, se, wi)
 			m.printMaze()