maubot/matrix/matrix.go

86 lines
2.1 KiB
Go
Raw Normal View History

2018-06-13 22:41:05 +00:00
// maubot - A plugin-based Matrix bot system written in Go.
// Copyright (C) 2018 Tulir Asokan
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
package matrix
import (
"maubot.xyz"
2018-06-13 22:41:05 +00:00
"maubot.xyz/database"
"maunium.net/go/gomatrix"
log "maunium.net/go/maulogger"
)
type Client struct {
*gomatrix.Client
syncer *MaubotSyncer
2018-06-13 22:41:05 +00:00
DB *database.MatrixClient
}
func NewClient(db *database.MatrixClient) (*Client, error) {
mxClient, err := gomatrix.NewClient(db.Homeserver, db.UserID, db.AccessToken)
if err != nil {
return nil, err
}
client := &Client{
Client: mxClient,
DB: db,
}
client.syncer = NewMaubotSyncer(client, client.Store)
client.Client.Syncer = client.syncer
client.AddEventHandler(maubot.StateMember, client.onJoin)
2018-06-13 22:41:05 +00:00
return client, nil
}
func (client *Client) ParseEvent(evt *gomatrix.Event) *Event {
return &Event{
Client: client,
Event: evt,
}
}
func (client *Client) AddEventHandler(evt maubot.EventType, handler maubot.EventHandler) {
client.syncer.OnEventType(evt, handler)
2018-06-13 22:41:05 +00:00
}
func (client *Client) onJoin(evt *maubot.Event) bool {
if !client.DB.AutoJoinRooms || evt.StateKey != client.DB.UserID {
return true
2018-06-13 22:41:05 +00:00
}
if evt.Content.Membership == "invite" {
2018-06-13 22:41:05 +00:00
client.JoinRoom(evt.RoomID)
return false
2018-06-13 22:41:05 +00:00
}
return true
2018-06-13 22:41:05 +00:00
}
func (client *Client) JoinRoom(roomID string) {
client.Client.JoinRoom(roomID, "", nil)
}
func (client *Client) Sync() {
go func() {
err := client.Client.Sync()
if err != nil {
log.Errorln("Sync() in client", client.UserID, "errored:", err)
}
}()
}