Initial commit

internal/service/policy.go

@@ -0,0 +1,143 @@
+package service
+
+import (
+	"errors"
+
+	"git.zkcoi.com/zkcoi/meshray/internal/model"
+	"git.zkcoi.com/zkcoi/meshray/internal/store/sqlite"
+	"gorm.io/gorm"
+)
+
+// PolicyService 策略管理服务
+type PolicyService struct {
+	store *sqlite.Store
+}
+
+// NewPolicyService 创建策略服务实例
+func NewPolicyService(store *sqlite.Store) *PolicyService {
+	return &PolicyService{store: store}
+}
+
+// ListPolicies 获取策略列表
+func (s *PolicyService) ListPolicies() ([]model.Policy, error) {
+	var policies []model.Policy
+	err := s.store.DB().Order("created_at desc").Find(&policies).Error
+	return policies, err
+}
+
+// GetPolicyByID 根据 ID 获取策略
+func (s *PolicyService) GetPolicyByID(id uint) (*model.Policy, error) {
+	var policy model.Policy
+	if err := s.store.DB().First(&policy, id).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, errors.New("策略不存在")
+		}
+		return nil, err
+	}
+	return &policy, nil
+}
+
+// CreatePolicy 创建策略
+func (s *PolicyService) CreatePolicy(req *model.Policy) (*model.Policy, error) {
+	if req.Name == "" {
+		return nil, errors.New("策略名称不能为空")
+	}
+
+	// 检查名称是否重复
+	var existing model.Policy
+	if err := s.store.DB().Where("name = ?", req.Name).First(&existing).Error; err == nil {
+		return nil, errors.New("策略名称已存在")
+	}
+
+	// 默认值
+	if req.Type == "" {
+		req.Type = "custom"
+	}
+	req.Enabled = true
+
+	if err := s.store.DB().Create(req).Error; err != nil {
+		return nil, err
+	}
+	return req, nil
+}
+
+// UpdatePolicy 更新策略
+func (s *PolicyService) UpdatePolicy(id uint, updates map[string]interface{}) (*model.Policy, error) {
+	var policy model.Policy
+	if err := s.store.DB().First(&policy, id).Error; err != nil {
+		return nil, errors.New("策略不存在")
+	}
+
+	// 系统默认策略不可编辑
+	if policy.Type == "system" {
+		return nil, errors.New("系统默认策略不可编辑")
+	}
+
+	if err := s.store.DB().Model(&policy).Updates(updates).Error; err != nil {
+		return nil, err
+	}
+	return &policy, nil
+}
+
+// DeletePolicy 删除策略
+func (s *PolicyService) DeletePolicy(id uint) error {
+	var policy model.Policy
+	if err := s.store.DB().First(&policy, id).Error; err != nil {
+		return errors.New("策略不存在")
+	}
+
+	// 系统默认策略不可删除
+	if policy.Type == "system" {
+		return errors.New("系统默认策略不可删除")
+	}
+
+	// 检查是否有关联的组网
+	var count int64
+	s.store.DB().Model(&model.Network{}).Where("policy_id = ?", id).Count(&count)
+	if count > 0 {
+		return errors.New("该策略正在被组网引用，无法删除")
+	}
+
+	return s.store.DB().Delete(&policy).Error
+}
+
+// InitializeDefaultPolicy 初始化默认策略
+func (s *PolicyService) InitializeDefaultPolicy() error {
+	// 检查是否已存在默认策略
+	var count int64
+	s.store.DB().Model(&model.Policy{}).Where("type = ?", "system").Count(&count)
+	if count > 0 {
+		return nil // 已存在默认策略
+	}
+
+	// 创建默认策略
+	defaultPolicy := &model.Policy{
+		Name:        "系统默认策略",
+		Type:        "system",
+		Description: "MeshRay 系统默认传输策略，包含基础的 P2P 直连和 TURN 中继功能",
+		Enabled:     true,
+		IsDefault:   true,
+		LayerConfig: `{
+  "tunnel": {"enabled": true},
+  "obfuscation": {"enabled": false, "method": "xor"},
+  "compression": {"enabled": false, "algorithm": "lz4"},
+  "mux": {"enabled": false, "protocol": "yamux", "streams": 4},
+  "fec": {"enabled": false, "mode": "rs", "redundancy": 20},
+  "qos": {"enabled": false},
+  "nat": {"enabled": true},
+  "relay": {"enabled": true}
+}`,
+		GlobalParams: `{
+  "encryption": "chacha20poly1305",
+  "ipv6_enabled": false,
+  "connect_timeout": 10,
+  "fallback_threshold": 2000
+}`,
+	}
+
+	if err := s.store.DB().Create(defaultPolicy).Error; err != nil {
+		return err
+	}
+
+	return nil
+}