Merge pull request #2 from muzzammilshahid/split-webrtc-wamp

Refactor into separate functions for WebRTC connection and WAMP session

client.go

@@ -18,7 +18,7 @@ type ClientConfig struct {
 	Serializer               xconn.WSSerializerSpec
 }
 
-func ConnectWebRTC(config *ClientConfig) (*xconn.Session, error) {
+func ConnectWebRTC(config *ClientConfig) (*WebRTCSession, error) {
 	session, err := xconn.Connect(context.Background(), config.URL, config.Realm)
 	if err != nil {
 		return nil, err
@@ -60,7 +60,19 @@ func ConnectWebRTC(config *ClientConfig) (*xconn.Session, error) {
 
 	channel := <-offerer.WaitReady()
 
-	peer := NewWebRTCPeer(channel)
+	return &WebRTCSession{
+		Channel:    channel,
+		Connection: offerer.connection,
+	}, nil
+}
+
+func ConnectWAMP(config *ClientConfig) (*xconn.Session, error) {
+	webRTCConnection, err := ConnectWebRTC(config)
+	if err != nil {
+		return nil, err
+	}
+
+	peer := NewWebRTCPeer(webRTCConnection.Channel)
 	base, err := xconn.Join(peer, config.Realm, config.Serializer.Serializer(), nil)
 	if err != nil {
 		return nil, err

cmd/client/main.go

@@ -20,7 +20,7 @@ func main() {
 		TopicAnswererOnCandidate: topicAnswererOnCandidate,
 		Serializer:               xconn.CBORSerializerSpec,
 	}
-	session, err := wamp_webrtc_go.ConnectWebRTC(config)
+	session, err := wamp_webrtc_go.ConnectWAMP(config)
 	if err != nil {
 		log.Fatal(err)
 	}

types.go

@@ -33,3 +33,8 @@ type ProviderConfig struct {
 	TopicPublishLocalCandidate  string
 	Serializer                  serializers.Serializer
 }
+
+type WebRTCSession struct {
+	Connection *webrtc.PeerConnection
+	Channel    *webrtc.DataChannel
+}