ABC 입력기 제한 기능 추가

kiding · kiding · commit d5a38cb8084c · 2023-09-12T21:25:38.000+09:00
암호 입력 필드에 포커스될 때 입력기가 ABC로 변경되는 문제 대응 (#20)
diff --git a/SokIM/AppDelegate.swift b/SokIM/AppDelegate.swift
@@ -51,6 +51,14 @@ class AppDelegate: NSObject, NSApplicationDelegate {
         var eventSpec = EventTypeSpec(eventClass: OSType(kEventClassKeyboard), eventKind: UInt32(kEventHotKeyPressed))
         InstallEventHandler(GetApplicationEventTarget(), eventHotKeyHandler, 1, &eventSpec, nil, &eventHandlerRef)
         registerEventHotKey(Preferences.rotateShortcut)
+
+        // 입력기가 변경되는 시점에 ABC 입력기 제한 로직 실행
+        NotificationCenter.default.addObserver(
+            self,
+            selector: #selector(suppressABC),
+            name: NSTextInputContext.keyboardSelectionDidChangeNotification,
+            object: nil
+        )
     }
 
     func applicationWillTerminate(_ aNotification: Notification) {
@@ -274,5 +282,54 @@ class AppDelegate: NSObject, NSApplicationDelegate {
             break
         }
     }
+
+    @objc private func suppressABC(_ aNotification: Notification) {
+        debug("\(String(describing: aNotification))")
+
+        guard Preferences.suppressABC == true else { return }
+
+        let current = TISCopyCurrentKeyboardInputSource()?.takeRetainedValue()
+        guard let current = current else {
+            warning("TISCopyCurrentKeyboardInputSource 실패")
+            return
+        }
+
+        let currentIDOpaque = TISGetInputSourceProperty(current, kTISPropertyInputSourceID)
+        guard let currentIDOpaque = currentIDOpaque else {
+            warning("TISGetInputSourceProperty 실패")
+            return
+        }
+        let currentID = Unmanaged<CFString>.fromOpaque(currentIDOpaque).takeUnretainedValue() as String
+
+        if currentID != "com.apple.keylayout.ABC" {
+            debug("현재 입력기 ABC 아님: \(currentID)")
+            return
+        }
+
+        let sokArray = TISCreateInputSourceList([
+            kTISPropertyInputSourceType: kTISTypeKeyboardInputMode,
+            kTISPropertyInputModeID: "com.kiding.inputmethod.sok.mode" as CFString
+        ] as CFDictionary, false)?.takeRetainedValue() as? [TISInputSource]
+        guard let sokArray = sokArray else {
+            warning("TISCreateInputSourceList 실패")
+            return
+        }
+
+        let sok = sokArray.first
+        guard let sok = sok else {
+            warning("sokArray.first 실패")
+            return
+        }
+
+        // "시스템 설정 > 암호" 필드에서는 무한 루프에 빠질 수 있음
+        DispatchQueue.main.asyncAfter(deadline: .now() + .milliseconds(100)) {
+            guard TISSelectInputSource(sok) == 0 else {
+                warning("TISSelectInputSource 실패")
+                return
+            }
+
+            debug("ABC 입력기 제한 성공")
+        }
+    }
 }
 // swiftlint:enable function_body_length
diff --git a/SokIM/Preferences.swift b/SokIM/Preferences.swift
@@ -27,6 +27,11 @@ struct Preferences {
         set(new) { UserDefaults.standard.set(new, forKey: "GraveOverWon") }
     }
 
+    static var suppressABC: Bool {
+        get { UserDefaults.standard.bool(forKey: "SuppressABC") }
+        set(new) { UserDefaults.standard.set(new, forKey: "SuppressABC") }
+    }
+
     static var debug: Bool {
         get { UserDefaults.standard.bool(forKey: "Debug") }
         set(new) { UserDefaults.standard.set(new, forKey: "Debug") }
diff --git a/SokIM/StatusBar.swift b/SokIM/StatusBar.swift
@@ -110,6 +110,13 @@ class StatusBar {
         graveItem.action = #selector(toggleGraveOverWon)
         menu.addItem(graveItem)
 
+        let abcItem = NSMenuItem()
+        abcItem.title = "ABC 입력기 제한"
+        abcItem.state = Preferences.suppressABC ? .on : .off
+        abcItem.target = self
+        abcItem.action = #selector(toggleSuppressABC)
+        menu.addItem(abcItem)
+
         let debugItem = NSMenuItem()
         debugItem.title = "디버그 모드"
         debugItem.state = Preferences.debug ? .on : .off
@@ -222,13 +229,18 @@ class StatusBar {
 
     /** 기타 설정 */
 
-    @objc func toggleDebug(sender: NSMenuItem) {
-        Preferences.debug = sender.state == .on ? false : true
-        sender.state = Preferences.debug ? .on : .off
-    }
-
     @objc func toggleGraveOverWon(sender: NSMenuItem) {
         Preferences.graveOverWon = sender.state == .on ? false : true
         sender.state = Preferences.graveOverWon ? .on : .off
     }
+
+    @objc func toggleSuppressABC(sender: NSMenuItem) {
+        Preferences.suppressABC = sender.state == .on ? false : true
+        sender.state = Preferences.suppressABC ? .on : .off
+    }
+
+    @objc func toggleDebug(sender: NSMenuItem) {
+        Preferences.debug = sender.state == .on ? false : true
+        sender.state = Preferences.debug ? .on : .off
+    }
 }
