forked from jshiffer/go-xmpp
8fb3e33a1f
If we always do this the address gets mangled, breaking the transport selection.
46 lines
1.3 KiB
Go
46 lines
1.3 KiB
Go
package xmpp
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"errors"
|
|
"strings"
|
|
)
|
|
|
|
var TLSNotSupported = errors.New("Transport does not support StartTLS")
|
|
|
|
type TransportConfiguration struct {
|
|
// Address is the XMPP Host and port to connect to. Host is of
|
|
// the form 'serverhost:port' i.e "localhost:8888"
|
|
Address string
|
|
ConnectTimeout int // Client timeout in seconds. Default to 15
|
|
// tls.Config must not be modified after having been passed to NewClient. Any
|
|
// changes made after connecting are ignored.
|
|
TLSConfig *tls.Config
|
|
}
|
|
|
|
type Transport interface {
|
|
Connect() error
|
|
DoesStartTLS() bool
|
|
StartTLS(domain string) error
|
|
|
|
IsSecure() bool
|
|
|
|
Read(p []byte) (n int, err error)
|
|
Write(p []byte) (n int, err error)
|
|
Close() error
|
|
}
|
|
|
|
// NewTransport creates a new Transport instance.
|
|
// The type of transport is determined by the address in the configuration:
|
|
// - if the address is a URL with the `ws` or `wss` scheme WebsocketTransport is used
|
|
// - in all other cases a XMPPTransport is used
|
|
// For XMPPTransport it is mandatory for the address to have a port specified.
|
|
func NewTransport(config TransportConfiguration) Transport {
|
|
if strings.HasPrefix(config.Address, "ws:") || strings.HasPrefix(config.Address, "wss:") {
|
|
return &WebsocketTransport{Config: config}
|
|
}
|
|
|
|
config.Address = ensurePort(config.Address, 5222)
|
|
return &XMPPTransport{Config: config}
|
|
}
|