package client

import (
	"fmt"
	"log"
	"net"
	"time"

	"NextConnect/utils"
)

func Start() {
	for {
		connectAndProxy()
		time.Sleep(5 * time.Second) // 自动重连间隔
	}
}

func connectAndProxy() {
	config := utils.LoadConfig()
	serverAddr := fmt.Sprintf("%s:%d", config.Client.ServerAddr, config.Client.ServerPort)
	
	var conn net.Conn
	var err error
	
	// 如果启用了加密，则使用TLS连接
	if config.Proxy.EncryptionEnabled {
		// 这里需要实现TLS连接
		// 暂时使用普通TCP连接
		conn, err = net.Dial("tcp", serverAddr)
		if err != nil {
			log.Printf("Failed to connect to server %s: %v", serverAddr, err)
			return
		}
	} else {
		conn, err = net.Dial("tcp", serverAddr)
		if err != nil {
			log.Printf("Failed to connect to server %s: %v", serverAddr, err)
			return
		}
	}
	
	defer conn.Close()
	
	log.Printf("Connected to server %s", serverAddr)
	
	// 连接认证
	if !authenticateWithServer(conn) {
		log.Println("Authentication with server failed")
		return
	}
	
	// 连接到本地服务
	localConn := connectToLocalService()
	if localConn == nil {
		log.Println("Failed to connect to local service")
		return
	}
	defer localConn.Close()
	
	// 开始数据转发
	forwardData(conn, localConn)
}

// 简单的服务器认证实现
func authenticateWithServer(conn net.Conn) bool {
	// 在实际应用中，这里应该实现更复杂的认证机制
	// 比如基于token、证书或用户名密码的认证
	return true
}