diff options
Diffstat (limited to 'WebKitTools/WebKitTestRunner/InjectedBundle/InjectedBundle.h')
-rw-r--r-- | WebKitTools/WebKitTestRunner/InjectedBundle/InjectedBundle.h | 78 |
1 files changed, 78 insertions, 0 deletions
diff --git a/WebKitTools/WebKitTestRunner/InjectedBundle/InjectedBundle.h b/WebKitTools/WebKitTestRunner/InjectedBundle/InjectedBundle.h new file mode 100644 index 0000000..33934cf --- /dev/null +++ b/WebKitTools/WebKitTestRunner/InjectedBundle/InjectedBundle.h @@ -0,0 +1,78 @@ +/* + * Copyright (C) 2010 Apple Inc. All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, + * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR + * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS + * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN + * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF + * THE POSSIBILITY OF SUCH DAMAGE. + */ + +#ifndef InjectedBundle_h +#define InjectedBundle_h + +#include "LayoutTestController.h" +#include <WebKit2/WKBase.h> +#include <WebKit2/WKBundleBase.h> +#include <wtf/HashMap.h> +#include <wtf/RefPtr.h> + +#include <sstream> + +namespace WTR { + +class InjectedBundlePage; + +class InjectedBundle { +public: + static InjectedBundle& shared(); + + // Initialize the InjectedBundle. + void initialize(WKBundleRef); + + void done(); + + LayoutTestController* layoutTestController() { return m_layoutTestController.get(); } + + std::ostringstream& os() { return m_outputStream; } + +private: + InjectedBundle(); + ~InjectedBundle(); + + static void _didCreatePage(WKBundleRef bundle, WKBundlePageRef page, const void* clientInfo); + static void _willDestroyPage(WKBundleRef bundle, WKBundlePageRef page, const void* clientInfo); + static void _didRecieveMessage(WKBundleRef bundle, WKStringRef message, const void *clientInfo); + + void didCreatePage(WKBundlePageRef page); + void willDestroyPage(WKBundlePageRef page); + void didRecieveMessage(WKStringRef message); + + void reset(); + + WKBundleRef m_bundle; + HashMap<WKBundlePageRef, InjectedBundlePage*> m_pages; + + RefPtr<LayoutTestController> m_layoutTestController; + + std::ostringstream m_outputStream; +}; + +} // namespace WTR + +#endif // InjectedBundle_h |