iotlcm/server/internal/logic/iot/t_sim_warning_config.go

// Package gen
// @Link  https://github.com/xiujiecn/xiu-admin
// @Copyright  Copyright (c) 2026 LiXiujie
// @Author  Lxj <li@xiujie.cn>
// @License  https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
// @AutoGenerate Version
package iot

import (
	"context"

	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/consts"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/logic/iot/simcardbatch/warningconfig"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/service"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/utility/iot_excelize"

	"github.com/gogf/gf/v2/errors/gerror"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/net/ghttp"
	"github.com/xuri/excelize/v2"
)

type sIotTSimWarningConfig struct{}

func NewIotTSimWarningConfig() *sIotTSimWarningConfig {
	return &sIotTSimWarningConfig{}
}

func init() {
	service.RegisterIotTSimWarningConfig(NewIotTSimWarningConfig())
}

func (s *sIotTSimWarningConfig) DownloadExcelTemplate(ctx context.Context, itemType int) error {
	f := excelize.NewFile()
	defer func() { _ = f.Close() }()

	processor := warningconfig.NewProcessor(itemType)
	headers := processor.Config().TemplateHeaders
	if err := f.SetSheetRow("Sheet1", "A1", &headers); err != nil {
		return err
	}
	if err := iot_excelize.ApplyHeaderColumnsTextStyle(f); err != nil {
		return err
	}
	fileName := "sim_warning_config_template"
	if len(headers) > 0 && headers[0] == "流量池编号" {
		fileName = "sim_flow_pool_warning_config_template"
	}
	return iot_excelize.WriteHTTP(ctx, f, fileName)
}

func (s *sIotTSimWarningConfig) SendEmailCode(ctx context.Context, in *model.TSimWarningConfigEmailCodeParam) error {
	return warningconfig.SendEmailCode(ctx, in)
}

func (s *sIotTSimWarningConfig) SubmitExcel(ctx context.Context, in *model.TSimWarningConfigExcelSubmitParam) (*model.TSimWarningConfigExcelSubmitRes, error) {
	if in == nil {
		in = &model.TSimWarningConfigExcelSubmitParam{}
	}
	file, err := warningConfigUploadFile(ctx)
	if err != nil {
		return nil, err
	}
	result, err := warningconfig.CreatePendingExcel(ctx, file, in)
	if err != nil {
		return nil, err
	}
	return (*model.TSimWarningConfigExcelSubmitRes)(warningConfigSubmitResult(result)), nil
}

func (s *sIotTSimWarningConfig) SubmitInput(ctx context.Context, in *model.TSimWarningConfigInputSubmitParam) (*model.TSimWarningConfigInputSubmitRes, error) {
	if in == nil {
		in = &model.TSimWarningConfigInputSubmitParam{}
	}
	result, err := warningconfig.CreatePendingInput(ctx, in)
	if err != nil {
		return nil, err
	}
	return (*model.TSimWarningConfigInputSubmitRes)(warningConfigSubmitResult(result)), nil
}

func (s *sIotTSimWarningConfig) SubmitRange(ctx context.Context, in *model.TSimWarningConfigRangeSubmitParam) (*model.TSimWarningConfigRangeSubmitRes, error) {
	if in == nil {
		in = &model.TSimWarningConfigRangeSubmitParam{}
	}
	result, err := warningconfig.CreatePendingRange(ctx, in)
	if err != nil {
		return nil, err
	}
	return (*model.TSimWarningConfigRangeSubmitRes)(warningConfigSubmitResult(result)), nil
}

func warningConfigUploadFile(ctx context.Context) (*ghttp.UploadFile, error) {
	r := g.RequestFromCtx(ctx)
	if r == nil {
		return nil, gerror.New("当前上下文不是HTTP请求")
	}
	file := r.GetUploadFile("file")
	if file == nil {
		return nil, gerror.New("请选择上传文件")
	}
	return file, nil
}

func warningConfigSubmitResult(result *warningconfig.PendingResult) *model.TSimWarningConfigSubmitRes {
	if result == nil {
		return &model.TSimWarningConfigSubmitRes{
			BatchType:        consts.BatchTypeSimWarningConfig,
			BatchAsyncStatus: consts.BatchAsyncStatusPending,
		}
	}
	return &model.TSimWarningConfigSubmitRes{
		BatchNo:          result.BatchNo,
		BatchType:        result.BatchType,
		BatchAsyncStatus: result.BatchAsyncStatus,
		Total:            result.Total,
		ValidCount:       result.ValidCount,
		FailCount:        result.FailCount,
	}
}