File indexing completed on 2024-05-12 04:58:29

 /* ============================================================
 * Falkon - Qt web browser
 * Copyright (C) 2015-2018 David Rosca <nowrep@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 "webhittestresult.h"
 #include "webpage.h"
 
 #include <QWebEngineContextMenuRequest>
 
 WebHitTestResult::WebHitTestResult(const WebPage *page, const QPoint &pos)
     : m_isNull(true)
     , m_isContentEditable(false)
     , m_isContentSelected(false)
     , m_mediaPaused(false)
     , m_mediaMuted(false)
     , m_pos(pos)
 {
     QString source = QL1S("(function() {"
                           "var e = document.elementFromPoint(%1, %2);"
                           "if (!e)"
                           "    return;"
                           "function isMediaElement(e) {"
                           "    return e.tagName.toLowerCase() == 'audio' || e.tagName.toLowerCase() == 'video';"
                           "}"
                           "function isEditableElement(e) {"
                           "    if (e.isContentEditable)"
                           "        return true;"
                           "    if (e.tagName.toLowerCase() == 'input' || e.tagName.toLowerCase() == 'textarea')"
                           "        return e.getAttribute('readonly') != 'readonly';"
                           "    return false;"
                           "}"
                           "function isSelected(e) {"
                           "    var selection = window.getSelection();"
                           "    if (selection.type != 'Range')"
                           "        return false;"
                           "    return window.getSelection().containsNode(e, true);"
                           "}"
                           "function attributeStr(e, a) {"
                           "    return e.getAttribute(a) || '';"
                           "}"
                           "var res = {"
                           "    baseUrl: document.baseURI,"
                           "    alternateText: e.getAttribute('alt'),"
                           "    boundingRect: '',"
                           "    imageUrl: '',"
                           "    contentEditable: isEditableElement(e),"
                           "    contentSelected: isSelected(e),"
                           "    linkTitle: '',"
                           "    linkUrl: '',"
                           "    mediaUrl: '',"
                           "    tagName: e.tagName.toLowerCase()"
                           "};"
                           "var r = e.getBoundingClientRect();"
                           "res.boundingRect = [r.top, r.left, r.width, r.height];"
                           "if (e.tagName.toLowerCase() == 'img')"
                           "    res.imageUrl = attributeStr(e, 'src').trim();"
                           "if (e.tagName.toLowerCase() == 'a') {"
                           "    res.linkTitle = e.text;"
                           "    res.linkUrl = attributeStr(e, 'href').trim();"
                           "}"
                           "while (e) {"
                           "    if (res.linkTitle == '' && e.tagName.toLowerCase() == 'a')"
                           "        res.linkTitle = e.text;"
                           "    if (res.linkUrl == '' && e.tagName.toLowerCase() == 'a')"
                           "        res.linkUrl = attributeStr(e, 'href').trim();"
                           "    if (res.mediaUrl == '' && isMediaElement(e)) {"
                           "        res.mediaUrl = e.currentSrc;"
                           "        res.mediaPaused = e.paused;"
                           "        res.mediaMuted = e.muted;"
                           "    }"
                           "    e = e.parentElement;"
                           "}"
                           "return res;"
                           "})()");
 
     auto *p = const_cast<WebPage*>(page);
     m_viewportPos = p->mapToViewport(m_pos);
     const QString &js = source.arg(m_viewportPos.x()).arg(m_viewportPos.y());
     init(p->url(), p->execJavaScript(js, WebPage::SafeJsWorld).toMap());
 }
 
 void WebHitTestResult::updateWithContextMenuData(const QWebEngineContextMenuRequest &data)
 {
     if (data.position() != m_pos) {
         return;
     }
 
     m_linkTitle = data.linkText();
     m_linkUrl = data.linkUrl();
     m_isContentEditable = data.isContentEditable();
     m_isContentSelected = !data.selectedText().isEmpty();
 
     switch (data.mediaType()) {
     case QWebEngineContextMenuRequest::MediaTypeImage:
         m_imageUrl = data.mediaUrl();
         break;
 
     case QWebEngineContextMenuRequest::MediaTypeVideo:
     case QWebEngineContextMenuRequest::MediaTypeAudio:
         m_mediaUrl = data.mediaUrl();
         break;
 
     default:
         break;
     }
 }
 
 QUrl WebHitTestResult::baseUrl() const
 {
     return m_baseUrl;
 }
 
 QString WebHitTestResult::alternateText() const
 {
     return m_alternateText;
 }
 
 QRect WebHitTestResult::boundingRect() const
 {
     return m_boundingRect;
 }
 
 QUrl WebHitTestResult::imageUrl() const
 {
     return m_imageUrl;
 }
 
 bool WebHitTestResult::isContentEditable() const
 {
     return m_isContentEditable;
 }
 
 bool WebHitTestResult::isContentSelected() const
 {
     return m_isContentSelected;
 }
 
 bool WebHitTestResult::isNull() const
 {
     return m_isNull;
 }
 
 QString WebHitTestResult::linkTitle() const
 {
     return m_linkTitle;
 }
 
 QUrl WebHitTestResult::linkUrl() const
 {
     return m_linkUrl;
 }
 
 QUrl WebHitTestResult::mediaUrl() const
 {
     return m_mediaUrl;
 }
 
 bool WebHitTestResult::mediaPaused() const
 {
     return m_mediaPaused;
 }
 
 bool WebHitTestResult::mediaMuted() const
 {
     return m_mediaMuted;
 }
 
 QPoint WebHitTestResult::pos() const
 {
     return m_pos;
 }
 
 QPointF WebHitTestResult::viewportPos() const
 {
     return m_viewportPos;
 }
 
 QString WebHitTestResult::tagName() const
 {
     return m_tagName;
 }
 
 void WebHitTestResult::init(const QUrl &url, const QVariantMap &map)
 {
     if (map.isEmpty())
         return;
 
     m_isNull = false;
     m_baseUrl = map.value(QSL("baseUrl")).toUrl();
     m_alternateText = map.value(QSL("alternateText")).toString();
     m_imageUrl = map.value(QSL("imageUrl")).toUrl();
     m_isContentEditable = map.value(QSL("contentEditable")).toBool();
     m_isContentSelected = map.value(QSL("contentSelected")).toBool();
     m_linkTitle = map.value(QSL("linkTitle")).toString();
     m_linkUrl = map.value(QSL("linkUrl")).toUrl();
     m_mediaUrl = map.value(QSL("mediaUrl")).toUrl();
     m_mediaPaused = map.value(QSL("mediaPaused")).toBool();
     m_mediaMuted = map.value(QSL("mediaMuted")).toBool();
     m_tagName = map.value(QSL("tagName")).toString();
 
     const QVariantList &rect = map.value(QSL("boundingRect")).toList();
     if (rect.size() == 4)
         m_boundingRect = QRect(rect.at(0).toInt(), rect.at(1).toInt(), rect.at(2).toInt(), rect.at(3).toInt());
 
     if (!m_imageUrl.isEmpty())
         m_imageUrl = url.resolved(m_imageUrl);
     if (!m_linkUrl.isEmpty())
         m_linkUrl = m_baseUrl.resolved(m_linkUrl);
     if (!m_mediaUrl.isEmpty())
         m_mediaUrl = url.resolved(m_mediaUrl);
 }