package client

import (
	"bufio"
	"crypto/tls"
	"crypto/x509"
	_ "embed"
	"fmt"
	"os"
	"strings"
	"time"

	"github.com/kyren223/eko/internal/utils/log"
)

//go:embed server.crt
var certPEM []byte

func Run() {
	certPool := x509.NewCertPool()
	if !certPool.AppendCertsFromPEM(certPEM) {
		log.Fatal("failed to append server certificate")
	}

	tlsConfig := &tls.Config{
		RootCAs: certPool,
		ServerName: "localhost",
	}

	log.Info("Client started, waiting for user input...")
	for {
		fmt.Print("> ")
		input, _ := bufio.NewReader(os.Stdin).ReadString('\n')
		input = strings.TrimSpace(input)
		if input == ":q" || input == "exit" || input == "quit" {
			break
		}
		err := processRequest(input, tlsConfig)
		if err != nil {
			log.Error("%v", err)
		}
	}
}

func processRequest(request string, tlsConfig *tls.Config) error {
	conn, err := tls.Dial("tcp", ":7223", tlsConfig)
	if err != nil {
		return fmt.Errorf("Unable to establish connection with server: %v", err)
	}
	defer conn.Close()

	conn.SetDeadline(time.Now().Add(time.Second))
	log.Info("Established connection to server: %v", conn.RemoteAddr().String())

	_, err = conn.Write([]byte(request))
	if err != nil {
		return fmt.Errorf("Unable to send request: %v", err)
	}

	buffer := make([]byte, 1024)
	n, err := conn.Read(buffer)
	if err != nil {
		return fmt.Errorf("Unable to receive response: %v", err)
	}

	log.Info("Response from server: %v", string(buffer[:n]))
	return nil
}