Add session close button

services/client.go

@@ -22,6 +22,10 @@ func NewClient(so socketio.Socket, session *Session) *Client {
 
 	c := &Client{SO: so}
 
+	so.On("session close", func() {
+		CloseSession(session)
+	})
+
 	so.On("terminal in", func(name, data string) {
 		// User wrote something on the terminal. Need to write it to the instance terminal
 		instance := GetInstance(session, name)

services/instance.go

@@ -37,8 +37,6 @@ func getDindImageName() string {
 }
 
 func NewInstance(session *Session) (*Instance, error) {
-
-	//TODO: Validate that a session can only have 5 instances
 	log.Printf("NewInstance - using image: [%s]\n", dindImage)
 	instance, err := CreateInstance(session.Id, dindImage)
 	instance.Session = session

services/session.go

@@ -50,6 +50,27 @@ func CreateWSServer() *socketio.Server {
 	return server
 }
 
+func CloseSession(s *Session) error {
+	s.Lock()
+	defer s.Unlock()
+	wsServer.BroadcastTo(s.Id, "session end")
+	log.Printf("Starting clean up of session [%s]\n", s.Id)
+	for _, i := range s.Instances {
+		i.Conn.Close()
+		if err := DeleteContainer(i.Name); err != nil {
+			log.Println(err)
+			return err
+		}
+	}
+	if err := DeleteNetwork(s.Id); err != nil {
+		log.Println(err)
+		return err
+	}
+	delete(sessions, s.Id)
+	log.Printf("Cleaned up session [%s]\n", s.Id)
+	return nil
+}
+
 func NewSession() (*Session, error) {
 	s := &Session{}
 	s.Id = uuid.NewV4().String()
@@ -60,22 +81,7 @@ func NewSession() (*Session, error) {
 
 	// Schedule cleanup of the session
 	time.AfterFunc(4*time.Hour, func() {
-		s = GetSession(s.Id)
-		s.Lock()
-		defer s.Unlock()
-		wsServer.BroadcastTo(s.Id, "session end")
-		log.Printf("Starting clean up of session [%s]\n", s.Id)
-		for _, i := range s.Instances {
-			i.Conn.Close()
-			if err := DeleteContainer(i.Name); err != nil {
-				log.Println(err)
-			}
-		}
-		if err := DeleteNetwork(s.Id); err != nil {
-			log.Println(err)
-		}
-		delete(sessions, s.Id)
-		log.Printf("Cleaned up session [%s]\n", s.Id)
+		CloseSession(s)
 	})
 
 	if err := CreateNetwork(s.Id); err != nil {