83 lines
2.6 KiB
Go
83 lines
2.6 KiB
Go
|
package gomatrix
|
||
|
|
|
||
|
|
import (
|
||
|
|
"html"
|
||
|
|
"regexp"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Event represents a single Matrix event.
|
||
|
|
type Event struct {
|
||
|
|
StateKey string `json:"state_key"` // 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 string `json:"type"` // The event type
|
||
|
|
Timestamp int `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 map[string]interface{} `json:"content"` // The JSON content of the event.
|
||
|
|
}
|
||
|
|
|
||
|
|
// Body returns the value of the "body" key in the event content if it is
|
||
|
|
// present and is a string.
|
||
|
|
func (event *Event) Body() (body string, ok bool) {
|
||
|
|
value, exists := event.Content["body"]
|
||
|
|
if !exists {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
body, ok = value.(string)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// MessageType returns the value of the "msgtype" key in the event content if
|
||
|
|
// it is present and is a string.
|
||
|
|
func (event *Event) MessageType() (msgtype string, ok bool) {
|
||
|
|
value, exists := event.Content["msgtype"]
|
||
|
|
if !exists {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
msgtype, ok = value.(string)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// TextMessage is the contents of a Matrix formated message event.
|
||
|
|
type TextMessage struct {
|
||
|
|
MsgType string `json:"msgtype"`
|
||
|
|
Body string `json:"body"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// ImageInfo contains info about an image
|
||
|
|
type ImageInfo struct {
|
||
|
|
Height uint `json:"h"`
|
||
|
|
Width uint `json:"w"`
|
||
|
|
Mimetype string `json:"mimetype"`
|
||
|
|
Size uint `json:"size"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// ImageMessage is an m.image event
|
||
|
|
type ImageMessage struct {
|
||
|
|
MsgType string `json:"msgtype"`
|
||
|
|
Body string `json:"body"`
|
||
|
|
URL string `json:"url"`
|
||
|
|
Info ImageInfo `json:"info"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// An HTMLMessage is the contents of a Matrix HTML formated message event.
|
||
|
|
type HTMLMessage struct {
|
||
|
|
Body string `json:"body"`
|
||
|
|
MsgType string `json:"msgtype"`
|
||
|
|
Format string `json:"format"`
|
||
|
|
FormattedBody string `json:"formatted_body"`
|
||
|
|
}
|
||
|
|
|
||
|
|
var htmlRegex = regexp.MustCompile("<[^<]+?>")
|
||
|
|
|
||
|
|
// GetHTMLMessage returns an HTMLMessage with the body set to a stripped version of the provided HTML, in addition
|
||
|
|
// to the provided HTML.
|
||
|
|
func GetHTMLMessage(msgtype, htmlText string) HTMLMessage {
|
||
|
|
return HTMLMessage{
|
||
|
|
Body: html.UnescapeString(htmlRegex.ReplaceAllLiteralString(htmlText, "")),
|
||
|
|
MsgType: msgtype,
|
||
|
|
Format: "org.matrix.custom.html",
|
||
|
|
FormattedBody: htmlText,
|
||
|
|
}
|
||
|
|
}
|