matrixbotlib/client.go

96 lines
2.2 KiB
Go

package matrixbotlib
import (
"io"
"os"
"strings"
"gopkg.in/yaml.v2"
"maunium.net/go/mautrix"
"maunium.net/go/mautrix/event"
"maunium.net/go/mautrix/id"
)
// NewMatrixClient returns a new logged in Mautrix Client struct
func NewMatrixClient(config *MatrixClientConfig, store mautrix.Storer) (*mautrix.Client, error) {
var client *mautrix.Client
var err error
// make sure username is lower case otherwise token login breaks
uname := strings.ToLower(config.Username)
if config.Token == "" {
client, err = mautrix.NewClient(config.Homeserver, "", "")
if err != nil {
return client, err
}
} else {
client, err = mautrix.NewClient(config.Homeserver, id.NewUserID(uname, config.Domain), config.Token)
if err != nil {
return client, err
}
}
client.Store = store
if config.Token == "" {
loginRes, err := client.Login(&mautrix.ReqLogin{
Type: "m.login.password",
Identifier: mautrix.UserIdentifier{Type: mautrix.IdentifierTypeUser, User: uname},
Password: config.Password,
StoreCredentials: true,
})
if err != nil {
return client, err
}
config.Token = loginRes.AccessToken
err = SyncToken(config)
if err != nil {
return nil, err
}
}
return client, err
}
func SyncToken(config *MatrixClientConfig) error {
file, err := os.OpenFile(config.filename, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0o600)
if err != nil {
return err
}
defer file.Close()
t := make(map[string]interface{})
data, err := io.ReadAll(file)
if err != nil {
return err
}
yaml.Unmarshal([]byte(data), &t)
t["token"] = config.Token
enc := yaml.NewEncoder(file)
err = enc.Encode(t)
if err != nil {
return err
}
return nil
}
// SetupAccountDataStore sets the client to use a AccountData store and filter appropriately
func SetupAccountDataStore(client *mautrix.Client, token string) error {
dataFilter := &mautrix.Filter{
AccountData: mautrix.FilterPart{
Limit: 20,
NotTypes: []event.Type{
event.NewEventType(token),
},
},
}
store := mautrix.NewAccountDataStore(token, client)
fID, err := client.CreateFilter(dataFilter)
if err != nil {
// don't want to continue if we can't keep state
return err
}
uid := client.UserID
store.SaveFilterID(uid, fID.FilterID)
client.Store = store
return nil
}