package message

import (
	"fmt"
	"io/ioutil"
	"local/truckstop/config"
	"log"
	"net/http"
	"regexp"
	"strings"
	"time"

	"github.com/matrix-org/gomatrix"
)

type Matrix struct {
	mock       bool
	homeserver string
	username   string
	token      string
	room       string
}

func NewMatrix() Matrix {
	conf := config.Get().Message.Matrix
	return Matrix{
		homeserver: conf.Homeserver,
		username:   conf.Username,
		token:      conf.Token,
		room:       conf.Room,
		mock:       conf.Mock,
	}
}

func (m Matrix) getclient() (*gomatrix.Client, error) {
	return gomatrix.NewClient(m.homeserver, m.username, m.token)
}

func (m Matrix) Receive() ([]Message, error) {
	if m.mock {
		log.Printf("matrix.Receive()")
		messages := make([]Message, 0)
		for k := range config.Get().Clients {
			messages = append(messages, Message{Timestamp: time.Now(), Sender: k, Content: "!state OH"})
			if k == "bel" {
				messages = append(messages, Message{Timestamp: time.Now(), Sender: k, Content: "!help"})
			}
			if k == "broc" {
				messages = append(messages, Message{Timestamp: time.Now(), Sender: k, Content: "!available 2148-10-" + fmt.Sprint(time.Now().Unix()%28)})
			}
		}
		return messages, nil
	}
	clients := config.Get().Clients
	matrixIDs := map[string]struct{}{}
	for k := range clients {
		matrixIDs[clients[k].IDs.Matrix] = struct{}{}
	}
	if len(matrixIDs) == 0 {
		return nil, nil
	}
	c, err := m.getclient()
	if err != nil {
		return nil, err
	}
	messages := make([]Message, 0)
	result, err := c.Messages(m.room, "", "", 'b', 50)
	if err != nil {
		return nil, err
	}
	for _, event := range result.Chunk {
		if _, ok := matrixIDs[event.Sender]; !ok {
			continue
		}
		switch event.Type {
		case "m.room.message":
			b, ok := event.Body()
			if ok {
				messages = append(messages, Message{Timestamp: time.Unix(0, event.Timestamp*int64(time.Millisecond)), Sender: event.Sender, Content: strings.TrimSpace(b)})
			}
		}
	}
	clientChange := regexp.MustCompile("@[a-z]+$")
	for i := range messages {
		if found := clientChange.FindString(messages[i].Content); found != "" {
			messages[i].Content = strings.TrimSpace(strings.ReplaceAll(messages[i].Content, found, ""))
			messages[i].Sender = found[1:]
		} else {
			for k, v := range config.Get().Clients {
				if v.IDs.Matrix == messages[i].Sender {
					messages[i].Sender = k
				}
			}
		}
		messages[i].Content = strings.TrimSpace(messages[i].Content)
	}
	return messages, nil
}

func (m Matrix) Send(text string) error {
	if m.mock {
		log.Printf("matrix.Send(%s)", text)
		return nil
	}
	c, err := m.getclient()
	if err != nil {
		return err
	}
	_, err = c.SendText(m.room, text)
	return err
}

func (m Matrix) SendImage(uri string) error {
	if m.mock {
		log.Printf("matrix.SendImage(%s)", uri)
		return nil
	}
	response, err := http.Get(uri)
	if err != nil {
		return err
	}
	if response.StatusCode != http.StatusOK {
		b, _ := ioutil.ReadAll(response.Body)
		response.Body.Close()
		return fmt.Errorf("failed to get %s: (%d) %s", uri, response.StatusCode, b)
	}
	b, err := ioutil.ReadAll(response.Body)
	response.Body.Close()
	if err != nil {
		return err
	}
	publicURI, err := UploadImage(b)
	if err != nil {
		return err
	}
	c, err := m.getclient()
	if err != nil {
		return err
	}
	resp, err := c.SendImage(m.room, "img", publicURI)
	log.Printf("sent image %s => %s: %+v", uri, publicURI, resp)
	return err
}