cids.go

package action

import (
	"encoding/json"
	"errors"
	"strconv"
	"strings"
)

type StemcellCID int
type VMCID int
type DiskCID int

func (stemcellCID *StemcellCID) UnmarshalJSON(data []byte) error {
	if stemcellCID == nil {
		return errors.New("StemcellCID: UnmarshalJSON on nil pointer")
	}

	dataString := strings.Trim(string(data), "\"")
	intValue, err := strconv.Atoi(dataString)
	if err != nil {
		return err
	}

	*stemcellCID = StemcellCID(intValue)

	return nil
}

func (stemcellCID StemcellCID) MarshalJSON() ([]byte, error) {
	return json.Marshal(int(stemcellCID))
}

func (vmCID *VMCID) UnmarshalJSON(data []byte) error {
	if vmCID == nil {
		return errors.New("VMCID: UnmarshalJSON on nil pointer")
	}

	dataString := strings.Trim(string(data), "\"")
	intValue, err := strconv.Atoi(dataString)
	if err != nil {
		return err
	}

	*vmCID = VMCID(intValue)

	return nil
}

func (vmCID VMCID) MarshalJSON() ([]byte, error) {
	return json.Marshal(int(vmCID))
}

func (diskCID *DiskCID) UnmarshalJSON(data []byte) error {
	if diskCID == nil {
		return errors.New("DiskCID: UnmarshalJSON on nil pointer")
	}

	dataString := strings.Trim(string(data), "\"")
	intValue, err := strconv.Atoi(dataString)
	if err != nil {
		return err
	}

	*diskCID = DiskCID(intValue)

	return nil
}

func (diskCID DiskCID) MarshalJSON() ([]byte, error) {
	return json.Marshal(int(diskCID))
}