Merge pull request #1917 from pedroSG94/feature/customamf0

Feature/customamf0

Author: pedroSG94

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

@@ -24,7 +24,7 @@ import javax.net.ssl.TrustManager
  * Created by pedro on 12/10/23.
  */
 class RtmpStreamClient(
-  private val rtmpClient: RtmpClient, 
+  private val rtmpClient: RtmpClient,
   private val streamClientListener: StreamClientListener?
 ): StreamBaseClient() {
 
@@ -174,4 +174,8 @@ class RtmpStreamClient(
   override fun setSocketType(type: SocketType) {
     rtmpClient.socketType = type
   }
+
+  fun setCustomAmfObject(amfObject: Map<String, Any>) {
+    rtmpClient.setCustomAmfObject(amfObject)
+  }
 }

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

@@ -88,6 +88,32 @@ open class AmfObject(private val properties: LinkedHashMap<AmfString, AmfData> =
     bodySize += data.getSize() + 1
   }
 
+  open fun setProperty(name: String, data: Any) {
+    val newValue: AmfData = when (data) {
+      is String   -> AmfString(data)
+      is Int      -> AmfNumber(data.toDouble())
+      is Long     -> AmfNumber(data.toDouble())
+      is Double   -> AmfNumber(data)
+      is Float    -> AmfNumber(data.toDouble())
+      is Boolean  -> AmfBoolean(data)
+      is AmfData  -> data
+      else        -> throw IllegalArgumentException(
+        "Unsupported value type: ${data::class.java.name}"
+      )
+    }
+
+    val existingEntry = properties.entries.find { it.key.value == name }
+
+    if (existingEntry != null) {
+      properties[existingEntry.key] = newValue
+      bodySize += newValue.getSize() - existingEntry.value.getSize()
+    } else {
+      val key = AmfString(name)
+      properties[key] = newValue
+      bodySize += key.getSize() + newValue.getSize() + 1
+    }
+  }
+
   fun getProperties() = properties
 
   @Throws(IOException::class)

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

@@ -61,6 +61,7 @@ abstract class CommandsManager {
   var readChunkSize = RtmpConfig.DEFAULT_CHUNK_SIZE
   var audioDisabled = false
   var videoDisabled = false
+  var customAmfObject: Map<String, Any> = emptyMap()
   private var bytesRead = 0
   private var acknowledgementSequence = 0
 

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

@@ -56,6 +56,11 @@ class CommandsManagerAmf0: CommandsManager() {
       }
     }
     connectInfo.setProperty("objectEncoding", 0.0)
+
+    // Inject other custom AMF fields as-is
+    customAmfObject.forEach { (key, value) ->
+      connectInfo.setProperty(key, value)
+    }
     connect.addData(connectInfo)
 
     connect.writeHeader(socket)

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

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