netmaker/models/mqtt.go

package models

import (
	"net"

	"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
)

// HostPeerUpdate - struct for host peer updates
type HostPeerUpdate struct {
	Host            Host                 `json:"host" bson:"host" yaml:"host"`
	Server          string               `json:"server" bson:"server" yaml:"server"`
	ServerVersion   string               `json:"serverversion" bson:"serverversion" yaml:"serverversion"`
	ServerAddrs     []ServerAddr         `json:"serveraddrs" bson:"serveraddrs" yaml:"serveraddrs"`
	NodePeers       []wgtypes.PeerConfig `json:"peers" bson:"peers" yaml:"peers"`
	Peers           []wgtypes.PeerConfig
	HostPeerIDs     HostPeerMap           `json:"hostpeerids" bson:"hostpeerids" yaml:"hostpeerids"`
	ProxyUpdate     ProxyManagerPayload   `json:"proxy_update" bson:"proxy_update" yaml:"proxy_update"`
	EgressInfo      map[string]EgressInfo `json:"egress_info" bson:"egress_info" yaml:"egress_info"` // map key is node ID
	IngressInfo     IngressInfo           `json:"ingress_info" bson:"ext_peers" yaml:"ext_peers"`
	PeerIDs         PeerMap               `json:"peerids" bson:"peerids" yaml:"peerids"`
	HostNetworkInfo HostInfoMap           `json:"host_network_info,omitempty" bson:"host_network_info,omitempty" yaml:"host_network_info,omitempty"`
}

// IngressInfo - struct for ingress info
type IngressInfo struct {
	ExtPeers     map[string]ExtClientInfo `json:"ext_peers" yaml:"ext_peers"`
	EgressRanges []string                 `json:"egress_ranges" yaml:"egress_ranges"`
}

// EgressInfo - struct for egress info
type EgressInfo struct {
	EgressID     string                   `json:"egress_id" yaml:"egress_id"`
	Network      net.IPNet                `json:"network" yaml:"network"`
	EgressGwAddr net.IPNet                `json:"egress_gw_addr" yaml:"egress_gw_addr"`
	GwPeers      map[string]PeerRouteInfo `json:"gateway_peers" yaml:"gateway_peers"`
	EgressGWCfg  EgressGatewayRequest     `json:"egress_gateway_cfg" yaml:"egress_gateway_cfg"`
}

// PeerRouteInfo - struct for peer info for an ext. client
type PeerRouteInfo struct {
	PeerAddr net.IPNet `json:"peer_addr" yaml:"peer_addr"`
	PeerKey  string    `json:"peer_key" yaml:"peer_key"`
	Network  string    `json:"network" yaml:"network"`
	Allow    bool      `json:"allow" yaml:"allow"`
	ID       string    `json:"id,omitempty" yaml:"id,omitempty"`
}

// ExtClientInfo - struct for ext. client and it's peers
type ExtClientInfo struct {
	IngGwAddr   net.IPNet                `json:"ingress_gw_addr" yaml:"ingress_gw_addr"`
	Network     net.IPNet                `json:"network" yaml:"network"`
	NetworkName string                   `json:"network_name"`
	Masquerade  bool                     `json:"masquerade" yaml:"masquerade"`
	ExtPeerAddr net.IPNet                `json:"ext_peer_addr" yaml:"ext_peer_addr"`
	ExtPeerKey  string                   `json:"ext_peer_key" yaml:"ext_peer_key"`
	Peers       map[string]PeerRouteInfo `json:"peers" yaml:"peers"`
}

// KeyUpdate - key update struct
type KeyUpdate struct {
	Network   string `json:"network" bson:"network"`
	Interface string `json:"interface" bson:"interface"`
}

// PeerMqActionType - peer update action type
type PeerMqActionType string

const (
	// AddPeer - peer mq action type for adding peers
	AddPeer PeerMqActionType = "ADD_PEER"
	// UpdatePeer - peer mq action type for updating peers
	UpdatePeer PeerMqActionType = "UPDATE_PEER"
	// RemovePeer - peer mq action type for removing peers
	RemovePeer PeerMqActionType = "REMOVE_PEER"
)

// PeerAction - struct for mq peer actions
type PeerAction struct {
	Action PeerMqActionType     `json:"action"`
	Peers  []wgtypes.PeerConfig `json:"peers"`
}

type FwActionType string

type FwAction struct {
	Action      FwActionType          `json:"action"`
	IsIngressGw bool                  `json:"is_ingress_gw"`
	IsEgressGw  bool                  `json:"is_egress_gw"`
	IngressInfo IngressInfo           `json:"ingress_info"`
	EgressInfo  map[string]EgressInfo `json:"egress_info"`
}

const (
	FwUpdate FwActionType = "FW_UPDATE"
)