Booking

interoperability/ocss/prices.go

@@ -1,5 +1,10 @@
 package ocss
 
+import (
+	"bytes"
+	"encoding/json"
+)
+
 type PriceType int64
 
 const (
@@ -8,6 +13,36 @@ const (
 	Unknown
 )
 
+var priceTypeToID = map[string]PriceType{
+	"FREE":    Free,
+	"PAYING":  Paying,
+	"UNKNOWN": Unknown,
+}
+
+var priceTypeToString = map[PriceType]string{
+	Free:    "FREE",
+	Paying:  "PAYING",
+	Unknown: "UNKNOWN",
+}
+
+func (s PriceType) MarshalJSON() ([]byte, error) {
+	buffer := bytes.NewBufferString(`"`)
+	buffer.WriteString(priceTypeToString[s])
+	buffer.WriteString(`"`)
+	return buffer.Bytes(), nil
+}
+
+func (bs *PriceType) UnmarshalJSON(b []byte) error {
+	var j string
+	err := json.Unmarshal(b, &j)
+	if err != nil {
+		return err
+	}
+	// Note that if the string cannot be found then it will be set to the zero value, 'Created' in this case.
+	*bs = priceTypeToID[j]
+	return nil
+}
+
 type Price struct {
 	Type     *PriceType `json:"type,omitempty"`
 	Amount   *float64   `json:"amount,omitempty"`