matterbridge/bridge/api/api.go

125 lines
2.6 KiB
Go
Raw Normal View History

2017-02-18 22:10:22 +00:00
package api
import (
"encoding/json"
2017-02-18 22:10:22 +00:00
"github.com/42wim/matterbridge/bridge/config"
log "github.com/Sirupsen/logrus"
"github.com/labstack/echo"
2017-06-05 22:05:32 +00:00
"github.com/labstack/echo/middleware"
2017-02-18 22:10:22 +00:00
"github.com/zfjagann/golang-ring"
"net/http"
"sync"
"time"
2017-02-18 22:10:22 +00:00
)
type Api struct {
Messages ring.Ring
sync.RWMutex
*config.BridgeConfig
2017-02-18 22:10:22 +00:00
}
type ApiMessage struct {
Text string `json:"text"`
Username string `json:"username"`
UserID string `json:"userid"`
2017-02-18 22:10:22 +00:00
Avatar string `json:"avatar"`
Gateway string `json:"gateway"`
2017-02-18 22:10:22 +00:00
}
var flog *log.Entry
var protocol = "api"
func init() {
flog = log.WithFields(log.Fields{"module": protocol})
}
func New(cfg *config.BridgeConfig) *Api {
b := &Api{BridgeConfig: cfg}
2017-02-18 22:10:22 +00:00
e := echo.New()
b.Messages = ring.Ring{}
b.Messages.SetCapacity(b.Config.Buffer)
2017-06-05 22:05:32 +00:00
if b.Config.Token != "" {
e.Use(middleware.KeyAuth(func(key string, c echo.Context) (bool, error) {
return key == b.Config.Token, nil
}))
}
2017-02-18 22:10:22 +00:00
e.GET("/api/messages", b.handleMessages)
e.GET("/api/stream", b.handleStream)
2017-02-18 22:10:22 +00:00
e.POST("/api/message", b.handlePostMessage)
go func() {
flog.Fatal(e.Start(b.Config.BindAddress))
2017-02-18 22:10:22 +00:00
}()
return b
}
func (b *Api) Connect() error {
return nil
}
func (b *Api) Disconnect() error {
return nil
}
func (b *Api) JoinChannel(channel config.ChannelInfo) error {
2017-02-18 22:10:22 +00:00
return nil
}
func (b *Api) Send(msg config.Message) (string, error) {
2017-02-18 22:10:22 +00:00
b.Lock()
defer b.Unlock()
// ignore delete messages
if msg.Event == config.EVENT_MSG_DELETE {
return "", nil
}
2017-02-18 22:10:22 +00:00
b.Messages.Enqueue(&msg)
return "", nil
2017-02-18 22:10:22 +00:00
}
func (b *Api) handlePostMessage(c echo.Context) error {
message := &ApiMessage{}
if err := c.Bind(message); err != nil {
return err
}
flog.Debugf("Sending message from %s on %s to gateway", message.Username, "api")
2017-02-18 22:10:22 +00:00
b.Remote <- config.Message{
Text: message.Text,
Username: message.Username,
UserID: message.UserID,
2017-02-18 22:10:22 +00:00
Channel: "api",
Avatar: message.Avatar,
Account: b.Account,
Gateway: message.Gateway,
Protocol: "api",
2017-02-18 22:10:22 +00:00
}
return c.JSON(http.StatusOK, message)
}
func (b *Api) handleMessages(c echo.Context) error {
b.Lock()
defer b.Unlock()
2017-06-05 21:08:36 +00:00
c.JSONPretty(http.StatusOK, b.Messages.Values(), " ")
2017-02-18 22:10:22 +00:00
b.Messages = ring.Ring{}
return nil
}
func (b *Api) handleStream(c echo.Context) error {
c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
c.Response().WriteHeader(http.StatusOK)
closeNotifier := c.Response().CloseNotify()
for {
select {
case <-closeNotifier:
return nil
default:
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)
}
}
}