From 4698326717368ee977dbf83fae98cd7a8ad52eee Mon Sep 17 00:00:00 2001
From: sithlord48 <sithlord48@gmail.com>
Date: Sat, 22 Nov 2025 12:15:23 -0500
Subject: [PATCH] refactor: MainWindow keep a local copy of the fingerprint
 This also makes it so MainWindow does not use the local_fingerprint file

---
 src/lib/gui/MainWindow.cpp | 51 ++++++--------------------------------
 src/lib/gui/MainWindow.h   |  2 +-
 2 files changed, 8 insertions(+), 45 deletions(-)

diff --git a/src/lib/gui/MainWindow.cpp b/src/lib/gui/MainWindow.cpp
index 761000a49..6a7d55e85 100644
--- a/src/lib/gui/MainWindow.cpp
+++ b/src/lib/gui/MainWindow.cpp
@@ -140,7 +140,7 @@ MainWindow::MainWindow()
 
   qDebug().noquote() << "active settings path:" << Settings::settingsPath();
 
-  // Force generation of SHA256 for the localhost
+  // Force generation of a certificate on the host
   if (TlsUtility::isEnabled()) {
     if (Settings::value(Settings::Security::KeySize).toInt() < 2048) {
       QMessageBox::information(
@@ -151,22 +151,11 @@ MainWindow::MainWindow()
     }
     if (!TlsUtility::isCertValid()) {
       generateCertificate();
-      return;
-    }
-
-    if (!QFile::exists(Settings::tlsLocalDb())) {
-      generateCertificate();
-      return;
-    }
-
-    FingerprintDatabase db;
-    db.read(Settings::tlsLocalDb());
-    if (db.fingerprints().isEmpty()) {
-      generateCertificate();
+    } else {
+      m_fingerprint = {Fingerprint::Type::SHA256, TlsUtility::certFingerprint()};
     }
   }
 }
-
 MainWindow::~MainWindow()
 {
   m_guiDupeChecker->close();
@@ -501,7 +490,7 @@ void MainWindow::resetCore()
 
 void MainWindow::showMyFingerprint()
 {
-  FingerprintDialog fingerprintDialog(this, localFingerprint());
+  FingerprintDialog fingerprintDialog(this, m_fingerprint);
   fingerprintDialog.exec();
 }
 
@@ -838,7 +827,7 @@ void MainWindow::checkFingerprint(const QString &line)
   }
 
   auto mode = isClient ? FingerprintDialogMode::Client : FingerprintDialogMode::Server;
-  FingerprintDialog fingerprintDialog(this, localFingerprint(), mode, sha256);
+  FingerprintDialog fingerprintDialog(this, m_fingerprint, mode, sha256);
 
   if (fingerprintDialog.exec() == QDialog::Accepted) {
     db.addTrusted(sha256);
@@ -1016,7 +1005,7 @@ void MainWindow::coreConnectionStateChanged(CoreConnectionState state)
 
 void MainWindow::updateLocalFingerprint()
 {
-  m_btnFingerprint->setVisible(TlsUtility::isEnabled() && QFile::exists(Settings::tlsLocalDb()));
+  m_btnFingerprint->setVisible(TlsUtility::isEnabled() && !m_fingerprint.data.isEmpty());
 }
 
 void MainWindow::hide()
@@ -1188,38 +1177,12 @@ bool MainWindow::generateCertificate()
     return false;
   }
 
-  if (TlsCertificate tls; !tls.generateFingerprint(certificate)) {
-    return false;
-  }
+  m_fingerprint = {Fingerprint::Type::SHA256, TlsUtility::certFingerprint()};
 
   updateLocalFingerprint();
   return true;
 }
 
-Fingerprint MainWindow::localFingerprint()
-{
-  if (!QFile::exists(Settings::tlsLocalDb())) {
-    if (generateCertificate())
-      return localFingerprint();
-  }
-
-  FingerprintDatabase db;
-  db.read(Settings::tlsLocalDb());
-  if (db.fingerprints().isEmpty()) {
-    if (generateCertificate())
-      return localFingerprint();
-  }
-
-  Fingerprint sha256Print;
-  for (const auto &f : std::as_const(db.fingerprints())) {
-    if (f.type == Fingerprint::Type::SHA256) {
-      sha256Print = f;
-      break;
-    }
-  }
-  return sha256Print;
-}
-
 void MainWindow::serverClientsChanged(const QStringList &clients)
 {
   if (m_coreProcess.mode() != CoreMode::Server || !m_coreProcess.isStarted())
diff --git a/src/lib/gui/MainWindow.h b/src/lib/gui/MainWindow.h
index 415f80bf3..255196d05 100644
--- a/src/lib/gui/MainWindow.h
+++ b/src/lib/gui/MainWindow.h
@@ -165,7 +165,7 @@ private:
    */
   bool generateCertificate();
 
-  Fingerprint localFingerprint();
+  Fingerprint m_fingerprint;
 
   void serverClientsChanged(const QStringList &clients);