Imported WebKit commit 7e538425aa020340619e927792f3d895061fb54b (http://svn.webkit.org/repository/webkit/trunk@116286)

1 files changed, 84 insertions, 0 deletions

diff --git a/Source/WebKit2/WebProcess/InjectedBundle/InjectedBundleDOMWindowExtension.cpp b/Source/WebKit2/WebProcess/InjectedBundle/InjectedBundleDOMWindowExtension.cpp
new file mode 100644
index 000000000..d448e84dc
--- /dev/null
+++ b/Source/WebKit2/WebProcess/InjectedBundle/InjectedBundleDOMWindowExtension.cpp
@@ -0,0 +1,84 @@
+/*
+ * Copyright (C) 2012 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. ``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
+ * 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.
+ */
+
+#include "config.h"
+#include "InjectedBundleDOMWindowExtension.h"
+
+#include "InjectedBundleScriptWorld.h"
+#include "WebFrame.h"
+#include "WebFrameLoaderClient.h"
+#include <WebCore/DOMWindowExtension.h>
+#include <WebCore/Frame.h>
+#include <wtf/HashMap.h>
+
+using namespace WebCore;
+
+namespace WebKit {
+
+typedef HashMap<WebCore::DOMWindowExtension*, InjectedBundleDOMWindowExtension*> ExtensionMap;
+static ExtensionMap& allExtensions()
+{
+    DEFINE_STATIC_LOCAL(ExtensionMap, map, ());
+    return map;
+}
+
+PassRefPtr<InjectedBundleDOMWindowExtension> InjectedBundleDOMWindowExtension::create(WebFrame* frame, InjectedBundleScriptWorld* world)
+{
+    return adoptRef(new InjectedBundleDOMWindowExtension(frame, world));
+}
+
+InjectedBundleDOMWindowExtension* InjectedBundleDOMWindowExtension::get(DOMWindowExtension* extension)
+{
+    ASSERT(allExtensions().contains(extension));
+    return allExtensions().get(extension);
+}
+
+InjectedBundleDOMWindowExtension::InjectedBundleDOMWindowExtension(WebFrame* frame, InjectedBundleScriptWorld* world)
+    : m_coreExtension(DOMWindowExtension::create(frame->coreFrame(), world->coreWorld()))
+{
+    allExtensions().add(m_coreExtension.get(), this);
+}
+
+InjectedBundleDOMWindowExtension::~InjectedBundleDOMWindowExtension()
+{
+    ASSERT(allExtensions().contains(m_coreExtension.get()));
+    allExtensions().remove(m_coreExtension.get());
+}
+
+WebFrame* InjectedBundleDOMWindowExtension::frame() const
+{
+    WebCore::Frame* frame = m_coreExtension->frame();
+    return frame ? static_cast<WebFrameLoaderClient*>(frame->loader()->client())->webFrame() : 0;
+}
+
+InjectedBundleScriptWorld* InjectedBundleDOMWindowExtension::world() const
+{
+    if (!m_world)
+        m_world = InjectedBundleScriptWorld::getOrCreate(m_coreExtension->world());
+        
+    return m_world.get();
+}
+
+} // namespace WebKit