mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2025-10-31 19:52:26 -05:00
start messing about with streaming api
This commit is contained in:
parent
82d9f88e42
commit
0cee5aa569
9 changed files with 241 additions and 2 deletions
46
internal/api/client/streaming/stream.go
Normal file
46
internal/api/client/streaming/stream.go
Normal file
|
|
@ -0,0 +1,46 @@
|
|||
package streaming
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/gorilla/websocket"
|
||||
)
|
||||
|
||||
func (m *Module) StreamGETHandler(c *gin.Context) {
|
||||
streamType := c.Query(StreamQueryKey)
|
||||
if streamType == "" {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": fmt.Sprintf("no stream type provided under query key %s", StreamQueryKey)})
|
||||
return
|
||||
}
|
||||
|
||||
accessToken := c.Query(AccessTokenQueryKey)
|
||||
if accessToken == "" {
|
||||
c.JSON(http.StatusUnauthorized, gin.H{"error": fmt.Sprintf("no access token provided under query key %s", AccessTokenQueryKey)})
|
||||
return
|
||||
}
|
||||
|
||||
account, err := m.processor.AuthorizeStreamingRequest(accessToken)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusUnauthorized, gin.H{"error": "could not authorize with given token"})
|
||||
return
|
||||
}
|
||||
|
||||
upgrader := websocket.Upgrader{
|
||||
HandshakeTimeout: 5 * time.Second,
|
||||
ReadBufferSize: 1024,
|
||||
WriteBufferSize: 1024,
|
||||
Subprotocols: []string{"wss"},
|
||||
}
|
||||
|
||||
conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if errWithCode := m.processor.StreamForAccount(conn, account, streamType); errWithCode != nil {
|
||||
c.JSON(errWithCode.Code(), errWithCode.Safe())
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue