package main

import (
	"bufio"
	"context"
	"fmt"
	"os"
	"strings"

	copilot "github.com/github/copilot-sdk/go"
	"github.com/github/copilot-sdk/go/samples/internal/sampleutil"
)

const blue = "\033[34m"
const reset = "\033[0m"

func main() {
	ctx := context.Background()
	cliPath, err := sampleutil.CLIPath()
	if err != nil {
		panic(err)
	}
	client := copilot.NewClient(&copilot.ClientOptions{Connection: copilot.StdioConnection{Path: cliPath}})
	if err := client.Start(ctx); err != nil {
		panic(err)
	}
	defer client.Stop()

	session, err := client.CreateSession(ctx, &copilot.SessionConfig{
		OnPermissionRequest: copilot.PermissionHandler.ApproveAll,
	})
	if err != nil {
		panic(err)
	}
	defer session.Disconnect()

	session.On(func(event copilot.SessionEvent) {
		var output string
		switch d := event.Data.(type) {
		case *copilot.AssistantReasoningData:
			output = fmt.Sprintf("[reasoning: %s]", d.Content)
		case *copilot.ToolExecutionStartData:
			output = fmt.Sprintf("[tool: %s]", d.ToolName)
		}
		if output != "" {
			fmt.Printf("%s%s%s\n", blue, output, reset)
		}
	})

	fmt.Println("Chat with Copilot (Ctrl+C to exit)")
	fmt.Println()
	scanner := bufio.NewScanner(os.Stdin)

	for {
		fmt.Print("You: ")
		if !scanner.Scan() {
			break
		}
		input := strings.TrimSpace(scanner.Text())
		if input == "" {
			continue
		}
		fmt.Println()

		reply, _ := session.SendAndWait(ctx, copilot.MessageOptions{Prompt: input})
		content := ""
		if reply != nil {
			if d, ok := reply.Data.(*copilot.AssistantMessageData); ok {
				content = d.Content
			}
		}
		fmt.Printf("\nAssistant: %s\n\n", content)
	}
}