/*
 * Copyright (C) 2013, 2017 Igalia S.L.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library 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
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public License
 * along with this library; see the file COPYING.LIB.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 */

#include "config.h"
#include "WebKitInjectedBundleClient.h"

#include "APIInjectedBundleClient.h"
#include "WebImage.h"
#include "WebKitPrivate.h"
#include "WebKitURIRequestPrivate.h"
#include "WebKitURIResponsePrivate.h"
#include "WebKitWebContextPrivate.h"
#include "WebKitWebResourcePrivate.h"
#include "WebKitWebViewPrivate.h"
#include <wtf/glib/GUniquePtr.h>

using namespace WebKit;
using namespace WebCore;

class WebKitInjectedBundleClient final : public API::InjectedBundleClient {
public:
    explicit WebKitInjectedBundleClient(WebKitWebContext* webContext)
        : m_webContext(webContext)
    {
    }

private:
    static void didReceiveWebViewMessageFromInjectedBundle(WebKitWebView* webView, const char* messageName, API::Dictionary& message)
    {
        if (g_str_equal(messageName, "DidInitiateLoadForResource")) {
            WebFrameProxy* frame = static_cast<WebFrameProxy*>(message.get(String::fromUTF8("Frame")));
            if (!frame)
                return;

            API::UInt64* resourceIdentifier = static_cast<API::UInt64*>(message.get(String::fromUTF8("Identifier")));
            if (!resourceIdentifier)
                return;

            API::URLRequest* webRequest = static_cast<API::URLRequest*>(message.get(String::fromUTF8("Request")));
            if (!webRequest)
                return;

            GRefPtr<WebKitURIRequest> request = adoptGRef(webkitURIRequestCreateForResourceRequest(webRequest->resourceRequest()));
            webkitWebViewResourceLoadStarted(webView, *frame, resourceIdentifier->value(), request.get());
        } else if (g_str_equal(messageName, "DidSendRequestForResource")) {
            API::UInt64* resourceIdentifier = static_cast<API::UInt64*>(message.get(String::fromUTF8("Identifier")));
            if (!resourceIdentifier)
                return;

            GRefPtr<WebKitWebResource> resource = webkitWebViewGetLoadingWebResource(webView, resourceIdentifier->value());
            if (!resource)
                return;

            API::URLRequest* webRequest = static_cast<API::URLRequest*>(message.get(String::fromUTF8("Request")));
            if (!webRequest)
                return;

            GRefPtr<WebKitURIRequest> request = adoptGRef(webkitURIRequestCreateForResourceRequest(webRequest->resourceRequest()));
            API::URLResponse* webRedirectResponse = static_cast<API::URLResponse*>(message.get(String::fromUTF8("RedirectResponse")));
            GRefPtr<WebKitURIResponse> redirectResponse = webRedirectResponse ? adoptGRef(webkitURIResponseCreateForResourceResponse(webRedirectResponse->resourceResponse())) : nullptr;
            webkitWebResourceSentRequest(resource.get(), request.get(), redirectResponse.get());
        } else if (g_str_equal(messageName, "DidReceiveResponseForResource")) {
            API::UInt64* resourceIdentifier = static_cast<API::UInt64*>(message.get(String::fromUTF8("Identifier")));
            if (!resourceIdentifier)
                return;

            GRefPtr<WebKitWebResource> resource = webkitWebViewGetLoadingWebResource(webView, resourceIdentifier->value());
            if (!resource)
                return;

            API::URLResponse* webResponse = static_cast<API::URLResponse*>(message.get(String::fromUTF8("Response")));
            if (!webResponse)
                return;

            GRefPtr<WebKitURIResponse> response = adoptGRef(webkitURIResponseCreateForResourceResponse(webResponse->resourceResponse()));
            webkitWebResourceSetResponse(resource.get(), response.get());
        } else if (g_str_equal(messageName, "DidReceiveContentLengthForResource")) {
            API::UInt64* resourceIdentifier = static_cast<API::UInt64*>(message.get(String::fromUTF8("Identifier")));
            if (!resourceIdentifier)
                return;

            GRefPtr<WebKitWebResource> resource = webkitWebViewGetLoadingWebResource(webView, resourceIdentifier->value());
            if (!resource)
                return;

            API::UInt64* contentLength = static_cast<API::UInt64*>(message.get(String::fromUTF8("ContentLength")));
            if (!contentLength)
                return;

            webkitWebResourceNotifyProgress(resource.get(), contentLength->value());
        } else if (g_str_equal(messageName, "DidFinishLoadForResource")) {
            API::UInt64* resourceIdentifier = static_cast<API::UInt64*>(message.get(String::fromUTF8("Identifier")));
            if (!resourceIdentifier)
                return;

            GRefPtr<WebKitWebResource> resource = webkitWebViewGetLoadingWebResource(webView, resourceIdentifier->value());
            if (!resource)
                return;

            webkitWebResourceFinished(resource.get());
            webkitWebViewRemoveLoadingWebResource(webView, resourceIdentifier->value());
        } else if (g_str_equal(messageName, "DidFailLoadForResource")) {
            API::UInt64* resourceIdentifier = static_cast<API::UInt64*>(message.get(String::fromUTF8("Identifier")));
            if (!resourceIdentifier)
                return;

            GRefPtr<WebKitWebResource> resource = webkitWebViewGetLoadingWebResource(webView, resourceIdentifier->value());
            if (!resource)
                return;

            API::Error* webError = static_cast<API::Error*>(message.get(String::fromUTF8("Error")));
            if (!webError)
                return;

            const ResourceError& platformError = webError->platformError();
            GUniquePtr<GError> resourceError(g_error_new_literal(g_quark_from_string(platformError.domain().utf8().data()),
                toWebKitError(platformError.errorCode()), platformError.localizedDescription().utf8().data()));
            if (platformError.tlsErrors())
                webkitWebResourceFailedWithTLSErrors(resource.get(), static_cast<GTlsCertificateFlags>(platformError.tlsErrors()), platformError.certificate());
            else
                webkitWebResourceFailed(resource.get(), resourceError.get());

            webkitWebViewRemoveLoadingWebResource(webView, resourceIdentifier->value());
#if PLATFORM(GTK)
        } else if (g_str_equal(messageName, "DidGetSnapshot")) {
            API::UInt64* callbackID = static_cast<API::UInt64*>(message.get("CallbackID"));
            if (!callbackID)
                return;

            WebImage* image = static_cast<WebImage*>(message.get("Snapshot"));
            webKitWebViewDidReceiveSnapshot(webView, callbackID->value(), image);
#endif
        }
    }

    void didReceiveMessageFromInjectedBundle(WebProcessPool&, const String& messageName, API::Object* messageBody) override
    {
        if (messageBody->type() != API::Object::Type::Dictionary)
            return;

        API::Dictionary& message = *static_cast<API::Dictionary*>(messageBody);

        CString messageNameUTF8 = messageName.utf8();
        if (g_str_has_prefix(messageNameUTF8.data(), "WebPage.")) {
            WebPageProxy* page = static_cast<WebPageProxy*>(message.get(String::fromUTF8("Page")));
            WebKitWebView* webView = webkitWebContextGetWebViewForPage(m_webContext, page);
            if (!webView)
                return;

            didReceiveWebViewMessageFromInjectedBundle(webView, messageNameUTF8.data() + strlen("WebPage."), message);
        }
    }

    RefPtr<API::Object> getInjectedBundleInitializationUserData(WebProcessPool&) override
    {
        GRefPtr<GVariant> data = webkitWebContextInitializeWebExtensions(m_webContext);
        GUniquePtr<gchar> dataString(g_variant_print(data.get(), TRUE));
        return API::String::create(String::fromUTF8(dataString.get()));
    }

    WebKitWebContext* m_webContext;
};

void attachInjectedBundleClientToContext(WebKitWebContext* webContext)
{
    webkitWebContextGetProcessPool(webContext).setInjectedBundleClient(makeUnique<WebKitInjectedBundleClient>(webContext));
}