stop function

Author: aeinstein

dist/webrtmp.min.js

@@ -34,6 +34,7 @@ class RTMPMessageHandler {
     chunk_stream_id = 2;
     trackedCommand = "";
     socket;
+    current_stream_id;
 
     /**
      *
@@ -126,6 +127,10 @@ class RTMPMessageHandler {
             Log.d(this.TAG, "AMF0", cmd);
 
             switch(cmd[0]) {
+            case "_error":
+                Log.e(this.TAG, cmd);
+                break;
+
             case "_result":
                 switch(this.trackedCommand){
                 case "connect":
@@ -139,7 +144,8 @@ class RTMPMessageHandler {
                 case "createStream":
                     Log.d(this.TAG,"got _result: " + cmd[3]);
                     if(cmd[3]) {
-                        postMessage(["RTMPStreamCreated"]);
+                        this.current_stream_id = cmd[4];
+                        postMessage(["RTMPStreamCreated", cmd[3], cmd[4]]);
                     }
                     break;
 
@@ -208,6 +214,14 @@ class RTMPMessageHandler {
         this._sendCommand(3, command);
     }
 
+    deleteStream(stream_id){
+        const command = new AMF0Object([
+            "deleteStream", 1, null, stream_id
+        ]);
+
+        this._sendCommand(3, command);
+    }
+
     /**
      *
      * @param {String} streamName
@@ -220,6 +234,10 @@ class RTMPMessageHandler {
         this._sendCommand(3, command);
     }
 
+    stop(){
+        this.deleteStream(this.current_stream_id);
+    }
+
     /**
      *
      * @param {boolean} enable

dist/webrtmp.min.js.map

@@ -131,6 +131,11 @@ class WebRTMP{
 		});
 	}
 
+	stop(){
+		this.wss.stop()
+		this._mediaElement.pause();
+	}
+
 	/**
 	 *
 	 * @param {String|null} host
@@ -161,14 +166,13 @@ class WebRTMP{
 	 */
 	connect(appName){
 		return new Promise((resolve, reject)=>{
-			this.wss.addEventListener("RTMPStreamCreated", ()=>{
-				Log.d(this.TAG,"RTMPStreamCreated");
+			this.wss.addEventListener("RTMPStreamCreated", (cmd, stream_id)=>{
+				Log.d(this.TAG,"RTMPStreamCreated: " + stream_id);
 				resolve();
 			});
 
 			this.wss.connect(appName);
 		})
-
 	}
 
 	pause(enable){

dist/webrtmp.worker.js

@@ -85,6 +85,10 @@ self.addEventListener('message', function(e) {
 			message_handler.play(data.streamName);
 			break;
 
+		case "stop":
+			message_handler.stop();
+			break;
+
         case "pause":
 			message_handler.pause(data.enable);
             break;

dist/webrtmp.worker.js.map

@@ -100,6 +100,10 @@ class WebRTMP_Controller {
 		this.WebRTMPWorker.postMessage({cmd: "play", streamName: streamName});
 	}
 
+	stop(){
+		this.WebRTMPWorker.postMessage({cmd: "stop"});
+	}
+
     pause(enable){
         this.WebRTMPWorker.postMessage({cmd: "pause", enable: enable});
     }