79 lines
1.9 KiB
Go
79 lines
1.9 KiB
Go
package libnmdc
|
|
|
|
import (
|
|
"crypto/rand"
|
|
)
|
|
|
|
type HubConnectionOptions struct {
|
|
Address HubAddress
|
|
SkipVerifyTLS bool // using a negative verb, because bools default to false
|
|
SkipAutoReconnect bool // as above
|
|
Self *UserInfo
|
|
NickPassword string
|
|
AdcPID string // blank: autogenerate
|
|
}
|
|
|
|
func NewPID() string {
|
|
pidBytes := make([]byte, 24)
|
|
n, err := rand.Read(pidBytes)
|
|
if err != nil {
|
|
panic(err) // Insufficient cryptographic randomness
|
|
}
|
|
|
|
if n != 24 {
|
|
panic("Insufficient cryptographic randomness")
|
|
}
|
|
|
|
return Base32(pidBytes)
|
|
}
|
|
|
|
func (this *HubConnectionOptions) prepareConnection() *HubConnection {
|
|
if this.Self.ClientTag == "" {
|
|
this.Self.ClientTag = DEFAULT_CLIENT_TAG
|
|
this.Self.ClientVersion = DEFAULT_CLIENT_VERSION
|
|
}
|
|
|
|
// Shouldn't be blank either
|
|
if this.Self.ClientVersion == "" {
|
|
this.Self.ClientVersion = "0"
|
|
}
|
|
|
|
if this.AdcPID == "" {
|
|
this.AdcPID = NewPID()
|
|
}
|
|
|
|
hc := HubConnection{
|
|
Hco: this,
|
|
HubName: DEFAULT_HUB_NAME,
|
|
State: CONNECTIONSTATE_DISCONNECTED,
|
|
users: make(map[string]UserInfo),
|
|
|
|
autoReconnect: !this.SkipAutoReconnect,
|
|
}
|
|
|
|
return &hc
|
|
}
|
|
|
|
// ConnectAsync connects to a hub server, and spawns a background goroutine to handle
|
|
// protocol messages. Events will be sent by channel to the supplied onEvent channel,
|
|
// the client is responsible for selecting off this.
|
|
func ConnectAsync(opts *HubConnectionOptions, onEvent chan HubEvent) *HubConnection {
|
|
hc := opts.prepareConnection()
|
|
hc.processEvent = func(ev HubEvent) {
|
|
onEvent <- ev
|
|
}
|
|
|
|
go hc.worker()
|
|
return hc
|
|
}
|
|
|
|
// ConnectSync connects to a hub server, and blocks forever to handle protocol messages.
|
|
// Client code should supply an event handling function as hco.OnEventSync.
|
|
func ConnectSync(opts *HubConnectionOptions, onEvent func(hub *HubConnection, ev HubEvent)) {
|
|
hc := opts.prepareConnection()
|
|
hc.processEvent = func(ev HubEvent) {
|
|
onEvent(hc, ev)
|
|
}
|
|
hc.worker()
|
|
}
|