widget/cocoa/nsSystemStatusBarCocoa.mm

   1 /* -*- Mode: c++; tab-width: 2; indent-tabs-mode: nil; -*- */
   2 /* This Source Code Form is subject to the terms of the Mozilla Public
   3  * License, v. 2.0. If a copy of the MPL was not distributed with this
   4  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   5
   6 #import <Cocoa/Cocoa.h>
   7
   8 #include "nsComponentManagerUtils.h"
   9 #include "nsSystemStatusBarCocoa.h"
  10 #include "NativeMenuMac.h"
  11 #include "nsObjCExceptions.h"
  12 #include "mozilla/dom/Element.h"
  13
  14 using mozilla::dom::Element;
  15 using mozilla::widget::NativeMenuMac;
  16
  17 NS_IMPL_ISUPPORTS(nsSystemStatusBarCocoa, nsISystemStatusBar)
  18
  19 NS_IMETHODIMP
  20 nsSystemStatusBarCocoa::AddItem(Element* aElement) {
  21   if (!aElement->IsAnyOfXULElements(nsGkAtoms::menu, nsGkAtoms::menupopup)) {
  22     return NS_ERROR_FAILURE;
  23   }
  24
  25   RefPtr<NativeMenuMac> menu = new NativeMenuMac(aElement);
  26
  27   nsCOMPtr<nsISupports> keyPtr = aElement;
  28   mItems.InsertOrUpdate(keyPtr, mozilla::MakeUnique<StatusItem>(menu));
  29
  30   return NS_OK;
  31 }
  32
  33 NS_IMETHODIMP
  34 nsSystemStatusBarCocoa::RemoveItem(Element* aElement) {
  35   mItems.Remove(aElement);
  36   return NS_OK;
  37 }
  38
  39 nsSystemStatusBarCocoa::StatusItem::StatusItem(NativeMenuMac* aMenu) : mMenu(aMenu) {
  40   NS_OBJC_BEGIN_TRY_ABORT_BLOCK;
  41
  42   MOZ_COUNT_CTOR(nsSystemStatusBarCocoa::StatusItem);
  43
  44   mStatusItem =
  45       [[NSStatusBar.systemStatusBar statusItemWithLength:NSSquareStatusItemLength] retain];
  46   mStatusItem.menu = mMenu->NativeNSMenu();
  47   mStatusItem.highlightMode = YES;
  48
  49   // We want the status item to get its image from menu item that mMenu was
  50   // initialized with. Icon loads are asynchronous, so we need to let the menu
  51   // know about the item so that it can update its icon as soon as it has
  52   // loaded.
  53   mMenu->SetContainerStatusBarItem(mStatusItem);
  54
  55   NS_OBJC_END_TRY_ABORT_BLOCK;
  56 }
  57
  58 nsSystemStatusBarCocoa::StatusItem::~StatusItem() {
  59   NS_OBJC_BEGIN_TRY_ABORT_BLOCK;
  60
  61   mMenu->SetContainerStatusBarItem(nil);
  62   [NSStatusBar.systemStatusBar removeStatusItem:mStatusItem];
  63   [mStatusItem release];
  64   mStatusItem = nil;
  65
  66   MOZ_COUNT_DTOR(nsSystemStatusBarCocoa::StatusItem);
  67
  68   NS_OBJC_END_TRY_ABORT_BLOCK;
  69 }