1 /*

     2  * Copyright (C) 2010 Apple Inc. All rights reserved.

     3  *

     4  * Redistribution and use in source and binary forms, with or without

     5  * modification, are permitted provided that the following conditions

     6  * are met:

     7  * 1. Redistributions of source code must retain the above copyright

     8  *    notice, this list of conditions and the following disclaimer.

     9  * 2. Redistributions in binary form must reproduce the above copyright

    10  *    notice, this list of conditions and the following disclaimer in the

    11  *    documentation and/or other materials provided with the distribution.

    12  *

    13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''

    14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,

    15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

    16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS

    17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR

    18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF

    19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS

    20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN

    21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)

    22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF

    23  * THE POSSIBILITY OF SUCH DAMAGE.

    24  */

    25 

    26 #include "InjectedBundle.h"

    27 

    28 #include "ActivateFonts.h"

    29 #include "InjectedBundlePage.h"

    30 #include <WebKit2/WKBundle.h>

    31 #include <WebKit2/WKBundlePage.h>

    32 #include <WebKit2/WKRetainPtr.h>

    33 #include <WebKit2/WKStringCF.h>

    34 #include <WebKit2/WebKit2.h>

    35 #include <wtf/RetainPtr.h>

    36 

    37 namespace WTR {

    38 

    39 InjectedBundle& InjectedBundle::shared()

    40 {

    41     static InjectedBundle& shared = *new InjectedBundle;

    42     return shared;

    43 }

    44 

    45 InjectedBundle::InjectedBundle()

    46     : m_bundle(0)

    47 {

    48 }

    49 

    50 void InjectedBundle::_didCreatePage(WKBundleRef bundle, WKBundlePageRef page, const void* clientInfo)

    51 {

    52     static_cast<InjectedBundle*>(const_cast<void*>(clientInfo))->didCreatePage(page);

    53 }

    54 

    55 void InjectedBundle::_willDestroyPage(WKBundleRef bundle, WKBundlePageRef page, const void* clientInfo)

    56 {

    57     static_cast<InjectedBundle*>(const_cast<void*>(clientInfo))->willDestroyPage(page);

    58 }

    59 

    60 void InjectedBundle::_didRecieveMessage(WKBundleRef bundle, WKStringRef message, const void *clientInfo)

    61 {

    62     static_cast<InjectedBundle*>(const_cast<void*>(clientInfo))->didRecieveMessage(message);

    63 }

    64 

    65 void InjectedBundle::initialize(WKBundleRef bundle)

    66 {

    67     m_bundle = bundle;

    68 

    69     WKBundleClient client = {

    70         0,

    71         this,

    72         _didCreatePage,

    73         _willDestroyPage,

    74         _didRecieveMessage

    75     };

    76     WKBundleSetClient(m_bundle, &client);

    77 

    78     activateFonts();

    79 }

    80 

    81 void InjectedBundle::done()

    82 {

    83     std::string output = m_outputStream.str();

    84     RetainPtr<CFStringRef> outputCFString(AdoptCF, CFStringCreateWithCString(0, output.c_str(), kCFStringEncodingUTF8));

    85     WKRetainPtr<WKStringRef> doneMessage(AdoptWK, WKStringCreateWithCFString(outputCFString.get()));

    86     WKBundlePostMessage(m_bundle, doneMessage.get());

    87 }

    88 

    89 void InjectedBundle::didCreatePage(WKBundlePageRef page)

    90 {

    91     // FIXME: we really need the main page ref to be sent over from the ui process

    92     m_mainPage = new InjectedBundlePage(page);

    93     m_pages.add(page, m_mainPage);

    94 }

    95 

    96 void InjectedBundle::willDestroyPage(WKBundlePageRef page)

    97 {

    98     delete m_pages.take(page);

    99 }

   100 

   101 void InjectedBundle::didRecieveMessage(WKStringRef message)

   102 {

   103     CFStringRef cfMessage = WKStringCopyCFString(0, message);

   104     if (CFEqual(cfMessage, CFSTR("BeginTest"))) {

   105         WKRetainPtr<WKStringRef> ackMessage(AdoptWK, WKStringCreateWithCFString(CFSTR("BeginTestAck")));

   106         WKBundlePostMessage(m_bundle, ackMessage.get());

   107 

   108         reset();

   109         return;

   110     }

   111 

   112     WKRetainPtr<WKStringRef> errorMessage(AdoptWK, WKStringCreateWithCFString(CFSTR("Error: Unknown.")));

   113     WKBundlePostMessage(m_bundle, errorMessage.get());

   114 }

   115 

   116 void InjectedBundle::reset()

   117 {

   118     m_outputStream.str("");

   119     m_layoutTestController = LayoutTestController::create(std::string(""));

   120 }

   121 

   122 } // namespace WTR