package locodedb

import (
	locodecolumn "github.com/nspcc-dev/neofs-node/pkg/util/locode/column"
	"github.com/pkg/errors"
)

// CountryCode represents country code for
// storage in the NeoFS location database.
type CountryCode locodecolumn.CountryCode

// CountryCodeFromString parses string UN/LOCODE country code
// and returns CountryCode.
func CountryCodeFromString(s string) (*CountryCode, error) {
	cc, err := locodecolumn.CountryCodeFromString(s)
	if err != nil {
		return nil, errors.Wrap(err, "could not parse country code")
	}

	return CountryFromColumn(cc)
}

// CountryFromColumn converts UN/LOCODE country code to CountryCode.
func CountryFromColumn(cc *locodecolumn.CountryCode) (*CountryCode, error) {
	return (*CountryCode)(cc), nil
}

func (c *CountryCode) String() string {
	syms := (*locodecolumn.CountryCode)(c).Symbols()
	return string(syms[:])
}