Give codecs complete control of their input files

1 files changed, 37 insertions, 0 deletions

diff --git a/src/codecs/include/source_buffer.hpp b/src/codecs/include/source_buffer.hpp
new file mode 100644
index 00000000..d0d7635a
--- /dev/null
+++ b/src/codecs/include/source_buffer.hpp
@@ -0,0 +1,37 @@
+/*
+ * Copyright 2023 jacqueline <me@jacqueline.id.au>
+ *
+ * SPDX-License-Identifier: GPL-3.0-only
+ */
+
+#pragma once
+
+#include <cstddef>
+#include <cstdint>
+#include <functional>
+
+#include "span.hpp"
+
+#include "codec.hpp"
+
+namespace codecs {
+
+class SourceBuffer {
+ public:
+  SourceBuffer();
+  ~SourceBuffer();
+
+  auto Refill(IStream* src) -> bool;
+  auto AddBytes(std::function<size_t(cpp::span<std::byte>)> writer) -> void;
+  auto ConsumeBytes(std::function<size_t(cpp::span<std::byte>)> reader) -> void;
+
+  SourceBuffer(const SourceBuffer&) = delete;
+  SourceBuffer& operator=(const SourceBuffer&) = delete;
+
+ private:
+  const cpp::span<std::byte> buffer_;
+  size_t bytes_in_buffer_;
+  size_t offset_of_bytes_;
+};
+
+}  // namespace codecs