package cmd

import (
	"fmt"

	cmdutil "git.mlow.ca/mlow/lmcli/pkg/cmd/util"
	"git.mlow.ca/mlow/lmcli/pkg/lmcli"
	"git.mlow.ca/mlow/lmcli/pkg/lmcli/model"
	"github.com/spf13/cobra"
)

func NewCmd(ctx *lmcli.Context) *cobra.Command {
	cmd := &cobra.Command{
		Use:   "new [message]",
		Short: "Start a new conversation",
		Long:  `Start a new conversation with the Large Language Model.`,
		RunE: func(cmd *cobra.Command, args []string) error {
			messageContents := inputFromArgsOrEditor(args, "# What would you like to say?\n", "")
			if messageContents == "" {
				return fmt.Errorf("No message was provided.")
			}

			conversation := &model.Conversation{}
			err := ctx.Store.SaveConversation(conversation)
			if err != nil {
				return fmt.Errorf("Could not save new conversation: %v", err)
			}

			messages := []model.Message{
				{
					ConversationID:  conversation.ID,
					Role:            model.MessageRoleSystem,
					Content: getSystemPrompt(ctx),
				},
				{
					ConversationID:  conversation.ID,
					Role:            model.MessageRoleUser,
					Content: messageContents,
				},
			}

			cmdutil.HandleConversationReply(ctx, conversation, true, messages...)

			title, err := cmdutil.GenerateTitle(ctx, conversation)
			if err != nil {
				lmcli.Warn("Could not generate title for conversation: %v\n", err)
			}

			conversation.Title = title

			err = ctx.Store.SaveConversation(conversation)
			if err != nil {
				lmcli.Warn("Could not save conversation after generating title: %v\n", err)
			}
			return nil
		},
	}

	return cmd
}