ocdp-go/backend/internal/adapter/output/helm/mock/helm_client_mock.go

package mock

import (
	"context"
	"fmt"
	"time"
	
	"github.com/ocdp/cluster-service/internal/domain/entity"
	"github.com/ocdp/cluster-service/internal/domain/repository"
)

// HelmClientMock Helm 客户端 Mock 实现
type HelmClientMock struct {
	// Mock 数据存储
	releases map[string]map[string]*entity.Instance    // clusterID -> releaseName -> instance
	history  map[string]map[string][]*entity.ReleaseHistory // clusterID -> releaseName -> []history
}

// NewHelmClientMock 创建 Mock 实现
func NewHelmClientMock() repository.HelmClient {
	return &HelmClientMock{
		releases: make(map[string]map[string]*entity.Instance),
		history:  make(map[string]map[string][]*entity.ReleaseHistory),
	}
}

func (c *HelmClientMock) Install(ctx context.Context, cluster *entity.Cluster, instance *entity.Instance) error {
	// 初始化集群数据
	if c.releases[cluster.ID] == nil {
		c.releases[cluster.ID] = make(map[string]*entity.Instance)
		c.history[cluster.ID] = make(map[string][]*entity.ReleaseHistory)
	}
	
	// 检查是否已存在
	key := fmt.Sprintf("%s/%s", instance.Namespace, instance.Name)
	if _, exists := c.releases[cluster.ID][key]; exists {
		return entity.ErrInstanceExists
	}
	
	// Mock 安装
	instance.Status = entity.StatusDeployed
	instance.Revision = 1
	instance.UpdatedAt = time.Now()
	
	c.releases[cluster.ID][key] = instance
	
	// 添加历史记录
	c.history[cluster.ID][key] = []*entity.ReleaseHistory{
		{
			Revision:    1,
			Updated:     time.Now(),
			Status:      entity.StatusDeployed,
			Chart:       fmt.Sprintf("%s-%s", instance.Chart, instance.Version),
			AppVersion:  instance.Version,
			Description: "Install complete",
		},
	}
	
	return nil
}

func (c *HelmClientMock) Upgrade(ctx context.Context, cluster *entity.Cluster, instance *entity.Instance) error {
	key := fmt.Sprintf("%s/%s", instance.Namespace, instance.Name)
	
	existing, exists := c.releases[cluster.ID][key]
	if !exists {
		return entity.ErrInstanceNotFound
	}
	
	// Mock 升级
	instance.Revision = existing.Revision + 1
	instance.Status = entity.StatusDeployed
	instance.UpdatedAt = time.Now()
	
	c.releases[cluster.ID][key] = instance
	
	// 添加历史记录
	history := &entity.ReleaseHistory{
		Revision:    instance.Revision,
		Updated:     time.Now(),
		Status:      entity.StatusDeployed,
		Chart:       fmt.Sprintf("%s-%s", instance.Chart, instance.Version),
		AppVersion:  instance.Version,
		Description: "Upgrade complete",
	}
	c.history[cluster.ID][key] = append(c.history[cluster.ID][key], history)
	
	return nil
}

func (c *HelmClientMock) Uninstall(ctx context.Context, cluster *entity.Cluster, releaseName, namespace string) error {
	key := fmt.Sprintf("%s/%s", namespace, releaseName)
	
	if _, exists := c.releases[cluster.ID][key]; !exists {
		return entity.ErrInstanceNotFound
	}
	
	// Mock 卸载
	delete(c.releases[cluster.ID], key)
	
	return nil
}

func (c *HelmClientMock) Rollback(ctx context.Context, cluster *entity.Cluster, releaseName, namespace string, revision int) error {
	key := fmt.Sprintf("%s/%s", namespace, releaseName)
	
	instance, exists := c.releases[cluster.ID][key]
	if !exists {
		return entity.ErrInstanceNotFound
	}
	
	// 检查历史记录是否存在
	histories := c.history[cluster.ID][key]
	if revision > len(histories) || revision < 1 {
		return fmt.Errorf("revision %d not found", revision)
	}
	
	// Mock 回滚
	instance.Revision = len(histories) + 1
	instance.Status = entity.StatusDeployed
	instance.UpdatedAt = time.Now()
	
	c.releases[cluster.ID][key] = instance
	
	// 添加回滚历史记录
	history := &entity.ReleaseHistory{
		Revision:    instance.Revision,
		Updated:     time.Now(),
		Status:      entity.StatusDeployed,
		Chart:       instance.Chart,
		AppVersion:  instance.Version,
		Description: fmt.Sprintf("Rollback to revision %d", revision),
	}
	c.history[cluster.ID][key] = append(c.history[cluster.ID][key], history)
	
	return nil
}

func (c *HelmClientMock) GetStatus(ctx context.Context, cluster *entity.Cluster, releaseName, namespace string) (*entity.Instance, error) {
	key := fmt.Sprintf("%s/%s", namespace, releaseName)
	
	instance, exists := c.releases[cluster.ID][key]
	if !exists {
		return nil, entity.ErrInstanceNotFound
	}
	
	return instance, nil
}

func (c *HelmClientMock) GetHistory(ctx context.Context, cluster *entity.Cluster, releaseName, namespace string) ([]*entity.ReleaseHistory, error) {
	key := fmt.Sprintf("%s/%s", namespace, releaseName)
	
	if _, exists := c.releases[cluster.ID][key]; !exists {
		return nil, entity.ErrInstanceNotFound
	}
	
	histories := c.history[cluster.ID][key]
	if histories == nil {
		return []*entity.ReleaseHistory{}, nil
	}
	
	return histories, nil
}

func (c *HelmClientMock) List(ctx context.Context, cluster *entity.Cluster, namespace string) ([]*entity.Instance, error) {
	clusterReleases := c.releases[cluster.ID]
	if clusterReleases == nil {
		return []*entity.Instance{}, nil
	}
	
	instances := make([]*entity.Instance, 0)
	for key, instance := range clusterReleases {
		// 如果指定了 namespace，只返回该 namespace 的
		if namespace != "" && namespace != "all" {
			keyNamespace := instance.Namespace
			if keyNamespace != namespace {
				continue
			}
		}
		instances = append(instances, c.releases[cluster.ID][key])
	}
	
	return instances, nil
}

func (c *HelmClientMock) GetValues(ctx context.Context, cluster *entity.Cluster, releaseName, namespace string) (map[string]interface{}, error) {
	key := fmt.Sprintf("%s/%s", namespace, releaseName)
	
	instance, exists := c.releases[cluster.ID][key]
	if !exists {
		return nil, entity.ErrInstanceNotFound
	}
	
	return instance.Values, nil
}