room.go

// Copyright © 2018 MTRNord <info@nordgedanken.de>
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package rooms

import (
	"github.com/Nordgedanken/Morpheusv2/pkg/matrix"
	"github.com/Nordgedanken/Morpheusv2/pkg/util"
	"github.com/matrix-org/gomatrix"
	"strings"
)

// Room holds the needed Room data and allows to work with that. It gets normally loaded from the cache
type Room struct {
	id       string
	aliases  []string
	name     string
	avatar   []byte
	topic    string
	messages []matrix.Message
}

// SetRoomID adds the roomID to the current Room
func (r *Room) SetRoomID(id string) {
	r.id = id
}

// SetRoomAliases adds the aliases to the current Room
func (r *Room) SetRoomAliases(aliases []string) {
	r.aliases = aliases
}

// SetName adds the name to the current Room
func (r *Room) SetName(name string) {
	r.name = name
}

// SetAvatar adds the avatar to the current Room
func (r *Room) SetAvatar(avatar []byte) {
	r.avatar = avatar
}

// SetTopic adds the topic to the current Room
func (r *Room) SetTopic(topic string) {
	r.topic = topic
}

// SetMessages adds the messages to the current Room
func (r *Room) SetMessages(messages []matrix.Message) {
	r.messages = messages
}

// GetRoomID returns the room ID from the current Room
func (r *Room) GetRoomID() string {
	return r.id
}

// GetRoomAliases returns the room aliases from the current Room
func (r *Room) GetRoomAliases() []string {
	return r.aliases
}

// GetName returns the name from the current Room
func (r *Room) GetName() (string, error) {
	if r.name == "" {
		type RespRoomName struct {
			Name string `json:"name"`
		}
		resp := &RespRoomName{}
		err := util.User.GetCli().StateEvent(r.id, "m.room.name", "", resp)
		if err != nil {
			return "", err
		}
		r.name = resp.Name
	}
	return r.name, nil
}

// GetAvatar returns the avatar from the current Room
func (r *Room) GetAvatar() ([]byte, error) {
	if r.avatar == nil {
		resp := &gomatrix.Event{}
		err := util.User.GetCli().StateEvent(r.id, "m.room.avatar", "", resp)
		if err != nil {
			return nil, err
		}
		var avatar []byte
		value, exists := resp.Content["url"]
		if !exists {
			return nil, nil
		}
		url, ok := value.(string)
		if !ok {
			return nil, nil
		}
		split := strings.Split(url, "/")
		servername := strings.TrimPrefix(split[0], "mxc://")
		mediaID := split[1]
		mediaURL := util.User.GetCli().BuildBaseURL("_matrix/media/r0/download", servername, mediaID)
		avatar, err = util.User.GetCli().MakeRequest("GET", mediaURL, nil, nil)
		if err != nil {
			return nil, err
		}
		r.avatar = avatar
	}
	return r.avatar, nil
}

// GetTopic returns the topic from the current Room
func (r *Room) GetTopic() (string, error) {
	if r.topic == "" {
		resp := &gomatrix.Event{}
		err := util.User.GetCli().StateEvent(r.id, "m.room.topic", "", resp)
		if err != nil {
			return "", err
		}
		value, exists := resp.Content["topic"]
		if !exists {
			return "", nil
		}
		topic, ok := value.(string)
		if !ok {
			return "", nil
		}
		r.topic = topic
	}
	return r.topic, nil
}

// GetMessages returns the messages from the current Room
func (r *Room) GetMessages() []matrix.Message {
	// TODO Lazy load from DB
	return r.messages
}