54 lines
918 B
Go
54 lines
918 B
Go
package handlers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"ytdlp-site/originals"
|
|
|
|
"github.com/labstack/echo/v4"
|
|
)
|
|
|
|
func VideosEvents(c echo.Context) error {
|
|
|
|
user, err := GetUser(c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
req := c.Request()
|
|
res := c.Response()
|
|
|
|
// Set headers for SSE
|
|
res.Header().Set(echo.HeaderContentType, "text/event-stream")
|
|
res.Header().Set("Cache-Control", "no-cache")
|
|
res.Header().Set("Connection", "keep-alive")
|
|
|
|
// Create a channel to signal client disconnect
|
|
done := req.Context().Done()
|
|
|
|
q := originals.Subscribe(user.Id)
|
|
defer originals.Unsubscribe(user.Id, q)
|
|
|
|
// Send SSE messages
|
|
for {
|
|
select {
|
|
case <-done:
|
|
return nil
|
|
default:
|
|
event := <-q.Ch
|
|
|
|
jsonData, err := json.Marshal(event)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
msg := fmt.Sprintf("data: %s\n\n", jsonData)
|
|
_, err = res.Write([]byte(msg))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
res.Flush()
|
|
}
|
|
}
|
|
}
|