module Messaging
  module Write
    module Substitute
      def self.build
        Substitute::Write.build.tap do |substitute_writer|
          sink = Messaging::Write.register_telemetry_sink(substitute_writer)
          substitute_writer.sink = sink
        end
      end

      class Write
        include Messaging::Write

        attr_accessor :sink

        def self.build(session: nil)
          new.tap do |instance|
            ::Telemetry.configure instance
          end
        end

        def writes(&blk)
          if blk.nil?
            return sink.written_records
          end

          sink.written_records.select do |record|
            blk.call(record.data.message, record.data.stream_name, record.data.expected_version, record.data.reply_stream_name)
          end
        end

        def written?(&blk)
          if blk.nil?
            return sink.recorded_written?
          end

          sink.recorded_written? do |record|
            blk.call(record.data.message, record.data.stream_name, record.data.expected_version, record.data.reply_stream_name)
          end
        end

        def replies(&blk)
          if blk.nil?
            return sink.replied_records
          end

          sink.replied_records.select do |record|
            blk.call(record.data.message, record.data.stream_name)
          end
        end

        def replied?(&blk)
          if blk.nil?
            return sink.recorded_replied?
          end

          sink.recorded_replied? do |record|
            blk.call(record.data.message, record.data.stream_name)
          end
        end
      end
    end
  end
end