mirror of
https://github.com/YspCoder/clawgo.git
synced 2026-04-29 05:47:37 +08:00
feat: add turn-ready node p2p config
This commit is contained in:
@@ -298,10 +298,17 @@ type GatewayNodesConfig struct {
|
||||
P2P GatewayNodesP2PConfig `json:"p2p,omitempty"`
|
||||
}
|
||||
|
||||
type GatewayICEConfig struct {
|
||||
URLs []string `json:"urls,omitempty"`
|
||||
Username string `json:"username,omitempty"`
|
||||
Credential string `json:"credential,omitempty"`
|
||||
}
|
||||
|
||||
type GatewayNodesP2PConfig struct {
|
||||
Enabled bool `json:"enabled"`
|
||||
Transport string `json:"transport,omitempty"`
|
||||
STUNServers []string `json:"stun_servers,omitempty"`
|
||||
Enabled bool `json:"enabled"`
|
||||
Transport string `json:"transport,omitempty"`
|
||||
STUNServers []string `json:"stun_servers,omitempty"`
|
||||
ICEServers []GatewayICEConfig `json:"ice_servers,omitempty"`
|
||||
}
|
||||
|
||||
type CronConfig struct {
|
||||
@@ -550,6 +557,7 @@ func DefaultConfig() *Config {
|
||||
Enabled: false,
|
||||
Transport: "websocket_tunnel",
|
||||
STUNServers: []string{},
|
||||
ICEServers: []GatewayICEConfig{},
|
||||
},
|
||||
},
|
||||
},
|
||||
|
||||
@@ -125,6 +125,26 @@ func Validate(cfg *Config) []error {
|
||||
errs = append(errs, fmt.Errorf("gateway.nodes.p2p.transport must be one of: websocket_tunnel, webrtc"))
|
||||
}
|
||||
errs = append(errs, validateNonEmptyStringList("gateway.nodes.p2p.stun_servers", cfg.Gateway.Nodes.P2P.STUNServers)...)
|
||||
for i, server := range cfg.Gateway.Nodes.P2P.ICEServers {
|
||||
prefix := fmt.Sprintf("gateway.nodes.p2p.ice_servers[%d]", i)
|
||||
errs = append(errs, validateNonEmptyStringList(prefix+".urls", server.URLs)...)
|
||||
needsAuth := false
|
||||
for _, raw := range server.URLs {
|
||||
u := strings.ToLower(strings.TrimSpace(raw))
|
||||
if strings.HasPrefix(u, "turn:") || strings.HasPrefix(u, "turns:") {
|
||||
needsAuth = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if needsAuth {
|
||||
if strings.TrimSpace(server.Username) == "" {
|
||||
errs = append(errs, fmt.Errorf("%s.username is required for turn/turns urls", prefix))
|
||||
}
|
||||
if strings.TrimSpace(server.Credential) == "" {
|
||||
errs = append(errs, fmt.Errorf("%s.credential is required for turn/turns urls", prefix))
|
||||
}
|
||||
}
|
||||
}
|
||||
if cfg.Cron.MinSleepSec <= 0 {
|
||||
errs = append(errs, fmt.Errorf("cron.min_sleep_sec must be > 0"))
|
||||
}
|
||||
|
||||
@@ -151,3 +151,29 @@ func TestValidateRejectsUnknownGatewayNodeP2PTransport(t *testing.T) {
|
||||
t.Fatalf("expected validation errors")
|
||||
}
|
||||
}
|
||||
|
||||
func TestValidateGatewayNodeP2PIceServersAllowsStunOnly(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
cfg := DefaultConfig()
|
||||
cfg.Gateway.Nodes.P2P.ICEServers = []GatewayICEConfig{
|
||||
{URLs: []string{"stun:stun.l.google.com:19302"}},
|
||||
}
|
||||
|
||||
if errs := Validate(cfg); len(errs) != 0 {
|
||||
t.Fatalf("expected config to be valid, got %v", errs)
|
||||
}
|
||||
}
|
||||
|
||||
func TestValidateGatewayNodeP2PIceServersRequireTurnCredentials(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
cfg := DefaultConfig()
|
||||
cfg.Gateway.Nodes.P2P.ICEServers = []GatewayICEConfig{
|
||||
{URLs: []string{"turn:turn.example.com:3478?transport=udp"}},
|
||||
}
|
||||
|
||||
if errs := Validate(cfg); len(errs) == 0 {
|
||||
t.Fatalf("expected validation errors")
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user