shop/relation/location.go

package relation

import (
	"github.com/gshopify/service-wrapper/model"
	"github.com/mailru/dbr"
	"gshopper.com/gshopify/shop/graphql/generated"
	"strings"
)

type Location struct {
	Id           string          `db:"id"`
	Name         string          `db:"name"`
	Address1     dbr.NullString  `db:"address_1"`
	Address2     dbr.NullString  `db:"address_2"`
	City         dbr.NullString  `db:"city"`
	Country      dbr.NullString  `db:"country"`
	CountryCode  dbr.NullString  `db:"country_code"`
	Latitude     dbr.NullFloat64 `db:"latitude"`
	Longitude    dbr.NullFloat64 `db:"longitude"`
	Phone        dbr.NullString  `db:"phone"`
	Province     dbr.NullString  `db:"province"`
	ProvinceCode dbr.NullString  `db:"province_code"`
	Zip          dbr.NullString  `db:"zip"`
}

func (l *Location) As() *generated.Location {
	var (
		fmtProvince = strings.Builder{}
		loc         = &generated.Location{
			Address: &generated.LocationAddress{},
			ID:      model.NewId(model.GidLocation, l.Id),
			Name:    l.Name,
		}
	)

	if l.Address1.Valid {
		loc.Address.Address1 = &l.Address1.String
		loc.Address.Formatted = append(loc.Address.Formatted, l.Address1.String)
	}

	if l.Address2.Valid {
		loc.Address.Address2 = &l.Address2.String
		loc.Address.Formatted = append(loc.Address.Formatted, l.Address2.String)
	}

	if l.City.Valid {
		loc.Address.City = &l.City.String
		loc.Address.Formatted = append(loc.Address.Formatted, l.City.String)
	}

	if l.ProvinceCode.Valid {
		loc.Address.ProvinceCode = &l.ProvinceCode.String
		fmtProvince.WriteString(l.ProvinceCode.String)
	}

	if l.Province.Valid {
		loc.Address.Province = &l.Province.String
		if fmtProvince.Len() == 0 {
			fmtProvince.WriteString(l.Province.String)
		}
	}

	if l.Zip.Valid {
		loc.Address.Zip = &l.Zip.String

		if fmtProvince.Len() == 0 {
			fmtProvince.WriteString(" ")
		}

		fmtProvince.WriteString(l.Zip.String)
	}

	if fmtProvince.Len() > 0 {
		loc.Address.Formatted = append(loc.Address.Formatted, fmtProvince.String())
	}

	if l.Country.Valid {
		loc.Address.Country = &l.Country.String
		loc.Address.Formatted = append(loc.Address.Formatted, l.Country.String)
	}

	if code := model.CountryCode(l.CountryCode.String); code.IsValid() {
		s := code.String()
		loc.Address.CountryCode = &s
	}

	if l.Latitude.Valid {
		loc.Address.Latitude = &l.Latitude.Float64
	}

	if l.Longitude.Valid {
		loc.Address.Longitude = &l.Longitude.Float64
	}

	if l.Phone.Valid {
		loc.Address.Phone = &l.Phone.String
	}

	return loc
}