package integration

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

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

// StartTestServerWithRoutes 启动测试服务器并注册路由
// 这是一个辅助函数，简化测试代码
func StartTestServerWithRoutes(t *testing.T, cfg *nnet.Config, setupRoutes func(nnet.Server)) *TestServer {
	// 如果地址为空或使用端口0，使用随机端口
	if cfg.Addr == "" || cfg.Addr == "tcp://:0" || cfg.Addr == "tcp://127.0.0.1:0" || cfg.Addr == "tcp://:6995" {
		listener, err := net.Listen("tcp", "127.0.0.1:0")
		require.NoError(t, err, "Failed to get random port")
		port := listener.Addr().(*net.TCPAddr).Port
		listener.Close()
		cfg.Addr = fmt.Sprintf("tcp://127.0.0.1:%d", port)
	}

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

	// 设置路由（在启动前）
	if setupRoutes != nil {
		setupRoutes(server)
	}

	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 within 3 seconds")

	// 等待服务器准备好接收连接
	require.Eventually(t, func() bool {
		addr := cfg.Addr
		if len(addr) > 6 && addr[:6] == "tcp://" {
			addr = addr[6:]
		}
		testConn, err := net.DialTimeout("tcp", addr, 500*time.Millisecond)
		if err != nil {
			return false
		}
		testConn.Close()
		return true
	}, 5*time.Second, 100*time.Millisecond, "Server should be ready to accept connections")

	// 额外等待，确保服务器完全准备好
	time.Sleep(100 * time.Millisecond)

	return ts
}

// TestServer 测试服务器包装
type TestServer struct {
	Server   nnet.Server
	Addr     string
	stopCh   chan struct{}
	stopOnce sync.Once
	wg       sync.WaitGroup
}

// Stop 停止测试服务器
func (ts *TestServer) Stop() {
	ts.stopOnce.Do(func() {
		close(ts.stopCh)
		if ts.Server != nil {
			ts.Server.Stop()
		}
		// 等待服务器停止
		ts.wg.Wait()
	})
}