summaryrefslogtreecommitdiffstats
path: root/bridge/api
diff options
context:
space:
mode:
authorhaykam821 <24855774+haykam821@users.noreply.github.com>2020-07-12 15:13:28 -0400
committerGitHub <noreply@github.com>2020-07-12 21:13:28 +0200
commit9b22f164970c1e01f4701db922aaddbe93999fb6 (patch)
tree6ec938e07e541462ac4ca22f0ae7df8a35a7f44c /bridge/api
parent2977a5957e98b7161669e72a00ecc0999e3566df (diff)
downloadmatterbridge-msglm-9b22f164970c1e01f4701db922aaddbe93999fb6.tar.gz
matterbridge-msglm-9b22f164970c1e01f4701db922aaddbe93999fb6.tar.bz2
matterbridge-msglm-9b22f164970c1e01f4701db922aaddbe93999fb6.zip
Add websocket to API (#970)
Co-authored-by: Qais Patankar <qaisjp@gmail.com>
Diffstat (limited to 'bridge/api')
-rw-r--r--bridge/api/api.go63
1 files changed, 59 insertions, 4 deletions
diff --git a/bridge/api/api.go b/bridge/api/api.go
index 38d1a4bd..62336881 100644
--- a/bridge/api/api.go
+++ b/bridge/api/api.go
@@ -8,6 +8,7 @@ import (
"github.com/42wim/matterbridge/bridge"
"github.com/42wim/matterbridge/bridge/config"
+ "github.com/gorilla/websocket"
"github.com/labstack/echo/v4"
"github.com/labstack/echo/v4/middleware"
ring "github.com/zfjagann/golang-ring"
@@ -51,6 +52,7 @@ func New(cfg *bridge.Config) bridge.Bridger {
e.GET("/api/health", b.handleHealthcheck)
e.GET("/api/messages", b.handleMessages)
e.GET("/api/stream", b.handleStream)
+ e.GET("/api/websocket", b.handleWebsocket)
e.POST("/api/message", b.handlePostMessage)
go func() {
if b.GetString("BindAddress") == "" {
@@ -113,13 +115,17 @@ func (b *API) handleMessages(c echo.Context) error {
return nil
}
-func (b *API) handleStream(c echo.Context) error {
- c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
- c.Response().WriteHeader(http.StatusOK)
- greet := config.Message{
+func (b *API) getGreeting() config.Message {
+ return config.Message{
Event: config.EventAPIConnected,
Timestamp: time.Now(),
}
+}
+
+func (b *API) handleStream(c echo.Context) error {
+ c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
+ c.Response().WriteHeader(http.StatusOK)
+ greet := b.getGreeting()
if err := json.NewEncoder(c.Response()).Encode(greet); err != nil {
return err
}
@@ -135,3 +141,52 @@ func (b *API) handleStream(c echo.Context) error {
time.Sleep(200 * time.Millisecond)
}
}
+
+func (b *API) handleWebsocketMessage(message config.Message) {
+ message.Channel = "api"
+ message.Protocol = "api"
+ message.Account = b.Account
+ message.ID = ""
+ message.Timestamp = time.Now()
+
+ b.Log.Debugf("Sending websocket message from %s on %s to gateway", message.Username, "api")
+ b.Remote <- message
+}
+
+func (b *API) writePump(conn *websocket.Conn) {
+ for {
+ msg := b.Messages.Dequeue()
+ if msg != nil {
+ err := conn.WriteJSON(msg)
+ if err != nil {
+ break
+ }
+ }
+ }
+}
+
+func (b *API) readPump(conn *websocket.Conn) {
+ for {
+ message := config.Message{}
+ err := conn.ReadJSON(&message)
+ if err != nil {
+ break
+ }
+ b.handleWebsocketMessage(message)
+ }
+}
+
+func (b *API) handleWebsocket(c echo.Context) error {
+ conn, err := websocket.Upgrade(c.Response().Writer, c.Request(), nil, 1024, 1024)
+ if err != nil {
+ return err
+ }
+
+ greet := b.getGreeting()
+ _ = conn.WriteJSON(greet)
+
+ go b.writePump(conn)
+ go b.readPump(conn)
+
+ return nil
+}