fix fuse error

Cargo.lock

@@ -639,7 +639,7 @@ dependencies = [
 
 [[package]]
 name = "markov"
-version = "0.4.1"
+version = "0.5.0"
 dependencies = [
  "async-compression",
  "cfg-if 1.0.0",

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "markov"
-version = "0.5.0"
+version = "0.5.1"
 description = "Generate string of text from Markov chain fed by stdin"
 authors = ["Avril <flanchan@cumallover.me>"]
 edition = "2018"

src/api.rs

@@ -51,10 +51,10 @@ async fn single_stream(host: IpAddr, num: Option<usize>, body: impl Buf) -> Resu
     cfg_if! {
 	if #[cfg(feature="split-newlines")] {
 	    for body in body.split('\n').filter(|line| !line.trim().is_empty()) {
-		feed::feed(&mut chain, body);
+		feed::feed(&mut chain, body, 1..);
 	    }
 	}else {
-	    feed::feed(&mut chain, body);
+	    feed::feed(&mut chain, body, 1..);
 	}
     }
     if_debug!{

src/chunking.rs

@@ -17,6 +17,8 @@ use tokio::{
 use futures::{
     stream::{
 	Stream,
+	StreamExt,
+	Fuse,
     },
 };
 use pin_project::pin_project;
@@ -27,7 +29,7 @@ pub struct StreamReader<I, T>
 where I: Stream<Item=T>
 {
     #[pin]
-    source: I,
+    source: Fuse<I>,
     buffer: Vec<u8>,
 }
 
@@ -43,13 +45,13 @@ where I: Stream<Item=T>,
     /// Consume into the original stream
     pub fn into_inner(self) -> I
     {
-	self.source
+	self.source.into_inner()
     }
     /// Create a new instance with a buffer capacity
     pub fn with_capacity(source: I, cap: usize) -> Self
     {
 	Self {
-	    source,
+	    source: source.fuse(),
 	    buffer: Vec::with_capacity(cap)
 	}
     }
@@ -57,7 +59,7 @@ where I: Stream<Item=T>,
     pub fn new(source: I) -> Self
     {
 	Self {
-	    source,
+	    source: source.fuse(),
 	    buffer: Vec::new(),
 	}
     }

src/feed.rs

@@ -1,13 +1,22 @@
 //! Feeding the chain
 use super::*;
 
-pub fn feed(chain: &mut Chain<String>, what: impl AsRef<str>)
+const FEED_BOUNDS: std::ops::RangeFrom<usize> = 2..; //TODO: Add to config somehow
+
+
+pub fn feed(chain: &mut Chain<String>, what: impl AsRef<str>, bounds: impl std::ops::RangeBounds<usize>) -> bool
 {
     let map = what.as_ref().split_whitespace()
         .filter(|word| !word.is_empty())
         .map(|s| s.to_owned()).collect::<Vec<_>>();
-    if map.len() > 0 {
+    debug_assert!(!bounds.contains(&0), "Cannot allow 0 size feeds");
+    if bounds.contains(&map.len()) {
 	chain.feed(map);
+	true
+    }
+    else {
+	debug!("Ignoring feed of invalid length {}", map.len());
+	false
     }
 }
 
@@ -38,10 +47,10 @@ pub async fn full(who: &IpAddr, state: State, body: impl Unpin + Stream<Item = R
 	    cfg_if! {
 		if #[cfg(feature="split-newlines")] {
 		    for buffer in buffer.split('\n').filter(|line| !line.trim().is_empty()) {
-			feed(&mut chain, buffer);
+			feed(&mut chain, buffer, FEED_BOUNDS);
 		    }
 		} else {
-		    feed(&mut chain, buffer);
+		    feed(&mut chain, buffer, FEED_BOUNDS);
 
 		}
 	    }
@@ -59,7 +68,7 @@ pub async fn full(who: &IpAddr, state: State, body: impl Unpin + Stream<Item = R
 		    #[cfg(not(feature="hog-buffer"))] 
 		    let mut chain = state.chain().write().await; // Acquire mutex once per line? Is this right?
 		    
-		    feed(&mut chain, line);
+		    feed(&mut chain, line, FEED_BOUNDS);
 		    info!("{} -> {:?}", who, line);
 		}
 		written+=line.len();

src/main.rs

@@ -140,7 +140,7 @@ async fn main() {
 		async move {
 		    feed::full(&host, state, buf).await
 			.map(|_| warp::reply::with_status(warp::reply(), status!(201)))
-			.map_err(warp::reject::custom)
+			.map_err(|_| warp::reject::not_found()) //(warp::reject::custom) //TODO: Recover rejection filter down below for custom error return
 		}
 	    })
 	    .with(warp::log("markov::put"));