Reset chat bot stream ever 90 minutes

tylertravisty · Feb 14, 2024 · 7d0e825 · 7d0e825
1 parent cdbaf8f
commit 7d0e825
Show file tree

Hide file tree

Showing 3 changed files with 65 additions and 14 deletions.
diff --git a/frontend/src/components/ChatBot.jsx b/frontend/src/components/ChatBot.jsx
@@ -1,7 +1,5 @@
 import { useEffect, useState } from 'react';
-
 import { Modal, SmallModal } from './Modal';
-
 import { LoginChatBot, UpdateChatBotUrl } from '../../wailsjs/go/main/App';
 
 import './ChatBot.css';

diff --git a/frontend/src/screens/Dashboard.jsx b/frontend/src/screens/Dashboard.jsx
@@ -122,6 +122,10 @@ function Dashboard() {
             // console.log('Query response error:', error);
             setActive(false);
         });
+
+        EventsOn('ChatBotChatStreamError', (error) => {
+            setError(error);
+        });
     }, []);
 
     const home = () => {

diff --git a/internal/chatbot/chatbot.go b/internal/chatbot/chatbot.go
@@ -21,14 +21,16 @@ import (
 )
 
 type ChatBot struct {
-	ctx        context.Context
-	client     *rumblelivestreamlib.Client
-	commands   map[string]chan rumblelivestreamlib.ChatView
-	commandsMu sync.Mutex
-	Cfg        config.ChatBot
-	logError   *log.Logger
-	messages   map[string]*message
-	messagesMu sync.Mutex
+	ctx                context.Context
+	cancelChatStream   context.CancelFunc
+	cancelChatStreamMu sync.Mutex
+	client             *rumblelivestreamlib.Client
+	commands           map[string]chan rumblelivestreamlib.ChatView
+	commandsMu         sync.Mutex
+	Cfg                config.ChatBot
+	logError           *log.Logger
+	messages           map[string]*message
+	messagesMu         sync.Mutex
 }
 
 type message struct {
@@ -389,22 +391,69 @@ func (cb *ChatBot) StartChatStream() error {
 		return fmt.Errorf("chatbot: error getting chat info: %v", err)
 	}
 
-	err = cb.client.StartChatStream(cb.handleChat, cb.handleError)
-	if err != nil {
-		return fmt.Errorf("chatbot: error starting chat stream: %v", err)
-	}
+	ctx, cancel := context.WithCancel(context.Background())
+	cb.cancelChatStreamMu.Lock()
+	cb.cancelChatStream = cancel
+	cb.cancelChatStreamMu.Unlock()
+
+	go cb.startChatStream(ctx)
+
+	// err = cb.client.StartChatStream(cb.handleChat, cb.handleError)
+	// if err != nil {
+	// 	return fmt.Errorf("chatbot: error starting chat stream: %v", err)
+	// }
 
 	return nil
 }
 
+func (cb *ChatBot) startChatStream(ctx context.Context) {
+	for {
+		err := cb.client.StartChatStream(cb.handleChat, cb.handleError)
+		if err != nil {
+			cb.logError.Println("error starting chat stream:", err)
+			runtime.EventsEmit(cb.ctx, "ChatBotChatStreamError", "Error starting chat stream.")
+			return
+		}
+		select {
+		case <-time.After(90 * time.Minute):
+			cb.client.StopChatStream()
+			break
+		case <-ctx.Done():
+			cb.client.StopChatStream()
+			return
+		}
+	}
+}
+
 func (cb *ChatBot) StopChatStream() error {
 	if cb.client == nil {
 		return fmt.Errorf("chatbot: client is nil")
 	}
 
 	// TODO: should a panic be caught here?
+	cb.cancelChatStreamMu.Lock()
+	if cb.cancelChatStream != nil {
+		cb.cancelChatStream()
+	} else {
+		cb.client.StopChatStream()
+	}
+	cb.cancelChatStreamMu.Unlock()
+
+	return nil
+}
+
+func (cb *ChatBot) RestartChatStream() error {
+	if cb.client == nil {
+		return fmt.Errorf("chatbot: client is nil")
+	}
+
 	cb.client.StopChatStream()
 
+	err := cb.client.StartChatStream(cb.handleChat, cb.handleError)
+	if err != nil {
+		return fmt.Errorf("chatbot: error starting chat stream: %v", err)
+	}
+
 	return nil
 }