From bace4fd0a8b4a82e4a27aae8cfbeb2f2a892a073 Mon Sep 17 00:00:00 2001
From: tobtoht <tob@featherwallet.org>
Date: Fri, 13 Jan 2023 14:16:36 +0100
Subject: [PATCH] wizard: added option to override embedded wallet creation
 date for 14/16-word seeds

---
 src/wizard/PageSetRestoreHeight.cpp  |  6 ++++++
 src/wizard/PageWalletRestoreSeed.cpp | 12 +++++++++---
 src/wizard/PageWalletRestoreSeed.ui  |  7 +++++++
 src/wizard/WalletWizard.cpp          |  2 +-
 src/wizard/WalletWizard.h            |  1 +
 5 files changed, 24 insertions(+), 4 deletions(-)

diff --git a/src/wizard/PageSetRestoreHeight.cpp b/src/wizard/PageSetRestoreHeight.cpp
index db399d28..5ec276f7 100644
--- a/src/wizard/PageSetRestoreHeight.cpp
+++ b/src/wizard/PageSetRestoreHeight.cpp
@@ -45,6 +45,12 @@ void PageSetRestoreHeight::initializePage() {
     ui->line_restoreHeight->setText("");
     ui->frame_scanWarning->hide();
     ui->frame_walletAgeWarning->hide();
+
+    if (m_fields->seedCreationDateOverridden) {
+        auto creationDate = QDateTime::fromSecsSinceEpoch(m_fields->seed.time);
+        ui->line_creationDate->setText(creationDate.toString("yyyy-MM-dd"));
+        this->onCreationDateEdited();
+    }
 }
 
 void PageSetRestoreHeight::onCreationDateEdited() {
diff --git a/src/wizard/PageWalletRestoreSeed.cpp b/src/wizard/PageWalletRestoreSeed.cpp
index 5ff08bf3..9c123d3a 100644
--- a/src/wizard/PageWalletRestoreSeed.cpp
+++ b/src/wizard/PageWalletRestoreSeed.cpp
@@ -57,8 +57,6 @@ PageWalletRestoreSeed::PageWalletRestoreSeed(WizardFields *fields, QWidget *pare
 
     connect(ui->seedBtnGroup, QOverload<QAbstractButton *>::of(&QButtonGroup::buttonClicked), this, &PageWalletRestoreSeed::onSeedTypeToggled);
     connect(ui->combo_seedLanguage, &QComboBox::currentTextChanged, this, &PageWalletRestoreSeed::onSeedLanguageChanged);
-
-    this->onSeedTypeToggled();
 }
 
 
@@ -68,18 +66,21 @@ void PageWalletRestoreSeed::onSeedTypeToggled() {
         m_fields->seedType = Seed::Type::POLYSEED;
         ui->seedEdit->setPlaceholderText("Enter 16 word seed..");
         ui->group_seedLanguage->hide();
+        ui->check_overrideCreationDate->setVisible(true);
     }
     if (ui->radio14->isChecked()) {
         m_mode = &m_tevador;
         m_fields->seedType = Seed::Type::TEVADOR;
         ui->seedEdit->setPlaceholderText("Enter 14 word seed..");
         ui->group_seedLanguage->hide();
+        ui->check_overrideCreationDate->setVisible(true);
     }
     else if (ui->radio25->isChecked()) {
         m_mode = &m_legacy;
         m_fields->seedType = Seed::Type::MONERO;
         ui->seedEdit->setPlaceholderText("Enter 25 word seed..");
         ui->group_seedLanguage->show();
+        ui->check_overrideCreationDate->setVisible(false);
     }
 
     ui->label_errorString->hide();
@@ -94,7 +95,7 @@ void PageWalletRestoreSeed::onSeedLanguageChanged(const QString &language) {
 }
 
 int PageWalletRestoreSeed::nextId() const {
-    if (m_mode == &m_legacy) {
+    if (m_mode == &m_legacy || ui->check_overrideCreationDate->isChecked()) {
         return WalletWizard::Page_SetRestoreHeight;
     }
 
@@ -107,6 +108,10 @@ void PageWalletRestoreSeed::initializePage() {
     ui->seedEdit->setStyleSheet("");
     ui->label_errorString->hide();
     ui->line_seedOffset->setText("");
+    ui->check_overrideCreationDate->setVisible(false);
+    ui->check_overrideCreationDate->setChecked(false);
+    ui->radio16->isChecked();
+    this->onSeedTypeToggled();
 }
 
 bool PageWalletRestoreSeed::validatePage() {
@@ -157,6 +162,7 @@ bool PageWalletRestoreSeed::validatePage() {
 
     m_fields->seed = _seed;
     m_fields->seedOffsetPassphrase = ui->line_seedOffset->text();
+    m_fields->seedCreationDateOverridden = ui->check_overrideCreationDate->isChecked();
 
     return true;
 }
diff --git a/src/wizard/PageWalletRestoreSeed.ui b/src/wizard/PageWalletRestoreSeed.ui
index 2c9b226f..3a82f7fc 100644
--- a/src/wizard/PageWalletRestoreSeed.ui
+++ b/src/wizard/PageWalletRestoreSeed.ui
@@ -111,6 +111,13 @@
      </property>
     </widget>
    </item>
+   <item>
+    <widget class="QCheckBox" name="check_overrideCreationDate">
+     <property name="text">
+      <string>Override embedded wallet creation date</string>
+     </property>
+    </widget>
+   </item>
    <item>
     <widget class="QLabel" name="label_errorString">
      <property name="text">
diff --git a/src/wizard/WalletWizard.cpp b/src/wizard/WalletWizard.cpp
index 1c9dd0ff..250437c9 100644
--- a/src/wizard/WalletWizard.cpp
+++ b/src/wizard/WalletWizard.cpp
@@ -132,7 +132,7 @@ void WalletWizard::onCreateWallet() {
         m_wizardFields.seed.restoreHeight = currentBlockHeight;
     }
 
-    if (m_wizardFields.mode == WizardMode::RestoreFromSeed && m_wizardFields.seedType == Seed::Type::MONERO) {
+    if (m_wizardFields.mode == WizardMode::RestoreFromSeed && (m_wizardFields.seedType == Seed::Type::MONERO || m_wizardFields.seedCreationDateOverridden)) {
         m_wizardFields.seed.setRestoreHeight(m_wizardFields.restoreHeight);
     }
 
diff --git a/src/wizard/WalletWizard.h b/src/wizard/WalletWizard.h
index e7300530..5a88f08a 100644
--- a/src/wizard/WalletWizard.h
+++ b/src/wizard/WalletWizard.h
@@ -32,6 +32,7 @@ struct WizardFields {
     QString walletDir;
     Seed seed;
     bool seedOffsetPassphraseEnabled = false;
+    bool seedCreationDateOverridden = false;
     QString seedOffsetPassphrase;
     QString seedLanguage = constants::seedLanguage;
     QString password;
-- 
2.52.0