package agent

import (
	"fmt"

	"github.com/mudler/local-agent-framework/llm"
	"github.com/sashabaranov/go-openai"
)

type llmOptions struct {
	APIURL string
	APIKey string
	Model  string
}

type options struct {
	LLMAPI                 llmOptions
	character              Character
	randomIdentityGuidance string
	randomIdentity         bool
}

type Agent struct {
	options   *options
	Character Character
	client    *openai.Client
}

type Option func(*options) error

func defaultOptions() *options {
	return &options{
		LLMAPI: llmOptions{
			APIURL: "http://localhost:8080",
			Model:  "echidna",
		},
		character: Character{
			Name:        "John Doe",
			Age:         0,
			Occupation:  "Unemployed",
			NowDoing:    "Nothing",
			DoingNext:   "Nothing",
			DoneHistory: []string{},
			Memories:    []string{},
			Hobbies:     []string{},
			MusicTaste:  []string{},
		},
	}
}

func newOptions(opts ...Option) (*options, error) {
	options := defaultOptions()
	for _, o := range opts {
		if err := o(options); err != nil {
			return nil, err
		}
	}
	return options, nil
}

func New(opts ...Option) (*Agent, error) {
	options, err := newOptions(opts...)
	if err != nil {
		return nil, err
	}

	client := llm.NewClient(options.LLMAPI.APIKey, options.LLMAPI.APIURL)
	a := &Agent{
		options:   options,
		client:    client,
		Character: options.character,
	}

	if a.options.randomIdentity {
		err = a.generateIdentity("")
	}

	return a, err
}

func WithLLMAPIURL(url string) Option {
	return func(o *options) error {
		o.LLMAPI.APIURL = url
		return nil
	}
}

func WithLLMAPIKey(key string) Option {
	return func(o *options) error {
		o.LLMAPI.APIKey = key
		return nil
	}
}

func WithModel(model string) Option {
	return func(o *options) error {
		o.LLMAPI.Model = model
		return nil
	}
}

func WithCharacter(c Character) Option {
	return func(o *options) error {
		o.character = c
		return nil
	}
}

func FromFile(path string) Option {
	return func(o *options) error {
		c, err := Load(path)
		if err != nil {
			return err
		}
		o.character = *c
		return nil
	}
}

func WithRandomIdentity(guidance ...string) Option {
	return func(o *options) error {
		o.randomIdentityGuidance = fmt.Sprint(guidance)
		o.randomIdentity = true
		return nil
	}
}