1- package com.github.siropkin.kursor.keyboardlayout
1+ package com.github.siropkin.kursor.keyboard
22
33import com.sun.jna.Platform
44import com.sun.jna.platform.win32.User32
55import com.sun.jna.platform.win32.WinDef
66import com.sun.jna.platform.win32.WinDef.HKL
77import java.awt.im.InputContext
8- import java.io.BufferedReader
9- import java.io.IOException
108
119
12- class KeyboardLayout {
13- private val unknown = " UNK"
14- private var linuxDistribution: String = System .getenv(" DESKTOP_SESSION" ? : " "
15- private var linuxDesktopGroup: String = System .getenv(" XDG_SESSION_TYPE" ? : " "
16- private var linuxKeyboardLayoutsCache: List <String > = emptyList()
10+ private const val UNKNOWN = " UNK"
1711
18- fun getLayoutInfo (): KeyboardLayoutInfo {
12+ class Keyboard {
13+ private var linuxConfig: LinuxConfig ? = null
14+
15+ fun getLayout (): KeyboardLayout {
16+ if (Platform .isLinux() && linuxConfig == null ) {
17+ linuxConfig = LinuxConfig ()
18+ }
1919 return when {
20- Platform .isLinux() -> getLinuxLayoutInfo ()
21- Platform .isMac() -> getMacLayoutInfo ()
22- Platform .isWindows() -> getWindowsLayoutInfo ()
23- else -> getUnknownLayoutInfo ()
20+ Platform .isLinux() -> getLinuxLayout ()
21+ Platform .isMac() -> getMacLayout ()
22+ Platform .isWindows() -> getWindowsLayout ()
23+ else -> getUnknownLayout ()
2424 }
2525 }
2626
27- private fun getUnknownLayoutInfo (): KeyboardLayoutInfo {
28- return KeyboardLayoutInfo (unknown, unknown, unknown )
27+ private fun getUnknownLayout (): KeyboardLayout {
28+ return KeyboardLayout ( UNKNOWN , UNKNOWN , UNKNOWN )
2929 }
3030
31- private fun getLinuxLayoutInfo (): KeyboardLayoutInfo {
31+ private fun getLinuxLayout (): KeyboardLayout {
3232 // InputContext.getInstance().locale is not working on Linux: it always returns "en_US"
3333 // This is not the ideal solution because it involves executing a shell command to know the current keyboard layout
3434 // which might affect the performance. And we have different commands for different Linux distributions.
3535 // But it is the only solution I found that works on Linux.
3636 // For Linux we know only keyboard layout and do not know keyboard language
37+ val config = linuxConfig ? : return getUnknownLayout()
3738 return when {
38- linuxDistribution == " ubuntu" -> getUbuntuLayoutInfo ()
39- linuxDesktopGroup == " wayland" -> getWaylandLayoutInfo ()
40- else -> getOtherLinuxLayoutInfo ()
39+ config.distribution == " ubuntu" -> getUbuntuLayout ()
40+ config.desktopGroup == " wayland" -> getWaylandLayout ()
41+ else -> getOtherLinuxLayout ()
4142 }
4243 }
4344
44- private fun getUbuntuLayoutInfo (): KeyboardLayoutInfo {
45+ private fun getUbuntuLayout (): KeyboardLayout {
4546 // Output example: [('xkb', 'us'), ('xkb', 'ru'), ('xkb', 'ca+eng')]
46- val split = executeNativeCommand(arrayOf(" gsettings" " get" " org.gnome.desktop.input-sources" " mru-sources"
47+ val split = Utils . executeNativeCommand(arrayOf(" gsettings" " get" " org.gnome.desktop.input-sources" " mru-sources"
4748 .substringAfter(" ('xkb', '"
4849 .substringBefore(" ')"
4950 .split(" +"
5051 val language = if (split.size > 1 ) split[1 ] else " "
5152 val country = split[0 ]
52- return KeyboardLayoutInfo (language, country, " "
53+ return KeyboardLayout (language, country, " "
5354 }
5455
55- private fun getWaylandLayoutInfo (): KeyboardLayoutInfo {
56+ private fun getWaylandLayout (): KeyboardLayout {
5657 // FIXME: Other Linux distribution commands not working "Wayland",
5758 // see: https://github.com/siropkin/kursor/issues/3
58- return getUnknownLayoutInfo ()
59+ return getUnknownLayout ()
5960 }
6061
61- private fun getOtherLinuxLayoutInfo (): KeyboardLayoutInfo {
62- if (linuxKeyboardLayoutsCache.isEmpty()) {
62+ private fun getOtherLinuxLayout (): KeyboardLayout {
63+ val config = linuxConfig ? : return getUnknownLayout()
64+ if (config.availableKeyboardLayouts.isEmpty()) {
6365 // Output example: rules: evdev
6466 // model: pc105
6567 // layout: us
6668 // options: grp:win_space_toggle,terminate:ctrl_alt_bksp
67- linuxKeyboardLayoutsCache = executeNativeCommand(arrayOf(" setxkbmap" " -query"
69+ config.availableKeyboardLayouts = Utils . executeNativeCommand(arrayOf(" setxkbmap" " -query"
6870 .substringAfter(" layout:"
6971 .substringBefore(" \n "
7072 .trim()
@@ -98,46 +100,45 @@ class KeyboardLayout {
98100 // Standby: 0 Suspend: 0 Off: 0
99101 // DPMS is Enabled
100102 // Monitor is On
101- val linuxCurrentKeyboardLayoutIndex = executeNativeCommand(arrayOf(" xset" " -q"
103+ val linuxCurrentKeyboardLayoutIndex = Utils . executeNativeCommand(arrayOf(" xset" " -q"
102104 .substringAfter(" LED mask:"
103105 .substringBefore(" \n "
104106 .trim()
105107 .substring(4 , 5 )
106108 .toInt(16 )
107109
108110 // Additional check to avoid out-of-bounds exception
109- if (linuxCurrentKeyboardLayoutIndex >= linuxKeyboardLayoutsCache .size) {
110- return getUnknownLayoutInfo ()
111+ if (linuxCurrentKeyboardLayoutIndex >= config.availableKeyboardLayouts .size) {
112+ return getUnknownLayout ()
111113 }
112114
113115 // This is a bad solution because it returns 0 if it's a default layout and 1 in other cases,
114116 // and if user has more than two layouts, we do not know which one is really on
115- if (linuxKeyboardLayoutsCache .size > 2 && linuxCurrentKeyboardLayoutIndex > 0 ) {
116- return getUnknownLayoutInfo ()
117+ if (config.availableKeyboardLayouts .size > 2 && linuxCurrentKeyboardLayoutIndex > 0 ) {
118+ return getUnknownLayout ()
117119 }
118120
119- val country = linuxKeyboardLayoutsCache [linuxCurrentKeyboardLayoutIndex]
120- return KeyboardLayoutInfo (" " " "
121+ val country = config.availableKeyboardLayouts [linuxCurrentKeyboardLayoutIndex]
122+ return KeyboardLayout (" " " "
121123 }
122124
123- private fun getMacLayoutInfo (): KeyboardLayoutInfo {
125+ private fun getMacLayout (): KeyboardLayout {
124126 val locale = InputContext .getInstance().locale
125- // Variant example for US: UserDefined_252
126- val variant = MacKeyboardVariants [locale.variant] ? : " "
127- return KeyboardLayoutInfo (locale.language, locale.country, variant)
127+ val localeVariant = locale.variant.removePrefix(" UserDefined_"
128+ val variant = MacStandardKeyboardVariants [localeVariant]
129+ ? : MacSogouPinyinVariants [localeVariant]
130+ ? : MacRimeSquirrelVariants [localeVariant]
131+ ? : " "
132+ return KeyboardLayout (locale.language, locale.country, variant)
128133 }
129134
130- private fun getWindowsLayoutInfo (): KeyboardLayoutInfo {
135+ private fun getWindowsLayout (): KeyboardLayout {
131136 val locale = InputContext .getInstance().locale
132137 // Standard locale object does not return correct info in case user set different keyboard inputs for one language
133138 // see: https://github.com/siropkin/kursor/issues/4
134139 val user32 = User32 .INSTANCE
135- val fgWindow: WinDef .HWND ? = user32.GetForegroundWindow () // Get the handle of the foreground window
136-
137- if (fgWindow == null ) {
138- return KeyboardLayoutInfo (locale.language, locale.country, " "
139- }
140-
140+ val fgWindow: WinDef .HWND = user32.GetForegroundWindow ()
141+ ? : return KeyboardLayout (locale.language, locale.country, " " // Get the handle of the foreground window
141142 val threadId = user32.GetWindowThreadProcessId (fgWindow, null ) // Get the thread ID of the foreground window
142143 val hkl: HKL = user32.GetKeyboardLayout (threadId) // Get the keyboard layout for the thread
143144 // FIXME: It should be a better way how to convert pointer to string
@@ -151,18 +152,7 @@ class KeyboardLayout {
151152 else -> layoutId.substring(2 ).padStart(8 , ' 0'
152153 }
153154 val variant = WindowsKeyboardVariants [layoutId.uppercase()] ? : " "
154- return KeyboardLayoutInfo (locale.language, locale.country, variant)
155- }
156-
157- private fun executeNativeCommand (command : Array <String >): String {
158- return try {
159- val process = Runtime .getRuntime().exec(command)
160- process.waitFor()
161- process.inputStream.bufferedReader().use(BufferedReader ::readText)
162- } catch (e: IOException ) {
163- e.printStackTrace()
164- " "
165- }
155+ return KeyboardLayout (locale.language, locale.country, variant)
166156 }
167157}
168158
0 commit comments