Product/relation/product_variant.go

package relation

import (
	"github.com/gshopify/service-wrapper/model"
	"github.com/gshopify/service-wrapper/scalar"
	"github.com/mailru/dbr"
	"gshopper.com/gshopify/products/graphql/generated"
	"time"
)

type ProductVariant struct {
	Id                  string          `db:"id"`
	ProductId           string          `db:"product_id"`
	InventoryItemId     string          `db:"inventory_item_id"`
	Price               float64         `db:"price"`
	CompareAtPrice      float64         `db:"compare_at_price"`
	Position            int8            `db:"position"`
	Image               dbr.NullString  `db:"image"`
	InventoryManagement string          `db:"inventory_management"`
	InventoryPolicy     string          `db:"inventory_policy"`
	Grams               dbr.NullFloat64 `db:"grams"`
	Weight              dbr.NullFloat64 `db:"weight"`
	WeightUnit          string          `db:"weight_unit"`
	CreatedAt           time.Time       `db:"created_at"`
	UpdatedAt           time.Time       `db:"updated_at"`
	PublishedAt         *time.Time      `db:"published_at"`
	DeletedAt           *time.Time      `db:"deleted_at"`
	Options             []struct {
		Field0 string
		Field1 int32
	} `db:"options"`

	Title             string         `db:"title"`
	Sku               dbr.NullString `db:"sku"`
	Barcode           dbr.NullString `db:"barcode"`
	RequiresShipping  bool           `db:"requires_shipping"`
	QuantityAvailable int            `db:"available"`
	ForSale           bool           `db:"for_sale"`
}

func (v *ProductVariant) As() *generated.ProductVariant {
	variant := generated.ProductVariant{
		AvailableForSale:    v.ForSale,
		CurrentlyNotInStock: v.QuantityAvailable <= 0,

		Image:                nil,
		SelectedOptions:      nil,
		UnitPrice:            nil,
		UnitPriceMeasurement: nil,

		CompareAtPrice: &generated.MoneyV2{
			Amount:       scalar.NewDecimal(v.CompareAtPrice),
			CurrencyCode: generated.CurrencyCodeUsd,
		},

		ID: model.NewId(model.GidVariant, v.Id),
		Price: &generated.MoneyV2{
			Amount:       scalar.NewDecimal(v.Price),
			CurrencyCode: generated.CurrencyCodeUsd,
		},
		Product:           &generated.Product{ID: model.NewId(model.GidProduct, v.ProductId)},
		QuantityAvailable: &v.QuantityAvailable,
		RequiresShipping:  v.RequiresShipping,
		Title:             v.Title,
		WeightUnit:        generated.WeightUnit(v.WeightUnit),
	}

	variant.Metafields = append(variant.Metafields, &generated.Metafield{
		Namespace: model.GidVariant.String(),
		Key:       "position",
		Type:      model.MetaFieldTypeNumberInteger.String(),
		Value:     model.MetaFieldValue(model.MetaFieldTypeNumberInteger, v.Position),
	})

	if v.Weight.Valid {
		s := v.Weight.Float64
		variant.Weight = &s
	}

	if v.Sku.Valid {
		s := v.Sku.String
		variant.Sku = &s
	}

	if v.Barcode.Valid {
		s := v.Barcode.String
		variant.Barcode = &s
	}

	return &variant
}