From 4a965846153abf30558c9aa20e5d410b3c837a09 Mon Sep 17 00:00:00 2001
From: laurens <miers132@gmail.com>
Date: Sun, 11 Oct 2020 17:52:56 +0200
Subject: [PATCH] state_pattern_types: add approval count of 2

Not sure this is the best way but hey, it works
---
 state_pattern_types/src/main.rs | 44 +++++++++++++++++++++++++++------
 1 file changed, 37 insertions(+), 7 deletions(-)

diff --git a/state_pattern_types/src/main.rs b/state_pattern_types/src/main.rs
index 4059d1d..1d7fe95 100644
--- a/state_pattern_types/src/main.rs
+++ b/state_pattern_types/src/main.rs
@@ -4,6 +4,7 @@ pub struct DraftPost {
 
 pub struct PendingReviewPost {
     content: String,
+    approve_count: i32,
 }
 
 pub struct Post {
@@ -18,6 +19,7 @@ impl DraftPost {
     pub fn request_review(self) -> PendingReviewPost {
         PendingReviewPost {
             content: self.content,
+            approve_count: 0,
         }
     }
 }
@@ -29,9 +31,18 @@ impl PendingReviewPost {
         }
     }
 
-    pub fn approve(self) -> Post {
-        Post {
-            content: self.content,
+    pub fn get_content(&self) -> &str {
+        ""
+    }
+
+    pub fn approve(&mut self) -> Option<Post> {
+        self.approve_count += 1;
+        if self.approve_count >= 2 {
+            Some(Post {
+                content: self.content.clone(),
+            })
+        } else {
+            None
         }
     }
 }
@@ -43,19 +54,38 @@ impl Post {
         }
     }
 
-    pub fn content(&self) -> &str {
+    pub fn get_content(&self) -> &str {
         &self.content
     }
 }
 
+fn _approve_post(post: &mut PendingReviewPost) -> Option<Post> {
+    let mut approve_count = 0;
+    for _ in 0..10 {
+        approve_count += 1;
+        match post.approve() {
+            Some(x) => {
+                println!("Post approved after {} approvals", approve_count);
+                return Some(x);
+            }
+            None => println!("Not enough approval counts"),
+        };
+    }
+
+    None
+}
+
 fn main() {
     let mut post = Post::new();
 
     post.add_text("I ate a salad for lunch today");
 
-    let post = post.request_review();
+    let mut post = post.request_review();
 
-    let post = post.approve();
+    let approv_res = _approve_post(&mut post);
 
-    assert_eq!("I ate a salad for lunch today", post.content());
+    match approv_res {
+        Some(post) => assert_eq!("I ate a salad for lunch today", post.get_content()),
+        None => panic!("Failed to approve post!"),
+    }
 }