add setCustomMetadata method to rtmp

library/src/main/java/com/pedro/library/util/streamclient/RtmpStreamClient.kt

@@ -213,6 +213,10 @@ class RtmpStreamClient(
     rtmpClient.setCustomAmfObject(amfObject)
   }
 
+  fun setCustomMetadata(metadata: Map<String, Any>) {
+    rtmpClient.setCustomMetadata(metadata)
+  }
+
   /**
    * Should notify onConnectionFailed if read packet from the server failed
    */

rtmp/src/main/java/com/pedro/rtmp/amf/v0/AmfObject.kt

@@ -60,7 +60,7 @@ open class AmfObject(private val properties: LinkedHashMap<AmfString, AmfData> =
   open fun setProperty(name: String, data: AmfData) = putProperty(name, data)
 
   open fun setProperty(name: String, data: Any) {
-    val newValue= when (data) {
+    val newValue = when (data) {
       is String -> AmfString(data)
       is Int -> AmfNumber(data.toDouble())
       is Long -> AmfNumber(data.toDouble())

rtmp/src/main/java/com/pedro/rtmp/amf/v3/Amf3Integer.kt

@@ -22,7 +22,7 @@ import java.io.OutputStream
 /**
  * Created by pedro on 29/04/21.
  */
-class Amf3Integer: Amf3Data() {
+class Amf3Integer(private val value: Int = 0): Amf3Data() {
 
   override fun readBody(input: InputStream) {
     TODO("Not yet implemented")

rtmp/src/main/java/com/pedro/rtmp/amf/v3/Amf3Object.kt

@@ -16,6 +16,7 @@
 
 package com.pedro.rtmp.amf.v3
 
+import com.pedro.rtmp.amf.v0.AmfNull
 import java.io.InputStream
 import java.io.OutputStream
 
@@ -35,43 +36,37 @@ open class Amf3Object(private val properties: HashMap<Amf3String, Amf3Data> = Li
     return null
   }
 
-  open fun setProperty(name: String, data: String) {
-    val key = Amf3String(name)
-    val value = Amf3String(data)
-    properties[key] = value
-    bodySize += key.getSize()
-    bodySize += value.getSize() + 1
-  }
+  open fun setProperty(name: String, data: String)  = putProperty(name, Amf3String(data))
 
-  open fun setProperty(name: String, data: Boolean) {
-    val key = Amf3String(name)
-    val value = if (data) Amf3True() else Amf3False()
-    properties[key] = value
-    bodySize += key.getSize()
-    bodySize += value.getSize() + 1
-  }
+  open fun setProperty(name: String, data: Boolean)  = putProperty(name, if (data) Amf3True() else Amf3False())
 
-  open fun setProperty(name: String, data: Amf3Data) {
-    val key = Amf3String(name)
-    properties[key] = data
-    bodySize += key.getSize()
-    bodySize += data.getSize() + 1
-  }
+  open fun setProperty(name: String, data: Amf3Data) = putProperty(name, data)
 
-  open fun setProperty(name: String) {
-    val key = Amf3String(name)
-    val value = Amf3Null()
-    properties[key] = value
-    bodySize += key.getSize()
-    bodySize += value.getSize() + 1
+  open fun setProperty(name: String) = putProperty(name, Amf3Null())
+
+  open fun setProperty(name: String, data: Double) = putProperty(name, Amf3Double(data))
+
+  open fun setProperty(name: String, data: Any) {
+    val newValue = when (data) {
+      is String -> Amf3String(data)
+      is Int -> Amf3Integer(data)
+      is Double -> Amf3Double(data)
+      is Float -> Amf3Double(data.toDouble())
+      is Boolean -> if (data) Amf3True() else Amf3False()
+      is Amf3Data -> data
+      else -> throw IllegalArgumentException("Unsupported value type: ${data::class.java.name}")
+    }
+    putProperty(name, newValue)
   }
 
-  open fun setProperty(name: String, data: Double) {
+  private fun putProperty(name: String, value: Amf3Data) {
     val key = Amf3String(name)
-    val value = Amf3Double(data)
-    properties[key] = value
-    bodySize += key.getSize()
-    bodySize += value.getSize() + 1
+    val previous = properties.put(key, value)
+    bodySize += if (previous != null) {
+      value.getSize() - previous.getSize()
+    } else {
+      key.getSize() + value.getSize() + 1
+    }
   }
 
 

rtmp/src/main/java/com/pedro/rtmp/rtmp/CommandsManager.kt

@@ -62,6 +62,7 @@ abstract class CommandsManager {
   var audioDisabled = false
   var videoDisabled = false
   var customAmfObject: Map<String, Any> = emptyMap()
+  var customMetadata: Map<String, Any> = emptyMap()
   private var bytesRead = 0
   private var lastAcknowledgementSequence = 0
 

rtmp/src/main/java/com/pedro/rtmp/rtmp/CommandsManagerAmf0.kt

@@ -131,6 +131,9 @@ class CommandsManagerAmf0: CommandsManager() {
       amfEcmaArray.setProperty("stereo", isStereo)
     }
     amfEcmaArray.setProperty("filesize", 0.0)
+    customMetadata.forEach { (key, value) ->
+      amfEcmaArray.setProperty(key, value)
+    }
     metadata.addData(amfEcmaArray)
 
     metadata.writeHeader(socket)

rtmp/src/main/java/com/pedro/rtmp/rtmp/CommandsManagerAmf3.kt

@@ -118,6 +118,9 @@ class CommandsManagerAmf3: CommandsManager() {
       amfEcmaArray.setProperty("stereo", isStereo)
     }
     amfEcmaArray.setProperty("filesize", 0.0)
+    customMetadata.forEach { (key, value) ->
+      amfEcmaArray.setProperty(key, value)
+    }
     metadata.addData(amfEcmaArray)
 
     metadata.writeHeader(socket)

rtmp/src/main/java/com/pedro/rtmp/rtmp/RtmpClient.kt

@@ -196,6 +196,10 @@ class RtmpClient(private val connectChecker: ConnectChecker) {
     commandsManager.customAmfObject = amfObject
   }
 
+  fun setCustomMetadata(metadata: Map<String, Any>) {
+    commandsManager.customMetadata = metadata
+  }
+
   fun setAuthorization(user: String?, password: String?) {
     commandsManager.setAuth(user, password)
   }