File indexing completed on 2024-12-22 04:41:15

 /* ============================================================
 * Falkon - Qt web browser
 * Copyright (C) 2018 Anmol Gautam <tarptaeya@gmail.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 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 "qmluserscripts.h"
 #include "mainapplication.h"
 #include <QWebEngineProfile>
 #include <QWebEngineScriptCollection>
 #include <QQmlEngine>
 
 QmlUserScripts::QmlUserScripts(QObject *parent)
     : QObject(parent)
 {
 }
 
 int QmlUserScripts::count() const
 {
     return mApp->webProfile()->scripts()->count();
 }
 
 int QmlUserScripts::size() const
 {
     return mApp->webProfile()->scripts()->count();
 }
 
 bool QmlUserScripts::empty() const
 {
     return mApp->webProfile()->scripts()->isEmpty();
 }
 
 QList<QObject*> QmlUserScripts::toQObjectList(const QList<QWebEngineScript> &list) const
 {
     QList<QObject*> userScriptList;
     userScriptList.reserve(list.size());
     for (const QWebEngineScript &script : list) {
         auto *userScript = new QmlUserScript();
         userScript->setWebEngineScript(script);
         userScriptList.append(userScript);
     }
     return userScriptList;
 }
 
 bool QmlUserScripts::contains(QObject *object) const
 {
     auto *userScript = qobject_cast<QmlUserScript*>(object);
     if (!userScript) {
         return false;
     }
     QWebEngineScript webEngineScript = userScript->webEngineScript();
     return mApp->webProfile()->scripts()->contains(webEngineScript);
 }
 
 QObject *QmlUserScripts::findScript(const QString &name) const
 {
     auto *qmlUserScript = new QmlUserScript();
 
     auto scripts = mApp->webProfile()->scripts()->find(name);
     if (!scripts.empty()) {
         qmlUserScript->setWebEngineScript(scripts.first());
     }
 
     return qmlUserScript;
 }
 
 QList<QObject*> QmlUserScripts::findScripts(const QString &name) const
 {
     QList<QWebEngineScript> list = mApp->webProfile()->scripts()->find(name);
     return toQObjectList(list);
 }
 
 void QmlUserScripts::remove(QObject *object) const
 {
     auto *userScript = qobject_cast<QmlUserScript*>(object);
     if (!userScript) {
         return;
     }
     QWebEngineScript webEngineScript = userScript->webEngineScript();
     mApp->webProfile()->scripts()->remove(webEngineScript);
 }
 
 QList<QObject*> QmlUserScripts::toList() const
 {
     QList<QWebEngineScript> list = mApp->webProfile()->scripts()->toList();
     return toQObjectList(list);
 }
 
 void QmlUserScripts::insert(QObject *object)
 {
     auto *userScript = qobject_cast<QmlUserScript*>(object);
     if (!userScript) {
         return;
     }
     QWebEngineScript webEngineScript = userScript->webEngineScript();
     mApp->webProfile()->scripts()->insert(webEngineScript);
 }