From 2058cb7ecb4660d0b308f9973735106c8a439e8b Mon Sep 17 00:00:00 2001
From: Carson McManus <carson.mcmanus1@gmail.com>
Date: Tue, 27 Dec 2016 21:58:12 -0500
Subject: [PATCH] fixed issue #11

---
 Program.cs | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/Program.cs b/Program.cs
index d78ab75..8e4919c 100644
--- a/Program.cs
+++ b/Program.cs
@@ -478,6 +478,7 @@ namespace SteamGuard
 			if (account.RefreshSession())
 			{
 				if (Verbose) Console.WriteLine("Session refreshed");
+				Manifest.SaveAccount(account, Manifest.Encrypted);
 			}
 			else
 			{
@@ -493,6 +494,7 @@ namespace SteamGuard
 				LoginResult loginResult = login.DoLogin();
 				if (loginResult == LoginResult.Need2FA && !string.IsNullOrEmpty(account.SharedSecret))
 				{
+					// if we need a 2fa code, and we can generate it, generate a 2fa code and log in.
 					if (Verbose) Console.WriteLine(loginResult);
 					TimeAligner.AlignTime();
 					login.TwoFactorCode = account.GenerateSteamGuardCode();
@@ -500,14 +502,19 @@ namespace SteamGuard
 					loginResult = login.DoLogin();
 				}
 				Console.WriteLine(loginResult);
+				if (loginResult == LoginResult.LoginOkay)
+				{
+					account.Session = login.Session;
+				}
 
 				if (account.RefreshSession())
 				{
 					if (Verbose) Console.WriteLine("Session refreshed");
+					Manifest.SaveAccount(account, Manifest.Encrypted);
 				}
 				else
 				{
-					Console.WriteLine("Failed to refresh session");
+					Console.WriteLine("Failed to refresh session, aborting...");
 					return;
 				}
 			}