//go:build server || (!server && !client)

package main

import (
	"log"
	"os"
	"os/signal"
	"syscall"

	internalprotocol "github.com/noahlann/nnet/internal/protocol/nnet"
	"github.com/noahlann/nnet/pkg/nnet"
)

// This example demonstrates protocol version management.
// It registers multiple versions of the nnet protocol and uses them based on configuration.
func main() {
	cfg := &nnet.Config{
		Addr: "tcp://:8082",
		// 指定应用层协议名称
		ApplicationProtocol: "nnet",
		Codec: &nnet.CodecConfig{
			// 与库默认保持一致：binary 为默认
			DefaultCodec:         "binary",
			EnableProtocolEncode: true, // 启用协议层编码
		},
	}

	srv, err := nnet.NewServer(cfg)
	if err != nil {
		log.Fatal("create server:", err)
	}

	// 注册多个版本的协议
	pm := srv.ProtocolManager()

	// 注册 v1.0 版本
	protoV1 := internalprotocol.NewNNetProtocol("1.0")
	if err := pm.Register(protoV1); err != nil {
		log.Fatal("register protocol v1.0:", err)
	}

	// 注册 v2.0 版本
	protoV2 := internalprotocol.NewNNetProtocol("2.0")
	if err := pm.Register(protoV2); err != nil {
		log.Fatal("register protocol v2.0:", err)
	}

	// 设置默认协议版本为 v1.0
	if err := pm.SetDefault("nnet", "1.0"); err != nil {
		log.Fatal("set default protocol:", err)
	}

	log.Println("Registered protocols:")
	log.Println("  - nnet v1.0 (default)")
	log.Println("  - nnet v2.0")

	// 注册路由处理器
	srv.Router().RegisterString("version", func(ctx nnet.Context) error {
		// 获取当前使用的协议信息
		proto := pm.GetDefault()
		if proto != nil {
			return ctx.Response().Write(map[string]any{
				"protocol": proto.Name(),
				"version":  proto.Version(),
			})
		}
		return ctx.Response().Write(map[string]any{
			"protocol": "unknown",
			"version":  "unknown",
		})
	})

	// 注册另一个路由，演示协议版本的使用
	srv.Router().RegisterString("echo", func(ctx nnet.Context) error {
		// 回显请求数据
		data := ctx.Request().Raw()
		return ctx.Response().Write(map[string]any{
			"echo":   string(data),
			"length": len(data),
		})
	})

	log.Println("protocol_version_server listening on :8082")
	log.Println("Protocol: nnet v1.0 (default)")
	if err := srv.Start(); err != nil {
		log.Fatal("start server:", err)
	}
	defer srv.Stop()

	quit := make(chan os.Signal, 1)
	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
	<-quit
}