File indexing completed on 2024-05-19 04:59:18

 /* ============================================================
 * Personal Information Manager plugin for Falkon
 * Copyright (C) 2012-2016 David Rosca <nowrep@gmail.com>
 * Copyright (C) 2012-2014 Mladen Pejaković <pejakm@autistici.org>
 *
 * 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 3 of the License, or
 * (at your option) any later version.
 *
 * 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 "PIM_handler.h"
 #include "PIM_settings.h"
 #include "webview.h"
 #include "webpage.h"
 #include "webhittestresult.h"
 
 #include <QApplication>
 #include <QSettings>
 #include <QLabel>
 #include <QToolTip>
 #include <QKeyEvent>
 
 PIM_Handler::PIM_Handler(const QString &sPath, QObject* parent)
     : QObject(parent)
     , m_settingsFile(sPath + QL1S("/extensions.ini"))
     , m_loaded(false)
 {
 }
 
 void PIM_Handler::loadSettings()
 {
     QSettings settings(m_settingsFile, QSettings::IniFormat);
 
     settings.beginGroup(QSL("PIM"));
     m_allInfo[PI_LastName] = settings.value(QSL("LastName"), QString()).toString();
     m_allInfo[PI_FirstName] = settings.value(QSL("FirstName"), QString()).toString();
     m_allInfo[PI_Email] = settings.value(QSL("Email"), QString()).toString();
     m_allInfo[PI_Mobile] = settings.value(QSL("Mobile"), QString()).toString();
     m_allInfo[PI_Phone] = settings.value(QSL("Phone"), QString()).toString();
     m_allInfo[PI_Address] = settings.value(QSL("Address"), QString()).toString();
     m_allInfo[PI_City] = settings.value(QSL("City"), QString()).toString();
     m_allInfo[PI_Zip] = settings.value(QSL("Zip"), QString()).toString();
     m_allInfo[PI_State] = settings.value(QSL("State"), QString()).toString();
     m_allInfo[PI_Country] = settings.value(QSL("Country"), QString()).toString();
     m_allInfo[PI_HomePage] = settings.value(QSL("HomePage"), QString()).toString();
     m_allInfo[PI_Special1] = settings.value(QSL("Special1"), QString()).toString();
     m_allInfo[PI_Special2] = settings.value(QSL("Special2"), QString()).toString();
     m_allInfo[PI_Special3] = settings.value(QSL("Special3"), QString()).toString();
     settings.endGroup();
 
     m_translations[PI_LastName] = tr("Last Name");
     m_translations[PI_FirstName] = tr("First Name");
     m_translations[PI_Email] = tr("E-mail");
     m_translations[PI_Mobile] = tr("Mobile");
     m_translations[PI_Phone] = tr("Phone");
     m_translations[PI_Address] = tr("Address");
     m_translations[PI_City] = tr("City");
     m_translations[PI_Zip] = tr("ZIP Code");
     m_translations[PI_State] = tr("State/Region");
     m_translations[PI_Country] = tr("Country");
     m_translations[PI_HomePage] = tr("Home Page");
     m_translations[PI_Special1] = tr("Custom 1");
     m_translations[PI_Special2] = tr("Custom 2");
     m_translations[PI_Special3] = tr("Custom 3");
 
     m_infoMatches[PI_LastName] << QSL("lastname") << QSL("surname");
     m_infoMatches[PI_FirstName] << QSL("firstname") << QSL("name");
     m_infoMatches[PI_Email] << QSL("email") << QSL("e-mail") << QSL("mail");
     m_infoMatches[PI_Mobile] << QSL("mobile") << QSL("mobilephone");
     m_infoMatches[PI_Phone] << QSL("phone") << QSL("telephone");
     m_infoMatches[PI_Address] << QSL("address");
     m_infoMatches[PI_City] << QSL("city");
     m_infoMatches[PI_Zip] << QSL("zip");
     m_infoMatches[PI_State] << QSL("state") << QSL("region");
     m_infoMatches[PI_Country] << QSL("country");
     m_infoMatches[PI_HomePage] << QSL("homepage") << QSL("www");
 
     m_loaded = true;
 }
 
 void PIM_Handler::showSettings(QWidget* parent)
 {
     if (!m_settings) {
         m_settings = new PIM_Settings(m_settingsFile, parent);
 
         connect(m_settings.data(), &QDialog::accepted, this, &PIM_Handler::loadSettings);
     }
 
     m_settings.data()->show();
     m_settings.data()->raise();
 }
 
 void PIM_Handler::populateWebViewMenu(QMenu* menu, WebView* view, const WebHitTestResult &hitTest)
 {
     m_view = view;
     m_clickedPos = hitTest.pos();
 
     if (!hitTest.isContentEditable()) {
         return;
     }
 
     if (!m_loaded) {
         loadSettings();
     }
 
     auto* pimMenu = new QMenu(tr("Insert Personal Information"));
     pimMenu->setIcon(QIcon(QStringLiteral(":/PIM/data/PIM.png")));
 
     if (!m_allInfo[PI_FirstName].isEmpty() && !m_allInfo[PI_LastName].isEmpty()) {
         const QString fullname = m_allInfo[PI_FirstName] + QLatin1Char(' ') + m_allInfo[PI_LastName];
         QAction* action = pimMenu->addAction(fullname, this, &PIM_Handler::pimInsert);
         action->setData(fullname);
     }
 
     for (int i = 0; i < PI_Max; ++i) {
         const QString info = m_allInfo[PI_Type(i)];
         if (info.isEmpty()) {
             continue;
         }
 
         QAction* action = pimMenu->addAction(info, this, &PIM_Handler::pimInsert);
         action->setData(info);
         action->setStatusTip(m_translations[PI_Type(i)]);
     }
 
     pimMenu->addSeparator();
     pimMenu->addAction(tr("Edit"), this, SLOT(showSettings()));
 
     menu->addMenu(pimMenu);
     menu->addSeparator();
 }
 
 bool PIM_Handler::keyPress(WebView* view, QKeyEvent* event)
 {
     if (!view) {
         return false;
     }
 
     bool isEnter = event->key() == Qt::Key_Return || event->key() == Qt::Key_Enter;
     bool isControlModifier = event->modifiers() & Qt::ControlModifier;
 
     if (!isEnter || !isControlModifier) {
         return false;
     }
 
     QString source = QL1S("var inputs = document.getElementsByTagName('input');"
                           "var table = %1;"
                           "for (var i = 0; i < inputs.length; ++i) {"
                           "    var input = inputs[i];"
                           "    if (input.type != 'text' || input.name == '')"
                           "        continue;"
                           "    for (var key in table) {"
                           "        if (!table.hasOwnProperty(key))"
                           "            continue;"
                           "        if (key == input.name || input.name.indexOf(key) != -1) {"
                           "            input.value = table[key];"
                           "            break;"
                           "        }"
                           "    }"
                           "}");
 
     view->page()->runJavaScript(source.arg(matchingJsTable()), WebPage::SafeJsWorld);
 
     return true;
 }
 
 void PIM_Handler::unloadPlugin()
 {
     delete m_settings.data();
 }
 
 void PIM_Handler::webPageCreated(WebPage* page)
 {
     connect(page, &QWebEnginePage::loadFinished, this, &PIM_Handler::pageLoadFinished);
 }
 
 void PIM_Handler::pimInsert()
 {
     if (!m_view || m_clickedPos.isNull())
         return;
 
     auto* action = qobject_cast<QAction*>(sender());
     if (!action)
         return;
 
     QString info = action->data().toString();
     info.replace(QLatin1Char('"'), QLatin1String("\\\""));
 
     QString source = QL1S("var e = document.elementFromPoint(%1, %2);"
                           "if (e) {"
                           "    var v = e.value.substring(0, e.selectionStart);"
                           "    v += \"%3\" + e.value.substring(e.selectionEnd);"
                           "    e.value = v;"
                           "}");
     source = source.arg(m_clickedPos.x()).arg(m_clickedPos.y()).arg(info);
     m_view->page()->runJavaScript(source, WebPage::SafeJsWorld);
 }
 
 void PIM_Handler::pageLoadFinished()
 {
     auto* page = qobject_cast<WebPage*>(sender());
     if (!page) {
         return;
     }
 
     if (!m_loaded) {
         loadSettings();
     }
 
     QString source = QL1S("var inputs = document.getElementsByTagName('input');"
                           "var table = %1;"
                           "for (var i = 0; i < inputs.length; ++i) {"
                           "    var input = inputs[i];"
                           "    if (input.type != 'text' || input.name == '')"
                           "        continue;"
                           "    for (var key in table) {"
                           "        if (!table.hasOwnProperty(key) || table[key] == '')"
                           "            continue;"
                           "        if (key == input.name || input.name.indexOf(key) != -1) {"
                           "            input.style['-webkit-appearance'] = 'none';"
                           "            input.style['-webkit-box-shadow'] = 'inset 0 0 2px 1px #EEE000';"
                           "            break;"
                           "        }"
                           "    }"
                           "}");
 
     page->runJavaScript(source.arg(matchingJsTable()), WebPage::SafeJsWorld);
 }
 
 QString PIM_Handler::matchingJsTable() const
 {
     QString values;
 
     QHashIterator<PI_Type, QStringList> i(m_infoMatches);
     while (i.hasNext()) {
         i.next();
 
         const auto ivalues = i.value();
         for (const QString &value : ivalues) {
             QString key = m_allInfo.value(i.key());
             key.replace(QL1C('"'), QL1S("\\\""));
             values.append(QSL("\"%1\":\"%2\",").arg(value, key));
         }
     }
 
     if (!values.isEmpty()) {
         values = values.left(values.size() - 1);
     }
 
     return QSL("{ %1 }").arg(values);
 }