// Package core 定义 VPN 协议统一抽象接口与运行时管理器。
// 所有协议（内置或自定义插件）只需实现 Protocol 接口即可接入平台。
package core

import (
	"fmt"
	"sync"

	"vpn-platform/backend/internal/model"
)

// Status 专线运行状态
type Status struct {
	Running bool   `json:"running"`
	State   string `json:"state"` // running / stopped / error / undeployed
	Detail  string `json:"detail"`
	RxBytes int64  `json:"rx_bytes"`
	TxBytes int64  `json:"tx_bytes"`
	// Peers 为协议可选的对端实时状态。WireGuard 等多对端协议使用它向
	// 管理端返回逐 Peer 的握手时间、端点和内核计数器，不能用隧道总流量冒充。
	Peers     []PeerStatus `json:"peers,omitempty"`
	UpdatedAt int64        `json:"updated_at"`
}

// PeerStatus 是多对端协议的真实运行时数据。RxBytes/TxBytes 来自协议
// 进程或内核的累计计数器；服务重启后计数器可能从零重新开始。
type PeerStatus struct {
	ID                string `json:"id"`
	Name              string `json:"name,omitempty"`
	PublicKey         string `json:"public_key"`
	Endpoint          string `json:"endpoint,omitempty"`
	AllowedIPs        string `json:"allowed_ips,omitempty"`
	LatestHandshakeAt int64  `json:"latest_handshake_at,omitempty"`
	Connected         bool   `json:"connected"`
	RxBytes           int64  `json:"rx_bytes"`
	TxBytes           int64  `json:"tx_bytes"`
}

// Protocol 统一协议接口 —— 自定义协议按此契约实现即可热插拔接入
type Protocol interface {
	// Name 协议标识，如 wireguard / ipsec / frp
	Name() string
	// DisplayName 展示名
	DisplayName() string
	// Validate 创建前校验参数，返回规范化后的配置
	Validate(cfg map[string]interface{}) (map[string]interface{}, error)
	// Create 生成服务端配置并落盘（不启动）
	Create(t *model.Tunnel) error
	// Start 启动实例
	Start(t *model.Tunnel) error
	// Stop 停止实例
	Stop(t *model.Tunnel) error
	// Status 查询运行状态与流量
	Status(t *model.Tunnel) (Status, error)
	// GenerateClientConfig 生成客户端配置（用于下载 / 二维码）
	GenerateClientConfig(t *model.Tunnel) (string, error)
	// Delete 清理实例配置与运行时资源
	Delete(t *model.Tunnel) error
	// Available 检测系统依赖是否就绪（wg / strongswan / frps 等）
	Available() (bool, string)
}

// Manager 协议注册中心 + 实例状态缓存
type Manager struct {
	mu        sync.RWMutex
	protocols map[string]Protocol
	statuses  map[uint]Status
}

func NewManager() *Manager {
	return &Manager{protocols: map[string]Protocol{}, statuses: map[uint]Status{}}
}

// Register 注册协议实现（内置协议与自定义插件共用入口）
func (m *Manager) Register(p Protocol) {
	m.mu.Lock()
	defer m.mu.Unlock()
	m.protocols[p.Name()] = p
}

// Get 获取协议实现
func (m *Manager) Get(name string) (Protocol, error) {
	m.mu.RLock()
	defer m.mu.RUnlock()
	p, ok := m.protocols[name]
	if !ok {
		return nil, fmt.Errorf("不支持的协议类型: %s", name)
	}
	return p, nil
}

// List 列出全部已注册协议
func (m *Manager) List() []Protocol {
	m.mu.RLock()
	defer m.mu.RUnlock()
	out := make([]Protocol, 0, len(m.protocols))
	for _, p := range m.protocols {
		out = append(out, p)
	}
	return out
}

// SetStatus 更新实例状态缓存（monitor 写入）
func (m *Manager) SetStatus(tunnelID uint, s Status) {
	m.mu.Lock()
	defer m.mu.Unlock()
	m.statuses[tunnelID] = s
}

// GetStatus 读取实例状态缓存
func (m *Manager) GetStatus(tunnelID uint) (Status, bool) {
	m.mu.RLock()
	defer m.mu.RUnlock()
	s, ok := m.statuses[tunnelID]
	return s, ok
}

// RemoveStatus 删除状态缓存
func (m *Manager) RemoveStatus(tunnelID uint) {
	m.mu.Lock()
	defer m.mu.Unlock()
	delete(m.statuses, tunnelID)
}