1 files changed, 179 insertions, 0 deletions
diff --git a/WebKit/win/WebCoreSupport/WebContextMenuClient.cpp b/WebKit/win/WebCoreSupport/WebContextMenuClient.cpp
new file mode 100644
index 0000000..349d1b3
--- /dev/null
+++ b/WebKit/win/WebCoreSupport/WebContextMenuClient.cpp
@@ -0,0 +1,179 @@
+/*
+ * Copyright (C) 2006, 2007 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 COMPUTER, 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 COMPUTER, 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 "WebContextMenuClient.h"
+
+#include "WebDownload.h"
+#include "WebElementPropertyBag.h"
+#include "WebLocalizableStrings.h"
+#include "WebView.h"
+
+#pragma warning(push, 0)
+#include <WebCore/ContextMenu.h>
+#include <WebCore/FrameLoader.h>
+#include <WebCore/FrameLoadRequest.h>
+#include <WebCore/Page.h>
+#include <WebCore/ResourceRequest.h>
+#include <WebCore/NotImplemented.h>
+#pragma warning(pop)
+
+#include <tchar.h>
+
+using namespace WebCore;
+
+WebContextMenuClient::WebContextMenuClient(WebView* webView)
+    : m_webView(webView)
+{
+}
+
+void WebContextMenuClient::contextMenuDestroyed()
+{
+    delete this;
+}
+
+static bool isPreInspectElementTagSafari(IWebUIDelegate* uiDelegate)
+{
+    if (!uiDelegate)
+        return false;
+
+    // We assume anyone who implements IWebUIDelegate2 also knows about the Inspect Element item.
+    COMPtr<IWebUIDelegate2> uiDelegate2;
+    if (SUCCEEDED(uiDelegate->QueryInterface(IID_IWebUIDelegate2, (void**)&uiDelegate2)))
+        return false;
+
+    TCHAR modulePath[MAX_PATH];
+    DWORD length = ::GetModuleFileName(0, modulePath, _countof(modulePath));
+    if (!length)
+        return false;
+
+    return String(modulePath, length).endsWith("Safari.exe", false);
+}
+
+static HMENU fixMenuReceivedFromOldSafari(IWebUIDelegate* uiDelegate, ContextMenu* originalMenu, HMENU menuFromClient)
+{
+    ASSERT_ARG(originalMenu, originalMenu);
+    if (!isPreInspectElementTagSafari(uiDelegate))
+        return menuFromClient;
+
+    int count = ::GetMenuItemCount(originalMenu->platformDescription());
+    if (count < 1)
+        return menuFromClient;
+
+    if (::GetMenuItemID(originalMenu->platformDescription(), count - 1) != WebMenuItemTagInspectElement)
+        return menuFromClient;
+
+    count = ::GetMenuItemCount(menuFromClient);
+    if (count < 1)
+        return menuFromClient;
+
+    if (::GetMenuItemID(menuFromClient, count - 1) == WebMenuItemTagInspectElement)
+        return menuFromClient;
+
+    originalMenu->setPlatformDescription(menuFromClient);
+    originalMenu->addInspectElementItem();
+    return originalMenu->platformDescription();
+}
+
+HMENU WebContextMenuClient::getCustomMenuFromDefaultItems(ContextMenu* menu)
+{
+    COMPtr<IWebUIDelegate> uiDelegate;
+    if (FAILED(m_webView->uiDelegate(&uiDelegate)))
+        return menu->platformDescription();
+
+    ASSERT(uiDelegate);
+
+    HMENU newMenu = 0;
+    COMPtr<WebElementPropertyBag> propertyBag;
+    propertyBag.adoptRef(WebElementPropertyBag::createInstance(menu->hitTestResult()));
+    // FIXME: We need to decide whether to do the default before calling this delegate method
+    if (FAILED(uiDelegate->contextMenuItemsForElement(m_webView, propertyBag.get(), (OLE_HANDLE)(ULONG64)menu->platformDescription(), (OLE_HANDLE*)&newMenu)))
+        return menu->platformDescription();
+    return fixMenuReceivedFromOldSafari(uiDelegate.get(), menu, newMenu);
+}
+
+void WebContextMenuClient::contextMenuItemSelected(ContextMenuItem* item, const ContextMenu* parentMenu)
+{
+    ASSERT(item->type() == ActionType || item->type() == CheckableActionType);
+
+    COMPtr<IWebUIDelegate> uiDelegate;
+    if (FAILED(m_webView->uiDelegate(&uiDelegate)))
+        return;
+
+    ASSERT(uiDelegate);
+
+    COMPtr<WebElementPropertyBag> propertyBag;
+    propertyBag.adoptRef(WebElementPropertyBag::createInstance(parentMenu->hitTestResult()));
+            
+    uiDelegate->contextMenuItemSelected(m_webView, item->releasePlatformDescription(), propertyBag.get());
+}
+
+void WebContextMenuClient::downloadURL(const KURL& url)
+{
+    COMPtr<IWebDownloadDelegate> downloadDelegate;
+    if (FAILED(m_webView->downloadDelegate(&downloadDelegate))) {
+        // If the WebView doesn't successfully provide a download delegate we'll pass a null one
+        // into the WebDownload - which may or may not decide to use a DefaultDownloadDelegate
+        LOG_ERROR("Failed to get downloadDelegate from WebView");
+        downloadDelegate = 0;
+    }
+
+    // Its the delegate's job to ref the WebDownload to keep it alive - otherwise it will be destroyed
+    // when this method returns
+    COMPtr<WebDownload> download;
+    download.adoptRef(WebDownload::createInstance(url, downloadDelegate.get()));
+    download->start();
+}
+
+void WebContextMenuClient::searchWithGoogle(const Frame* frame)
+{
+    String searchString = frame->selectedText();
+    searchString.stripWhiteSpace();
+    String encoded = encodeWithURLEscapeSequences(searchString);
+    encoded.replace("%20", "+");
+    
+    String url("http://www.google.com/search?q=");
+    url.append(encoded);
+    url.append("&ie=UTF-8&oe=UTF-8");
+
+    ResourceRequest request = ResourceRequest(url);
+    if (Page* page = frame->page())
+        page->mainFrame()->loader()->urlSelected(FrameLoadRequest(request), 0, false, true);
+}
+
+void WebContextMenuClient::lookUpInDictionary(Frame*)
+{
+    notImplemented();
+}
+
+void WebContextMenuClient::speak(const String&)
+{
+    notImplemented();
+}
+
+void WebContextMenuClient::stopSpeaking()
+{
+    notImplemented();
+}