concepts: rustfmt

[learn-rust.git] / concepts / src / lib.rs

diff --git a/concepts/src/lib.rs b/concepts/src/lib.rs
index bb15eeb814ea45d9aee64ed43497f586df33ba65..f2b1c100c043a0f567771536c36df6fa32ffd5c7 100644 (file)
--- a/concepts/src/lib.rs
+++ b/concepts/src/lib.rs
@@ -1,11 +1,10 @@
+use std::collections::HashMap;
 use std::fs::File;
 use std::io::prelude::*;
-use std::collections::HashMap;
 
 pub fn read_file(path: &str) -> Result<String, std::io::Error> {
     let mut s = String::new();
-    File::open(path)?
-        .read_to_string(&mut s)?;
+    File::open(path)?.read_to_string(&mut s)?;
     Ok(s)
 }
 
@@ -71,7 +70,10 @@ where
     V: Copy,
 {
     pub fn new(calc: T) -> Cacher<T, A, V> {
-        Cacher { calc, values: HashMap::new() }
+        Cacher {
+            calc,
+            values: HashMap::new(),
+        }
     }
 
     pub fn value(&mut self, arg: A) -> V {
@@ -87,7 +89,7 @@ where
 }
 
 pub struct Counter5 {
-    count: u32
+    count: u32,
 }
 
 impl Counter5 {
@@ -137,11 +139,16 @@ impl Post {
     pub fn approve(&mut self) {
         self.state = self.state.approve();
     }
+
+    pub fn reject(&mut self) {
+        self.state = self.state.reject();
+    }
 }
 
 trait State {
     fn request_review(&self) -> Box<dyn State>;
-    fn approve(&self) -> Box<dyn State>;
+    fn approve(&mut self) -> Box<dyn State>;
+    fn reject(&self) -> Box<dyn State>;
 
     #[allow(unused_variables)]
     fn content<'a>(&self, post: &'a Post) -> &'a str {
@@ -152,23 +159,40 @@ trait State {
 struct Draft {}
 impl State for Draft {
     fn request_review(&self) -> Box<dyn State> {
-        Box::new(PendingReview {})
+        Box::new(PendingReview { acks: 0 })
     }
 
-    fn approve(&self) -> Box<dyn State> {
+    fn approve(&mut self) -> Box<dyn State> {
         // don't change state
         Box::new(Self {})
     }
+
+    fn reject(&self) -> Box<dyn State> {
+        Box::new(Self {})
+    }
+}
+
+struct PendingReview {
+    acks: u32,
 }
 
-struct PendingReview {}
 impl State for PendingReview {
     fn request_review(&self) -> Box<dyn State> {
-        Box::new(Self {})
+        Box::new(Self { acks: self.acks })
     }
 
-    fn approve(&self) -> Box<dyn State> {
-        Box::new(Published {})
+    fn approve(&mut self) -> Box<dyn State> {
+        if self.acks >= 1 {
+            Box::new(Published {})
+        } else {
+            Box::new(Self {
+                acks: self.acks + 1,
+            })
+        }
+    }
+
+    fn reject(&self) -> Box<dyn State> {
+        Box::new(Draft {})
     }
 }
 
@@ -178,11 +202,66 @@ impl State for Published {
         Box::new(Self {})
     }
 
-    fn approve(&self) -> Box<dyn State> {
+    fn approve(&mut self) -> Box<dyn State> {
         Box::new(Published {})
     }
 
+    fn reject(&self) -> Box<dyn State> {
+        Box::new(Self {})
+    }
+
     fn content<'a>(&self, post: &'a Post) -> &'a str {
         &post.content
     }
 }
+
+// state encoded as types; this is the "approved" state
+pub struct TPost {
+    content: String,
+}
+
+impl TPost {
+    pub fn new() -> TPostDraft {
+        TPostDraft {
+            content: String::new(),
+        }
+    }
+
+    pub fn content(&self) -> &str {
+        &self.content
+    }
+}
+
+pub struct TPostDraft {
+    content: String,
+}
+
+impl TPostDraft {
+    pub fn add_text(&mut self, text: &str) {
+        self.content.push_str(text);
+    }
+
+    pub fn request_review(self) -> TPostReview {
+        TPostReview {
+            content: self.content,
+        }
+    }
+}
+
+pub struct TPostReview {
+    content: String,
+}
+
+impl TPostReview {
+    pub fn approve(self) -> TPost {
+        TPost {
+            content: self.content,
+        }
+    }
+
+    pub fn reject(self) -> TPostDraft {
+        TPostDraft {
+            content: self.content,
+        }
+    }
+}