package sms77api

import (
	"context"
	"encoding/json"
)

type PricingResource resource

type CountryNetwork struct {
	Comment     string   `json:"comment,omitempty"`
	Features    []string `json:"features,omitempty"`
	Mcc         string   `json:"mcc,omitempty"`
	Mncs        []string `json:"mncs,omitempty"`
	NetworkName string   `json:"networkName,omitempty"`
	Price       float64  `json:"price,omitempty"`
}

type CountryPricing struct {
	CountryCode   string           `json:"countryCode,omitempty"`
	CountryName   string           `json:"countryName,omitempty"`
	CountryPrefix string           `json:"countryPrefix,omitempty"`
	Networks      []CountryNetwork `json:"networks,omitempty"`
}

type PricingFormat string

const (
	PricingFormatCsv  PricingFormat = "csv"
	PricingFormatJson PricingFormat = "json"
)

type PricingParams struct {
	Country string `json:"country,omitempty"`
}

type PricingResponse struct {
	CountCountries int64            `json:"countCountries"`
	CountNetworks  int64            `json:"countNetworks"`
	Countries      []CountryPricing `json:"countries"`
}

type PricingApiParams struct {
	PricingParams
	Format PricingFormat `json:"format,omitempty"`
}

type PricingCsvHeader int

const (
	PricingHeaderCountryCode PricingCsvHeader = iota
	PricingHeaderCountryName
	PricingHeaderCountryPrefix
	PricingHeaderMcc
	PricingHeaderMncs
	PricingHeaderNetworkName
	PricingHeaderPrice
	PricingHeaderFeatures
	PricingHeaderComment
)

var PricingCsvHeaders = map[PricingCsvHeader]string{
	PricingHeaderCountryCode:   "countryCode",
	PricingHeaderCountryName:   "countryName",
	PricingHeaderCountryPrefix: "countryPrefix",
	PricingHeaderMcc:           "mcc",
	PricingHeaderMncs:          "mnc",
	PricingHeaderNetworkName:   "networkName",
	PricingHeaderPrice:         "price",
	PricingHeaderFeatures:      "features",
	PricingHeaderComment:       "comment",
}

type PricingCsvColumn int

const (
	PricingColumnCountryCode PricingCsvColumn = iota
	PricingColumnCountryName
	PricingColumnCountryPrefix
	PricingColumnMcc
	PricingColumnNetworkName
	PricingColumnPrice
	PricingColumnFeatures
	PricingColumnComment
	PricingColumnMncs
)

const EndpointPricing = "pricing"

func (api *PricingResource) Csv(p PricingParams) (string, error) {
	return api.CsvContext(context.Background(), p)
}

func (api *PricingResource) CsvContext(ctx context.Context, p PricingParams) (string, error) {
	res, err := api.client.request(ctx, EndpointPricing, "GET", PricingApiParams{
		PricingParams: p,
		Format:        PricingFormatCsv,
	})

	if err != nil {
		return "", err
	}

	return res, nil
}

func (api *PricingResource) Json(p PricingParams) (*PricingResponse, error) {
	return api.JsonContext(context.Background(), p)
}
func (api *PricingResource) JsonContext(ctx context.Context, p PricingParams) (*PricingResponse, error) {
	res, err := api.client.request(ctx, EndpointPricing, "GET", PricingApiParams{PricingParams: p})

	if err != nil {
		return nil, err
	}

	var js = &PricingResponse{}

	if err := json.Unmarshal([]byte(res), &js); err != nil {
		return nil, err
	}

	return js, nil
}