65 lines
954 B
Go
65 lines
954 B
Go
package notifications
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
const (
|
|
IMAGE_TYPE = "image"
|
|
LIST_TYPE = "list"
|
|
)
|
|
|
|
type ImageNotification struct {
|
|
Type string
|
|
|
|
ImageID uuid.UUID
|
|
ImageName string
|
|
|
|
Status string
|
|
}
|
|
|
|
type ListNotification struct {
|
|
Type string
|
|
|
|
ListID uuid.UUID
|
|
Name string
|
|
|
|
Status string
|
|
}
|
|
|
|
type Notification struct {
|
|
image *ImageNotification
|
|
list *ListNotification
|
|
}
|
|
|
|
func GetImageNotification(image ImageNotification) Notification {
|
|
return Notification{
|
|
image: &image,
|
|
}
|
|
}
|
|
|
|
func GetListNotification(list ListNotification) Notification {
|
|
return Notification{
|
|
list: &list,
|
|
}
|
|
}
|
|
|
|
func (n Notification) MarshalJSON() ([]byte, error) {
|
|
if n.image != nil {
|
|
return json.Marshal(n.image)
|
|
}
|
|
|
|
if n.list != nil {
|
|
return json.Marshal(n.list)
|
|
}
|
|
|
|
return nil, fmt.Errorf("no image or list present")
|
|
}
|
|
|
|
func (n *Notification) UnmarshalJSON(data []byte) error {
|
|
return fmt.Errorf("unimplemented")
|
|
}
|