opamp

type AgentConnectionState struct {
	AgentID  string
	Conn     opamp.Connection
	SendLock sync.Mutex
}

type AgentState struct {
	SequenceNum uint64                  `json:"sequenceNum" yaml:"sequenceNum" mapstructure:"sequenceNum"`
	Status      protobufs.AgentToServer `json:"status,omitempty" yaml:"status,omitempty" mapstructure:"status"`
}

type Connection interface {
	opamp.Connection
}

type Connections[S any] interface {
	// Connected returns true if the agent with the specified agentID is connected
	Connected(agentID string) bool

	// ConnectedAgentIDs returns the IDs of all connected agents
	ConnectedAgentIDs(context.Context) []string

	// ConnectedAgentsCount returns the number of connected agents
	ConnectedAgentsCount(context.Context) int

	// OnConnecting should be called within the opamp OnConnecting callback. It adds the association of the accountID to
	// the agentID after it has been determined. The associated opamp.Connection is not yet available. It will be added in
	// OnMessage.
	OnConnecting(ctx context.Context, agentID string) S

	// OnMessage should be called within the opamp OnMessage callback. It adds the association of the opamp.Connection to
	// the connectionState.
	OnMessage(agentID string, conn opamp.Connection) (S, error)

	// OnConnectionClose removes the connection and returns the state and the count of remaining agents connected.
	OnConnectionClose(conn opamp.Connection) (state S, count int)

	// StateForAgentID returns the current state for the specified agentID or nil if there is no connection
	StateForAgentID(agentID string) S

	// StateForConnection returns the current state for the specified [opamp.Connection] or nil if there is no connection
	StateForConnection(opamp.Connection) S
}

type SerializedAgentState struct {
	SequenceNum uint64 `json:"sequenceNum" yaml:"sequenceNum" mapstructure:"sequenceNum"`
	Status      string `json:"status,omitempty" yaml:"status,omitempty" mapstructure:"status"`
}