package integration

import (
	"fmt"
	"net"
	"testing"
	"time"

	"github.com/noahlann/nnet/pkg/nnet"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

// TestSimpleEcho 最简单的测试：服务器echo客户端发送的数据
func TestSimpleEcho(t *testing.T) {
	// 获取随机端口
	listener, err := net.Listen("tcp", "127.0.0.1:0")
	require.NoError(t, err)
	port := listener.Addr().(*net.TCPAddr).Port
	listener.Close()

	cfg := &nnet.Config{
		Addr: fmt.Sprintf("tcp://127.0.0.1:%d", port),
		Codec: &nnet.CodecConfig{
			DefaultCodec:         "json",
			EnableProtocolEncode: false, // 明确禁用协议编码
		},
		Logger: &nnet.LoggerConfig{
			Level:  "debug", // 使用debug级别查看详细日志
			Format: "text",
			Output: "stdout",
		},
	}

	// 创建服务器
	server, err := nnet.NewServer(cfg)
	require.NoError(t, err)

	// 注册一个简单的echo路由
	server.Router().RegisterString("echo", func(ctx nnet.Context) error {
		data := ctx.Request().Raw()
		t.Logf("Server received: %q", string(data))
		// 直接写入原始数据
		return ctx.Response().WriteBytes(data)
	})

	// 启动服务器
	ts := &TestServer{
		Server: server,
		Addr:   cfg.Addr,
		stopCh: make(chan struct{}),
	}

	ts.wg.Add(1)
	go func() {
		defer ts.wg.Done()
		if err := server.Start(); err != nil {
			t.Logf("Server error: %v", err)
		}
	}()

	// 等待服务器启动
	require.Eventually(t, func() bool {
		return server.Started()
	}, 3*time.Second, 50*time.Millisecond, "Server should start")

	// 等待服务器准备好
	require.Eventually(t, func() bool {
		testConn, err := net.DialTimeout("tcp", fmt.Sprintf("127.0.0.1:%d", port), 500*time.Millisecond)
		if err != nil {
			return false
		}
		testConn.Close()
		return true
	}, 5*time.Second, 100*time.Millisecond, "Server should be ready")

	defer CleanupTestServer(t, ts)

	// 创建客户端
	client := NewTestClient(t, ts.Addr, nil)
	defer CleanupTestClient(t, client)

	// 连接客户端
	ConnectTestClient(t, client)

	// 等待一下
	time.Sleep(500 * time.Millisecond)

	// 发送请求
	testData := "echo\n"
	t.Logf("Client sending: %q", testData)
	
	// 使用Send和Receive分开测试
	err = client.Send([]byte(testData))
	require.NoError(t, err, "Send should succeed")

	// 接收响应
	resp, err := client.Receive()
	if err != nil {
		t.Logf("Receive error: %v", err)
		// 如果Receive失败，尝试Request
		resp, err = client.Request([]byte(testData), 3*time.Second)
	}
	
	require.NoError(t, err, "Should receive response")
	t.Logf("Client received: %q", string(resp))
	assert.NotEmpty(t, resp, "Response should not be empty")
}