From 4b4d95f960fc2d9009e740bfa4add79dd70cef77 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kl=C3=A1ben=20Szabolcs=20Bence=20=28Tudi20=29?=
 <13798839+Tudi20@users.noreply.github.com>
Date: Sun, 13 Jan 2019 05:07:25 +0100
Subject: [PATCH] Fixes for the bugs
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Kláben Szabolcs Bence (Tudi20) <13798839+Tudi20@users.noreply.github.com>
---
 Universal THCRAP Launcher/Form1.cs | 57 +++++++++++++++++-------------
 1 file changed, 33 insertions(+), 24 deletions(-)

diff --git a/Universal THCRAP Launcher/Form1.cs b/Universal THCRAP Launcher/Form1.cs
index 9a4a6e7..96493dc 100644
--- a/Universal THCRAP Launcher/Form1.cs	
+++ b/Universal THCRAP Launcher/Form1.cs	
@@ -128,17 +128,8 @@ private void Form1_Load(object sender, EventArgs e)
             checkBox1.Checked = Configuration1.ExitAfterStartup;
 
             //Update Display favourites
-            foreach (var variable in Favourites1.Patches)
-            {
-                var index = listBox1.FindStringExact(variable);
-                listBox1.Items[index] += " ★";
-            }
-
-            foreach (var variable in Favourites1.Games)
-            {
-                var index = listBox2.FindStringExact(variable);
-                listBox2.Items[index] += " ★";
-            }
+            AddStars(listBox1, Favourites1.Patches);
+            AddStars(listBox2, Favourites1.Games);
 
             #endregion
 
@@ -165,8 +156,15 @@ private void Form1_Load(object sender, EventArgs e)
                 if (Configuration1.IsDescending[i] == "false")
                 {
                     if (i == 0)
+                    {
                         SortListBoxItems(ref listBox1);
-                    else SortListBoxItems(ref listBox2);
+                        sort_az_button1.BackgroundImage = _sortDescending;
+                    }
+                    else
+                    {
+                        SortListBoxItems(ref listBox2);
+                        sort_az_button2.BackgroundImage = _sortDescending;
+                    }
                 }
                 else if (i == 0)
                 {
@@ -182,6 +180,7 @@ private void Form1_Load(object sender, EventArgs e)
             //Default favourite button state
             for (var i = 0; i < 2; i++)
                 if (Configuration1.OnlyFavourites[i] == "true")
+                {
                     if (i == 0)
                     {
                         star_button1.BackgroundImage = _starHollow;
@@ -202,6 +201,12 @@ private void Form1_Load(object sender, EventArgs e)
                                 listBox2.Items.RemoveAt(n);
                         }
                     }
+                }
+                else
+                {
+                    if (i == 0) star_button1.BackgroundImage = _star;
+                    else star_button2.BackgroundImage = _star;
+                }
 
             //Default exe type button state
             filterByType_button.BackgroundImage = _gameAndCustom;
@@ -329,7 +334,7 @@ private void StartThcrap()
         /// <summary>
         ///     Handles starting thcrap with enter and favouring when pressing f
         /// </summary>
-        private new void KeyPress(object sender, KeyPressEventArgs e)
+        private void KeyPress(object sender, KeyPressEventArgs e)
         {
             switch (e.KeyChar)
             {
@@ -435,6 +440,15 @@ private void sort_az_button2_Click(object sender, EventArgs e)
             ReadConfig();
         }
 
+        private static void AddStars(ListBox listBox, IEnumerable<string> list)
+        {
+            foreach (var variable in list)
+            {
+                var index = listBox.FindStringExact(variable);
+                if (index != -1) listBox.Items[index] += " ★";
+            }
+        }
+
         private void star_button1_Click(object sender, EventArgs e)
         {
             if (!star_button1.BackgroundImage.Equals(_starHollow))
@@ -442,7 +456,7 @@ private void star_button1_Click(object sender, EventArgs e)
                 star_button1.BackgroundImage = _starHollow;
                 for (var n = listBox1.Items.Count - 1; n >= 0; --n)
                 {
-                    var filterItem = "★";
+                    const char filterItem = '★';
                     if (!listBox1.Items[n].ToString().Contains(filterItem))
                         listBox1.Items.RemoveAt(n);
                 }
@@ -455,11 +469,7 @@ private void star_button1_Click(object sender, EventArgs e)
                 listBox1.Items.Clear();
                 foreach (var s in _jsFiles) listBox1.Items.Add(s);
 
-                foreach (var variable in Favourites1.Patches)
-                {
-                    var index = listBox1.FindStringExact(variable);
-                    listBox1.Items[index] += " ★";
-                }
+                AddStars(listBox1, Favourites1.Patches);
 
                 Configuration1.OnlyFavourites[0] = "false";
                 ReadConfig();
@@ -486,11 +496,7 @@ private void star_button2_Click(object sender, EventArgs e)
                 listBox2.Items.Clear();
                 foreach (var s in _gamesList) listBox2.Items.Add(s);
 
-                foreach (var variable in Favourites1.Games)
-                {
-                    var index = listBox2.FindStringExact(variable);
-                    listBox2.Items[index] += " ★";
-                }
+                AddStars(listBox2, Favourites1.Games);
 
                 Configuration1.OnlyFavourites[1] = "false";
                 ReadConfig();
@@ -538,6 +544,7 @@ private void filterByType_button_Click(object sender, EventArgs e)
                 foreach (var item in _gamesList)
                     if (!item.Contains("_custom"))
                         listBox2.Items.Add(item);
+                AddStars(listBox2, Favourites1.Games);
                 return;
             }
 
@@ -548,6 +555,7 @@ private void filterByType_button_Click(object sender, EventArgs e)
                 foreach (var item in _gamesList)
                     if (item.Contains("_custom"))
                         listBox2.Items.Add(item);
+                AddStars(listBox2, Favourites1.Games);
                 return;
             }
 
@@ -556,6 +564,7 @@ private void filterByType_button_Click(object sender, EventArgs e)
                 filterByType_button.BackgroundImage = _gameAndCustom;
                 listBox2.Items.Clear();
                 foreach (var item in _gamesList) listBox2.Items.Add(item);
+                AddStars(listBox2, Favourites1.Games);
             }
         }
     }