Add option to disable the console broadcasts

This commit is contained in:
2024-09-24 15:17:58 +02:00
parent cc28f0f3c2
commit 7445627f7e
4 changed files with 16 additions and 6 deletions

View File

@@ -386,8 +386,13 @@ var clients = make(map[string]map[*websocket.Conn]bool)
var mu sync.Mutex
func handleWebSocket(w http.ResponseWriter, r *http.Request) {
fmt.Println(config.EnableConsole)
if !config.EnableConsole {
http.Error(w, "Console output is disabled", http.StatusForbidden)
return
}
filename := r.URL.Query().Get("filename")
fmt.Println("Filename:", filename)
if filename == "" {
http.Error(w, "Filename is required", http.StatusBadRequest)
return
@@ -423,11 +428,13 @@ func handleWebSocket(w http.ResponseWriter, r *http.Request) {
}
func broadcast(filename string, message []byte) {
if !config.EnableConsole {
return
}
mu.Lock()
defer mu.Unlock()
logger.LogInfo("Broadcast", fmt.Sprintf("Broadcasting message for filename: %s", filename))
for client := range clients[filename] {
if err := client.WriteMessage(websocket.TextMessage, message); err != nil {
client.Close()