payload.go RawHistoryFolder

1package common
2
3import (
4	"encoding/json"
5	"fmt"
6)
7
8type RequestPayload struct {
9	Type    string  `json:"type"`
10	Payload Payload `json:"payload"`
11}
12
13type Payload interface {
14	GetType() string
15}
16
17func (p *RequestPayload) UnmarshalJSON(data []byte) error {
18	var raw map[string]json.RawMessage
19	err := json.Unmarshal(data, &raw)
20	if err != nil {
21		return err
22	}
23
24	if err := json.Unmarshal(raw["type"], &p.Type); err != nil {
25		return err
26	}
27
28	switch p.Type {
29	case "ping":
30		var payload PingPayload
31		if err := json.Unmarshal(raw["payload"], &payload); err != nil {
32			return err
33		}
34		p.Payload = payload
35	case "push":
36		var payload PushPayload
37		if err := json.Unmarshal(raw["payload"], &payload); err != nil {
38			return err
39		}
40		p.Payload = payload
41	}
42	return nil
43}
44
45type PingPayload struct {
46	Message string `json:"message"`
47}
48
49func (p PingPayload) GetType() string {
50	return "ping"
51}
52
53func (p PushPayload) GetType() string {
54	return "push"
55}
56
57func NewPingPayload(data map[string]interface{}) (PingPayload, error) {
58	message, ok := data["message"].(string)
59	if !ok {
60		return PingPayload{}, fmt.Errorf("invalid data for message")
61	}
62	return PingPayload{Message: message}, nil
63}
64
65// What do we need inside the push payload :
66// the ports as a string (80:80/tcp, 443:443/tcp)
67// the image+tag as a string (nginx:latest)
68// the .tar container image as a byte array
69
70type PushPayload struct {
71	Ports        string `json:"ports"`
72	TargetDomain string `json:"targetdomain"`
73	ImageName    string `json:"imagename"`
74	Data         []byte `json:"data"`
75}