The Tor Browser / file revision

 /* -*- Mode: c++; tab-width: 2; indent-tabs-mode: nil; -*- */

 /* This Source Code Form is subject to the terms of the Mozilla Public

  * License, v. 2.0. If a copy of the MPL was not distributed with this

  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

 #import <Cocoa/Cocoa.h>

 #include "nsStandaloneNativeMenu.h"

 #include "nsMenuUtilsX.h"

 #include "nsIDOMElement.h"

 #include "nsIMutationObserver.h"

 #include "nsGkAtoms.h"

 #include "nsObjCExceptions.h"

 NS_IMPL_ISUPPORTS(nsStandaloneNativeMenu, nsIMutationObserver, nsIStandaloneNativeMenu)

 nsStandaloneNativeMenu::nsStandaloneNativeMenu()

 : mMenu(nullptr)

 {

 }

 nsStandaloneNativeMenu::~nsStandaloneNativeMenu()

 {

   if (mMenu) delete mMenu;

 }

 NS_IMETHODIMP

 nsStandaloneNativeMenu::Init(nsIDOMElement * aDOMElement)

 {

   NS_ASSERTION(mMenu == nullptr, "nsNativeMenu::Init - mMenu not null!");

   nsresult rv;

   nsCOMPtr<nsIContent> content = do_QueryInterface(aDOMElement, &rv);

   NS_ENSURE_SUCCESS(rv, rv);

   nsIAtom * tag = content->Tag();

   if (!content->IsXUL() ||

       (tag != nsGkAtoms::menu && tag != nsGkAtoms::menupopup))

     return NS_ERROR_FAILURE;

   rv = nsMenuGroupOwnerX::Create(content);

   if (NS_FAILED(rv))

     return rv;

   mMenu = new nsMenuX();

   rv = mMenu->Create(this, this, content);

   if (NS_FAILED(rv)) {

     delete mMenu;

     mMenu = nullptr;

     return rv;

   }

   return NS_OK;

 }

 static void

 UpdateMenu(nsMenuX * aMenu)

 {

   aMenu->MenuOpened();

   aMenu->MenuClosed();

   uint32_t itemCount = aMenu->GetItemCount();

   for (uint32_t i = 0; i < itemCount; i++) {

     nsMenuObjectX * menuObject = aMenu->GetItemAt(i);

     if (menuObject->MenuObjectType() == eSubmenuObjectType) {

       UpdateMenu(static_cast<nsMenuX*>(menuObject));

     }

   }

 }

 NS_IMETHODIMP

 nsStandaloneNativeMenu::MenuWillOpen(bool * aResult)

 {

   NS_ASSERTION(mMenu != nullptr, "nsStandaloneNativeMenu::OnOpen - mMenu is null!");

   // Force an update on the mMenu by faking an open/close on all of

   // its submenus.

   UpdateMenu(mMenu);

   *aResult = true;

   return NS_OK;

 }

 NS_IMETHODIMP

 nsStandaloneNativeMenu::GetNativeMenu(void ** aVoidPointer)

 {

   if (mMenu) {

     *aVoidPointer = mMenu->NativeData();

     [[(NSObject *)(*aVoidPointer) retain] autorelease];

     return NS_OK;

   }  else {

     *aVoidPointer = nullptr;

     return NS_ERROR_NOT_INITIALIZED;

   }

 }

 static NSMenuItem *

 NativeMenuItemWithLocation(NSMenu * currentSubmenu, NSString * locationString)

 {

   NSArray * indexes = [locationString componentsSeparatedByString:@"|"];

   NSUInteger indexCount = [indexes count];

   if (indexCount == 0)

     return nil;

   for (NSUInteger i = 0; i < indexCount; i++) {

     NSInteger targetIndex = [[indexes objectAtIndex:i] integerValue];

     NSInteger itemCount = [currentSubmenu numberOfItems];

     if (targetIndex < itemCount) {

       NSMenuItem* menuItem = [currentSubmenu itemAtIndex:targetIndex];

       // If this is the last index, just return the menu item.

       if (i == (indexCount - 1))

         return menuItem;

       // If this is not the last index, find the submenu and keep going.

       if ([menuItem hasSubmenu])

         currentSubmenu = [menuItem submenu];

       else

         return nil;

     }

   }

   return nil;

 }

 NS_IMETHODIMP

 nsStandaloneNativeMenu::ActivateNativeMenuItemAt(const nsAString& indexString)

 {

   NS_OBJC_BEGIN_TRY_ABORT_BLOCK_NSRESULT;

   if (!mMenu)

     return NS_ERROR_NOT_INITIALIZED;

   NSString * locationString = [NSString stringWithCharacters:reinterpret_cast<const unichar*>(indexString.BeginReading())

                                                       length:indexString.Length()];

   NSMenu * menu = static_cast<NSMenu *> (mMenu->NativeData());

   NSMenuItem * item = NativeMenuItemWithLocation(menu, locationString);

   // We can't perform an action on an item with a submenu, that will raise

   // an obj-c exception.

   if (item && ![item hasSubmenu]) {

     NSMenu * parent = [item menu];

     if (parent) {

       // NSLog(@"Performing action for native menu item titled: %@\n",

       //       [[currentSubmenu itemAtIndex:targetIndex] title]);

       [parent performActionForItemAtIndex:[parent indexOfItem:item]];

       return NS_OK;

     }

   }

   return NS_ERROR_FAILURE;

   NS_OBJC_END_TRY_ABORT_BLOCK_NSRESULT;

 }

 NS_IMETHODIMP

 nsStandaloneNativeMenu::ForceUpdateNativeMenuAt(const nsAString& indexString)

 {

   if (!mMenu)

     return NS_ERROR_NOT_INITIALIZED;

   NSString* locationString = [NSString stringWithCharacters:reinterpret_cast<const unichar*>(indexString.BeginReading())

                                                      length:indexString.Length()];

   NSArray* indexes = [locationString componentsSeparatedByString:@"|"];

   unsigned int indexCount = [indexes count];

   if (indexCount == 0)

     return NS_OK;

   nsMenuX* currentMenu = mMenu;

   // now find the correct submenu

   for (unsigned int i = 1; currentMenu && i < indexCount; i++) {

     int targetIndex = [[indexes objectAtIndex:i] intValue];

     int visible = 0;

     uint32_t length = currentMenu->GetItemCount();

     for (unsigned int j = 0; j < length; j++) {

       nsMenuObjectX* targetMenu = currentMenu->GetItemAt(j);

       if (!targetMenu)

         return NS_OK;

       if (!nsMenuUtilsX::NodeIsHiddenOrCollapsed(targetMenu->Content())) {

         visible++;

         if (targetMenu->MenuObjectType() == eSubmenuObjectType && visible == (targetIndex + 1)) {

           currentMenu = static_cast<nsMenuX*>(targetMenu);

           // fake open/close to cause lazy update to happen

           currentMenu->MenuOpened();

           currentMenu->MenuClosed();

           break;

         }

       }

     }

   }

   return NS_OK;

 }