Added preprocessor to enable undo or not

Refactored to use HangmanMemento with internal property

Author: ardalis

DesignPatternsInCSharp/Memento/HangmanGameWithUndo.cs

@@ -2,22 +2,18 @@
 {
     public class HangmanGameWithUndo : HangmanGame
     {
-        public IMemento CreateSetPoint()
+        public HangmanMemento CreateSetPoint()
         {
             var guesses = PreviousGuesses.ToArray();
-            return new HangmanState() { State = guesses };
+            return new HangmanMemento() { Guesses = guesses };
         }
 
-        public void ResumeFrom(IMemento memento)
+        public void ResumeFrom(HangmanMemento memento)
         {
-            var guesses = (char[])memento.State;
+            var guesses = memento.Guesses;
             PreviousGuesses.Clear();
             PreviousGuesses.AddRange(guesses);
         }
-
-        public class HangmanState : IMemento
-        {
-            public object State { get; set; }
-        }
     }
+
 }

DesignPatternsInCSharp/Memento/HangmanMemento.cs

@@ -0,0 +1,7 @@
+namespace DesignPatternsInCSharp.Memento
+{
+    public sealed class HangmanMemento
+    {
+        internal char[] Guesses { get; set; }
+    }
+}

DesignPatternsInCSharp/Memento/IMemento.cs

@@ -2,7 +2,7 @@
 
 namespace DesignPatternsInCSharp.Memento.Tests
 {
-    public class HangmaneGameConstructor
+    public class HangmaneGamConstructor
     {
         private const string TEST_SECRET_WORD = "TEST";
 

DesignPatternsInCSharp/Memento/Tests/HangmaneGameConstructor.cs renamed to DesignPatternsInCSharp/Memento/Tests/HangmanGameConstructor.cs

@@ -1,4 +1,6 @@
-using DesignPatternsInCSharp.Memento;
+#define SupportUndo
+
+using DesignPatternsInCSharp.Memento;
 using System;
 using System.Collections.Generic;
 
@@ -11,11 +13,14 @@ class Program
     {
         static void Main(string[] args)
         {
-            //var game = new HangmanGame();
+
+#if SupportUndo
             var game = new HangmanGameWithUndo();
-            var gameHistory = new Stack<IMemento>();
+            var gameHistory = new Stack<HangmanMemento>();
             gameHistory.Push(game.CreateSetPoint());
-
+#else
+            var game = new HangmanGame();
+#endif
 
             while (!game.IsOver)
             {
@@ -26,10 +31,16 @@ static void Main(string[] args)
                 Console.WriteLine(game.CurrentMaskedWord);
                 Console.WriteLine($"Previous Guesses: {String.Join(',', game.PreviousGuesses.ToArray())}");
                 Console.WriteLine($"Guesses Left: {game.GuessesRemaining}");
+
+#if SupportUndo
                 Console.WriteLine("Guess (a-z or '-' to undo last guess): ");
+#else
+                Console.WriteLine("Guess (a-z): ");
+#endif
 
                 var entry = char.ToUpperInvariant(Console.ReadKey().KeyChar);
 
+#if SupportUndo
                 if(entry == '-')
                 {
                     if(gameHistory.Count > 1)
@@ -40,8 +51,11 @@ static void Main(string[] args)
                         continue;
                     }
                 }
+#endif
                 game.Guess(entry);
+#if SupportUndo
                 gameHistory.Push(game.CreateSetPoint());
+#endif
                 Console.WriteLine();
             }