rust wrappers

Author: mikea

kj-rs/io/bridge.c++

@@ -8,6 +8,7 @@
 using namespace kj_rs;
 
 namespace kj_rs_io {
+namespace ffi {
 
 CxxAsyncInputStream::CxxAsyncInputStream(kj::Own<kj::AsyncInputStream> stream)
     : stream(kj::mv(stream)) {}
@@ -127,4 +128,5 @@ void CxxAsyncIoStream::abort_read() {
   stream->abortRead();
 }
 
+}  // namespace ffi
 }  // namespace kj_rs_io

kj-rs/io/bridge.h

@@ -7,7 +7,8 @@
 
 namespace kj_rs_io {
 
-// Forward declarations
+// Forward declarations  
+namespace ffi {
 struct CxxAsyncOutputStream;
 
 // Wrapper structs for KJ async stream types to provide controlled FFI interface
@@ -63,4 +64,71 @@ struct CxxAsyncIoStream {
   void abort_read();
 };
 
+} // namespace ffi
+
+// C++ classes that wrap Rust FFI types and implement KJ interfaces
+namespace ffi {
+
+// Forward declarations for Rust FFI types  
+struct RustAsyncInputStream;
+struct RustAsyncOutputStream; 
+struct RustAsyncIoStream;
+
+// C++ classes that wrap Rust FFI types and implement KJ interfaces
+
+class RustInputStreamWrapper : public kj::AsyncInputStream {
+public:
+  explicit RustInputStreamWrapper(::rust::Box<RustAsyncInputStream> rust_stream);
+  virtual ~RustInputStreamWrapper() = default;
+
+  // kj::AsyncInputStream interface
+  kj::Promise<size_t> tryRead(void* buffer, size_t minBytes, size_t maxBytes) override;
+  kj::Maybe<uint64_t> tryGetLength() override;
+  kj::Promise<uint64_t> pumpTo(kj::AsyncOutputStream& output, uint64_t amount) override;
+
+private:
+  ::rust::Box<RustAsyncInputStream> rust_stream_;
+};
+
+class RustOutputStreamWrapper : public kj::AsyncOutputStream {
+public:
+  explicit RustOutputStreamWrapper(::rust::Box<RustAsyncOutputStream> rust_stream);
+  virtual ~RustOutputStreamWrapper() = default;
+
+  // kj::AsyncOutputStream interface
+  kj::Promise<void> write(kj::ArrayPtr<const kj::byte> buffer) override;
+  kj::Promise<void> write(kj::ArrayPtr<const kj::ArrayPtr<const kj::byte>> pieces) override;
+  kj::Maybe<kj::Promise<uint64_t>> tryPumpFrom(kj::AsyncInputStream& input, uint64_t amount) override;
+  kj::Promise<void> whenWriteDisconnected() override;
+
+private:
+  ::rust::Box<RustAsyncOutputStream> rust_stream_;
+};
+
+class RustIoStreamWrapper : public kj::AsyncIoStream {
+public:
+  explicit RustIoStreamWrapper(::rust::Box<RustAsyncIoStream> rust_stream);
+  virtual ~RustIoStreamWrapper() = default;
+
+  // kj::AsyncInputStream interface
+  kj::Promise<size_t> tryRead(void* buffer, size_t minBytes, size_t maxBytes) override;
+  kj::Maybe<uint64_t> tryGetLength() override;
+  kj::Promise<uint64_t> pumpTo(kj::AsyncOutputStream& output, uint64_t amount) override;
+
+  // kj::AsyncOutputStream interface
+  kj::Promise<void> write(kj::ArrayPtr<const kj::byte> buffer) override;
+  kj::Promise<void> write(kj::ArrayPtr<const kj::ArrayPtr<const kj::byte>> pieces) override;
+  kj::Maybe<kj::Promise<uint64_t>> tryPumpFrom(kj::AsyncInputStream& input, uint64_t amount) override;
+  kj::Promise<void> whenWriteDisconnected() override;
+
+  // kj::AsyncIoStream interface
+  void shutdownWrite() override;
+  void abortRead() override;
+
+private:
+  ::rust::Box<RustAsyncIoStream> rust_stream_;
+};
+
+}  // namespace ffi
+
 }  // namespace kj_rs_io