| /* |
| * 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(); |
| DeprecatedString encoded = KURL::encode_string(searchString.deprecatedString()); |
| encoded.replace(DeprecatedString("%20"), DeprecatedString("+")); |
| |
| String url("http://www.google.com/search?q="); |
| url.append(String(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(); |
| } |