Subscribe the output controller to notifications directly. Application delegate has...

[enjoyable.git] / NJOutputSwitchMouseMode.m

diff --git a/NJOutputSwitchMouseMode.m b/NJOutputSwitchMouseMode.m
index 3332a91..130a2a0 100644 (file)
--- a/NJOutputSwitchMouseMode.m
+++ b/NJOutputSwitchMouseMode.m
@@ -7,8 +7,8 @@
 
 #import "NJOutputSwitchMouseMode.h"
 
-#import "ApplicationController.h"
-#import "NJInputController.h"
+#import "EnjoyableApplicationDelegate.h"
+#import "NJDeviceController.h"
 
 @implementation NJOutputSwitchMouseMode
 
@@ -17,7 +17,7 @@
 }
 
 - (NSDictionary *)serialize {
-    return @{ @"type": @"switch mouse mode" };
+    return @{ @"type": self.class.serializationCode };
 }
 
 + (NJOutput *)outputDeserialize:(NSDictionary *)serialization
@@ -27,8 +27,8 @@
 - (void)trigger {
     // FIXME: It's hacky to get at the controller this way, but it's
     // also hacky to pass it. Shouldn't need to do either.
-    ApplicationController *ac = NSApplication.sharedApplication.delegate;
-    NJInputController *jc = ac.inputController;
+    EnjoyableApplicationDelegate *ac = NSApplication.sharedApplication.delegate;
+    NJDeviceController *jc = ac.inputController;
     jc.frontWindowOnly = !jc.frontWindowOnly;
 }