From 0146d9e05d516c81a4bc517d882a7d0f3fe01512 Mon Sep 17 00:00:00 2001
From: giongto35 <giongto35@yahoo.com>
Date: Fri, 12 Apr 2019 02:39:19 +0800
Subject: [PATCH] Restart room after close

---
 main.go        | 28 +++++++++++++++++++++++++---
 ui/gameview.go |  7 ++++++-
 2 files changed, 31 insertions(+), 4 deletions(-)

diff --git a/main.go b/main.go
index fb488b29..ab5d40c3 100644
--- a/main.go
+++ b/main.go
@@ -85,7 +85,10 @@ func getWeb(w http.ResponseWriter, r *http.Request) {
 
 // init initilizes a room returns roomID
 func initRoom(roomID, gameName string) string {
-	roomID = generateRoomID()
+	// if no roomID is given, generate it
+	if roomID == "" {
+		roomID = generateRoomID()
+	}
 	imageChannel := make(chan *image.RGBA, 100)
 	inputChannel := make(chan int, 100)
 	closedChannel := make(chan bool)
@@ -101,10 +104,29 @@ func initRoom(roomID, gameName string) string {
 	return roomID
 }
 
+// isRoomRunning check if there is any running sessions.
+// TODO: If we remove sessions from room anytime a session is closed, we can check if the sessions list is empty or not.
+func isRoomRunning(roomID string) bool {
+	// If no roomID is registered
+	if _, ok := rooms[roomID]; !ok {
+		return false
+	}
+
+	// If there is running session
+	for _, s := range rooms[roomID].rtcSessions {
+		if !s.IsClosed() {
+			return true
+		}
+	}
+	return false
+}
+
 // startSession handles one session call
 func startSession(webRTC *webrtc.WebRTC, gameName string, roomID string, playerIndex int) string {
-	// If the roomID is empty, we spawn a new room
-	if roomID == "" {
+	// If the roomID is empty,
+	// or the roomID doesn't have any running sessions (room was closed)
+	// we spawn a new room
+	if roomID == "" || !isRoomRunning(roomID) {
 		roomID = initRoom(roomID, gameName)
 	}
 
diff --git a/ui/gameview.go b/ui/gameview.go
index f20d595d..a1bdbba7 100644
--- a/ui/gameview.go
+++ b/ui/gameview.go
@@ -84,7 +84,12 @@ func (view *GameView) Enter() {
 	// view.console.SetAudioSampleRate(view.director.audio.sampleRate)
 
 	// load state
-	view.console.Reset()
+	if err := view.console.LoadState(savePath(view.hash)); err == nil {
+		return
+	} else {
+		view.console.Reset()
+	}
+	//view.console.Reset()
 
 	// load sram
 	cartridge := view.console.Cartridge