1/*
2 * Copyright (C) 2006, 2007 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 COMPUTER, 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 COMPUTER, 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 "WebContextMenuClient.h"
28
29#include "WebElementPropertyBag.h"
30#include "WebLocalizableStrings.h"
31#include "WebView.h"
32
33#include <WebCore/ContextMenu.h>
34#include <WebCore/Event.h>
35#include <WebCore/Frame.h>
36#include <WebCore/FrameLoader.h>
37#include <WebCore/FrameLoadRequest.h>
38#include <WebCore/Page.h>
39#include <WebCore/ResourceRequest.h>
40#include <WebCore/NotImplemented.h>
41
42#include <tchar.h>
43
44using namespace WebCore;
45
46WebContextMenuClient::WebContextMenuClient(WebView* webView)
47    : m_webView(webView)
48{
49}
50
51void WebContextMenuClient::contextMenuDestroyed()
52{
53    delete this;
54}
55
56static bool isPreInspectElementTagSafari(IWebUIDelegate* uiDelegate)
57{
58    if (!uiDelegate)
59        return false;
60
61    TCHAR modulePath[MAX_PATH];
62    DWORD length = ::GetModuleFileName(0, modulePath, _countof(modulePath));
63    if (!length)
64        return false;
65
66    return String(modulePath, length).endsWith("Safari.exe", false);
67}
68
69static HMENU fixMenuReceivedFromOldSafari(IWebUIDelegate* uiDelegate, ContextMenu* originalMenu, HMENU menuFromClient)
70{
71    ASSERT_ARG(originalMenu, originalMenu);
72    if (!isPreInspectElementTagSafari(uiDelegate))
73        return menuFromClient;
74
75    int count = ::GetMenuItemCount(originalMenu->platformDescription());
76    if (count < 1)
77        return menuFromClient;
78
79    if (::GetMenuItemID(originalMenu->platformDescription(), count - 1) != WebMenuItemTagInspectElement)
80        return menuFromClient;
81
82    count = ::GetMenuItemCount(menuFromClient);
83    if (count < 1)
84        return menuFromClient;
85
86    if (::GetMenuItemID(menuFromClient, count - 1) == WebMenuItemTagInspectElement)
87        return menuFromClient;
88
89    originalMenu->setPlatformDescription(menuFromClient);
90    originalMenu->addInspectElementItem();
91    return originalMenu->platformDescription();
92}
93
94HMENU WebContextMenuClient::getCustomMenuFromDefaultItems(ContextMenu* menu)
95{
96    COMPtr<IWebUIDelegate> uiDelegate;
97    if (FAILED(m_webView->uiDelegate(&uiDelegate)))
98        return menu->platformDescription();
99
100    ASSERT(uiDelegate);
101
102    HMENU newMenu = 0;
103    COMPtr<WebElementPropertyBag> propertyBag;
104    propertyBag.adoptRef(WebElementPropertyBag::createInstance(menu->hitTestResult()));
105    // FIXME: We need to decide whether to do the default before calling this delegate method
106    if (FAILED(uiDelegate->contextMenuItemsForElement(m_webView, propertyBag.get(), (OLE_HANDLE)(ULONG64)menu->platformDescription(), (OLE_HANDLE*)&newMenu)))
107        return menu->platformDescription();
108    return fixMenuReceivedFromOldSafari(uiDelegate.get(), menu, newMenu);
109}
110
111void WebContextMenuClient::contextMenuItemSelected(ContextMenuItem* item, const ContextMenu* parentMenu)
112{
113    ASSERT(item->type() == ActionType || item->type() == CheckableActionType);
114
115    COMPtr<IWebUIDelegate> uiDelegate;
116    if (FAILED(m_webView->uiDelegate(&uiDelegate)))
117        return;
118
119    ASSERT(uiDelegate);
120
121    COMPtr<WebElementPropertyBag> propertyBag;
122    propertyBag.adoptRef(WebElementPropertyBag::createInstance(parentMenu->hitTestResult()));
123
124    uiDelegate->contextMenuItemSelected(m_webView, item->releasePlatformDescription(), propertyBag.get());
125}
126
127void WebContextMenuClient::downloadURL(const KURL& url)
128{
129    m_webView->downloadURL(url);
130}
131
132void WebContextMenuClient::searchWithGoogle(const Frame* frame)
133{
134    String searchString = frame->selectedText();
135    searchString.stripWhiteSpace();
136    String encoded = encodeWithURLEscapeSequences(searchString);
137    encoded.replace("%20", "+");
138
139    String url("http://www.google.com/search?q=");
140    url.append(encoded);
141    url.append("&ie=UTF-8&oe=UTF-8");
142
143    ResourceRequest request = ResourceRequest(url);
144    if (Page* page = frame->page())
145        page->mainFrame()->loader()->urlSelected(request, String(), 0, false, false, true, SendReferrer);
146}
147
148void WebContextMenuClient::lookUpInDictionary(Frame*)
149{
150    notImplemented();
151}
152
153void WebContextMenuClient::speak(const String&)
154{
155    notImplemented();
156}
157
158void WebContextMenuClient::stopSpeaking()
159{
160    notImplemented();
161}
162
163bool WebContextMenuClient::isSpeaking()
164{
165    notImplemented();
166    return false;
167}
168