mirror of
https://github.com/henrygd/beszel.git
synced 2025-12-11 22:05:33 +00:00
7d6230de74
- add one minute charts - update disk io to use bytes - update hub and agent connection interfaces / handlers to be more flexible - change agent cache to use cache time instead of session id - refactor collection of metrics which require deltas to track separately per cache time
108 lines
3.3 KiB
Go
108 lines
3.3 KiB
Go
package ws
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"github.com/fxamacker/cbor/v2"
|
|
"github.com/henrygd/beszel/internal/common"
|
|
"github.com/henrygd/beszel/internal/entities/system"
|
|
"github.com/lxzan/gws"
|
|
"golang.org/x/crypto/ssh"
|
|
)
|
|
|
|
// ResponseHandler defines interface for handling agent responses
|
|
type ResponseHandler interface {
|
|
Handle(agentResponse common.AgentResponse) error
|
|
HandleLegacy(rawData []byte) error
|
|
}
|
|
|
|
// BaseHandler provides a default implementation that can be embedded to make HandleLegacy optional
|
|
// type BaseHandler struct{}
|
|
|
|
// func (h *BaseHandler) HandleLegacy(rawData []byte) error {
|
|
// return errors.New("legacy format not supported")
|
|
// }
|
|
|
|
////////////////////////////////////////////////////////////////////////////
|
|
////////////////////////////////////////////////////////////////////////////
|
|
////////////////////////////////////////////////////////////////////////////
|
|
|
|
// systemDataHandler implements ResponseHandler for system data requests
|
|
type systemDataHandler struct {
|
|
data *system.CombinedData
|
|
}
|
|
|
|
func (h *systemDataHandler) HandleLegacy(rawData []byte) error {
|
|
return cbor.Unmarshal(rawData, h.data)
|
|
}
|
|
|
|
func (h *systemDataHandler) Handle(agentResponse common.AgentResponse) error {
|
|
if agentResponse.SystemData != nil {
|
|
*h.data = *agentResponse.SystemData
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// RequestSystemData requests system metrics from the agent and unmarshals the response.
|
|
func (ws *WsConn) RequestSystemData(ctx context.Context, data *system.CombinedData, options common.DataRequestOptions) error {
|
|
if !ws.IsConnected() {
|
|
return gws.ErrConnClosed
|
|
}
|
|
|
|
req, err := ws.requestManager.SendRequest(ctx, common.GetData, options)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
handler := &systemDataHandler{data: data}
|
|
return ws.handleAgentRequest(req, handler)
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////
|
|
////////////////////////////////////////////////////////////////////////////
|
|
////////////////////////////////////////////////////////////////////////////
|
|
|
|
// fingerprintHandler implements ResponseHandler for fingerprint requests
|
|
type fingerprintHandler struct {
|
|
result *common.FingerprintResponse
|
|
}
|
|
|
|
func (h *fingerprintHandler) HandleLegacy(rawData []byte) error {
|
|
return cbor.Unmarshal(rawData, h.result)
|
|
}
|
|
|
|
func (h *fingerprintHandler) Handle(agentResponse common.AgentResponse) error {
|
|
if agentResponse.Fingerprint != nil {
|
|
*h.result = *agentResponse.Fingerprint
|
|
return nil
|
|
}
|
|
return errors.New("no fingerprint data in response")
|
|
}
|
|
|
|
// GetFingerprint authenticates with the agent using SSH signature and returns the agent's fingerprint.
|
|
func (ws *WsConn) GetFingerprint(ctx context.Context, token string, signer ssh.Signer, needSysInfo bool) (common.FingerprintResponse, error) {
|
|
if !ws.IsConnected() {
|
|
return common.FingerprintResponse{}, gws.ErrConnClosed
|
|
}
|
|
|
|
challenge := []byte(token)
|
|
signature, err := signer.Sign(nil, challenge)
|
|
if err != nil {
|
|
return common.FingerprintResponse{}, err
|
|
}
|
|
|
|
req, err := ws.requestManager.SendRequest(ctx, common.CheckFingerprint, common.FingerprintRequest{
|
|
Signature: signature.Blob,
|
|
NeedSysInfo: needSysInfo,
|
|
})
|
|
if err != nil {
|
|
return common.FingerprintResponse{}, err
|
|
}
|
|
|
|
var result common.FingerprintResponse
|
|
handler := &fingerprintHandler{result: &result}
|
|
err = ws.handleAgentRequest(req, handler)
|
|
return result, err
|
|
}
|