-
Notifications
You must be signed in to change notification settings - Fork 0
/
YRSolcounterPref.m
92 lines (73 loc) · 3.19 KB
/
YRSolcounterPref.m
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
//
// YRSolcounterPref.m
// Solcounter
//
// Created by elliottcable on 14 768 ſ.
// Copyright 2010 yreality. All rights reserved.
//
#import "YRSolcounterPref.h"
// Some declarations for private CoreMenuExtra functions:
typedef struct _OpaqueMenuExtra OpaqueMenuExtra;
OSStatus CoreMenuExtraGetMenuExtra(CFStringRef identifier, OpaqueMenuExtra **menuExtraOut);
OSStatus CoreMenuExtraAddMenuExtra(CFURLRef path, SInt32 position, void *_0, void *_1, void *_2, void *_3);
OSStatus CoreMenuExtraRemoveMenuExtra(OpaqueMenuExtra *menuExtraIn, void *_0);
#define CMEGetExtra(identifier, menuExtraOut) \
CoreMenuExtraGetMenuExtra(identifier, menuExtraOut)//;
#define CMEAddExtra(path, position) \
CoreMenuExtraAddMenuExtra(path, position, NULL, NULL, NULL, NULL)//;
#define CMERemoveExtra(menuExtraIn) \
CoreMenuExtraRemoveMenuExtra(menuExtraIn, NULL)//;
// (From http://www.cocoadev.com/index.pl?CoreMenuExtra)
@implementation YRSolcounterPref
-(void) mainViewDidLoad {
NSString *extraPath = [[self bundle] pathForResource: @"Solcounter"
ofType: @"menu"
inDirectory: @""];
if ([self isExtraLoaded: extraPath])
[extraEnabled setState: NSOnState];
}
-(IBAction) toggleExtra: (id)sender {
NSString *extraPath = [[self bundle] pathForResource: @"Solcounter"
ofType: @"menu"
inDirectory: @""];
if (([sender state] == NSOnState) && ![self isExtraLoaded: extraPath]) {
[self loadExtra: extraPath];
if (![self isExtraLoaded: extraPath])
[sender setState: NSOffState];
} else //»
if (([sender state] == NSOffState) && [self isExtraLoaded: extraPath]) {
[self removeExtra: extraPath];
if ([self isExtraLoaded: extraPath])
[sender setState: NSOnState];
}
}
-(void) loadExtra: (NSString*)extraPath { int sleepCount;
NSURL *extraURL = [NSURL fileURLWithPath: extraPath];
CMEAddExtra((CFURLRef)extraURL, -1);
sleepCount = 0;
while (sleepCount < 1000000) {
if ([self isExtraLoaded: extraPath]) return;
usleep(sleepCount += 250000); }
NSString *crackerPath = [[self bundle] pathForResource: @"MenuCracker"
ofType: @"menu"
inDirectory: @""];
if (extraPath != crackerPath && ![self isExtraLoaded: extraPath])
[self loadExtra: crackerPath];
CMEAddExtra((CFURLRef)extraURL, -1);
sleepCount = 0;
while (sleepCount < 1000000) {
if ([self isExtraLoaded: extraPath]) return;
usleep(sleepCount += 250000); }
}
-(void) removeExtra: (NSString*)extraPath { OpaqueMenuExtra *extra;
NSString *extraID = [[NSBundle bundleWithPath: extraPath]
objectForInfoDictionaryKey: @"CFBundleIdentifier"];
if ((CMEGetExtra((CFStringRef)extraID, &extra) == 0) && extra)
CMERemoveExtra(extra);
}
-(BOOL) isExtraLoaded: (NSString*)extraPath { OpaqueMenuExtra *extra;
NSString *extraID = [[NSBundle bundleWithPath: extraPath]
objectForInfoDictionaryKey: @"CFBundleIdentifier"];
return CMEGetExtra((CFStringRef) extraID, &extra) == 0 && extra;
}
@end