mirror of
https://github.com/netbirdio/netbird.git
synced 2026-04-16 07:16:38 +00:00
63c83aa8d2
CLI: new expose command to publish a local port with flags for PIN, password, user groups, custom domain, name prefix and protocol (HTTP default). Management/API: create/renew/stop expose sessions (streamed status), automatic naming/domain, TTL renewals, background expiration, new management RPCs and client methods. UI/API: account settings now include peer_expose_enabled and peer_expose_groups; new activity codes for peer expose events.
96 lines
2.4 KiB
Go
96 lines
2.4 KiB
Go
package expose
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
mgm "github.com/netbirdio/netbird/shared/management/client"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
const renewTimeout = 10 * time.Second
|
|
|
|
// Response holds the response from exposing a service.
|
|
type Response struct {
|
|
ServiceName string
|
|
ServiceURL string
|
|
Domain string
|
|
}
|
|
|
|
type Request struct {
|
|
NamePrefix string
|
|
Domain string
|
|
Port uint16
|
|
Protocol int
|
|
Pin string
|
|
Password string
|
|
UserGroups []string
|
|
}
|
|
|
|
type ManagementClient interface {
|
|
CreateExpose(ctx context.Context, req mgm.ExposeRequest) (*mgm.ExposeResponse, error)
|
|
RenewExpose(ctx context.Context, domain string) error
|
|
StopExpose(ctx context.Context, domain string) error
|
|
}
|
|
|
|
// Manager handles expose session lifecycle via the management client.
|
|
type Manager struct {
|
|
mgmClient ManagementClient
|
|
ctx context.Context
|
|
}
|
|
|
|
// NewManager creates a new expose Manager using the given management client.
|
|
func NewManager(ctx context.Context, mgmClient ManagementClient) *Manager {
|
|
return &Manager{mgmClient: mgmClient, ctx: ctx}
|
|
}
|
|
|
|
// Expose creates a new expose session via the management server.
|
|
func (m *Manager) Expose(ctx context.Context, req Request) (*Response, error) {
|
|
log.Infof("exposing service on port %d", req.Port)
|
|
resp, err := m.mgmClient.CreateExpose(ctx, toClientExposeRequest(req))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
log.Infof("expose session created for %s", resp.Domain)
|
|
|
|
return fromClientExposeResponse(resp), nil
|
|
}
|
|
|
|
func (m *Manager) KeepAlive(ctx context.Context, domain string) error {
|
|
ticker := time.NewTicker(10 * time.Second)
|
|
defer ticker.Stop()
|
|
defer m.stop(domain)
|
|
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
log.Infof("context canceled, stopping keep alive for %s", domain)
|
|
|
|
return nil
|
|
case <-ticker.C:
|
|
if err := m.renew(ctx, domain); err != nil {
|
|
log.Errorf("renewing expose session for %s: %v", domain, err)
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// renew extends the TTL of an active expose session.
|
|
func (m *Manager) renew(ctx context.Context, domain string) error {
|
|
renewCtx, cancel := context.WithTimeout(ctx, renewTimeout)
|
|
defer cancel()
|
|
return m.mgmClient.RenewExpose(renewCtx, domain)
|
|
}
|
|
|
|
// stop terminates an active expose session.
|
|
func (m *Manager) stop(domain string) {
|
|
stopCtx, cancel := context.WithTimeout(m.ctx, renewTimeout)
|
|
defer cancel()
|
|
err := m.mgmClient.StopExpose(stopCtx, domain)
|
|
if err != nil {
|
|
log.Warnf("Failed stopping expose session for %s: %v", domain, err)
|
|
}
|
|
}
|