package main

import (
	"flag"
	"fmt"
	"os"
	"time"

	"github.com/noahlann/nnet/pkg/nnet"
)

func main() {
	var (
		addr     = flag.String("addr", "tcp://127.0.0.1:6995", "Server address")
		protocol = flag.String("protocol", "", "Transport protocol (tcp, udp, websocket, unix, serial). If empty, auto-detect from addr")
		timeout  = flag.Duration("timeout", 10*time.Second, "Connection timeout")
		req      = flag.String("request", "", "Send a request and wait for response (overrides -message)")
		reqTO    = flag.Duration("request-timeout", 2*time.Second, "Request timeout when using -request")
		autoRec  = flag.Bool("auto-reconnect", false, "Enable auto reconnect")
		recIntv  = flag.Duration("reconnect-interval", 1*time.Second, "Reconnect interval")
		recMax   = flag.Int("reconnect-max", 0, "Max reconnect attempts (0 means infinite)")
		message  = flag.String("message", "", "Message to send")
	)
	flag.Parse()

	// 创建客户端配置
	config := &nnet.ClientConfig{
		Addr:                 *addr,
		TransportProtocol:    *protocol, // 如果为空，会根据地址前缀自动识别
		ConnectTimeout:       *timeout,
		ReadTimeout:          30 * time.Second,
		WriteTimeout:         30 * time.Second,
		AutoReconnect:        *autoRec,
		ReconnectInterval:    *recIntv,
		MaxReconnectAttempts: *recMax,
	}

	// 创建客户端
	c := nnet.NewClient(config)
	defer c.Close()

	// 连接到服务器
	if err := c.Connect(); err != nil {
		fmt.Printf("Failed to connect: %v\n", err)
		os.Exit(1)
	}
	defer c.Disconnect()

	// 发送消息
	if *req != "" {
		resp, err := c.Request([]byte(*req), *reqTO)
		if err != nil {
			fmt.Printf("Request error: %v\n", err)
			os.Exit(1)
		}
		fmt.Printf("Response: %s\n", string(resp))
	} else if *message != "" {
		if err := c.Send([]byte(*message)); err != nil {
			fmt.Printf("Failed to send message: %v\n", err)
			os.Exit(1)
		}
		fmt.Printf("Sent: %s\n", *message)

		// 接收响应
		response, err := c.Receive()
		if err != nil {
			fmt.Printf("Failed to receive response: %v\n", err)
			os.Exit(1)
		}
		fmt.Printf("Received: %s\n", string(response))
	} else {
		fmt.Println("Client connected successfully. Use -message to send a message.")
	}
}