dom/html/HTMLMenuElement.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "mozilla/dom/HTMLMenuElement.h"
   8
   9 #include "mozilla/BasicEvents.h"
  10 #include "mozilla/EventDispatcher.h"
  11 #include "mozilla/dom/DocumentInlines.h"
  12 #include "mozilla/dom/HTMLMenuElementBinding.h"
  13 #include "mozilla/dom/HTMLMenuItemElement.h"
  14 #include "nsIMenuBuilder.h"
  15 #include "nsAttrValueInlines.h"
  16 #include "nsContentUtils.h"
  17 #include "nsIURI.h"
  18
  19 #define HTMLMENUBUILDER_CONTRACTID "@mozilla.org/content/html-menu-builder;1"
  20
  21 NS_IMPL_NS_NEW_HTML_ELEMENT(Menu)
  22
  23 namespace mozilla {
  24 namespace dom {
  25
  26 enum MenuType : uint8_t { MENU_TYPE_CONTEXT = 1, MENU_TYPE_TOOLBAR };
  27
  28 static const nsAttrValue::EnumTable kMenuTypeTable[] = {
  29     {"context", MENU_TYPE_CONTEXT},
  30     {"toolbar", MENU_TYPE_TOOLBAR},
  31     {nullptr, 0}};
  32
  33 static const nsAttrValue::EnumTable* kMenuDefaultType = &kMenuTypeTable[1];
  34
  35 enum SeparatorType { ST_TRUE_INIT = -1, ST_FALSE = 0, ST_TRUE = 1 };
  36
  37 HTMLMenuElement::HTMLMenuElement(
  38     already_AddRefed<mozilla::dom::NodeInfo>&& aNodeInfo)
  39     : nsGenericHTMLElement(std::move(aNodeInfo)), mType(MENU_TYPE_TOOLBAR) {}
  40
  41 HTMLMenuElement::~HTMLMenuElement() {}
  42
  43 NS_IMPL_ELEMENT_CLONE(HTMLMenuElement)
  44
  45 void HTMLMenuElement::SendShowEvent() {
  46   nsCOMPtr<Document> document = GetComposedDoc();
  47   if (!document) {
  48     return;
  49   }
  50
  51   WidgetEvent event(true, eShow);
  52   event.mFlags.mBubbles = false;
  53   event.mFlags.mCancelable = false;
  54
  55   RefPtr<nsPresContext> presContext = document->GetPresContext();
  56   if (!presContext) {
  57     return;
  58   }
  59
  60   nsEventStatus status = nsEventStatus_eIgnore;
  61   EventDispatcher::Dispatch(static_cast<nsIContent*>(this), presContext, &event,
  62                             nullptr, &status);
  63 }
  64
  65 already_AddRefed<nsIMenuBuilder> HTMLMenuElement::CreateBuilder() {
  66   if (mType != MENU_TYPE_CONTEXT) {
  67     return nullptr;
  68   }
  69
  70   nsCOMPtr<nsIMenuBuilder> builder =
  71       do_CreateInstance(HTMLMENUBUILDER_CONTRACTID);
  72   NS_WARNING_ASSERTION(builder, "No builder available");
  73   return builder.forget();
  74 }
  75
  76 void HTMLMenuElement::Build(nsIMenuBuilder* aBuilder) {
  77   if (!aBuilder) {
  78     return;
  79   }
  80
  81   BuildSubmenu(EmptyString(), this, aBuilder);
  82 }
  83
  84 nsresult HTMLMenuElement::AfterSetAttr(int32_t aNameSpaceID, nsAtom* aName,
  85                                        const nsAttrValue* aValue,
  86                                        const nsAttrValue* aOldValue,
  87                                        nsIPrincipal* aSubjectPrincipal,
  88                                        bool aNotify) {
  89   if (aNameSpaceID == kNameSpaceID_None && aName == nsGkAtoms::type) {
  90     if (aValue) {
  91       mType = aValue->GetEnumValue();
  92     } else {
  93       mType = kMenuDefaultType->value;
  94     }
  95   }
  96
  97   return nsGenericHTMLElement::AfterSetAttr(
  98       aNameSpaceID, aName, aValue, aOldValue, aSubjectPrincipal, aNotify);
  99 }
 100
 101 bool HTMLMenuElement::ParseAttribute(int32_t aNamespaceID, nsAtom* aAttribute,
 102                                      const nsAString& aValue,
 103                                      nsIPrincipal* aMaybeScriptedPrincipal,
 104                                      nsAttrValue& aResult) {
 105   if (aNamespaceID == kNameSpaceID_None && aAttribute == nsGkAtoms::type) {
 106     return aResult.ParseEnumValue(aValue, kMenuTypeTable, false,
 107                                   kMenuDefaultType);
 108   }
 109
 110   return nsGenericHTMLElement::ParseAttribute(aNamespaceID, aAttribute, aValue,
 111                                               aMaybeScriptedPrincipal, aResult);
 112 }
 113
 114 void HTMLMenuElement::BuildSubmenu(const nsAString& aLabel,
 115                                    nsIContent* aContent,
 116                                    nsIMenuBuilder* aBuilder) {
 117   aBuilder->OpenContainer(aLabel);
 118
 119   int8_t separator = ST_TRUE_INIT;
 120   TraverseContent(aContent, aBuilder, separator);
 121
 122   if (separator == ST_TRUE) {
 123     aBuilder->UndoAddSeparator();
 124   }
 125
 126   aBuilder->CloseContainer();
 127 }
 128
 129 // static
 130 bool HTMLMenuElement::CanLoadIcon(nsIContent* aContent,
 131                                   const nsAString& aIcon) {
 132   if (aIcon.IsEmpty()) {
 133     return false;
 134   }
 135
 136   Document* doc = aContent->OwnerDoc();
 137
 138   nsCOMPtr<nsIURI> uri;
 139   nsContentUtils::NewURIWithDocumentCharset(getter_AddRefs(uri), aIcon, doc,
 140                                             aContent->GetBaseURI());
 141
 142   if (!uri) {
 143     return false;
 144   }
 145
 146   return nsContentUtils::CanLoadImage(uri, aContent, doc,
 147                                       aContent->NodePrincipal());
 148 }
 149
 150 void HTMLMenuElement::TraverseContent(nsIContent* aContent,
 151                                       nsIMenuBuilder* aBuilder,
 152                                       int8_t& aSeparator) {
 153   nsCOMPtr<nsIContent> child;
 154   for (child = aContent->GetFirstChild(); child;
 155        child = child->GetNextSibling()) {
 156     nsGenericHTMLElement* element = nsGenericHTMLElement::FromNode(child);
 157     if (!element) {
 158       continue;
 159     }
 160
 161     if (child->IsHTMLElement(nsGkAtoms::menuitem)) {
 162       HTMLMenuItemElement* menuitem = HTMLMenuItemElement::FromNode(child);
 163
 164       if (menuitem->IsHidden()) {
 165         continue;
 166       }
 167
 168       nsAutoString label;
 169       menuitem->GetLabel(label);
 170       if (label.IsEmpty()) {
 171         continue;
 172       }
 173
 174       nsAutoString icon;
 175       menuitem->GetIcon(icon);
 176
 177       aBuilder->AddItemFor(menuitem, CanLoadIcon(child, icon));
 178
 179       aSeparator = ST_FALSE;
 180     } else if (child->IsHTMLElement(nsGkAtoms::hr)) {
 181       aBuilder->AddSeparator();
 182     } else if (child->IsHTMLElement(nsGkAtoms::menu) && !element->IsHidden()) {
 183       if (child->AsElement()->HasAttr(kNameSpaceID_None, nsGkAtoms::label)) {
 184         nsAutoString label;
 185         child->AsElement()->GetAttr(kNameSpaceID_None, nsGkAtoms::label, label);
 186
 187         BuildSubmenu(label, child, aBuilder);
 188
 189         aSeparator = ST_FALSE;
 190       } else {
 191         AddSeparator(aBuilder, aSeparator);
 192
 193         TraverseContent(child, aBuilder, aSeparator);
 194
 195         AddSeparator(aBuilder, aSeparator);
 196       }
 197     }
 198   }
 199 }
 200
 201 inline void HTMLMenuElement::AddSeparator(nsIMenuBuilder* aBuilder,
 202                                           int8_t& aSeparator) {
 203   if (aSeparator) {
 204     return;
 205   }
 206
 207   aBuilder->AddSeparator();
 208   aSeparator = ST_TRUE;
 209 }
 210
 211 JSObject* HTMLMenuElement::WrapNode(JSContext* aCx,
 212                                     JS::Handle<JSObject*> aGivenProto) {
 213   return HTMLMenuElement_Binding::Wrap(aCx, this, aGivenProto);
 214 }
 215
 216 }  // namespace dom
 217 }  // namespace mozilla