package fsdb

import (
	"fmt"

	"github.com/danielmiessler/fabric/common"
	goopenai "github.com/sashabaranov/go-openai"
)

type SessionsEntity struct {
	*StorageEntity
}

func (o *SessionsEntity) Get(name string) (session *Session, err error) {
	session = &Session{Name: name}

	if o.Exists(name) {
		err = o.LoadAsJson(name, &session.Messages)
	} else {
		fmt.Printf("Creating new session: %s\n", name)
	}
	return
}

func (o *SessionsEntity) PrintSession(name string) (err error) {
	if o.Exists(name) {
		var session Session
		if err = o.LoadAsJson(name, &session.Messages); err == nil {
			fmt.Println(session.String())
		}
	}
	return
}

func (o *SessionsEntity) SaveSession(session *Session) (err error) {
	return o.SaveAsJson(session.Name, session.Messages)
}

type Session struct {
	Name     string
	Messages []*goopenai.ChatCompletionMessage

	vendorMessages []*goopenai.ChatCompletionMessage
}

func (o *Session) IsEmpty() bool {
	return len(o.Messages) == 0
}

func (o *Session) Append(messages ...*goopenai.ChatCompletionMessage) {
	if o.vendorMessages != nil {
		for _, message := range messages {
			o.Messages = append(o.Messages, message)
			o.appendVendorMessage(message)
		}
	} else {
		o.Messages = append(o.Messages, messages...)
	}
}

func (o *Session) GetVendorMessages() (ret []*goopenai.ChatCompletionMessage) {
	if len(o.vendorMessages) == 0 {
		for _, message := range o.Messages {
			o.appendVendorMessage(message)
		}
	}
	ret = o.vendorMessages
	return
}

func (o *Session) appendVendorMessage(message *goopenai.ChatCompletionMessage) {
	if message.Role != common.ChatMessageRoleMeta {
		o.vendorMessages = append(o.vendorMessages, message)
	}
}

func (o *Session) GetLastMessage() (ret *goopenai.ChatCompletionMessage) {
	if len(o.Messages) > 0 {
		ret = o.Messages[len(o.Messages)-1]
	}
	return
}

func (o *Session) String() (ret string) {
	for _, message := range o.Messages {
		ret += fmt.Sprintf("\n--- \n[%v]\n%v", message.Role, message.Content)
		if message.MultiContent != nil {
			for _, part := range message.MultiContent {
				if part.Type == goopenai.ChatMessagePartTypeImageURL {
					ret += fmt.Sprintf("\n%v: %v", part.Type, *part.ImageURL)
				} else if part.Type == goopenai.ChatMessagePartTypeText {
					ret += fmt.Sprintf("\n%v: %v", part.Type, part.Text)
				}
			}
		}
	}
	return
}