better handling of the state machine object

Breeze/src/Breeze.TumbleBit.Client/IStateMachine.cs

@@ -8,10 +8,10 @@
         void Save();
 
         /// <summary>
-        /// Loads the state of the current tumbling session.
+        /// Loads the saved state of the tumbling execution to the file system.
         /// </summary>
         /// <returns></returns>
-        IStateMachine Load();
+        void LoadStateFromMemory();
 
         /// <summary>
         /// Deletes the state of the current tumbling session..

Breeze/src/Breeze.TumbleBit.Client/TumbleBitManager.cs

@@ -34,8 +34,7 @@ namespace Breeze.TumbleBit.Client
             this.network = network;
             this.logger = loggerFactory.CreateLogger(this.GetType().FullName);
 
-            // load the persisted tumbling state 
-            this.tumblingState = TumblingState.LoadState();
+            this.tumblingState = new TumblingState(loggerFactory);
         }
 
         /// <inheritdoc />
@@ -49,12 +48,11 @@ namespace Breeze.TumbleBit.Client
                 throw new Exception($"The tumbler is on network {this.TumblerParameters.Network} while the wallet is on network {this.network}.");
             }
             
-            if (this.tumblingState == null)
-            {
-                this.tumblingState = new TumblingState();
-            }
+            // load the current tumbling state fromt he file system
+            this.tumblingState.LoadStateFromMemory();
             
             // update and save the state
+            this.tumblingState.TumblerUri = serverAddress;
             this.tumblingState.TumblerParameters = this.TumblerParameters;
             this.tumblingState.Save();
 

Breeze/src/Breeze.TumbleBit.Client/TumblingState.cs

@@ -15,6 +15,8 @@ namespace Breeze.TumbleBit.Client
 {
     public class TumblingState : IStateMachine
     {
+        private readonly ILogger logger;
+
         private const string StateFileName = "tumblebit_state.json";
 
         [JsonProperty("tumblerParameters")]
@@ -23,21 +25,26 @@ namespace Breeze.TumbleBit.Client
         [JsonProperty("tumblerUri")]
         public Uri TumblerUri { get; set; }
 
-        [JsonProperty("lastBlockReceivedHeight")]
+        [JsonProperty("lastBlockReceivedHeight", DefaultValueHandling = DefaultValueHandling.Ignore)]
         public int LastBlockReceivedHeight { get; set; }
 
-        [JsonProperty("originWalletName")]
+        [JsonProperty("originWalletName", NullValueHandling = NullValueHandling.Ignore)]
         public string OriginWalletName { get; set; }
 
-        [JsonProperty("destinationWalletName")]
+        [JsonProperty("destinationWalletName", NullValueHandling = NullValueHandling.Ignore)]
         public string DestinationWalletName { get; set; }
 
-        [JsonProperty("sessions")]
+        [JsonProperty("sessions", NullValueHandling = NullValueHandling.Ignore)]
         public IList<Session> Sessions { get; set; }
 
+        [JsonConstructor]
         public TumblingState()
         {
-            this.Sessions = new List<Session>();
+        }
+
+        public TumblingState(ILoggerFactory loggerFactory)
+        {
+            this.logger = loggerFactory.CreateLogger(this.GetType().FullName);
         }
 
         /// <inheritdoc />
@@ -47,9 +54,23 @@ namespace Breeze.TumbleBit.Client
         }
 
         /// <inheritdoc />
-        public IStateMachine Load()
+        public void LoadStateFromMemory()
+        {
+            var stateFilePath = GetStateFilePath();
+            if (!File.Exists(stateFilePath))
             {
-            return LoadState();
+                return;
+            }
+
+            // load the file from the local system
+            var savedState = JsonConvert.DeserializeObject<TumblingState>(File.ReadAllText(stateFilePath));
+
+            this.Sessions = savedState.Sessions ?? new List<Session>();
+            this.OriginWalletName = savedState.OriginWalletName;
+            this.DestinationWalletName = savedState.DestinationWalletName;
+            this.LastBlockReceivedHeight = savedState.LastBlockReceivedHeight;
+            this.TumblerParameters = savedState.TumblerParameters;
+            this.TumblerUri = savedState.TumblerUri;            
         }
 
         /// <inheritdoc />
@@ -64,24 +85,31 @@ namespace Breeze.TumbleBit.Client
         {
             // get the next cycle to be started
             var cycle = this.TumblerParameters.CycleGenerator.GetRegistratingCycle(this.LastBlockReceivedHeight);
-            var lastCycleStarted = this.Sessions.Max(s => s.StartCycle);
 
-            // check if we need to start a new session starting from the registration cycle
-            // TODO remove the limitation to have only 1 session
-            if (lastCycleStarted != cycle.Start && this.Sessions.Count == 0)
-            {                
-                if (this.Sessions.SingleOrDefault(s => s.StartCycle == cycle.Start) == null)
+            // create a new session if allowed
+            if (this.Sessions.Count == 0)
             {
                 this.CreateNewSession(cycle.Start);
             }
+            else
+            {
+                // TODO remove the limitation to have only 1 session
+                //var lastCycleStarted = this.Sessions.Max(s => s.StartCycle);
+
+                //// check if we need to start a new session starting from the registration cycle
+                //if (lastCycleStarted != cycle.Start)
+                //{
+                //    if (this.Sessions.SingleOrDefault(s => s.StartCycle == cycle.Start) == null)
+                //    {
+                //        this.CreateNewSession(cycle.Start);
+                //    }
+                //}
             }
 
             // get a list of cycles we expect to have at this height
             var cycles = this.TumblerParameters.CycleGenerator.GetCycles(this.LastBlockReceivedHeight);
             var existingSessions = cycles.SelectMany(c => this.Sessions.Where(s => s.StartCycle == c.Start)).ToList();
             foreach (var existingSession in existingSessions)
-            {
-                try
             {                
                 // create a new session to be updated
                 var session = new Session();
@@ -96,7 +124,7 @@ namespace Breeze.TumbleBit.Client
                     session.SolverClientSession = new SolverClientSession(this.TumblerParameters.CreateSolverParamaters(), existingSession.SolverClientState);
 
                 // update the session
-                    session.Update();
+                this.MoveToNextPhase(session);
 
                 // replace the updated session in the list of existing sessions
                 int index = this.Sessions.IndexOf(existingSession);
@@ -107,12 +135,11 @@ namespace Breeze.TumbleBit.Client
 
                 this.Save();              
             }
-                catch (Exception)
-                {
-                    throw;
         }
 
-            }
+        public void MoveToNextPhase(Session session)
+        {
+            this.logger.LogInformation($"Entering next phase for cycle {session.StartCycle}.");
         }
 
         public void CreateNewSession(int start)
@@ -121,22 +148,6 @@ namespace Breeze.TumbleBit.Client
             this.Save();
         }
 
-        /// <summary>
-        /// Loads the saved state of the tumbling execution to the file system.
-        /// </summary>
-        /// <returns></returns>
-        public static TumblingState LoadState()
-        {
-            var stateFilePath = GetStateFilePath();
-            if (!File.Exists(stateFilePath))
-            {
-                return null;
-            }
-
-            // load the file from the local system
-            return JsonConvert.DeserializeObject<TumblingState>(File.ReadAllText(stateFilePath));
-        }
-
         /// <summary>
         /// Gets the file path of the file containing the state of the tumbling execution.
         /// </summary>
@@ -179,10 +190,5 @@ namespace Breeze.TumbleBit.Client
 
         [JsonIgnore]
         public PromiseClientSession PromiseClientSession { get; set; }
-
-        public void Update()
-        {
-
-        }
     }
 }