options.go

package wgephemeralpeer

import "net/netip"

type Option func(*ephemeralPeer)

// WithMcEliece460896Round3 uses the key encapsulation method
// McEliece460896-Round3 when negotiating a PSK for the ephemeral peer
func WithMcEliece460896Round3() Option {
	return func(ep *ephemeralPeer) {
		ep.kemSchemes = append(ep.kemSchemes, schemeMcEliece460896Round3)
	}
}

// WithKyber1024 uses the key encapsulation method Kyber1024 when negotiating a
// PSK for the ephemeral peer
func WithKyber1024() Option {
	return func(ep *ephemeralPeer) {
		ep.kemSchemes = append(ep.kemSchemes, schemeKyber1024)
	}
}

// WithMLKEM1024 uses the key encapsulation method ML-KEM-1024 when negotiating a
// PSK for the ephemeral peer
func WithMLKEM1024() Option {
	return func(ep *ephemeralPeer) {
		ep.kemSchemes = append(ep.kemSchemes, schemeMLKEM1024)
	}
}

// WithAPIAddress sets the address used to connect to the gRPC API.
func WithAPIAddress(apiAddress netip.AddrPort) Option {
	return func(ep *ephemeralPeer) {
		ep.apiAddress = apiAddress
	}
}