X-Git-Url: https://piware.de/gitweb/?p=learn-rust.git;a=blobdiff_plain;f=concepts%2Fsrc%2Flib.rs;h=b0839ad391739cb186adcf2bb81c97484c5a3dfe;hp=bb15eeb814ea45d9aee64ed43497f586df33ba65;hb=1962a5412d16f34c738e9f81bb5271da043a7505;hpb=c2e012ab539dff44f0690e5d7cab4880400454fe

diff --git a/concepts/src/lib.rs b/concepts/src/lib.rs
index bb15eeb..b0839ad 100644
--- a/concepts/src/lib.rs
+++ b/concepts/src/lib.rs
@@ -137,11 +137,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 +157,38 @@ 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,10 +198,14 @@ 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
     }