package connection

import (
	"net"
	"sync"
	"time"

	"github.com/rs/xid"
)

// UDPConnection UDP连接包装器
type UDPConnection struct {
	id         string
	addr       *net.UDPAddr
	conn       *net.UDPConn
	mu         sync.RWMutex
	createdAt  time.Time
	lastActive time.Time
	attributes map[string]interface{}
}

// NewUDPConnection 创建UDP连接
// 如果id为空，将自动生成唯一的xid
func NewUDPConnection(id string, addr *net.UDPAddr, conn *net.UDPConn) *UDPConnection {
	if id == "" {
		id = xid.New().String()
	}
	return &UDPConnection{
		id:         id,
		addr:       addr,
		conn:       conn,
		createdAt:  time.Now(),
		lastActive: time.Now(),
		attributes: make(map[string]interface{}),
	}
}

// ID 获取连接ID
func (c *UDPConnection) ID() string {
	return c.id
}

// RemoteAddr 获取远程地址
func (c *UDPConnection) RemoteAddr() string {
	if c.addr != nil {
		return c.addr.String()
	}
	return ""
}

// LocalAddr 获取本地地址
func (c *UDPConnection) LocalAddr() string {
	if c.conn != nil {
		return c.conn.LocalAddr().String()
	}
	return ""
}

// Write 写入数据
// 如果设置了gnet_conn属性，使用gnet.Conn的Write方法（gnet会自动处理UDP目标地址）
// 否则使用传统的WriteToUDP方法
func (c *UDPConnection) Write(data []byte) error {
	c.mu.Lock()
	c.lastActive = time.Now()
	
	// 检查是否有gnet.Conn（通过属性传递）
	gnetConn := c.attributes["gnet_conn"]
	c.mu.Unlock()

	// 如果使用gnet，通过gnet.Conn的Write方法发送
	// gnet会自动处理UDP的目标地址（因为gnet.Conn已经绑定了远程地址）
	if gnetConn != nil {
		// gnet.Conn应该有Write方法
		if writer, ok := gnetConn.(interface {
			Write([]byte) error
		}); ok {
			return writer.Write(data)
		}
		// 如果gnet.Conn有SendTo方法（UDP专用），优先使用
		if sendTo, ok := gnetConn.(interface {
			SendTo([]byte) error
		}); ok {
			return sendTo.SendTo(data)
		}
	}

	// 回退到传统的WriteToUDP
	if c.conn != nil && c.addr != nil {
		_, err := c.conn.WriteToUDP(data, c.addr)
		return err
	}
	return nil
}

// Close 关闭连接（UDP连接通常不需要关闭）
func (c *UDPConnection) Close() error {
	// UDP是无连接的，这里不做实际关闭操作
	return nil
}

// SetAttribute 设置属性
func (c *UDPConnection) SetAttribute(key string, value interface{}) {
	c.mu.Lock()
	defer c.mu.Unlock()
	if c.attributes == nil {
		c.attributes = make(map[string]interface{})
	}
	c.attributes[key] = value
}

// GetAttribute 获取属性
func (c *UDPConnection) GetAttribute(key string) interface{} {
	c.mu.RLock()
	defer c.mu.RUnlock()
	if c.attributes == nil {
		return nil
	}
	return c.attributes[key]
}

// LastActive 获取最后活动时间
func (c *UDPConnection) LastActive() time.Time {
	c.mu.RLock()
	defer c.mu.RUnlock()
	return c.lastActive
}

// UpdateActive 更新活动时间
func (c *UDPConnection) UpdateActive() {
	c.mu.Lock()
	defer c.mu.Unlock()
	c.lastActive = time.Now()
}