package console_new

import (
	"strings"

	gollm "gitea.stevedudenhoeffer.com/steve/go-llm"
)

type execution struct {
	Command               string
	Output                string
	WhatILearned          []string
	WhatIStillNeedToLearn []string
}

const kMaxLenCommandSummary = 200
const kMaxLenCommandOutputSummary = 200

func (e execution) ToGeneralMessageHistory() gollm.Message {
	if len(e.Command) > kMaxLenCommandSummary {
		e.Command = e.Command[:kMaxLenCommandSummary] + "... (truncated)"
	}

	if len(e.Output) > kMaxLenCommandOutputSummary {
		e.Output = e.Output[:kMaxLenCommandOutputSummary] + "... (truncated)"
	}

	text := "# " + e.Command + "\n" + e.Output

	return gollm.Message{
		Role: gollm.RoleUser,
		Text: text,
	}
}

func (e execution) ToDetailedMessageHistory() gollm.Message {
	prompt := "$ "
	if strings.HasPrefix(e.Command, "sudo ") {
		prompt = "# "
		e.Command = e.Command[5:]
	}

	text := prompt + strings.TrimSpace(e.Command) + "\n" + e.Output

	if len(e.WhatILearned) > 0 {
		text += "\n\nWhat I learned:\n" + strings.Join(e.WhatILearned, "\n")
	} else {
		text += "\n\nI didn't learn anything new."
	}

	if len(e.WhatIStillNeedToLearn) > 0 {
		text += "\n\nWhat I still need to learn:\n" + strings.Join(e.WhatIStillNeedToLearn, "\n")
	} else {
		text += "\n\nI don't need to learn anything else."
	}

	return gollm.Message{
		Role: gollm.RoleUser,
		Text: text,
	}
}

type executions []execution

func (e executions) ToGeneralMessageHistory() []gollm.Message {
	var messages []gollm.Message

	for _, v := range e {
		messages = append(messages, v.ToGeneralMessageHistory())
	}

	return messages
}

func (e executions) ToGeneralButLastMessageHistory() []gollm.Message {
	var messages []gollm.Message

	for i, v := range e {
		if i == len(e)-1 {
			messages = append(messages, v.ToDetailedMessageHistory())
			break
		}

		messages = append(messages, v.ToGeneralMessageHistory())
	}

	return messages
}
func (e executions) ToDetailedMessageHistory() []gollm.Message {
	var messages []gollm.Message

	for _, v := range e {
		messages = append(messages, v.ToDetailedMessageHistory())
	}

	return messages
}