libnmdc/HubConnectionOptions.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),
		userSIDs: make(map[string]string),

		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()
}