Small fixes
This commit is contained in:
parent
3a27831112
commit
feaf75b327
5 changed files with 53 additions and 29 deletions
|
@ -40,6 +40,10 @@ func (bot *Bot) initClients() {
|
|||
func (bot *Bot) startClients() {
|
||||
log.Debugln("Starting Matrix syncer")
|
||||
for _, client := range bot.Clients {
|
||||
go func() {
|
||||
client.SetAvatarURL(client.DB.AvatarURL)
|
||||
client.SetDisplayName(client.DB.DisplayName)
|
||||
}()
|
||||
if client.DB.Sync {
|
||||
client.Sync()
|
||||
}
|
||||
|
|
30
matrix.go
30
matrix.go
|
@ -57,9 +57,11 @@ type EventHandler func(*Event) bool
|
|||
|
||||
type MatrixClient interface {
|
||||
AddEventHandler(EventType, EventHandler)
|
||||
GetEvent(string, string) *Event
|
||||
}
|
||||
|
||||
type EventFuncs interface {
|
||||
MarkRead() error
|
||||
Reply(string) (string, error)
|
||||
ReplyContent(Content) (string, error)
|
||||
SendMessage(string) (string, error)
|
||||
|
@ -93,32 +95,32 @@ type Content struct {
|
|||
|
||||
MsgType MessageType `json:"msgtype"`
|
||||
Body string `json:"body"`
|
||||
Format string `json:"format"`
|
||||
FormattedBody string `json:"formatted_body"`
|
||||
Format string `json:"format,omitempty"`
|
||||
FormattedBody string `json:"formatted_body,omitempty"`
|
||||
|
||||
Info FileInfo `json:"info"`
|
||||
URL string `json:"url"`
|
||||
Info FileInfo `json:"info,omitempty"`
|
||||
URL string `json:"url,omitempty"`
|
||||
|
||||
Membership string `json:"membership"`
|
||||
Membership string `json:"membership,omitempty"`
|
||||
|
||||
RelatesTo RelatesTo `json:"m.relates_to"`
|
||||
RelatesTo RelatesTo `json:"m.relates_to,omitempty"`
|
||||
}
|
||||
|
||||
type FileInfo struct {
|
||||
MimeType string `json:"mimetype"`
|
||||
ThumbnailInfo *FileInfo `json:"thumbnail_info"`
|
||||
ThumbnailURL string `json:"thumbnail_url"`
|
||||
Height int `json:"h"`
|
||||
Width int `json:"w"`
|
||||
Size int `json:"size"`
|
||||
MimeType string `json:"mimetype,omitempty"`
|
||||
ThumbnailInfo *FileInfo `json:"thumbnail_info,omitempty"`
|
||||
ThumbnailURL string `json:"thumbnail_url,omitempty"`
|
||||
Height int `json:"h,omitempty"`
|
||||
Width int `json:"w,omitempty"`
|
||||
Size int `json:"size,omitempty"`
|
||||
}
|
||||
|
||||
type RelatesTo struct {
|
||||
InReplyTo InReplyTo `json:"m.in_reply_to"`
|
||||
InReplyTo InReplyTo `json:"m.in_reply_to,omitempty"`
|
||||
}
|
||||
|
||||
type InReplyTo struct {
|
||||
EventID string `json:"event_id"`
|
||||
// Not required, just for future-proofing
|
||||
RoomID string `json:"room_id"`
|
||||
RoomID string `json:"room_id,omitempty"`
|
||||
}
|
||||
|
|
|
@ -65,6 +65,10 @@ func (evt *Event) Interface() *maubot.Event {
|
|||
return mbEvent
|
||||
}
|
||||
|
||||
func (evt *Event) MarkRead() error {
|
||||
return evt.Client.MarkRead(evt.RoomID, evt.ID)
|
||||
}
|
||||
|
||||
func (evt *Event) Reply(text string) (string, error) {
|
||||
return evt.SendRawEvent(maubot.EventMessage,
|
||||
SetReply(
|
||||
|
|
|
@ -57,7 +57,21 @@ func (client *Client) ParseEvent(evt *gomatrix.Event) *Event {
|
|||
}
|
||||
|
||||
func (client *Client) AddEventHandler(evt maubot.EventType, handler maubot.EventHandler) {
|
||||
client.syncer.OnEventType(evt, handler)
|
||||
client.syncer.OnEventType(evt, func(evt *maubot.Event) bool {
|
||||
if evt.Sender == client.UserID {
|
||||
return false
|
||||
}
|
||||
return handler(evt)
|
||||
})
|
||||
}
|
||||
|
||||
func (client *Client) GetEvent(roomID, eventID string) *maubot.Event {
|
||||
evt, err := client.Client.GetEvent(roomID, eventID)
|
||||
if err != nil {
|
||||
log.Warnf("Failed to get event %s @ %s: %v", eventID, roomID, err)
|
||||
return nil
|
||||
}
|
||||
return client.ParseEvent(evt).Interface()
|
||||
}
|
||||
|
||||
func (client *Client) onJoin(evt *maubot.Event) bool {
|
||||
|
|
|
@ -41,7 +41,7 @@ func TrimReplyFallbackText(text string) string {
|
|||
for len(lines) > 0 && strings.HasPrefix(lines[0], "> ") {
|
||||
lines = lines[1:]
|
||||
}
|
||||
return strings.Join(lines, "\n")
|
||||
return strings.TrimSpace(strings.Join(lines, "\n"))
|
||||
}
|
||||
|
||||
func RemoveReplyFallback(evt *maubot.Event) {
|
||||
|
|
Loading…
Reference in a new issue