1/*
2 * Copyright (C) 2012 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. ``AS IS'' AND ANY
14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#include "config.h"
27#include "InjectedBundleDOMWindowExtension.h"
28
29#include "InjectedBundleScriptWorld.h"
30#include "WebFrame.h"
31#include "WebFrameLoaderClient.h"
32#include <WebCore/DOMWindowExtension.h>
33#include <WebCore/DOMWrapperWorld.h>
34#include <WebCore/Frame.h>
35#include <WebCore/FrameLoader.h>
36#include <wtf/HashMap.h>
37#include <wtf/NeverDestroyed.h>
38
39using namespace WebCore;
40
41namespace WebKit {
42
43typedef HashMap<WebCore::DOMWindowExtension*, InjectedBundleDOMWindowExtension*> ExtensionMap;
44static ExtensionMap& allExtensions()
45{
46    static NeverDestroyed<ExtensionMap> map;
47    return map;
48}
49
50PassRefPtr<InjectedBundleDOMWindowExtension> InjectedBundleDOMWindowExtension::create(WebFrame* frame, InjectedBundleScriptWorld* world)
51{
52    return adoptRef(new InjectedBundleDOMWindowExtension(frame, world));
53}
54
55InjectedBundleDOMWindowExtension* InjectedBundleDOMWindowExtension::get(DOMWindowExtension* extension)
56{
57    ASSERT(allExtensions().contains(extension));
58    return allExtensions().get(extension);
59}
60
61InjectedBundleDOMWindowExtension::InjectedBundleDOMWindowExtension(WebFrame* frame, InjectedBundleScriptWorld* world)
62    : m_coreExtension(DOMWindowExtension::create(frame->coreFrame(), world->coreWorld()))
63{
64    allExtensions().add(m_coreExtension.get(), this);
65}
66
67InjectedBundleDOMWindowExtension::~InjectedBundleDOMWindowExtension()
68{
69    ASSERT(allExtensions().contains(m_coreExtension.get()));
70    allExtensions().remove(m_coreExtension.get());
71}
72
73WebFrame* InjectedBundleDOMWindowExtension::frame() const
74{
75    Frame* frame = m_coreExtension->frame();
76    if (!frame)
77        return nullptr;
78
79    return WebFrame::fromCoreFrame(*frame);
80}
81
82InjectedBundleScriptWorld* InjectedBundleDOMWindowExtension::world() const
83{
84    if (!m_world)
85        m_world = InjectedBundleScriptWorld::getOrCreate(m_coreExtension->world());
86
87    return m_world.get();
88}
89
90} // namespace WebKit
91