maubot/matrix.go

// 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 maubot

type EventType string
type MessageType string

// State events
const (
	StateAliases        EventType = "m.room.aliases"
	StateCanonicalAlias           = "m.room.canonical_alias"
	StateCreate                   = "m.room.create"
	StateJoinRules                = "m.room.join_rules"
	StateMember                   = "m.room.member"
	StatePowerLevels              = "m.room.power_levels"
	StateRoomName                 = "m.room.name"
	StateTopic                    = "m.room.topic"
	StateRoomAvatar               = "m.room.avatar"
	StatePinnedEvents             = "m.room.pinned_events"
)

// Message events
const (
	EventRedaction EventType = "m.room.redaction"
	EventMessage             = "m.room.message"
	EventSticker             = "m.sticker"
)

// Msgtypes
const (
	MsgText     MessageType = "m.text"
	MsgEmote                = "m.emote"
	MsgNotice               = "m.notice"
	MsgImage                = "m.image"
	MsgLocation             = "m.location"
	MsgVideo                = "m.video"
	MsgAudio                = "m.audio"
)

const FormatHTML = "org.matrix.custom.html"

type EventHandler func(*Event) bool

type MatrixClient interface {
	AddEventHandler(EventType, EventHandler)
}

type EventFuncs interface {
	Reply(string) (string, error)
	ReplyContent(Content) (string, error)
	SendMessage(string) (string, error)
	SendContent(Content) (string, error)
	SendRawEvent(EventType, interface{}) (string, error)
}

type Event struct {
	EventFuncs

	StateKey   string                 `json:"state_key,omitempty"` // The state key for the event. Only present on State Events.
	Sender     string                 `json:"sender"`              // The user ID of the sender of the event
	Type       EventType              `json:"type"`                // The event type
	Timestamp  int64                  `json:"origin_server_ts"`    // The unix timestamp when this message was sent by the origin server
	ID         string                 `json:"event_id"`            // The unique ID of this event
	RoomID     string                 `json:"room_id"`             // The room the event was sent to. May be nil (e.g. for presence)
	Content    Content                `json:"content"`
	Redacts    string                 `json:"redacts,omitempty"`  // The event ID that was redacted if a m.room.redaction event
	Unsigned   Unsigned               `json:"unsigned,omitempty"` // Unsigned content set by own homeserver.
}

type Unsigned struct {
	PrevContent    Content                `json:"prev_content,omitempty"`
	PrevSender     string                 `json:"prev_sender,omitempty"`
	ReplacesState  string                 `json:"replaces_state,omitempty"`
	Age            int64                  `json:"age"`
}

type Content struct {
	Raw map[string]interface{} `json:"-"`

	MsgType       MessageType `json:"msgtype"`
	Body          string      `json:"body"`
	Format        string      `json:"format"`
	FormattedBody string      `json:"formatted_body"`

	Info FileInfo `json:"info"`
	URL  string   `json:"url"`

	Membership string `json:"membership"`

	RelatesTo RelatesTo `json:"m.relates_to"`
}

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"`
}

type RelatesTo struct {
	InReplyTo InReplyTo `json:"m.in_reply_to"`
}

type InReplyTo struct {
	EventID string `json:"event_id"`
	// Not required, just for future-proofing
	RoomID string `json:"room_id"`
}