Easybridge has an avatar

1 files changed, 54 insertions, 0 deletions

diff --git a/mxlib/mediaobject.go b/mxlib/mediaobject.go
new file mode 100644
index 0000000..8a730d2
--- /dev/null
+++ b/mxlib/mediaobject.go
@@ -0,0 +1,54 @@
+package mxlib
+
+import (
+	"io"
+	"fmt"
+	"net/url"
+	"net/http"
+)
+
+type MediaObject struct {
+	mxClient *Client
+	filename string
+	size int64
+	mimetype string
+	MxcServer string
+	MxcMediaId string
+}
+
+func (m *MediaObject) Filename() string {
+	return m.filename
+}
+
+func (m *MediaObject) Size() int64 {
+	return m.size
+}
+
+func (m *MediaObject) Mimetype() string {
+	return m.mimetype
+}
+
+func (m *MediaObject) Read() (io.ReadCloser, error) {
+	req, err := http.NewRequest("GET", m.URL(), nil)
+	if err != nil {
+		return nil, err
+	}
+
+	req.Header.Add("Authorization", "Bearer "+m.mxClient.Token)
+
+	resp, err := m.mxClient.httpClient.Do(req)
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("HTTP error %d", resp.StatusCode)
+	}
+
+	return resp.Body, nil
+}
+
+func (m *MediaObject) URL() string {
+	return fmt.Sprintf("%s/_matrix/media/r0/download/%s/%s/%s",
+		m.mxClient.Server, m.MxcServer, m.MxcMediaId, url.QueryEscape(m.filename))
+}