First Run Security Question Setup

VoxDroid · Dec 20, 2023 · 41885d7 · 41885d7
1 parent 6233b5c
commit 41885d7
Show file tree

Hide file tree

Showing 5 changed files with 1,201 additions and 8 deletions.
diff --git a/Login.cs b/Login.cs
@@ -625,6 +625,7 @@ private void loginbutton_Click(object sender, EventArgs e)
             lblError.Text = string.Empty;
             string username1 = username.Text;
             string password1 = password.Text;
+
             if (IsDatabaseConnected())
             {
                 if (string.IsNullOrWhiteSpace(username1) || string.IsNullOrWhiteSpace(password1))
@@ -644,11 +645,33 @@ private void loginbutton_Click(object sender, EventArgs e)
                         lblError.Enabled = false;
                         lblError.Visible = false;
                         lblError.Text = string.Empty;
-                        Main mainForm = new Main();
-                        mainForm.Show();
-                        mainForm.LoggedInUser = currentlyLoggedInUser;
 
-                        this.Hide();
+                        bool securityQuestionsExist = CheckSecurityQuestionsExist();
+
+                        if (securityQuestionsExist)
+                        {
+                            Main mainForm = new Main();
+                            mainForm.Show();
+                            mainForm.LoggedInUser = currentlyLoggedInUser;
+
+                            this.Hide();
+                        }
+                        else
+                        {
+                            DialogResult result = MessageBox.Show("It seems like it's your first time logging in. Security questions are required before continuing. Would you like to set them up now?", "Security Questions Setup Required", MessageBoxButtons.YesNo);
+
+                            if (result == DialogResult.Yes)
+                            {
+                                SetupSQ setupForm = new SetupSQ();
+                                setupForm.ShowDialog();
+                            }
+                            else
+                            {
+                                lblError.Enabled = true;
+                                lblError.Visible = true;
+                                lblError.Text = "First Set Up is Required.";
+                            }
+                        }
                     }
                     else
                     {
@@ -667,6 +690,32 @@ private void loginbutton_Click(object sender, EventArgs e)
             }
         }
 
+        private bool CheckSecurityQuestionsExist()
+        {
+            string sqlSelect = "SELECT * FROM securityquestions WHERE sqs_id = 1";
+
+            try
+            {
+                using (MySqlConnection connection = new MySqlConnection(connectionString))
+                {
+                    connection.Open();
+
+                    using (MySqlCommand selectCommand = new MySqlCommand(sqlSelect, connection))
+                    {
+                        using (MySqlDataReader reader = selectCommand.ExecuteReader())
+                        {
+                            return reader.Read();
+                        }
+                    }
+                }
+            }
+            catch (Exception ex)
+            {
+                MessageBox.Show($"Error: {ex.Message}", "Error", MessageBoxButtons.OK, MessageBoxIcon.Error);
+                return false;
+            }
+        }
+
         private string AuthenticateUser(string username, string password)
         {
             try
@@ -736,6 +785,7 @@ private void LoginEnter()
             lblError.Text = string.Empty;
             string username1 = username.Text;
             string password1 = password.Text;
+
             if (IsDatabaseConnected())
             {
                 if (string.IsNullOrWhiteSpace(username1) || string.IsNullOrWhiteSpace(password1))
@@ -755,11 +805,33 @@ private void LoginEnter()
                         lblError.Enabled = false;
                         lblError.Visible = false;
                         lblError.Text = string.Empty;
-                        Main mainForm = new Main();
-                        mainForm.Show();
-                        mainForm.LoggedInUser = currentlyLoggedInUser;
 
-                        this.Hide();
+                        bool securityQuestionsExist = CheckSecurityQuestionsExist();
+
+                        if (securityQuestionsExist)
+                        {
+                            Main mainForm = new Main();
+                            mainForm.Show();
+                            mainForm.LoggedInUser = currentlyLoggedInUser;
+
+                            this.Hide();
+                        }
+                        else
+                        {
+                            DialogResult result = MessageBox.Show("It seems like it's your first time logging in. Security questions are required before continuing. Would you like to set them up now?", "Security Questions Setup Required", MessageBoxButtons.YesNo);
+
+                            if (result == DialogResult.Yes)
+                            {
+                                SetupSQ setupForm = new SetupSQ();
+                                setupForm.ShowDialog();
+                            }
+                            else
+                            {
+                                lblError.Enabled = true;
+                                lblError.Visible = true;
+                                lblError.Text = "First Set Up is Required.";
+                            }
+                        }
                     }
                     else
                     {

diff --git a/SPAAT.csproj b/SPAAT.csproj
@@ -559,6 +559,12 @@
     <Compile Include="Recovery.Designer.cs">
       <DependentUpon>Recovery.cs</DependentUpon>
     </Compile>
+    <Compile Include="SetupSQ.cs">
+      <SubType>Form</SubType>
+    </Compile>
+    <Compile Include="SetupSQ.Designer.cs">
+      <DependentUpon>SetupSQ.cs</DependentUpon>
+    </Compile>
     <Compile Include="SubPages\ModBudMan.cs">
       <SubType>UserControl</SubType>
     </Compile>
@@ -648,6 +654,9 @@
     <EmbeddedResource Include="Recovery.resx">
       <DependentUpon>Recovery.cs</DependentUpon>
     </EmbeddedResource>
+    <EmbeddedResource Include="SetupSQ.resx">
+      <DependentUpon>SetupSQ.cs</DependentUpon>
+    </EmbeddedResource>
     <EmbeddedResource Include="SubPages\ModBudMan.resx">
       <DependentUpon>ModBudMan.cs</DependentUpon>
     </EmbeddedResource>