package Messages

import (
	"encoding/json"
	"net/http"

	"git.tovijaeschke.xyz/tovi/Capsule/Backend/Database"
	"git.tovijaeschke.xyz/tovi/Capsule/Backend/Models"

	"github.com/gorilla/mux"
)

// Messages gets messages by the associationKey
func Messages(w http.ResponseWriter, r *http.Request) {
	var (
		messages       []Models.Message
		message        Models.Message
		urlVars        map[string]string
		associationKey string
		returnJSON     []byte
		i              int
		ok             bool
		err            error
	)

	urlVars = mux.Vars(r)
	associationKey, ok = urlVars["associationKey"]
	if !ok {
		http.Error(w, "Not Found", http.StatusNotFound)
		return
	}

	messages, err = Database.GetMessagesByAssociationKey(associationKey)
	if !ok {
		http.Error(w, "Not Found", http.StatusNotFound)
		return
	}

	for i, message = range messages {
		if message.MessageData.AttachmentID == nil {
			continue
		}

		messages[i].MessageData.Attachment.ImageLink = message.MessageData.Attachment.FilePath
	}

	returnJSON, err = json.MarshalIndent(messages, "", "  ")
	if err != nil {
		http.Error(w, "Error", http.StatusInternalServerError)
		return
	}

	w.WriteHeader(http.StatusOK)
	w.Write(returnJSON)
}