get_available_numbers.go

package sms_activate_go

import (
	"encoding/json"
	"io"
	"net/http"
	"strconv"
	"strings"

	"github.com/google/go-querystring/query"
)

const (
	numsStatusAction = "getNumbersStatus"
	russiaID         = 0
	ukraineID        = 1
	kazakhstanID     = 2
)

// GetAvailableNumbers returns available phone numbers to rent by country.
// (It is also possible to add certain mobile operators. But only for countries with ID 0, 1, 2 - Russia, Ukraine, Kazakhstan, respectively)
//
// Example
//
//	av, err := client.GetAvailableNumbers(0)
//	if err != nil {
//	    log.Fatal(err)
//	}
//	for k, v := range av {
//	    fmt.Printf("Service: %s, Count: %s\n", k, v)
//	}
func (act *SMSActivate) GetAvailableNumbers(country int, operator ...string) (map[string]string, error) {
	if country < allCountries || country > maxAvailableCountries {
		return nil, BadCountryNum
	}

	req, _ := http.NewRequest(http.MethodGet, act.BaseURL.String(), nil)

	numsReq := baseRequest{
		APIKey: apiKeyQuery,
		Action: numsStatusAction,
	}
	val, err := query.Values(numsReq)
	if err != nil {
		return nil, err
	}
	if country > allCountries {
		val.Add(countryQuery, strconv.Itoa(country))
	}

	if country == russiaID || country == ukraineID || country == kazakhstanID {
		var operators string
		operators = strings.Join(operator, ",")
		val.Add(operatorQuery, operators)
	}
	req.URL.RawQuery = val.Encode()

	resp, err := act.httpClient.Do(req)
	if err != nil {
		return nil, err
	}
	defer resp.Body.Close()

	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return nil, err
	}
	switch string(body) {
	case badKeyMsg:
		return nil, ErrBadKey
	case errorSQLMsg:
		return nil, ErrSQL
	}

	var data map[string]string

	err = json.Unmarshal(body, &data)
	if err != nil {
		return nil, err
	}
	return data, nil
}