package lifecycle

import (
	"context"
	"strings"

	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/consts"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/batchimport"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"

	"github.com/gogf/gf/v2/database/gdb"
	"github.com/gogf/gf/v2/encoding/gjson"
	"github.com/gogf/gf/v2/errors/gerror"
	"github.com/gogf/gf/v2/frame/g"
)

type Processor struct {
	config OperationConfig
}

func NewProcessor(operation Operation) (*Processor, error) {
	config, err := operation.Config()
	if err != nil {
		return nil, err
	}
	return &Processor{config: config}, nil
}

func (p *Processor) Config() batchimport.ProcessorConfig {
	return batchimport.ProcessorConfig{
		Code:            "sim_card_" + p.config.Code,
		BatchType:       p.config.BatchType,
		BatchNoPrefix:   p.config.BatchPrefix,
		TaskName:        p.config.TaskName,
		ExportScene:     p.config.ExportScene,
		ImportDir:       p.config.ImportDir,
		ResultDir:       p.config.ResultDir,
		MaxFileSize:     consts.SimCardLifecycleMaxFileSize,
		MaxRows:         consts.SimCardBatchMaxRows,
		TemplateHeaders: []string{"ICCID", "MSISDN", "IMSI"},
		RequiredHeaders: []string{"ICCID"},
	}
}

func (p *Processor) ParseRows(ctx context.Context, filePath string) ([]batchimport.Row, error) {
	return ParseExcelFileRows(ctx, filePath)
}

func (p *Processor) ValidateRows(_ context.Context, rows []batchimport.Row) error {
	return validateBasicRows(rows)
}

func (p *Processor) Execute(context.Context, gdb.TX, *batchimport.SubmitRequest, []batchimport.Row) error {
	return gerror.New("停复机批次只能由异步执行器执行")
}

func (p *Processor) BuildBatchInfo(_ context.Context, req batchimport.SubmitRequest, result *batchimport.Result, importOss *model.SysOssSaveContentModel, resultOss *model.SysOssSaveContentModel, downloadID int64) *gjson.Json {
	info := g.Map{
		"operation": p.config.Code,
	}
	if p.config.OperType != "" {
		info["operType"] = p.config.OperType
	}
	if p.config.Operation == OperationRestart {
		info["stopOperType"] = consts.SimCardStatusOperTypeStop
		info["resumeOperType"] = consts.SimCardStatusOperTypeStart
		info["resumeDelaySeconds"] = 300
		info["restart"] = g.Map{"phase": "created"}
	}
	if options, ok := req.BizParams.(*SubmitOptions); ok && options != nil {
		if remark := strings.TrimSpace(options.Remark); remark != "" {
			info["remark"] = remark
		}
	}
	if result != nil && result.ImportFileOssId > 0 {
		info["importFileOssId"] = result.ImportFileOssId
	} else if importOss != nil && importOss.OssId > 0 {
		info["importFileOssId"] = importOss.OssId
	}
	if resultOss != nil {
		info["resultFileOssId"] = resultOss.OssId
	}
	if downloadID > 0 {
		info["resultDownloadId"] = downloadID
	}
	return gjson.New(info)
}