From 5980183a33bd75235c3dcffdf6e0027462493749 Mon Sep 17 00:00:00 2001
From: Romain Vimont <rom@rom1v.com>
Date: Sun, 11 Apr 2021 15:01:05 +0200
Subject: [PATCH] Expose recorder as packet sink

Make recorder implement the packet sink trait.

This will allow the stream to push packets without depending on the
concrete sink type.
---
 app/src/recorder.c | 29 +++++++++++++++++++++++++++++
 app/src/recorder.h |  3 +++
 2 files changed, 32 insertions(+)

diff --git a/app/src/recorder.c b/app/src/recorder.c
index 387d8f79..fabbc95c 100644
--- a/app/src/recorder.c
+++ b/app/src/recorder.c
@@ -5,6 +5,9 @@
 
 #include "util/log.h"
 
+/** Downcast packet_sink to recorder */
+#define DOWNCAST(SINK) container_of(SINK, struct recorder, packet_sink)
+
 static const AVRational SCRCPY_TIME_BASE = {1, 1000000}; // timestamps in us
 
 static const AVOutputFormat *
@@ -312,6 +315,24 @@ recorder_push(struct recorder *recorder, const AVPacket *packet) {
     return true;
 }
 
+static bool
+recorder_packet_sink_open(struct sc_packet_sink *sink, const AVCodec *codec) {
+    struct recorder *recorder = DOWNCAST(sink);
+    return recorder_open(recorder, codec);
+}
+
+static void
+recorder_packet_sink_close(struct sc_packet_sink *sink) {
+    struct recorder *recorder = DOWNCAST(sink);
+    recorder_close(recorder);
+}
+
+static bool
+recorder_packet_sink_push(struct sc_packet_sink *sink, const AVPacket *packet) {
+    struct recorder *recorder = DOWNCAST(sink);
+    return recorder_push(recorder, packet);
+}
+
 bool
 recorder_init(struct recorder *recorder,
               const char *filename,
@@ -346,6 +367,14 @@ recorder_init(struct recorder *recorder,
     recorder->header_written = false;
     recorder->previous = NULL;
 
+    static const struct sc_packet_sink_ops ops = {
+        .open = recorder_packet_sink_open,
+        .close = recorder_packet_sink_close,
+        .push = recorder_packet_sink_push,
+    };
+
+    recorder->packet_sink.ops = &ops;
+
     return true;
 }
 
diff --git a/app/src/recorder.h b/app/src/recorder.h
index 97e499bc..4991f0cf 100644
--- a/app/src/recorder.h
+++ b/app/src/recorder.h
@@ -8,6 +8,7 @@
 
 #include "coords.h"
 #include "scrcpy.h"
+#include "trait/packet_sink.h"
 #include "util/queue.h"
 #include "util/thread.h"
 
@@ -19,6 +20,8 @@ struct record_packet {
 struct recorder_queue QUEUE(struct record_packet);
 
 struct recorder {
+    struct sc_packet_sink packet_sink; // packet sink trait
+
     char *filename;
     enum sc_record_format format;
     AVFormatContext *ctx;