From e913951b9149607453c13682d7a54d81d59b05f9 Mon Sep 17 00:00:00 2001
From: Tony Garnock-Jones <tonyg@leastfixedpoint.com>
Date: Mon, 2 Aug 2021 21:49:40 +0200
Subject: [PATCH] TextWriter::encode for Vec<u8>

---
 .../rust/preserves/src/value/packed/writer.rs |  5 ++---
 .../rust/preserves/src/value/text/writer.rs   | 20 +++++++++++++++++++
 2 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/implementations/rust/preserves/src/value/packed/writer.rs b/implementations/rust/preserves/src/value/packed/writer.rs
index 64be487..810a529 100644
--- a/implementations/rust/preserves/src/value/packed/writer.rs
+++ b/implementations/rust/preserves/src/value/packed/writer.rs
@@ -16,14 +16,13 @@ use super::super::writer::{Writer, CompoundWriter, varint};
 
 pub struct PackedWriter<W: io::Write>(Suspendable<W>);
 
-impl PackedWriter<Vec<u8>> {
+impl PackedWriter<&mut Vec<u8>> {
     pub fn encode<D: Embeddable, N: NestedValue<D>, Enc: DomainEncode<D>>(
         enc: &mut Enc,
         v: &N,
     ) -> io::Result<Vec<u8>> {
         let mut buf: Vec<u8> = Vec::new();
-        let w = &mut PackedWriter::new(&mut buf);
-        w.write(enc, v)?;
+        PackedWriter::new(&mut buf).write(enc, v)?;
         Ok(buf)
     }
 
diff --git a/implementations/rust/preserves/src/value/text/writer.rs b/implementations/rust/preserves/src/value/text/writer.rs
index ae2bc4f..401889d 100644
--- a/implementations/rust/preserves/src/value/text/writer.rs
+++ b/implementations/rust/preserves/src/value/text/writer.rs
@@ -1,3 +1,8 @@
+use crate::value::DomainEncode;
+use crate::value::Embeddable;
+use crate::value::IOValue;
+use crate::value::IOValueDomainCodec;
+use crate::value::NestedValue;
 use crate::value::Writer;
 use crate::value::suspendable::Suspendable;
 use crate::value::writer::CompoundWriter;
@@ -28,6 +33,21 @@ impl std::default::Default for CommaStyle {
     }
 }
 
+impl TextWriter<&mut Vec<u8>> {
+    pub fn encode<D: Embeddable, N: NestedValue<D>, Enc: DomainEncode<D>>(
+        enc: &mut Enc,
+        v: &N,
+    ) -> io::Result<Vec<u8>> {
+        let mut buf: Vec<u8> = Vec::new();
+        TextWriter::new(&mut buf).write(enc, v)?;
+        Ok(buf)
+    }
+
+    pub fn encode_iovalue(v: &IOValue) -> io::Result<Vec<u8>> {
+        Self::encode(&mut IOValueDomainCodec, v)
+    }
+}
+
 impl<W: io::Write> TextWriter<W> {
     pub fn new(w: W) -> Self {
         TextWriter {