hue hue hue

effects/serializable.go

@@ -0,0 +1,56 @@
+package effects
+
+import (
+	"encoding/json"
+	"errors"
+	lucifer3 "git.aiterp.net/lucifer3/server"
+)
+
+type serializedEffect struct {
+	Manual   *Manual   `json:"manual,omitempty"`
+	Gradient *Gradient `json:"gradient,omitempty"`
+	Pattern  *Pattern  `json:"pattern,omitempty"`
+	Random   *Random   `json:"random,omitempty"`
+}
+
+type Serializable struct {
+	lucifer3.Effect
+}
+
+func (s *Serializable) UnmarshalJSON(raw []byte) error {
+	value := serializedEffect{}
+	err := json.Unmarshal(raw, &value)
+	if err != nil {
+		return err
+	}
+
+	switch {
+	case value.Manual != nil:
+		s.Effect = *value.Manual
+	case value.Gradient != nil:
+		s.Effect = *value.Gradient
+	case value.Pattern != nil:
+		s.Effect = *value.Pattern
+	case value.Random != nil:
+		s.Effect = *value.Random
+	default:
+		return errors.New("unsupported effect")
+	}
+
+	return nil
+}
+
+func (s *Serializable) MarshalJSON() ([]byte, error) {
+	switch effect := s.Effect.(type) {
+	case Manual:
+		return json.Marshal(serializedEffect{Manual: &effect})
+	case Gradient:
+		return json.Marshal(serializedEffect{Gradient: &effect})
+	case Pattern:
+		return json.Marshal(serializedEffect{Pattern: &effect})
+	case Random:
+		return json.Marshal(serializedEffect{Random: &effect})
+	default:
+		panic(s.Effect.EffectDescription() + "is not understood by serializer")
+	}
+}