cefengine.cpp

#include "cefengine.h"

extern QVariantMap settingsData;
extern QVariantMap notificationsData;
extern CefString historyData;
extern SignalBroker* signalBroker;

CefEngine::CefEngine() : CefApp()
{

}

void CefEngine::AddRef() const {
    CefRefCount::AddRef();
}

bool CefEngine::Release() const {
    return CefRefCount::Release();
}

bool CefEngine::HasOneRef() const {
    return CefRefCount::HasOneRef();
}

void CefEngine::OnContextInitialized() {
    CefRegisterSchemeHandlerFactory("theweb", "theweb", new theWebSchemeHandler());
}

void CefEngine::OnContextCreated(Browser browser, CefRefPtr<CefFrame> frame, CefRefPtr<CefV8Context> context) {
    //Get browser settings
    CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create("theWebSettings_get");
    browser.get()->SendProcessMessage(PID_BROWSER, message);

    if (QUrl(QString::fromStdString(frame.get()->GetURL().ToString())).scheme() == "theweb") {
        {
            //Register the theWebSettingsObject JavaScript object
            CefRefPtr<theWebSettingsAccessor> accessor = new theWebSettingsAccessor(browser);
            CefRefPtr<CefV8Value> JsObject = CefV8Value::CreateObject(accessor, new V8Interceptor);
            JsObject.get()->SetValue("dnt", V8_ACCESS_CONTROL_DEFAULT, V8_PROPERTY_ATTRIBUTE_NONE);
            JsObject.get()->SetValue("home", V8_ACCESS_CONTROL_DEFAULT, V8_PROPERTY_ATTRIBUTE_NONE);
            JsObject.get()->SetValue("toolbar", V8_ACCESS_CONTROL_DEFAULT, V8_PROPERTY_ATTRIBUTE_NONE);
            JsObject.get()->SetValue("tabText", V8_ACCESS_CONTROL_DEFAULT, V8_PROPERTY_ATTRIBUTE_NONE);
            JsObject.get()->SetValue("malwareProtect", V8_ACCESS_CONTROL_DEFAULT, V8_PROPERTY_ATTRIBUTE_NONE);
            JsObject.get()->SetValue("tabPreview", V8_ACCESS_CONTROL_DEFAULT, V8_PROPERTY_ATTRIBUTE_NONE);
            JsObject.get()->SetValue("history", V8_ACCESS_CONTROL_DEFAULT, V8_PROPERTY_ATTRIBUTE_NONE);
            JsObject.get()->SetValue("resetBrowser", CefV8Value::CreateFunction("resetBrowser", new V8Function([=]() {
                CefRefPtr<CefV8Value> returnVal;
                CefRefPtr<CefV8Exception> exception;
                context.get()->Eval("confirm(\"Reset theWeb?:Here\'s what we\'ll do.\\n"
                                    "- Reset all settings back to defaults\\n"
                                    "- Erase all history items\\n"
                                    "- Clear the cache\\n"
                                    "- Erase all cookies\\n"
                                    "\\n"
                                    "Are you sure that you want to reset theWeb?:Continue:Cancel\")", "", 0, returnVal, exception);
                if (returnVal.get()->GetBoolValue()) {
                    context.get()->Eval("confirm(\"Reset theWeb?:We're ready to reset theWeb. Once this is done, theWeb will exit. Is this OK?:Reset:Cancel\")", "", 0, returnVal, exception);
                }
            })), V8_PROPERTY_ATTRIBUTE_NONE);
            JsObject.get()->SetValue("proxySettings", CefV8Value::CreateFunction("proxySettings", new V8Function([=]() {
                CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create("showProxy");
                browser.get()->SendProcessMessage(PID_BROWSER, message);
            })), V8_PROPERTY_ATTRIBUTE_NONE);
            JsObject.get()->SetValue("clearData", CefV8Value::CreateFunction("clearData", new V8Function([=](CefV8ValueList &arguments) {
                QVector<CefRefPtr<CefV8Value>> args = QVector<CefRefPtr<CefV8Value>>::fromStdVector(arguments);
                if (args.size() == 3) {
                    //Erase History
                    if (args[0].get()->GetBoolValue()) {
                        CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create("clearData_History");
                        browser.get()->SendProcessMessage(PID_BROWSER, message);
                    }
                    //Erase Cache
                    if (args[1].get()->GetBoolValue()) {
                        qDebug() << "Erase Cache";
                    }

                    //Erase Cookies
                    if (args[2].get()->GetBoolValue()) {
                        qDebug() << "Erase Cookies";
                    }
                }
            })), V8_PROPERTY_ATTRIBUTE_NONE);
            JsObject.get()->SetValue("goToCredits", CefV8Value::CreateFunction("goToCredits", new V8Function([=]() {
                CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create("goToCredits");
                browser.get()->SendProcessMessage(PID_BROWSER, message);
            })), V8_PROPERTY_ATTRIBUTE_NONE);

            context.get()->GetGlobal()->SetValue("theWebSettingsObject", JsObject, V8_PROPERTY_ATTRIBUTE_NONE);
        }
    }

    {
        {
            //Register the Notifications JavaScript Object
            CefRefPtr<CefV8Value> notificationObject = CefV8Value::CreateFunction("NotificationConstructor", new V8Function([=](CefV8ValueList &arguments) {
                QVector<CefRefPtr<CefV8Value>> args = QVector<CefRefPtr<CefV8Value>>::fromStdVector(arguments);

                if (args.count() < 1) {
                    throw new std::exception;
                }
                CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create("jsNotifications_post");
                message.get()->GetArgumentList().get()->SetString(0, QUrl(QString::fromStdString(frame.get()->GetURL().ToString())).host().toStdString());
                message.get()->GetArgumentList().get()->SetString(1, args.at(0).get()->GetStringValue());
                browser.get()->SendProcessMessage(PID_BROWSER, message);
            }));

            notificationObject.get()->SetValue("permission", CefV8Value::CreateString(notificationsData.value(QUrl(QString::fromStdString(frame.get()->GetURL().ToString())).host(), "default").toString().toStdString()), V8_PROPERTY_ATTRIBUTE_NONE);
            notificationObject.get()->SetValue("requestPermission", CefV8Value::CreateFunction("requestPermission", new V8Function((std::function<CefRefPtr<CefV8Value>()>) [=]() {
                if (notificationObject.get()->GetValue("permission").get()->GetStringValue() == "default") {
                    if (notificationRequestPromise == NULL) {
                        qDebug() << context.get()->GetGlobal().get()->SetValue("theWebNotificationRequest", V8_ACCESS_CONTROL_DEFAULT, V8_PROPERTY_ATTRIBUTE_NONE);
                        CefRefPtr<CefV8Exception> exception;
                        context.get()->Eval("new Promise(function(resolve, reject) {"
                                                "theWebNotificationRequest = resolve;"
                                            "});", "", 0, notificationRequestPromise, exception);
                        notificationRequestResolver = context.get()->GetGlobal().get()->GetValue("theWebNotificationRequest");

                        //Ask user permission to show notifications
                        CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create("jsNotificationRequest");
                        message.get()->GetArgumentList().get()->SetString(0, QUrl(QString::fromStdString(frame.get()->GetURL().ToString())).host().toStdString());
                        browser.get()->SendProcessMessage(PID_BROWSER, message);

                        notificationRequestFrame = frame;

                        return notificationRequestPromise;
                    } else {
                        return CefV8Value::CreateNull();
                    }
                } else {
                    return CefV8Value::CreateNull();
                }
            })), V8_PROPERTY_ATTRIBUTE_NONE);

            context.get()->GetGlobal().get()->SetValue("Notification", notificationObject, V8_PROPERTY_ATTRIBUTE_NONE);
        }
    }
}

void CefEngine::OnContextReleased(Browser browser, CefRefPtr<CefFrame> frame, CefRefPtr<CefV8Context> context) {
    if (this->videoFrame.get() != NULL) {
        //Stop MPRIS because the browser just closed
        if (this->videoFrame.get()->GetIdentifier() == frame.get()->GetIdentifier()) {
            CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create("mprisDoStop");
            browser.get()->SendProcessMessage(PID_BROWSER, message);
            this->videoFrame = NULL;
        }
    }
}

bool CefEngine::OnProcessMessageReceived(CefRefPtr<CefBrowser> browser, CefProcessId source_process, CefRefPtr<CefProcessMessage> message) {
    if (message.get()->GetName() == "theWebSettings_reply") {
        CefRefPtr<CefListValue> args = message.get()->GetArgumentList();
        {
            CefRefPtr<CefDictionaryValue> settingsDictionary = args.get()->GetDictionary(0);
            std::vector<CefString> keys;
            settingsDictionary.get()->GetKeys(keys);
            for (CefString key : keys) {
                //Convert CEF types into Qt types
                if (settingsDictionary.get()->GetType(key) == VTYPE_BOOL) {
                    settingsData.insert(QString::fromStdString(key.ToString()), settingsDictionary.get()->GetBool(key));
                } else if (settingsDictionary.get()->GetType(key) == VTYPE_STRING) {
                    settingsData.insert(QString::fromStdString(key.ToString()), QString::fromStdString(settingsDictionary.get()->GetString(key).ToString()));
                }
            }

            historyData = args.get()->GetString(2);
        }

        {
            CefRefPtr<CefDictionaryValue> notificationsDictionary = args.get()->GetDictionary(1);
            std::vector<CefString> keys;
            notificationsDictionary.get()->GetKeys(keys);
            for (CefString key : keys) {
                //Convert CEF types into Qt types
                notificationsData.insert(QString::fromStdString(key.ToString()), QString::fromStdString(notificationsDictionary.get()->GetString(key).ToString()));
            }
        }
    } else if (message.get()->GetName() == "mprisCheck") {
        //Iterate over all frames
        std::vector<int64> frameIdentifiers;
        browser.get()->GetFrameIdentifiers(frameIdentifiers);
        for (unsigned long i = 0; i < frameIdentifiers.size(); i++) {
            //Check if we should check the browser
            bool doCheck = false;
            if (this->videoFrame == NULL) { //If there is no browser playing media, check the browser
                doCheck = true;
            } else if (this->videoFrame.get()->GetIdentifier() == browser.get()->GetFrame(frameIdentifiers[i]).get()->GetIdentifier() &&
                       browser.get()->GetIdentifier() == this->videoFrame.get()->GetBrowser().get()->GetIdentifier()) {
                //If this is the same browser playing media, check the browser
                doCheck = true;
            }

            if (doCheck) {
                //Enter the V8 Context
                CefRefPtr<CefV8Context> context = browser.get()->GetFrame(frameIdentifiers[i]).get()->GetV8Context();
                context.get()->Enter();

                //Get all <video> elements
                CefRefPtr<CefV8Value> returnVal;
                CefRefPtr<CefV8Exception> exception;
                context.get()->Eval("document.getElementsByTagName('video').length;", "", 0, returnVal, exception);

                if (returnVal.get() == NULL) {
                    mprisElementTagType = "";
                } else {
                    if (returnVal.get()->GetIntValue() > 0) {
                        //Video found.
                        mprisElementTagType = "video";
                    } else {
                        //No <video> elements. Get all <audio> elements
                        context.get()->Eval("document.getElementsByTagName('audio').length;", "", 0, returnVal, exception);
                        if (returnVal.get()->GetIntValue() > 0) {
                            //Audio found.
                            mprisElementTagType = "audio";
                        } else {
                            mprisElementTagType = "";
                        }
                    }
                }

                if (mprisElementTagType != "") {
                    this->videoContext = context;
                    this->videoFrame = browser.get()->GetFrame(frameIdentifiers[i]);

                    CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create("mprisStart");
                    browser.get()->SendProcessMessage(PID_BROWSER, message);

                    //Get if the video is paused
                    context.get()->Eval("document.getElementsByTagName('" + mprisElementTagType.toStdString() + "')[0].paused", "", 0, returnVal, exception);
                    videoPlaying = !returnVal.get()->GetBoolValue();

                    CefString title, artist, album;

                    //Provide extra information for videos from youtube.com
                    if (QUrl(QString::fromStdString(videoFrame.get()->GetURL().ToString())).host().startsWith("www.youtube.com")) {
                        context.get()->Eval("document.getElementById('eow-title').innerHTML", "", 0, returnVal, exception);
                        if (returnVal.get() != NULL) {
                            title = QString::fromStdString(returnVal.get()->GetStringValue().ToString()).trimmed().replace("&amp;", "&")
                                    .replace("&lt;", "<").replace("&gt;", ">").toStdString();
                        }

                        context.get()->Eval("document.getElementById('watch7-user-header').childNodes[3].childNodes[1].innerHTML", "", 0, returnVal, exception);
                        if (returnVal.get() != NULL) {
                            artist = QString::fromStdString(returnVal.get()->GetStringValue().ToString()).replace("&amp;", "&")
                                    .replace("&lt;", "<").replace("&gt;", ">").toStdString();
                        }

                        album = QUrl(QString::fromStdString(videoFrame.get()->GetURL().ToString())).host().toStdString();
                    }

                    {
                        CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create("mprisData");
                        message.get()->GetArgumentList().get()->SetBool(0, videoPlaying);
                        message.get()->GetArgumentList().get()->SetString(1, title);
                        message.get()->GetArgumentList().get()->SetString(2, artist);
                        message.get()->GetArgumentList().get()->SetString(3, album);
                        browser.get()->SendProcessMessage(PID_BROWSER, message);
                    }
                } else {
                    CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create("mprisStop");
                    //message.get()->GetArgumentList().get()->SetBool(0, browser.get()->GetFrame(frameIdentifiers[i]).get()->IsMain());
                    browser.get()->SendProcessMessage(PID_BROWSER, message);
                    this->videoFrame = NULL;
                }

                //Exit the V8 Context
                context.get()->Exit();
            }
        }
    } else if (message.get()->GetName() == "mprisPlayPause") {
        //Enter the V8 Context
        CefRefPtr<CefV8Context> context = videoFrame.get()->GetV8Context();
        context.get()->Enter();

        CefRefPtr<CefV8Value> returnVal;
        CefRefPtr<CefV8Exception> exception;
        if (videoPlaying) {
            //The media is playing, pause the media.
            context.get()->Eval("document.getElementsByTagName('" + mprisElementTagType.toStdString() + "')[0].pause()", "", 0, returnVal, exception);
        } else {
            //The media is paused, play the media.
            context.get()->Eval("document.getElementsByTagName('" + mprisElementTagType.toStdString() + "')[0].play()", "", 0, returnVal, exception);
        }

        //Exit the V8 Context
        context.get()->Exit();

        //Do MPRIS Check
        CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create("mprisCheck");
        browser.get()->SendProcessMessage(PID_RENDERER, message);
    } else if (message.get()->GetName() == "mprisPlay") {
        if (!videoPlaying) {
            //Enter the V8 Context
            CefRefPtr<CefV8Context> context = videoFrame.get()->GetV8Context();
            context.get()->Enter();

            //The media is paused, play the media.
            CefRefPtr<CefV8Value> returnVal;
            CefRefPtr<CefV8Exception> exception;
            context.get()->Eval("document.getElementsByTagName('" + mprisElementTagType.toStdString() + "')[0].play()", "", 0, returnVal, exception);

            //Exit the V8 Context
            context.get()->Exit();
        }

        //Do MPRIS Check
        CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create("mprisCheck");
        browser.get()->SendProcessMessage(PID_RENDERER, message);
    } else if (message.get()->GetName() == "mprisPause") {
        if (videoPlaying) {
            //Enter the V8 Context
            CefRefPtr<CefV8Context> context = videoFrame.get()->GetV8Context();
            context.get()->Enter();

            //The media is playing, pause the media.
            CefRefPtr<CefV8Value> returnVal;
            CefRefPtr<CefV8Exception> exception;
            context.get()->Eval("document.getElementsByTagName('" + mprisElementTagType.toStdString() + "')[0].pause()", "", 0, returnVal, exception);

            //Exit the V8 Context
            context.get()->Exit();
        }

        //Do MPRIS Check
        CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create("mprisCheck");
        browser.get()->SendProcessMessage(PID_RENDERER, message);
    } else if (message.get()->GetName() == "mprisBack") {
        //Enter the V8 Context
        CefRefPtr<CefV8Context> context = videoFrame.get()->GetV8Context();
        context.get()->Enter();

        //Set the media to 0 seconds
        CefRefPtr<CefV8Value> returnVal;
        CefRefPtr<CefV8Exception> exception;
        context.get()->Eval("document.getElementsByTagName('" + mprisElementTagType.toStdString() + "')[0].currentTime = 0", "", 0, returnVal, exception);

        //Exit the V8 Context
        context.get()->Exit();

        //Do MPRIS Check
        CefRefPtr<CefProcessMessage> message = CefProcessMessage::Create("mprisCheck");
        browser.get()->SendProcessMessage(PID_RENDERER, message);
    } else if (message.get()->GetName() == "jsNotificationRequest_Reply") {
        //Enter the V8 Context
        CefRefPtr<CefV8Context> context = notificationRequestFrame->GetV8Context();
        context.get()->Enter();

        CefRefPtr<CefProcessMessage> setMessage = CefProcessMessage::Create("jsNotifications_set");
        setMessage.get()->GetArgumentList().get()->SetString(0, QUrl(QString::fromStdString(notificationRequestFrame.get()->GetURL().ToString())).host().toStdString());

        CefV8ValueList args;
        if (message.get()->GetArgumentList().get()->GetBool(0)) {
            args.push_back(CefV8Value::CreateString("granted"));
            context.get()->GetGlobal().get()->GetValue("Notification").get()->SetValue("permission", CefV8Value::CreateString("granted"), V8_PROPERTY_ATTRIBUTE_NONE);
            notificationsData.insert(QUrl(QString::fromStdString(notificationRequestFrame.get()->GetURL().ToString())).host(), "granted");
            setMessage.get()->GetArgumentList().get()->SetString(1, "granted");
        } else {
            args.push_back(CefV8Value::CreateString("denied"));
            context.get()->GetGlobal().get()->GetValue("Notification").get()->SetValue("permission", CefV8Value::CreateString("denied"), V8_PROPERTY_ATTRIBUTE_NONE);
            notificationsData.insert(QUrl(QString::fromStdString(notificationRequestFrame.get()->GetURL().ToString())).host(), "denied");
            setMessage.get()->GetArgumentList().get()->SetString(1, "denied");
        }
        notificationRequestResolver.get()->ExecuteFunction(NULL, args);

        browser.get()->SendProcessMessage(PID_RENDERER, setMessage);

        //notificationRequestPromise.get()->GetValue("resolve").get()->ExecuteFunction(NULL, args);
        /*if (message.get()->GetArgumentList().get()->GetBool(0)) {
            browser.get()->GetMainFrame().get()->ExecuteJavaScript("theWebNotificationRequest(\"granted\")", "", 0);
        } else {
            browser.get()->GetMainFrame().get()->ExecuteJavaScript("theWebNotificationRequest(\"denied\")", "", 0);
        }*/

        //Exit the V8 Context
        context.get()->Exit();
        notificationRequestPromise = NULL;
    }
    return true;
}

bool CefEngine::OnPrintDialog(bool has_selection, CefRefPtr<CefPrintDialogCallback> callback) {
    emit signalBroker->PrintDialog(this->printingBrowser, this->printer, has_selection, callback);
    return true;
}

void CefEngine::OnPrintStart(Browser browser) {
    if (printer == NULL) {
        printer = new QPrinter;
    }
    this->printingBrowser = browser;
}

void CefEngine::OnPrintSettings(CefRefPtr<CefPrintSettings> settings, bool get_defaults) {
    if (get_defaults) {
        getCefPrinterSettings(this->printer, settings);
    } else {
        printer->setCollateCopies(settings.get()->WillCollate());
        printer->setOrientation(settings.get()->IsLandscape() ? QPrinter::Landscape : QPrinter::Portrait);
    }
}

void CefEngine::OnPrintReset() {
    this->printingBrowser = NULL;
}

bool CefEngine::OnPrintJob(const CefString &document_name, const CefString &pdf_file_path, CefRefPtr<CefPrintJobCallback> callback) {
    QString path = QString::fromStdString(pdf_file_path);

    //Set up the printer
    printer->setDocName(QString::fromStdString(document_name));
    Poppler::Document *document = Poppler::Document::load(QString::fromStdString(pdf_file_path));

    //Make the output look a bit better
    document->setRenderHint(Poppler::Document::Antialiasing);
    document->setRenderHint(Poppler::Document::TextAntialiasing);

    //Start to print the document
    QPrinter::Margins margins = printer->margins();
    QPainter* painter = new QPainter(this->printer);

    //Iterate over all pages
    for (int i = 0; i < document->numPages(); i++) {
        //Render page as an image
        QImage render = document->page(i)->renderToImage(printer->resolution(), printer->resolution());
        //render.save("/home/victor/printTest.png");
        //document->page(i)->renderToPainter(painter, printer->resolution(), printer->resolution());

        //Draw image onto printer
        painter->drawImage(margins.left, margins.top, render);

        //Check if we need to feed a new page
        if (i != document->numPages() - 1) {
            //Feed a new page
            printer->newPage();
        }
    }

    //Complete the printing
    painter->end();

    //Tell CEF to continue
    callback.get()->Continue();
    return true;
}

CefRefPtr<CefPrintSettings> CefEngine::getCefPrinterSettings(QPrinter* printer, CefRefPtr<CefPrintSettings> currentSettings) {
    currentSettings.get()->SetDeviceName(printer->printerName().toStdString());
    currentSettings.get()->SetCollate(printer->collateCopies());
    currentSettings.get()->SetOrientation(printer->orientation() == QPrinter::Landscape);
    currentSettings.get()->SetDPI(printer->resolution());
    currentSettings.get()->SetCopies(printer->numCopies());
    currentSettings.get()->SetColorModel(printer->colorMode() == QPrinter::Color ? COLOR_MODEL_COLOR : COLOR_MODEL_GRAY);
    currentSettings.get()->SetDuplexMode(printer->doubleSidedPrinting() ? DUPLEX_MODE_LONG_EDGE : DUPLEX_MODE_SIMPLEX);

    CefSize size;
    CefRect area;

    QRect paperSize = printer->paperRect();
    size.Set(paperSize.width(), paperSize.height());

    QPrinter::Margins margins = printer->margins();
    area.Set(margins.left, margins.top, paperSize.width() - margins.left - margins.right, paperSize.height() - margins.top - margins.bottom);

    currentSettings.get()->SetPrinterPrintableArea(size, area, true);

    return currentSettings;
}