Improve session message handling

app/src/Media.ts

@@ -1,5 +1,7 @@
 "use strict";
 
+import { Channel } from "castv2";
+
 import Session from "./Session";
 
 import { Message
@@ -27,7 +29,7 @@ export default class Media {
     private mediaSessionId: number;
     private referenceId: string;
     private session: Session;
-    private channel: any;
+    private channel: Channel;
     private sendMessageCallback: SendMessageCallback;
 
     constructor (

app/src/Session.ts

@@ -37,10 +37,12 @@ export default class Session {
           , port: number
           , appId: string
           , sessionId: number
+          , referenceId: string
           , sendMessageCallback: SendMessageCallback) {
 
-        this.sendMessageCallback = sendMessageCallback;
         this.sessionId = sessionId;
+        this.referenceId = referenceId;
+        this.sendMessageCallback = sendMessageCallback;
 
         this.client = new Client();
 

app/src/main.ts

@@ -97,6 +97,7 @@ async function handleMessage (message: Message) {
                       , message.data.port
                       , message.data.appId
                       , message.data.sessionId
+                      , message._id
                       , sendMessage));
             }
         }