/*
   SPDX-FileCopyrightText: 2016-2025 Laurent Montel <montel@kde.org>

   SPDX-License-Identifier: LGPL-2.0-or-later
*/
#include "webenginescript.h"
using namespace Qt::Literals::StringLiterals;

using namespace WebEngineViewer;

QString WebEngineScript::findAllImages()
{
    const QString source = QStringLiteral(
        "(function() {"
        "var out = [];"
        "var imgs = document.getElementsByTagName('img');"
        "for (var i = 0; i < imgs.length; ++i) {"
        "    var e = imgs[i];"
        "    out.push({"
        "        src: e.src"
        "    });"
        "}"
        "return out;"
        "})()");

    return source;
}

QString WebEngineScript::findAllScripts()
{
    const QString source = QStringLiteral(
        "(function() {"
        "var out = [];"
        "var scripts = document.getElementsByTagName('script');"
        "for (var i = 0; i < scripts.length; ++i) {"
        "    var e = scripts[i];"
        "    out.push({"
        "        src: e.src"
        "    });"
        "}"
        "return out;"
        "})()");

    return source;
}

QString WebEngineScript::findAllAnchors()
{
    const QString source = QStringLiteral(
        "(function() {"
        "var out = [];"
        "var anchors = document.getElementsByTagName('a');"
        "for (var i = 0; i < anchors.length; ++i) {"
        "    var r = anchors[i].getBoundingClientRect();"
        "    out.push({"
        "        src: anchors[i].href,"
        "        title: anchors[i].title,"
        "        boudingRect: [r.top, r.left, r.width, r.height]"
        "    });"
        "}"
        "return out;"
        "})()");

    return source;
}

QString WebEngineScript::findAllAnchorsAndForms()
{
    const QString source = QStringLiteral(
        "(function() {"
        "var res = [];"
        "var out = [];"
        "var anchors = document.getElementsByTagName('a');"
        "for (var i = 0; i < anchors.length; ++i) {"
        "    out.push({"
        "        src: anchors[i].href,"
        "        title: anchors[i].title,"
        "        text: anchors[i].innerText"
        "    });"
        "}"
        "var forms = document.getElementsByTagName('form');"
        "res.push({"
        "    anchors: out,"
        "    forms:  forms.length"
        "    });"
        "return res;"
        "})()");

    return source;
}

QString WebEngineScript::setElementByIdVisible(const QString &elementStr, bool visibility)
{
    if (visibility) {
        const QString source = QStringLiteral(
                                   "var element = document.getElementById('%1'); "
                                   "if (element) { "
                                   "    element.style.removeProperty( 'display' );"
                                   "}")
                                   .arg(elementStr);
        return source;
    } else {
        const QString source = QStringLiteral(
                                   "var element = document.getElementById('%1'); "
                                   "if (element) { "
                                   "    element.style.display = \"none\";"
                                   "}")
                                   .arg(elementStr);
        return source;
    }
}

QString WebEngineScript::searchElementPosition(const QString &elementStr)
{
    const QString source = QStringLiteral(
                               "var element = document.getElementById('%1'); "
                               "if (element) { "
                               "    var geometry = element.getBoundingClientRect(); "
                               "    [(geometry.left + window.scrollX), (geometry.top + window.scrollY)]; "
                               "}")
                               .arg(elementStr);
    return source;
}

static QString scrollTop()
{
    return u"document.documentElement.scrollTop"_s;
}

QString WebEngineScript::scrollPercentage(int percent)
{
    const QString source = u"var current = "_s + scrollTop()
        + QStringLiteral(
              ";"
              "var docElement = document.documentElement;"
              "var height = docElement.clientHeight;"
              "var newPosition = current + height * %1 /100;"
              "window.scrollTo(window.scrollX, newPosition);")
              .arg(percent);
    return source;
}

QString WebEngineScript::scrollUp(int pixel)
{
    const QString source = u"window.scrollBy(0, %1);"_s.arg(-pixel);
    return source;
}

QString WebEngineScript::scrollDown(int pixel)
{
    const QString source = u"window.scrollBy(0, %1);"_s.arg(pixel);
    return source;
}

QString WebEngineScript::scrollToPosition(const QPoint &pos)
{
    const QString source = u"window.scrollTo(%1, %2); [window.scrollX, window.scrollY];"_s.arg(pos.x()).arg(pos.y());
    return source;
}

QString WebEngineScript::removeStyleToElement(const QString &elementStr)
{
    const QString source = QStringLiteral(
                               "var element = document.getElementById('%1'); "
                               "if (element) { "
                               "    element.removeAttribute(\"style\");"
                               "}")
                               .arg(elementStr);
    return source;
}

QString WebEngineScript::setStyleToElement(const QString &elementStr, const QString &style)
{
    const QString source = QStringLiteral(
                               "var element = document.getElementById('%1'); "
                               "if (element) { "
                               "    element.style = '%2';"
                               "}")
                               .arg(elementStr, style);
    return source;
}

QString WebEngineScript::scrollToRelativePosition(qreal pos)
{
    const QString source = u"window.scrollTo(window.scrollX, %1); [window.scrollX, window.scrollY];"_s.arg(pos);
    return source;
}

QString WebEngineScript::isScrolledToBottom()
{
    return QStringLiteral(
               "(function() { "
               "var docElement = document.documentElement;"
               "var viewportHeight = docElement.clientHeight;"
               "var isAtBottom = ")
        + scrollTop()
        + QStringLiteral(
               " + viewportHeight >= document.body.scrollHeight;"
               "return Boolean(isAtBottom); "
               "}());");
}
