From 33ddfcae792b8cb3a52db8c2fd680cb12e6a5451 Mon Sep 17 00:00:00 2001
From: kidfearless <kidfearless1@gmail.com>
Date: Mon, 31 Aug 2020 09:28:00 -0600
Subject: [PATCH] Implement import and export

---
 MainWindow.xaml    |   8 ++--
 MainWindow.xaml.cs | 114 +++++++++++++++++++++++++++++++++------------
 2 files changed, 87 insertions(+), 35 deletions(-)

diff --git a/MainWindow.xaml b/MainWindow.xaml
index c38d697..b0cc987 100644
--- a/MainWindow.xaml
+++ b/MainWindow.xaml
@@ -14,11 +14,11 @@
 		<Button x:Name="BtnAddCheater" Content="Add Cheater" HorizontalAlignment="Left" Margin="113,58,0,0" VerticalAlignment="Top" Height="66" Width="120" RenderTransformOrigin="1.635,0.554" Click="AddCheaterButtonClicked"/>
 		<Button x:Name="BtnImport" Content="Import" HorizontalAlignment="Left" Margin="398,58,0,0" VerticalAlignment="Top" RenderTransformOrigin="0.52,1.485" Height="66" Width="120" Click="ImportClicked"/>
 		<Button x:Name="BtnExport" Content="Export" HorizontalAlignment="Left" Margin="537,58,0,0" VerticalAlignment="Top" Height="66" Width="120" Click="ExportClicked"/>
-		<TextBox x:Name="TxtSteamID" HorizontalAlignment="Left" Margin="113,10,0,0" Text="STEAM:0:1:12354847" TextWrapping="Wrap" VerticalAlignment="Top" Width="120" GotFocus="SteamID_GotFocus" Height="24" />
+		<TextBox x:Name="TxtSteamID" HorizontalAlignment="Left" Margin="113,10,0,0" Text="STEAM:0:1:12354847" TextWrapping="Wrap" VerticalAlignment="Top" Width="120" GotFocus="SteamID_GotFocus" LostFocus="SteamID_LostFocus"  Height="24" />
 		<Button x:Name="BtnAutoStatus" Content="Status Key Bind" HorizontalAlignment="Left" Margin="256,58,0,0" VerticalAlignment="Top" Height="66" Width="120" RenderTransformOrigin="1.635,0.554" Click="StatusButtonClicked"/>
-		<TextBox x:Name="TxtCheats" HorizontalAlignment="Left" Margin="256,10,0,0" Text="Spinbot, Aimbot, Wallhacks" TextWrapping="Wrap" VerticalAlignment="Top" Width="120" Height="24" GotFocus="Cheats_GotFocus"/>
-		<TextBox x:Name="TxtThreatLevel" HorizontalAlignment="Left" Margin="398,10,0,0" Text="1-10" TextWrapping="Wrap" VerticalAlignment="Top" Width="120" Height="24" GotFocus="ThreatLevel_GotFocus"/>
-		<TextBox x:Name="TxtName" HorizontalAlignment="Left" Margin="537,10,0,0" Text="The Suspect" TextWrapping="Wrap" VerticalAlignment="Top" Width="120" Height="24" GotFocus="Name_GotFocus"/>
+		<TextBox x:Name="TxtCheats" HorizontalAlignment="Left" Margin="256,10,0,0" Text="Spinbot, Aimbot, Wallhacks" TextWrapping="Wrap" VerticalAlignment="Top" Width="120" Height="24" GotFocus="Cheats_GotFocus" LostFocus="Cheats_LostFocus"/>
+		<TextBox x:Name="TxtThreatLevel" HorizontalAlignment="Left" Margin="398,10,0,0" Text="1-10" TextWrapping="Wrap" VerticalAlignment="Top" Width="120" Height="24" GotFocus="ThreatLevel_GotFocus" LostFocus="ThreatLevel_LostFocus"/>
+		<TextBox x:Name="TxtName" HorizontalAlignment="Left" Margin="537,10,0,0" Text="The Suspect" TextWrapping="Wrap" VerticalAlignment="Top" Width="120" Height="24" GotFocus="Name_GotFocus" LostFocus="Name_LostFocus"/>
 		<ScrollViewer Background="Black" x:Name="ScrollPlayers" 
 				  HorizontalScrollBarVisibility="Auto" Margin="32,148,32,48">
 			<StackPanel x:Name="StackPanel" Background="#FF232323" Margin="0,0,0,0">
diff --git a/MainWindow.xaml.cs b/MainWindow.xaml.cs
index b0196de..bea68bd 100644
--- a/MainWindow.xaml.cs
+++ b/MainWindow.xaml.cs
@@ -87,7 +87,7 @@ public MainWindow()
 			CommunityURLRegex = new Regex(@"(\d+)");
 			CommunityProfilePictureRegex = new Regex(string.Format(@"<link rel={0}image_src{0} href={0}(.*){0}>", '"'));
 
-			if(!Directory.Exists(Path.GetDirectoryName(Constants.DatabasePath)))
+			if (!Directory.Exists(Path.GetDirectoryName(Constants.DatabasePath)))
 			{
 				Directory.CreateDirectory(Path.GetDirectoryName(Constants.DatabasePath));
 			}
@@ -120,8 +120,8 @@ public async void ShowToastAsync(Cheater cheater)
 			var notificationManager = new NotificationManager();
 			await notificationManager.ShowAsync(new NotificationContent()
 			{
-				Title =		$"Hacker {cheater.LastKnownName} Found In Game",
-				Message =	$"Threat Level: {cheater.ThreatLevel}\n" +
+				Title = $"Hacker {cheater.LastKnownName} Found In Game",
+				Message = $"Threat Level: {cheater.ThreatLevel}\n" +
 							$"Known Cheats: {cheater.CheatList}",
 				Type = NotificationType.Error
 			});
@@ -145,7 +145,7 @@ private void Timer_CheckStatus(object sender, System.Timers.ElapsedEventArgs e)
 					Keyboard.SendKey(KeyConvert.ToDirectXKeyCode(StatusKey), true, Keyboard.InputType.Keyboard);
 				}
 			}
-			catch (Exception _)	{}
+			catch (Exception _) { }
 		}
 
 		/// <summary>
@@ -157,7 +157,7 @@ private void ReadConsole(object nill)
 			// Open the file in a way that won't bother csgo.
 			using var stream = File.Open(Path.Combine(path, "console.log"), FileMode.Open, FileAccess.ReadWrite, FileShare.ReadWrite);
 			using var reader = new StreamReader(stream, true);
-			if(reader.BaseStream.Length < 0)
+			if (reader.BaseStream.Length < 0)
 			{
 				return;
 			}
@@ -171,7 +171,7 @@ private void ReadConsole(object nill)
 			{
 				var mapMatch = MapNameRegex.Match(line);
 				// The first group match will be the current map on the server
-				if(mapMatch.Success && mapMatch.Groups[1].Value != CurrentMap)
+				if (mapMatch.Success && mapMatch.Groups[1].Value != CurrentMap)
 				{
 					AlertedPlayers.Clear();
 					CurrentMap = mapMatch.Groups[1].Value;
@@ -216,7 +216,7 @@ private void ReadConsole(object nill)
 			stream.SetLength(0);
 			stream.Close();
 
-			if(tempUsers.Count != Users.Count && tempUsers.Count != 0)
+			if (tempUsers.Count != Users.Count && tempUsers.Count != 0)
 			{
 				Users = tempUsers;
 				// can only update controls from the main thread, which we are not in. So we invoke an inline function to do it for us.
@@ -238,12 +238,30 @@ private void ExportClicked(object sender, RoutedEventArgs e)
 			var dialog = new SaveFileDialog();
 			dialog.FileName = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.MyDocuments), "exported_cheaters.sq3");
 			dialog.DefaultExt = ".sq3";
-			if((bool)dialog.ShowDialog())
+			dialog.Filter = "sqlite files (*.sq3)|*.sq3|All Files (*.*)|*.*";
+
+			if ((bool)dialog.ShowDialog())
 			{
-				File.Copy(Constants.DatabasePath, dialog.FileName);
+				// Try to save the database file
+				try
+				{
+					if(File.Exists(dialog.FileName))
+					{
+						File.Delete(dialog.FileName);
+					}
+					File.Copy(Constants.DatabasePath, dialog.FileName);
+				}
+				// If we fail try to save it under a unique name
+				catch
+				{
+					try
+					{
+						File.Copy(Constants.DatabasePath, dialog.FileName + "." + Guid.NewGuid());
+					}
+					catch { }
+				}
 			}
 
-			//dialog.Filter = "sqlite files (*.sq3)|*.sq3|All Files (*.*)|*.*";
 		}
 
 		private void ImportClicked(object sender, RoutedEventArgs e)
@@ -251,7 +269,7 @@ private void ImportClicked(object sender, RoutedEventArgs e)
 			var dialog = new OpenFileDialog();
 			dialog.Filter = "sqlite files (*.sq3)|*.sq3|All Files (*.*)|*.*";
 			dialog.Multiselect = false;
-			if((bool)dialog.ShowDialog())
+			if ((bool)dialog.ShowDialog())
 			{
 				using var otherDB = new SQLiteConnection(dialog.FileName, Constants.Flags);
 				var cheaterList = otherDB.Table<Cheater>().ToList();
@@ -283,25 +301,33 @@ private void StatusButtonClicked(object sender, RoutedEventArgs e)
 		/// <param name="e"></param>
 		private void AddCheaterButtonClicked(object sender, RoutedEventArgs e)
 		{
-			SteamID steamAccount;
-			if (TxtSteamID.Text.Contains("community"))
+			try
 			{
-				steamAccount = new SteamID(CommunityURLRegex.Match(TxtSteamID.Text).Groups[1].Value);
+				SteamID steamAccount;
+				if (TxtSteamID.Text.Contains("community"))
+				{
+					var steam64 = CommunityURLRegex.Match(TxtSteamID.Text).Groups[1].Value;
+					steamAccount = new SteamID(Convert.ToUInt64(steam64));
+				}
+				else
+				{
+					steamAccount = new SteamID(TxtSteamID.Text);
+				}
+
+				using var db = new DatabaseConnection();
+				int.TryParse(TxtThreatLevel.Text.Trim(), out int threat);
+				db.InsertOrReplace(new Cheater()
+				{
+					AccountID = steamAccount.AccountID,
+					ThreatLevel = threat,
+					CheatList = TxtCheats.Text,
+					LastKnownName = TxtName.Text
+				});
 			}
-			else
+			catch
 			{
-				steamAccount = new SteamID(TxtSteamID.Text);
-			}
 
-			using var db = new DatabaseConnection();
-			int.TryParse(TxtThreatLevel.Text.Trim(), out int threat);
-			db.Insert(new Cheater()
-			{
-				AccountID = steamAccount.AccountID,
-				ThreatLevel = threat,
-				CheatList = TxtCheats.Text,
-				LastKnownName = TxtName.Text
-			});
+			}
 
 			// reset the default placeholders
 			TxtCheats.Text = "Spinbot, Aimbot, Wallhacks";
@@ -344,6 +370,13 @@ private void SteamID_GotFocus(object sender, RoutedEventArgs e)
 				TxtSteamID.Text = "";
 			}
 		}
+		private void SteamID_LostFocus(object sender, RoutedEventArgs e)
+		{
+			if (TxtSteamID.Text == "")
+			{
+				TxtSteamID.Text = "STEAM:0:1:12354847";
+			}
+		}
 
 		private void ThreatLevel_GotFocus(object sender, RoutedEventArgs e)
 		{
@@ -352,6 +385,13 @@ private void ThreatLevel_GotFocus(object sender, RoutedEventArgs e)
 				TxtThreatLevel.Text = "";
 			}
 		}
+		private void ThreatLevel_LostFocus(object sender, RoutedEventArgs e)
+		{
+			if (TxtThreatLevel.Text == "")
+			{
+				TxtThreatLevel.Text = "1-10";
+			}
+		}
 
 		private void Name_GotFocus(object sender, RoutedEventArgs e)
 		{
@@ -360,6 +400,13 @@ private void Name_GotFocus(object sender, RoutedEventArgs e)
 				TxtName.Text = "";
 			}
 		}
+		private void Name_LostFocus(object sender, RoutedEventArgs e)
+		{
+			if (TxtName.Text == "")
+			{
+				TxtName.Text = "The Suspect";
+			}
+		}
 
 		private void Cheats_GotFocus(object sender, RoutedEventArgs e)
 		{
@@ -368,7 +415,13 @@ private void Cheats_GotFocus(object sender, RoutedEventArgs e)
 				TxtCheats.Text = "";
 			}
 		}
-
+		private void Cheats_LostFocus(object sender, RoutedEventArgs e)
+		{
+			if (TxtCheats.Text == "")
+			{
+				TxtCheats.Text = "Spinbot, Aimbot, Wallhacks";
+			}
+		}
 		#endregion
 
 		/// <summary>
@@ -435,7 +488,7 @@ private Grid CreateUserRow(string name, SteamID steamid)
 			Label lName = new Label()
 			{
 				Content = name.Replace("_", "__"),
-				Margin = new Thickness(0.0, 48/4, 48/4, 5.0)
+				Margin = new Thickness(0.0, 48 / 4, 48 / 4, 5.0)
 			};
 			// single underscores have special meanings and have to be escaped
 			Label steamID = new Label()
@@ -540,10 +593,10 @@ private string GetProfilePictureURL(string url)
 				}
 
 				string line;
-				while((line = readStream.ReadLine()) != null)
+				while ((line = readStream.ReadLine()) != null)
 				{
 					var match = CommunityProfilePictureRegex.Match(line);
-					if(match.Success)
+					if (match.Success)
 					{
 						response.Close();
 						readStream.Close();
@@ -561,7 +614,6 @@ private string GetProfilePictureURL(string url)
 
 		private Image GetProfilePicture(SteamID steam)
 		{
-
 			var url = $"http://steamcommunity.com/profiles/{steam.ConvertToUInt64()}";
 			var profilePicURL = GetProfilePictureURL(url);
 			var image = CreateImage(profilePicURL);