add end of seek event

membraneframework · Jun 27, 2024 · 349b0b3 · 349b0b3
1 parent 1a1d3f9
commit 349b0b3
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/lib/membrane_file/end_of_seek_event.ex b/lib/membrane_file/end_of_seek_event.ex
@@ -0,0 +1,10 @@
+defmodule Membrane.File.EndOfSeekEvent do
+  @moduledoc """
+  An event sent by the `Membrane.File.Source` with `seekable?: true` option,
+  after it sent all the data requested by `Membrane.File.SeekSourceEvent` or
+  whole file was read.
+  """
+  @derive Membrane.EventProtocol
+
+  defstruct []
+end
diff --git a/lib/membrane_file/source.ex b/lib/membrane_file/source.ex
@@ -10,8 +10,7 @@ defmodule Membrane.File.Source do
   use Membrane.Source
 
   alias Membrane.{Buffer, RemoteStream}
-  alias Membrane.File.NewSeekEvent
-  alias Membrane.File.SeekSourceEvent
+  alias Membrane.File.{EndOfSeekEvent, NewSeekEvent, SeekSourceEvent}
 
   @common_file Membrane.File.CommonFileBehaviour.get_impl()
 
@@ -185,8 +184,10 @@ defmodule Membrane.File.Source do
     actions =
       buffer_actions ++
         cond do
-          state.should_send_eos? and (state.size_to_read == 0 or supplied_size < to_supply_size) ->
-            [end_of_stream: :output]
+          state.size_to_read == 0 or supplied_size < to_supply_size ->
+            eos = if state.should_send_eos?, do: [end_of_stream: :output], else: []
+            end_of_seek = if state.seekable?, do: [event: {:output, %EndOfSeekEvent{}}], else: []
+            end_of_seek ++ eos
 
           to_supply_size == supplied_size ->
             redemand