package helpers

import (
	"forge.redroom.link/yves/meowlib/client"
)

func PrepareUserMessage(message string, srvuid string, peer_idx int, replyToUid string, filelist []string) ([]byte, string, error) {

	peer := client.GetConfig().GetIdentity().Peers[peer_idx]
	srv, err := client.GetConfig().GetIdentity().MessageServers.LoadServer(srvuid)
	if err != nil {
		return nil, "PrepareServerMessage : LoadServer", err
	}
	// Creating User message
	usermessage, err := peer.BuildSimpleUserMessage([]byte(message))
	if err != nil {
		return nil, "PrepareServerMessage : BuildSimpleUserMessage", err
	}
	for _, file := range filelist {
		err = usermessage.AddFile(file, client.GetConfig().Chunksize)
		if err != nil {
			return nil, "PrepareServerMessage : AddFile", err
		}
	}
	usermessage.Status.AnswerToUuid = replyToUid
	// Prepare cyphered + packed user message
	packedMsg, err := peer.ProcessOutboundUserMessage(usermessage)
	if err != nil {
		return nil, "PrepareServerMessage : ProcessOutboundUserMessage", err
	}
	// Creating Server message for transporting the user message
	toServerMessage := srv.BuildToServerMessageFromUserMessage(packedMsg)
	data, err := srv.ProcessOutboundMessage(toServerMessage)
	if err != nil {
		return nil, "PrepareServerMessage : ProcessOutboundMessage", err
	}
	// Store message
	err = peer.StoreMessage(usermessage, filelist)
	if err != nil {
		return nil, "PrepareServerMessage : StoreMessage", err
	}
	return data, "", nil
}

func SendAck(messageUid string, srvuid string, peer_idx int, received int64, processed int64) ([]byte, string, error) {

	peer := client.GetConfig().GetIdentity().Peers[peer_idx]
	srv, err := client.GetConfig().GetIdentity().MessageServers.LoadServer(srvuid)
	if err != nil {
		return nil, "PrepareServerMessage : LoadServer", err
	}
	// Creating User message
	usermessage, err := peer.BuildSimpleUserMessage(nil)
	if err != nil {
		return nil, "PrepareServerMessage : BuildSimpleUserMessage", err
	}
	usermessage.Status.Uuid = messageUid
	usermessage.Status.Received = uint64(received)
	usermessage.Status.Processed = uint64(processed)
	// Prepare cyphered + packed user message
	packedMsg, err := peer.ProcessOutboundUserMessage(usermessage)
	if err != nil {
		return nil, "PrepareServerMessage : ProcessOutboundUserMessage", err
	}
	// Creating Server message for transporting the user message
	toServerMessage := srv.BuildToServerMessageFromUserMessage(packedMsg)
	data, err := srv.ProcessOutboundMessage(toServerMessage)
	if err != nil {
		return nil, "PrepareServerMessage : ProcessOutboundMessage", err
	}
	// Store message
	err = peer.StoreMessage(usermessage, nil)
	if err != nil {
		return nil, "PrepareServerMessage : StoreMessage", err
	}
	return data, "", nil
}