proto

type AgentRule struct {
	ID         string    `json:"id"`
	Proto      Proto     `json:"proto"`
	ListenPort PortRange `json:"listen_port"`
	Target     string    `json:"target"`
	Enabled    bool      `json:"enabled"`
}

type ChangeKind string

type FieldChange struct {
	Field    string
	Old, New string
}

type Heartbeat struct {
	Generation uint64       `json:"generation"` // 最後に受け取って処理した世代(部分失敗でも進める)
	Tunnel     TunnelStatus `json:"tunnel"`
	Rules      []RuleStatus `json:"rules"`
}

type MergeBlocker string

type Message struct {
	Type      string     `json:"type"`
	PublicKey string     `json:"public_key,omitempty"` // MsgPublicKey:wg 公開鍵(base64)
	State     *State     `json:"state,omitempty"`      // MsgState
	Heartbeat *Heartbeat `json:"heartbeat,omitempty"`  // MsgHeartbeat

	// ProtocolMin/ProtocolMax/Capabilities は MsgPublicKey に載る版と機能の交渉(仕様 7a.6 節)。
	// ポインタと *[]string にしてあるのは、フィールドが無いこと(legacy v0 の agent)と、
	// 空配列(版はあるが追加の機能は無い)を JSON の上で区別するため。encoding/json の
	// omitempty は空スライスも「空」として省いてしまうので、[]string のままでは区別できない
	ProtocolMin  *int      `json:"protocol_min,omitempty"`
	ProtocolMax  *int      `json:"protocol_max,omitempty"`
	Capabilities *[]string `json:"capabilities,omitempty"`
}

type Negotiated struct {
	Legacy       bool
	Version      int      // 選んだ版(Legacy なら 0)
	AgentMin     int      // agent が宣言した protocol_min(Legacy なら 0)
	AgentMax     int      // agent が宣言した protocol_max(Legacy なら 0)
	Capabilities []string // agent が宣言した capabilities(Legacy か、宣言が無ければ nil)
}

type PortRange struct {
	Lo, Hi uint16
}

type Proto string

type ProtocolRange struct {
	Min int
	Max int
}

type Rate struct {
	Count uint64
	Unit  RateUnit
}

type RateUnit string

type Reserved map[uint16]string

type Rule struct {
	ID            string         `json:"id"`
	Agent         string         `json:"agent"`
	Group         string         `json:"group"` // 任意。ルールを束ねるフラットなラベル 1 つ(仕様 5.3)
	Note          string         `json:"note"`  // 任意。何のためのルールかの自由記述
	Proto         Proto          `json:"proto"`
	ListenPort    PortRange      `json:"listen_port"`
	Target        string         `json:"target"` // host:port。範囲のときは先頭ポートに対応する
	VPSMode       VPSMode        `json:"vps_mode"`
	ProxyProtocol bool           `json:"proxy_protocol"`
	SourceAllow   []netip.Prefix `json:"source_allow"`
	SourceDeny    []netip.Prefix `json:"source_deny"`
	NewFlowRate   *Rate          `json:"new_flow_rate"`
	PacketRate    *Rate          `json:"packet_rate"`
	PerSourceRate *Rate          `json:"per_source_rate"`
	Enabled       bool           `json:"enabled"`
}

type RuleChange struct {
	Kind         ChangeKind
	Rule         Rule
	Previous     Rule
	FieldChanges []FieldChange
}

type RuleStatus struct {
	ID     string `json:"id"`
	State  string `json:"state"` // ok | error
	Reason string `json:"reason,omitempty"`
}

type State struct {
	Generation uint64      `json:"generation"`
	WG         WGConfig    `json:"wg"`
	Rules      []AgentRule `json:"rules"`

	// ServerProtocolVersion/ServerCapabilities は、その stream 接続で選んだ版と server の機能
	// (仕様 7a.6 節)。agent が legacy v0 なら vpsd はこのフィールドを載せない(nil のまま)。
	// *[]string にしてあるのは Message.Capabilities と同じ理由(空配列と不在の区別)
	ServerProtocolVersion *int      `json:"server_protocol_version,omitempty"`
	ServerCapabilities    *[]string `json:"server_capabilities,omitempty"`

	// AgentDisabled は、server がこのエージェントを無効にしていることを示す(仕様 5.1 節)。無効の間、
	// Rules はすべて enabled:false の写しで届く。エージェントが止まるのはその enabled:false による
	// ものであり、このフィールドはエージェント自身の診断のためだけにある。守りには使わない。
	// 加算のフィールドで、有効なら省く。旧い版のエージェントは読み飛ばす(Go の encoding/json は
	// 構造体に無いフィールドを無視する。仕様 7a.6 節)ので、版と機能の交渉は要らない
	AgentDisabled bool `json:"agent_disabled,omitempty"`
}

type TunnelStatus struct {
	State         string    `json:"state"` // ok | error
	Reason        string    `json:"reason,omitempty"`
	Endpoint      string    `json:"endpoint,omitempty"` // 解決したエンドポイント(ip:port)
	LastHandshake time.Time `json:"last_handshake,omitempty"`
}

type VPSMode string

type WGConfig struct {
	ServerPubkey     string `json:"server_pubkey"`
	Endpoint         string `json:"endpoint"` // host:port
	Address          string `json:"address"`  // 10.200.0.2/24 の形
	MTU              int    `json:"mtu"`
	Keepalive        int    `json:"keepalive"`          // 秒
	UDPTimeout       int    `json:"udp_timeout"`        // VPS の nf_conntrack_udp_timeout(秒)
	UDPTimeoutStream int    `json:"udp_timeout_stream"` // VPS の nf_conntrack_udp_timeout_stream(秒)
}