arikawa/utils/json/option/custom.go

package option

import (
	"strconv"

	"github.com/diamondburned/arikawa/v2/discord"
)

// ================================ Seconds ================================

// Seconds is the option type for discord.Seconds.
type Seconds = *discord.Seconds

// ZeroSeconds are 0 Seconds.
var ZeroSeconds = NewSeconds(0)

// NewString creates a new Seconds with the value of the passed discord.Seconds.
func NewSeconds(s discord.Seconds) Seconds { return &s }

// ================================ Color ================================

// Color is the option type for discord.Color.
type Color = *discord.Color

// NewString creates a new Color with the value of the passed discord.Color.
func NewColor(s discord.Color) Color { return &s }

// ================================ NullableColor ================================

// Nullable is a nullable version of discord.Color.
type NullableColor = *NullableColorData

type NullableColorData struct {
	Val  discord.Color
	Init bool
}

// NullColor serializes to JSON null.
var NullColor = &NullableColorData{}

// NewNullableColor creates a new non-null NullableColor using the value of the
// passed discord.Color.
func NewNullableColor(v discord.Color) NullableColor {
	return &NullableColorData{
		Val:  v,
		Init: true,
	}
}

func (i NullableColorData) MarshalJSON() ([]byte, error) {
	if !i.Init {
		return []byte("null"), nil
	}
	return []byte(strconv.FormatUint(uint64(i.Val), 10)), nil
}

func (i *NullableColorData) UnmarshalJSON(json []byte) error {
	s := string(json)

	if s == "null" {
		*i = *NullColor
		return nil
	}

	v, err := strconv.ParseUint(s, 10, 32)

	i.Val = discord.Color(v)
	i.Init = true

	return err
}