Handle FCPublish command (#8)

Author: gBillal

lib/ex_rtmp/chunk.ex

@@ -50,14 +50,8 @@ defmodule ExRTMP.Chunk do
     msg_stream_id =
       if chunk.message_stream_id, do: <<chunk.message_stream_id::integer-32-little>>, else: <<>>
 
-    [
-      <<chunk.fmt::2, encode_stream_id(chunk.stream_id)::bitstring>>,
-      timestamp,
-      length,
-      type_id,
-      msg_stream_id,
-      chunk.payload
-    ]
+    <<chunk.fmt::2, encode_stream_id(chunk.stream_id)::bitstring, timestamp::binary,
+      length::binary, type_id::binary, msg_stream_id::binary, chunk.payload::binary>>
   end
 
   defp parse_stream_id(data) do

lib/ex_rtmp/message.ex

@@ -6,7 +6,7 @@ defmodule ExRTMP.Message do
   require Logger
 
   alias __MODULE__.Command.NetConnection.{Connect, CreateStream, Response}
-  alias __MODULE__.Command.NetStream.{DeleteStream, OnStatus, Play, Publish}
+  alias __MODULE__.Command.NetStream.{DeleteStream, FCPublish, OnStatus, Play, Publish}
   alias __MODULE__.Metadata
   alias __MODULE__.UserControl.Event
   alias ExRTMP.Chunk
@@ -239,6 +239,9 @@ defmodule ExRTMP.Message do
         ["publish", transaction_id, nil, name, type] ->
           Publish.new(transaction_id, name, type)
 
+        ["FCPublish", transaction_id, nil, name] ->
+          FCPublish.new(transaction_id, name)
+
         ["deleteStream", transaction_id, nil, stream_id] ->
           DeleteStream.new(transaction_id, stream_id)
 

lib/ex_rtmp/message/command/net_stream/fc_publish.ex

@@ -0,0 +1,30 @@
+defmodule ExRTMP.Message.Command.NetStream.FCPublish do
+  @moduledoc """
+  Struct representing an "FCPublish" command message for RTMP NetStream.
+  """
+
+  @type t :: %__MODULE__{
+          transaction_id: number(),
+          name: String.t()
+        }
+
+  defstruct [:transaction_id, :name]
+
+  @spec new(number(), String.t()) :: t()
+  def new(transaction_id, name) do
+    %__MODULE__{transaction_id: transaction_id, name: name}
+  end
+
+  defimpl ExRTMP.Message.Serializer do
+    alias ExRTMP.AMF0
+
+    def serialize(publish) do
+      IO.iodata_to_binary([
+        AMF0.serialize("FCPublish"),
+        AMF0.serialize(publish.transaction_id),
+        AMF0.serialize(nil),
+        AMF0.serialize(publish.name)
+      ])
+    end
+  end
+end

lib/ex_rtmp/message/command/net_stream/publish.ex

@@ -1,5 +1,7 @@
 defmodule ExRTMP.Message.Command.NetStream.Publish do
-  @moduledoc false
+  @moduledoc """
+  Struct representing a "publish" command message for RTMP NetStream.
+  """
 
   @type publishing_type :: :live | :record | :append
 

lib/ex_rtmp/server/client_session.ex

@@ -12,7 +12,7 @@ defmodule ExRTMP.Server.ClientSession do
   alias ExRTMP.Message
   alias ExRTMP.Message.Command.NetConnection
   alias ExRTMP.Message.Command.NetConnection.{CreateStream, Response}
-  alias ExRTMP.Message.Command.NetStream.{DeleteStream, Play, Publish, OnStatus}
+  alias ExRTMP.Message.Command.NetStream.{DeleteStream, FCPublish, Play, Publish, OnStatus}
   alias ExRTMP.Message.Metadata
 
   @default_acknowledgement_size 3_000_000
@@ -237,6 +237,10 @@ defmodule ExRTMP.Server.ClientSession do
         %CreateStream{} ->
           handle_create_stream_message(message.payload, state)
 
+        %FCPublish{transaction_id: id} ->
+          send_messages(state, [Message.command(Response.ok(id))])
+          {[], state}
+
         %Publish{} ->
           handle_publish_message(message.payload, message.stream_id, state)