package nodeprocessor

import (
	"context"
	"fmt"
	"strings"

	"github.com/usual2970/certimate/internal/deployer"
	"github.com/usual2970/certimate/internal/domain"
	"github.com/usual2970/certimate/internal/repository"
)

type deployNode struct {
	node       *domain.WorkflowNode
	outputRepo WorkflowOutputRepository
	*Logger
}

func NewDeployNode(node *domain.WorkflowNode) *deployNode {
	return &deployNode{
		node:       node,
		Logger:     NewLogger(node),
		outputRepo: repository.NewWorkflowOutputRepository(),
	}
}

func (d *deployNode) Run(ctx context.Context) error {
	d.AddOutput(ctx, d.node.Name, "开始执行")
	// 检查是否部署过（部署过则直接返回，和 v0.2 暂时保持一致）
	output, err := d.outputRepo.GetByNodeId(ctx, d.node.Id)
	if err != nil && !domain.IsRecordNotFound(err) {
		d.AddOutput(ctx, d.node.Name, "查询部署记录失败", err.Error())
		return err
	}
	// 获取部署对象
	// 获取证书
	certSource := d.node.GetConfigString("certificate")

	certSourceSlice := strings.Split(certSource, "#")
	if len(certSourceSlice) != 2 {
		d.AddOutput(ctx, d.node.Name, "证书来源配置错误", certSource)
		return fmt.Errorf("证书来源配置错误: %s", certSource)
	}

	cert, err := d.outputRepo.GetCertificateByNodeId(ctx, certSourceSlice[0])
	if err != nil {
		d.AddOutput(ctx, d.node.Name, "获取证书失败", err.Error())
		return err
	}

	// 未部署过，开始部署
	// 部署过但是证书更新了，重新部署
	// 部署过且证书未更新，直接返回

	if d.deployed(output) && cert.CreatedAt.Before(output.UpdatedAt) {
		d.AddOutput(ctx, d.node.Name, "已部署过且证书未更新")
		return nil
	}

	deploy, err := deployer.NewWithDeployNode(d.node, struct {
		Certificate string
		PrivateKey  string
	}{Certificate: cert.Certificate, PrivateKey: cert.PrivateKey})
	if err != nil {
		d.AddOutput(ctx, d.node.Name, "获取部署对象失败", err.Error())
		return err
	}

	// 部署
	if err := deploy.Deploy(ctx); err != nil {
		d.AddOutput(ctx, d.node.Name, "部署失败", err.Error())
		return err
	}

	d.AddOutput(ctx, d.node.Name, "部署成功")

	// 记录部署结果
	outputId := ""
	if output != nil {
		outputId = output.Id
	}
	output = &domain.WorkflowOutput{
		Meta:       domain.Meta{Id: outputId},
		WorkflowId: GetWorkflowId(ctx),
		NodeId:     d.node.Id,
		Node:       d.node,
		Succeeded:  true,
	}

	if err := d.outputRepo.Save(ctx, output, nil, nil); err != nil {
		d.AddOutput(ctx, d.node.Name, "保存部署记录失败", err.Error())
		return err
	}

	d.AddOutput(ctx, d.node.Name, "保存部署记录成功")

	return nil
}

func (d *deployNode) deployed(output *domain.WorkflowOutput) bool {
	return output != nil && output.Succeeded
}