3
0
mirror of https://github.com/42wim/matterbridge.git synced 2024-12-24 20:22:39 +01:00
matterbridge/bridge/api/api.go

208 lines
5.0 KiB
Go
Raw Normal View History

2017-02-18 23:10:22 +01:00
package api
import (
"encoding/json"
"net/http"
"sync"
"time"
"gopkg.in/olahol/melody.v1"
2018-02-27 00:33:21 +01:00
"github.com/42wim/matterbridge/bridge"
2017-02-18 23:10:22 +01:00
"github.com/42wim/matterbridge/bridge/config"
"github.com/labstack/echo/v4"
"github.com/labstack/echo/v4/middleware"
ring "github.com/zfjagann/golang-ring"
2017-02-18 23:10:22 +01:00
)
type API struct {
2017-02-18 23:10:22 +01:00
Messages ring.Ring
sync.RWMutex
*bridge.Config
mrouter *melody.Melody
2017-02-18 23:10:22 +01:00
}
type Message struct {
2017-02-18 23:10:22 +01:00
Text string `json:"text"`
Username string `json:"username"`
UserID string `json:"userid"`
2017-02-18 23:10:22 +01:00
Avatar string `json:"avatar"`
Gateway string `json:"gateway"`
2017-02-18 23:10:22 +01:00
}
func New(cfg *bridge.Config) bridge.Bridger {
b := &API{Config: cfg}
2017-02-18 23:10:22 +01:00
e := echo.New()
2018-02-21 00:49:10 +01:00
e.HideBanner = true
e.HidePort = true
b.mrouter = melody.New()
b.mrouter.HandleMessage(func(s *melody.Session, msg []byte) {
message := config.Message{}
err := json.Unmarshal(msg, &message)
if err != nil {
b.Log.Errorf("failed to decode message from byte[] '%s'", string(msg))
return
}
b.handleWebsocketMessage(message, s)
})
b.mrouter.HandleConnect(func(session *melody.Session) {
greet := b.getGreeting()
data, err := json.Marshal(greet)
if err != nil {
b.Log.Errorf("failed to encode message '%v'", greet)
return
}
err = session.Write(data)
if err != nil {
b.Log.Errorf("failed to write message '%s'", string(data))
return
}
// TODO: send message history buffer from `b.Messages` here
})
2017-02-18 23:10:22 +01:00
b.Messages = ring.Ring{}
if b.GetInt("Buffer") != 0 {
b.Messages.SetCapacity(b.GetInt("Buffer"))
}
if b.GetString("Token") != "" {
2017-06-06 00:05:32 +02:00
e.Use(middleware.KeyAuth(func(key string, c echo.Context) (bool, error) {
return key == b.GetString("Token"), nil
2017-06-06 00:05:32 +02:00
}))
}
// Set RemoteNickFormat to a sane default
if !b.IsKeySet("RemoteNickFormat") {
b.Log.Debugln("RemoteNickFormat is unset, defaulting to \"{NICK}\"")
b.Config.Config.Viper().Set(b.GetConfigKey("RemoteNickFormat"), "{NICK}")
}
2018-11-07 09:11:59 +01:00
e.GET("/api/health", b.handleHealthcheck)
2017-02-18 23:10:22 +01:00
e.GET("/api/messages", b.handleMessages)
e.GET("/api/stream", b.handleStream)
e.GET("/api/websocket", b.handleWebsocket)
2017-02-18 23:10:22 +01:00
e.POST("/api/message", b.handlePostMessage)
go func() {
if b.GetString("BindAddress") == "" {
2018-02-27 00:33:21 +01:00
b.Log.Fatalf("No BindAddress configured.")
2018-02-21 00:49:10 +01:00
}
b.Log.Infof("Listening on %s", b.GetString("BindAddress"))
b.Log.Fatal(e.Start(b.GetString("BindAddress")))
2017-02-18 23:10:22 +01:00
}()
return b
}
func (b *API) Connect() error {
2017-02-18 23:10:22 +01:00
return nil
}
func (b *API) Disconnect() error {
2017-02-18 23:10:22 +01:00
return nil
}
func (b *API) JoinChannel(channel config.ChannelInfo) error {
2017-02-18 23:10:22 +01:00
return nil
}
func (b *API) Send(msg config.Message) (string, error) {
2017-02-18 23:10:22 +01:00
b.Lock()
defer b.Unlock()
// ignore delete messages
if msg.Event == config.EventMsgDelete {
return "", nil
}
b.Log.Debugf("enqueueing message from %s on ring buffer", msg.Username)
b.Messages.Enqueue(msg)
data, err := json.Marshal(msg)
if err != nil {
b.Log.Errorf("failed to encode message '%s'", msg)
}
_ = b.mrouter.Broadcast(data)
return "", nil
2017-02-18 23:10:22 +01:00
}
func (b *API) handleHealthcheck(c echo.Context) error {
2018-11-07 09:11:59 +01:00
return c.String(http.StatusOK, "OK")
}
func (b *API) handlePostMessage(c echo.Context) error {
message := config.Message{}
if err := c.Bind(&message); err != nil {
2017-02-18 23:10:22 +01:00
return err
}
// these values are fixed
message.Channel = "api"
message.Protocol = "api"
message.Account = b.Account
message.ID = ""
message.Timestamp = time.Now()
2018-02-27 00:33:21 +01:00
b.Log.Debugf("Sending message from %s on %s to gateway", message.Username, "api")
b.Remote <- message
2017-02-18 23:10:22 +01:00
return c.JSON(http.StatusOK, message)
}
func (b *API) handleMessages(c echo.Context) error {
2017-02-18 23:10:22 +01:00
b.Lock()
defer b.Unlock()
2017-06-05 23:08:36 +02:00
c.JSONPretty(http.StatusOK, b.Messages.Values(), " ")
2017-02-18 23:10:22 +01:00
b.Messages = ring.Ring{}
return nil
}
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
}
c.Response().Flush()
for {
// TODO: this causes issues, messages should be broadcasted to all connected clients
msg := b.Messages.Dequeue()
if msg != nil {
if err := json.NewEncoder(c.Response()).Encode(msg); err != nil {
return err
}
c.Response().Flush()
}
time.Sleep(200 * time.Millisecond)
}
}
func (b *API) handleWebsocketMessage(message config.Message, s *melody.Session) {
message.Channel = "api"
message.Protocol = "api"
message.Account = b.Account
message.ID = ""
message.Timestamp = time.Now()
data, err := json.Marshal(message)
if err != nil {
b.Log.Errorf("failed to encode message for loopback '%v'", message)
return
}
_ = b.mrouter.BroadcastOthers(data, s)
b.Log.Debugf("Sending websocket message from %s on %s to gateway", message.Username, "api")
b.Remote <- message
}
func (b *API) handleWebsocket(c echo.Context) error {
err := b.mrouter.HandleRequest(c.Response(), c.Request())
if err != nil {
b.Log.Errorf("error in websocket handling '%v'", err)
return err
}
return nil
}