mirror of
https://github.com/netbirdio/netbird.git
synced 2024-11-25 17:43:38 +01:00
4587f7686e
* feat: basic management service implementation [FAILING TESTS] * test: fix healthcheck test * test: #39 add peer registration endpoint test * feat: #39 add setup key handling * feat: #39 add peer management store persistence * refactor: extract config read/write to the utility package * refactor: move file contents copy to the utility package * refactor: use Accounts instead of Users in the Store * feature: add management server Docker file * refactor: introduce datadir instead of config * chore: use filepath.Join to concat filepaths instead of string concat * refactor: move stop channel to the root * refactor: move stop channel to the root * review: fix PR review notes Co-authored-by: braginini <hello@wiretrustee.com>
40 lines
1.0 KiB
Go
40 lines
1.0 KiB
Go
package management
|
|
|
|
import (
|
|
"context"
|
|
"github.com/wiretrustee/wiretrustee/management/proto"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
// Server an instance of a Management server
|
|
type Server struct {
|
|
Store *Store
|
|
}
|
|
|
|
// NewServer creates a new Management server
|
|
func NewServer(dataDir string) (*Server, error) {
|
|
store, err := NewStore(dataDir)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &Server{
|
|
Store: store,
|
|
}, nil
|
|
}
|
|
|
|
// RegisterPeer adds a peer to the Store. Returns 404 in case the provided setup key doesn't exist
|
|
func (s *Server) RegisterPeer(ctx context.Context, req *proto.RegisterPeerRequest) (*proto.RegisterPeerResponse, error) {
|
|
|
|
err := s.Store.AddPeer(req.SetupKey, req.Key)
|
|
if err != nil {
|
|
return &proto.RegisterPeerResponse{}, status.Errorf(404, "provided setup key doesn't exists")
|
|
}
|
|
|
|
return &proto.RegisterPeerResponse{}, nil
|
|
}
|
|
|
|
// IsHealthy indicates whether the service is healthy
|
|
func (s *Server) IsHealthy(ctx context.Context, req *proto.Empty) (*proto.Empty, error) {
|
|
return &proto.Empty{}, nil
|
|
}
|