findnotificationwe.h

Go to the documentation of this file.

/**********************************************************************
 * LeechCraft - modular cross-platform feature rich internet client.
 * Copyright (C) 2006-2014  Georg Rudoy
 *
 * Distributed under the Boost Software License, Version 1.0.
 * (See accompanying file LICENSE or copy at https://www.boost.org/LICENSE_1_0.txt)
 **********************************************************************/

#pragma once

#include "findnotification.h"
#include <QWebEnginePage>
#include <QWebEngineView>
#include <QWebEngineFindTextResult>

namespace LC::Util
{
    class UTIL_GUI_API FindNotificationWE : public FindNotification
    {
        QWebEngineView * const WebView_;
        QString PreviousFindText_;
    public:
        FindNotificationWE (const ICoreProxy_ptr& proxy, QWebEngineView *near)
        : FindNotification { proxy, near }
        , WebView_ { near }
        {
        }

        static QWebEnginePage::FindFlags ToPageFlags (FindFlags findFlags)
        {
            QWebEnginePage::FindFlags pageFlags;
            auto check = [&pageFlags, findFlags] (FindFlag ourFlag, QWebEnginePage::FindFlag pageFlag)
            {
                if (findFlags & ourFlag)
                    pageFlags |= pageFlag;
            };
            check (FindCaseSensitively, QWebEnginePage::FindCaseSensitively);
            check (FindBackwards, QWebEnginePage::FindBackward);
            return pageFlags;
        }
    private:
        void ClearFindResults ()
        {
            PreviousFindText_.clear ();
            WebView_->page ()->findText ({});
        }
    protected:
        void HandleNext (const QString& text, FindFlags findFlags) override
        {
            if (PreviousFindText_ != text)
            {
                ClearFindResults ();
                PreviousFindText_ = text;
            }

            WebView_->page ()->findText (text, ToPageFlags (findFlags),
                    [this] (const QWebEngineFindTextResult& found) { SetSuccessful (found.numberOfMatches ()); });
        }

        void Reject () override
        {
            FindNotification::Reject ();
            ClearFindResults ();
        }
    };
}