Implement contextmenu attribute with basic support of <menu>

Source/core/html/HTMLMenuElement.cpp

 /*
  * Copyright (C) 1999 Lars Knoll (knoll@kde.org)
  *           (C) 1999 Antti Koivisto (koivisto@kde.org)
  * Copyright (C) 2010 Apple Inc. All rights reserved.
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Library General Public
  * License as published by the Free Software Foundation; either
  * version 2 of the License, or (at your option) any later version.
  *
  * This library is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  * Library General Public License for more details.
  *
  * You should have received a copy of the GNU Library General Public License
  * along with this library; see the file COPYING.LIB.  If not, write to
  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  * Boston, MA 02110-1301, USA.
  *
  */
 
 #include "config.h"
+
 #include "core/html/HTMLMenuElement.h"
 
-#include "HTMLNames.h"
+#include "core/dom/ElementTraversal.h"
+#include "core/events/RelatedEvent.h"
+#include "core/html/HTMLMenuItemElement.h"
+#include "core/page/ContextMenuController.h"
+#include "core/page/Page.h"
 
 namespace WebCore {
 
 using namespace HTMLNames;
 
 inline HTMLMenuElement::HTMLMenuElement(Document& document)
     : HTMLElement(menuTag, document)
 {
     ScriptWrappable::init(this);
 }
 
 PassRefPtrWillBeRawPtr<HTMLMenuElement> HTMLMenuElement::create(Document& document)
 {
     return adoptRefWillBeRefCountedGarbageCollected(new HTMLMenuElement(document));
 }
 
+CustomContextMenuProvider& HTMLMenuElement::ensureMenuProvider()
+{
+    if (!m_menuProvider) {
+        ContextMenu menu;
+        populateContextMenuItems(this, menu);
+        m_menuProvider = CustomContextMenuProvider::create(this, menu.items());
+    }
+    return *m_menuProvider;
 }
+
+void HTMLMenuElement::populateContextMenuItems(HTMLMenuElement* menu, ContextMenu& contextMenu)
+{
+    Element* nextElement = ElementTraversal::firstWithin(*menu);
+    while (nextElement) {
+        if (nextElement->hasTagName(HTMLNames::menuTag)) {
+            ContextMenu subMenu;
+            populateContextMenuItems(toHTMLMenuElement(nextElement), subMenu);
+            contextMenu.appendItem(ContextMenuItem(SubmenuType, ContextMenuItemCustomTagNoAction, nextElement->fastGetAttribute(HTMLNames::labelAttr), &subMenu));
+            nextElement = ElementTraversal::nextSibling(*nextElement);
+        } else if (nextElement->hasTagName(HTMLNames::menuitemTag)) {
+            m_menuItems.append(toHTMLElement(nextElement));
+            contextMenu.appendItem(ContextMenuItem(ActionType, static_cast<ContextMenuAction>(ContextMenuItemBaseCustomTag + m_menuItems.size() - 1), nextElement->fastGetAttribute(HTMLNames::labelAttr)));
+            if (ContextMenuItemBaseCustomTag + m_menuItems.size() >= ContextMenuItemLastCustomTag)
+                break;
+            nextElement = ElementTraversal::next(*nextElement, menu);
+        }
+    }
+}
+
+void HTMLMenuElement::defaultEventHandler(Event* event)
+{
+    if (event->type() == EventTypeNames::show && event->isRelatedEvent()) {
+        m_subject = toHTMLElement(toRelatedEvent(event)->relatedTarget()->toNode());
+        ensureMenuProvider();
+        event->setDefaultHandled();
+    }
+}
+
+HTMLElement* HTMLMenuElement::menuItemAt(unsigned menuId)
+{
+    if (menuId < ContextMenuItemBaseCustomTag || menuId > ContextMenuItemLastCustomTag)
+        return 0;
+    int itemNumber = menuId - ContextMenuItemBaseCustomTag;
+    return m_menuItems[itemNumber];
+}
+
+}