package integration

import (
	"testing"
	"time"

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

// StartTestServer 启动测试服务器（简化版本，使用helper.go中的辅助函数）
func StartTestServer(t *testing.T, cfg *nnet.Config) *TestServer {
	return StartTestServerWithRoutes(t, cfg, nil)
}

// NewTestClient 创建测试客户端
func NewTestClient(t *testing.T, addr string, cfg *nnet.ClientConfig) nnet.Client {
	if cfg == nil {
		cfg = &nnet.ClientConfig{}
	}
	cfg.Addr = addr
	if cfg.ConnectTimeout == 0 {
		cfg.ConnectTimeout = 3 * time.Second
	}
	if cfg.ReadTimeout == 0 {
		cfg.ReadTimeout = 3 * time.Second
	}
	if cfg.WriteTimeout == 0 {
		cfg.WriteTimeout = 3 * time.Second
	}

	client := nnet.NewClient(cfg)
	require.NotNil(t, client, "Failed to create client")

	return client
}

// ConnectTestClient 连接测试客户端
func ConnectTestClient(t *testing.T, client nnet.Client) {
	err := client.Connect()
	require.NoError(t, err, "Failed to connect client")
	assert.True(t, client.IsConnected(), "Client should be connected")
}

// RequestWithTimeout 发送请求并等待响应
func RequestWithTimeout(t *testing.T, client nnet.Client, data []byte, timeout time.Duration) []byte {
	if timeout == 0 {
		timeout = 5 * time.Second // 默认超时时间
	}

	resp, err := client.Request(data, timeout)
	if err != nil {
		t.Logf("Request failed: %v, data: %q", err, string(data))
	}
	require.NoError(t, err, "Request should succeed")
	return resp
}

// CleanupTestServer 清理测试服务器
func CleanupTestServer(t *testing.T, ts *TestServer) {
	if ts != nil {
		ts.Stop()
	}
}

// CleanupTestClient 清理测试客户端
func CleanupTestClient(t *testing.T, client nnet.Client) {
	if client != nil {
		if client.IsConnected() {
			client.Disconnect()
		}
		client.Close()
	}
}

// WaitForServer 等待服务器启动
func WaitForServer(t *testing.T, server nnet.Server, timeout time.Duration) {
	require.Eventually(t, func() bool {
		return server.Started()
	}, timeout, 100*time.Millisecond, "Server should start within timeout")
}

// ExtractPort 从地址中提取端口
func ExtractPort(addr string) string {
	// 从 "tcp://:6995" 提取 ":6995" 或从 "tcp://127.0.0.1:6995" 提取 ":6995"
	// 简化实现：假设地址格式正确
	if len(addr) > 6 && addr[:6] == "tcp://" {
		hostPort := addr[6:]
		// 查找最后一个 ":"
		lastColon := -1
		for i := len(hostPort) - 1; i >= 0; i-- {
			if hostPort[i] == ':' {
				lastColon = i
				break
			}
		}
		if lastColon >= 0 {
			return hostPort[lastColon:]
		}
	}
	return ""
}