1 files changed, 100 insertions, 0 deletions
diff --git a/src/gui/TotpDialog.cpp b/src/gui/TotpDialog.cpp
new file mode 100644
index 000000000..2eb1c9e2c
--- /dev/null
+++ b/src/gui/TotpDialog.cpp
@@ -0,0 +1,100 @@
+/*
+ *  Copyright (C) 2017 Weslly Honorato <weslly@protonmail.com>
+ *
+ *  This program is free software: you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation, either version 2 or (at your option)
+ *  version 3 of the License.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "TotpDialog.h"
+#include "ui_TotpDialog.h"
+
+#include "core/Config.h"
+#include "core/Entry.h"
+#include "gui/DatabaseWidget.h"
+#include "gui/Clipboard.h"
+
+#include <QTimer>
+#include <QDateTime>
+#include <QPushButton>
+
+
+TotpDialog::TotpDialog(DatabaseWidget* parent, Entry* entry)
+    : QDialog(parent)
+    , m_ui(new Ui::TotpDialog())
+{
+    m_entry = entry;
+    m_parent = parent;
+    m_step = m_entry->totpStep();
+
+    m_ui->setupUi(this);
+
+    uCounter = resetCounter();
+    updateProgressBar();
+
+    QTimer *timer = new QTimer(this);
+    connect(timer, SIGNAL(timeout()), this, SLOT(updateProgressBar()));
+    connect(timer, SIGNAL(timeout()), this, SLOT(updateSeconds()));
+    timer->start(m_step * 10);
+
+    updateTotp();
+
+    setAttribute(Qt::WA_DeleteOnClose);
+
+    m_ui->buttonBox->button(QDialogButtonBox::Ok)->setText(tr("Copy"));
+
+    connect(m_ui->buttonBox, SIGNAL(rejected()), SLOT(close()));
+    connect(m_ui->buttonBox, SIGNAL(accepted()), SLOT(copyToClipboard()));
+}
+
+void TotpDialog::copyToClipboard()
+{
+    clipboard()->setText(m_entry->totp());
+    if (config()->get("MinimizeOnCopy").toBool()) {
+        m_parent->window()->showMinimized();
+    }
+}
+
+void TotpDialog::updateProgressBar()
+{
+    if (uCounter < 100) {
+        m_ui->progressBar->setValue(100 - uCounter);
+        m_ui->progressBar->update();
+        uCounter++;
+    } else {
+        updateTotp();
+        uCounter = resetCounter();
+    }
+}
+
+
+void TotpDialog::updateSeconds()
+{
+    uint epoch = QDateTime::currentDateTime().toTime_t() - 1;
+    m_ui->timerLabel->setText(tr("Expires in") + " <b>" + QString::number(m_step - (epoch % m_step)) + "</b> " + tr("seconds"));
+}
+
+void TotpDialog::updateTotp()
+{
+    m_ui->totpLabel->setText(m_entry->totp());
+}
+
+double TotpDialog::resetCounter()
+{
+    uint epoch = QDateTime::currentDateTime().toTime_t();
+    double counter = qRound(static_cast<double>(epoch % m_step) / m_step * 100);
+    return counter;
+}
+
+TotpDialog::~TotpDialog()
+{
+}