package upstreamimpl

import (
	"context"
	"encoding/json"

	"git.5iot.com/iotlcm-public/iotlcm-upstream-chan/internal/consts"
	"git.5iot.com/iotlcm-public/iotlcm-upstream-chan/internal/model"

	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/util/gconv"
)

type ydDataDiagnosisRequestFunc func(ctx context.Context, operation, url string, params g.Map) (*ydHTTPResponse, error)

type ydDataDiagnosisResp struct {
	Status  string `json:"status"`
	Message string `json:"message"`
}

// QuerySimDataDiagnosis 物联卡达量限速档位查询
func (s *sSimYD) QuerySimDataDiagnosis(ctx context.Context, input *model.SimDataDiagnosisInput) (model.SimDataDiagnosisOutput, error) {
	return queryYdSimDataDiagnosis(ctx, input, s.getUrl(ctx, "ec/query/sim-data-diagnosis"), s.getTransId, s.requestYd)
}

// QuerySimDataDiagnosis 物联卡达量限速档位查询
func (s *sSimYDMock) QuerySimDataDiagnosis(ctx context.Context, input *model.SimDataDiagnosisInput) (model.SimDataDiagnosisOutput, error) {
	return queryYdSimDataDiagnosis(ctx, input, s.getUrl(ctx, "ec/query/sim-data-diagnosis"), s.getTransId, s.requestYd)
}

// QuerySimDataDiagnosis 物联卡达量限速档位查询
func (s *sSimDX) QuerySimDataDiagnosis(ctx context.Context, input *model.SimDataDiagnosisInput) (model.SimDataDiagnosisOutput, error) {
	items := make(model.SimDataDiagnosisOutput, 0)
	if input == nil {
		return items, nil
	}
	for _, simKey := range input.SimKeys {
		items = append(items, &model.SimBaseRes[any]{
			SimKey:  simKey,
			Code:    consts.ApiInvokeException,
			Message: "电信官方接口暂不提供当前达量限速查询能力",
		})
	}
	return items, nil
}

func queryYdSimDataDiagnosis(
	ctx context.Context,
	input *model.SimDataDiagnosisInput,
	url string,
	getTransID func(context.Context) string,
	request ydDataDiagnosisRequestFunc,
) (model.SimDataDiagnosisOutput, error) {
	items := make(model.SimDataDiagnosisOutput, 0)
	if input == nil {
		return items, nil
	}
	for _, simKey := range input.SimKeys {
		params := g.Map{"transid": getTransID(ctx)}
		paramName, paramValue := pickSimQueryParam(simKey)
		if paramName == "" {
			items = append(items, &model.SimBaseRes[any]{
				SimKey:  simKey,
				Code:    consts.ApiStatusError,
				Message: "missing iccid/msisdn/imsi",
			})
			continue
		}
		params[paramName] = paramValue

		resp, err := request(ctx, "QuerySimDataDiagnosis", url, params)
		if err != nil {
			g.Log().Errorf(ctx, "QuerySimDataDiagnosis get error %+v,url %s, params %+v", err, url, params)
			items = append(items, &model.SimBaseRes[any]{
				SimKey:  simKey,
				Code:    consts.ApiStatusError,
				Message: err.Error(),
			})
			continue
		}

		body := resp.ReadAllString()
		var ydResp ydDataDiagnosisResp
		if err = json.Unmarshal([]byte(body), &ydResp); err != nil {
			g.Log().Errorf(ctx, "QuerySimDataDiagnosis unmarshal error %+v,url %s, params %+v, body %s", err, url, params, body)
			items = append(items, &model.SimBaseRes[any]{
				SimKey:  simKey,
				Code:    consts.ApiStatusError,
				Message: err.Error(),
			})
			continue
		}

		items = append(items, &model.SimBaseRes[any]{
			SimKey:  simKey,
			Code:    gconv.Int(ydResp.Status),
			Message: ydResp.Message,
			ResData: json.RawMessage(body),
		})
	}
	return items, nil
}