1 files changed, 101 insertions, 0 deletions
diff --git a/Framework/MASShortcutMonitor.m b/Framework/MASShortcutMonitor.m
new file mode 100644
index 0000000..cb89ce1
--- /dev/null
+++ b/Framework/MASShortcutMonitor.m
@@ -0,0 +1,101 @@
+#import "MASShortcutMonitor.h"
+#import "MASHotKey.h"
+
+@interface MASShortcutMonitor ()
+@property(assign) EventHandlerRef eventHandlerRef;
+@property(strong) NSMutableDictionary *hotKeys;
+@end
+
+static OSStatus MASCarbonEventCallback(EventHandlerCallRef, EventRef, void*);
+
+@implementation MASShortcutMonitor
+
+#pragma mark Initialization
+
+- (instancetype) init
+{
+    self = [super init];
+    [self setHotKeys:[NSMutableDictionary dictionary]];
+    EventTypeSpec hotKeyPressedSpec = { .eventClass = kEventClassKeyboard, .eventKind = kEventHotKeyPressed };
+    OSStatus status = InstallEventHandler(GetEventDispatcherTarget(), MASCarbonEventCallback,
+        1, &hotKeyPressedSpec, (__bridge void*)self, &_eventHandlerRef);
+    if (status != noErr) {
+        return nil;
+    }
+    return self;
+}
+
+- (void) dealloc
+{
+    if (_eventHandlerRef) {
+        RemoveEventHandler(_eventHandlerRef);
+        _eventHandlerRef = NULL;
+    }
+}
+
++ (instancetype) sharedMonitor
+{
+    static dispatch_once_t once;
+    static MASShortcutMonitor *sharedInstance;
+    dispatch_once(&once, ^{
+        sharedInstance = [[self alloc] init];
+    });
+    return sharedInstance;
+}
+
+#pragma mark Registration
+
+- (void) registerShortcut: (MASShortcut*) shortcut withAction: (dispatch_block_t) action
+{
+    MASHotKey *hotKey = [MASHotKey registeredHotKeyWithShortcut:shortcut];
+    [hotKey setAction:action];
+    [_hotKeys setObject:hotKey forKey:shortcut];
+}
+
+- (void) unregisterShortcut: (MASShortcut*) shortcut
+{
+    [_hotKeys removeObjectForKey:shortcut];
+}
+
+- (void) unregisterAllShortcuts
+{
+    [_hotKeys removeAllObjects];
+}
+
+- (BOOL) isShortcutRegistered: (MASShortcut*) shortcut
+{
+    return !![_hotKeys objectForKey:shortcut];
+}
+
+#pragma mark Event Handling
+
+- (void) handleEvent: (EventRef) event
+{
+    if (GetEventClass(event) != kEventClassKeyboard) {
+        return;
+    }
+
+    EventHotKeyID hotKeyID;
+    OSStatus status = GetEventParameter(event, kEventParamDirectObject, typeEventHotKeyID, NULL, sizeof(hotKeyID), NULL, &hotKeyID);
+    if (status != noErr || hotKeyID.signature != MASHotKeySignature) {
+        return;
+    }
+
+    [_hotKeys enumerateKeysAndObjectsUsingBlock:^(MASShortcut *shortcut, MASHotKey *hotKey, BOOL *stop) {
+        if (hotKeyID.id == [hotKey carbonID]) {
+            if ([hotKey action]) {
+                dispatch_async(dispatch_get_main_queue(), [hotKey action]);
+            }
+            *stop = YES;
+        }
+    }];
+}
+
+@end
+
+static OSStatus MASCarbonEventCallback(EventHandlerCallRef _, EventRef event, void *context)
+{
+    MASShortcutMonitor *dispatcher = (__bridge id)context;
+    [dispatcher handleEvent:event];
+    return noErr;
+}