package unsub

import (
	"context"
	"strings"

	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/consts"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/dao"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model/entity"
)

type Repository struct{}

func NewRepository() *Repository {
	return &Repository{}
}

func (r *Repository) GetSourceBatch(ctx context.Context, batchNo string) (*entity.BatchMain, error) {
	batch := &entity.BatchMain{}
	columns := dao.BatchMain.Columns()
	err := dao.BatchMain.Ctx(ctx).
		Fields(
			columns.Id,
			columns.BatchNo,
			columns.BatchType,
			columns.BatchStatus,
			columns.RunResult,
			columns.TenantId,
			columns.DeptId,
			columns.CreatedDept,
			columns.CreatedBy,
		).
		Where(columns.BatchNo, strings.TrimSpace(batchNo)).
		Scan(batch)
	if err != nil {
		return nil, err
	}
	if batch.Id == 0 {
		return nil, nil
	}
	return batch, nil
}

func (r *Repository) GetSourceBatchRows(ctx context.Context, batchNo string) ([]*Row, error) {
	var details []*entity.BatchDetail
	columns := dao.BatchDetail.Columns()
	err := dao.BatchDetail.Ctx(ctx).
		Fields(columns.RowNo, columns.ItemId, columns.Snapshot).
		Where(columns.BatchNo, strings.TrimSpace(batchNo)).
		Where(columns.BatchType, consts.BatchTypeOutbound).
		WhereIn(columns.DetailStatus, []int{consts.BatchDetailStatusSuccess, consts.BatchAsyncStatusSuccess}).
		OrderAsc(columns.RowNo).
		Scan(&details)
	if err != nil {
		return nil, err
	}

	rows := make([]*Row, 0, len(details))
	for _, detail := range details {
		if detail == nil {
			continue
		}
		row := &Row{RowNo: detail.RowNo, Iccid: strings.TrimSpace(detail.ItemId)}
		if detail.Snapshot != nil {
			if row.Iccid == "" {
				row.Iccid = strings.TrimSpace(detail.Snapshot.Get("iccid").String())
			}
			if row.Iccid == "" {
				row.Iccid = strings.TrimSpace(detail.Snapshot.Get("ICCID").String())
			}
			row.Msisdn = strings.TrimSpace(detail.Snapshot.Get("msisdn").String())
			if row.Msisdn == "" {
				row.Msisdn = strings.TrimSpace(detail.Snapshot.Get("MSISDN").String())
			}
			row.Imsi = strings.TrimSpace(detail.Snapshot.Get("imsi").String())
			if row.Imsi == "" {
				row.Imsi = strings.TrimSpace(detail.Snapshot.Get("IMSI").String())
			}
		}
		rows = append(rows, row)
	}
	return rows, nil
}