1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
|
/* -*- 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_INHERITED(nsStandaloneNativeMenu, nsMenuGroupOwnerX,
nsIMutationObserver, nsIStandaloneNativeMenu)
nsStandaloneNativeMenu::nsStandaloneNativeMenu()
: mMenu(nullptr)
, mContainerStatusBarItem(nil)
{
}
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);
if (!content->IsAnyOfXULElements(nsGkAtoms::menu, 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;
}
mMenu->SetupIcon();
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;
}
void
nsStandaloneNativeMenu::IconUpdated()
{
if (mContainerStatusBarItem) {
[mContainerStatusBarItem setImage:[mMenu->NativeMenuItem() image]];
}
}
void
nsStandaloneNativeMenu::SetContainerStatusBarItem(NSStatusItem* aItem)
{
mContainerStatusBarItem = aItem;
IconUpdated();
}
|
