add base64 input for ‘set’

2 files changed, 45 insertions, 0 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 9914f77..2dee9ca 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -571,9 +571,40 @@ impl Palette
         Self::from_buffered_reader(&mut reader)
     }
 
+    fn from_bytes(b: &[u8]) -> io::Result<Self>
+    {
+        if b.len() != PALETTE_SIZE * 3 {
+            return Err(io::Error::new(
+                io::ErrorKind::Other,
+                format!(
+                    "expected {} B of data, got {}",
+                    PALETTE_SIZE * 3,
+                    b.len()
+                ),
+            ));
+        }
+
+        let mut res = Self::new();
+        res.0.copy_from_slice(&b);
+
+        Ok(res)
+    }
+
     /* [Palette::from_stdin] */
 
     pub fn base64(&self) -> String { base64::encode(&self.0) }
+
+    pub fn from_base64(b64: &str) -> io::Result<Self>
+    {
+        base64::decode(b64.as_bytes())
+            .map_err(|e| {
+                io::Error::new(
+                    io::ErrorKind::Other,
+                    format!("failed to decode input as base64: {}", e),
+                )
+            })
+            .and_then(|b| Self::from_bytes(&b))
+    }
 } /* [impl Palette] */
 
 impl fmt::Display for Palette
diff --git a/src/vtcol.rs b/src/vtcol.rs
index 0a198e6..8161241 100644
--- a/src/vtcol.rs
+++ b/src/vtcol.rs
@@ -76,6 +76,15 @@ impl<'a> Job
                             .value_name("PATH")
                             .help("apply scheme from file")
                             .takes_value(true),
+                    )
+                    .arg(
+                        Arg::with_name("base64")
+                            .short("b")
+                            .long("base64")
+                            .help("base64 encoded binary input")
+                            .value_name("DATA")
+                            .required(false)
+                            .takes_value(true),
                     ),
             )
             .subcommand(
@@ -174,6 +183,11 @@ impl<'a> Job
             },
             ("list", _) => Ok(Self::List),
             ("set", Some(subm)) => {
+                if let Some(b64) = subm.value_of("base64") {
+                    let scheme =
+                        Palette::from_base64(&b64).map(Scheme::from)?;
+                    return Ok(Self::Set(scheme));
+                }
                 let scheme = match subm.value_of("scheme") {
                     Some("-") => Self::read_scheme_from_stdin(),
                     Some(name) => {