package logic

import (
	"context"

	"muyu-apiserver/pkg/tenantctx"
	"muyu-apiserver/rpc/production/internal/svc"
	"muyu-apiserver/rpc/production/pb"

	"github.com/zeromicro/go-zero/core/logx"
)

type CompleteStepLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewCompleteStepLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CompleteStepLogic {
	return &CompleteStepLogic{ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx)}
}

func (l *CompleteStepLogic) CompleteStep(in *pb.CompleteStepReq) (*pb.Empty, error) {
	tenantId := tenantctx.ExtractTenantId(l.ctx)

	// Validate plan exists and belongs to tenant
	plan, err := l.svcCtx.PlanModel.FindOneByPlanId(l.ctx, in.PlanId)
	if err != nil {
		return nil, ErrPlanNotFound
	}
	if plan.TenantId != tenantId {
		return nil, ErrPermissionDenied
	}

	// Find steps for this plan and check the target step
	steps, err := l.svcCtx.ProcessStepModel.FindByPlanId(l.ctx, in.PlanId)
	if err != nil {
		return nil, ErrStepNotFound
	}

	found := false
	for _, s := range steps {
		if s.StepType == in.StepType {
			found = true
			if s.Status == 2 {
				return nil, ErrStepAlreadyComplete
			}
			break
		}
	}
	if !found {
		return nil, ErrStepNotFound
	}

	operatorId := in.OperatorId
	if operatorId == "" {
		operatorId = tenantId
	}

	// Update step status to 2 (completed)
	if err := l.svcCtx.ProcessStepModel.UpdateStepStatus(l.ctx, in.PlanId, in.StepType, 2, operatorId); err != nil {
		return nil, err
	}

	return &pb.Empty{}, nil
}