feat: support to change user dir through registry

chore: change log dir to %TEMP%\rime.rabbit
fix: first run does not load rabbit config
fix: too frequently setting ascii mode

Author: amorphobia

Lib/RabbitCommon.ahk

@@ -89,12 +89,37 @@ CreateTraits() {
     traits.distribution_code_name := RABBIT_CODE_NAME
     traits.distribution_version := RABBIT_VERSION
     traits.app_name := "rime.rabbit"
-    traits.shared_data_dir := "Data"
-    traits.user_data_dir := "Rime"
+    traits.shared_data_dir := RabbitSharedDataPath()
+    traits.user_data_dir := RabbitUserDataPath()
+    traits.log_dir := RabbitLogPath()
 
     return traits
 }
 
+RabbitUserDataPath() {
+    try {
+        local dir := RegRead("HKEY_CURRENT_USER\Software\Rime\Rabbit", "RimeUserDir")
+    }
+    if dir && Type(dir) = "String" {
+        size := DllCall("ExpandEnvironmentStrings", "Str", dir, "Ptr", 0, "UInt", 0)
+        path := Buffer(size * 2, 0)
+        DllCall("ExpandEnvironmentStrings", "Str", dir, "Ptr", path, "UInt", path.Size)
+        return StrGet(path)
+    }
+    return A_ScriptDir . "\Rime"
+}
+
+RabbitSharedDataPath() {
+    return A_ScriptDir . "\Data"
+}
+
+RabbitLogPath() {
+    path := A_Temp . "\rime.rabbit"
+    if !DirExist(path)
+        DirCreate(path)
+    return path
+}
+
 OnRimeMessage(context_object, session_id, message_type, message_value) {
     msg_type := StrGet(message_type, "UTF-8")
     msg_value := StrGet(message_value, "UTF-8")

Lib/RabbitTrayMenu.ahk

@@ -25,7 +25,6 @@ global TRAY_ASCII_MODE := 0
 global TRAY_FULL_SHAPE := 0
 global TRAY_ASCII_PUNCT := 0
 
-SetupTrayMenu()
 UpdateTrayIcon()
 
 SetupTrayMenu() {
@@ -38,7 +37,7 @@ SetupTrayMenu() {
 
     A_TrayMenu.Add()
 
-    A_TrayMenu.Add("用户文件夹", (*) => Run(A_ScriptDir . "\Rime"))
+    A_TrayMenu.Add("用户文件夹", (*) => Run(rime.get_user_data_dir_s()))
     A_TrayMenu.Add("脚本文件夹", (*) => Run(A_ScriptDir))
 
     A_TrayMenu.Add()

Rabbit.ahk

@@ -52,6 +52,11 @@ RabbitMain(args) {
         }
     }
 
+    ; TODO: better handling of first run
+    local first_run := !FileExist(RabbitUserDataPath() . "\default.custom.yaml")
+                    || !FileExist(RabbitUserDataPath() . "\rabbit.custom.yaml")
+                    || !FileExist(RabbitUserDataPath() . "\user.yaml")
+
     rabbit_traits := CreateTraits()
     global rime
     rime.setup(rabbit_traits)
@@ -60,7 +65,10 @@ RabbitMain(args) {
 
     local m := (args.Length == 0) ? RABBIT_PARTIAL_MAINTENANCE : args[1]
     if m != RABBIT_NO_MAINTENANCE {
-        if rime.start_maintenance(m == RABBIT_FULL_MAINTENANCE)
+        if first_run {
+            SetDefaultKeyboard(layout)
+            Deploy()
+        } else if rime.start_maintenance(m == RABBIT_FULL_MAINTENANCE)
             rime.join_maintenance_thread()
     } else {
         TrayTip()
@@ -87,6 +95,7 @@ RabbitMain(args) {
 
         UpdateTrayTip(schema_name, ascii_mode, full_shape, ascii_punct)
     }
+    SetupTrayMenu()
     OnMessage(AHK_NOTIFYICON, ClickHandler.Bind())
     if !RabbitConfig.global_ascii
         SetTimer(UpdateWinAscii)
@@ -427,23 +436,28 @@ UpdateWinAscii(target := false, use_target := false, proc_name := "", by_tray_ic
             return
     }
     RabbitGlobals.active_win := proc_name
+    ; TODO: current state might not be accurate due to non-atomic
+    current := !!rime.get_option(session_id, "ascii_mode")
     if use_target {
         ; force to use passed target
-        RabbitGlobals.process_ascii[proc_name] := target
+        RabbitGlobals.process_ascii[proc_name] := !!target
     } else if RabbitGlobals.process_ascii.Has(proc_name) {
         ; not first time to active window, restore the ascii_mode
         target := RabbitGlobals.process_ascii[proc_name]
-        rime.set_option(session_id, "ascii_mode", target)
+        if current !== target
+            rime.set_option(session_id, "ascii_mode", target)
     } else if RabbitConfig.preset_process_ascii.Has(proc_name) {
         ; in preset, set ascii_mode as preset
         target := RabbitConfig.preset_process_ascii[proc_name]
-        RabbitGlobals.process_ascii[proc_name] := target
-        rime.set_option(session_id, "ascii_mode", target)
+        RabbitGlobals.process_ascii[proc_name] := !!target
+        if current !== target
+            rime.set_option(session_id, "ascii_mode", target)
     } else {
         ; not in preset, set ascii_mode to false
         target := false
-        RabbitGlobals.process_ascii[proc_name] := target
-        rime.set_option(session_id, "ascii_mode", target)
+        RabbitGlobals.process_ascii[proc_name] := !!target
+        if current !== target
+            rime.set_option(session_id, "ascii_mode", target)
     }
     UpdateTrayTip(, target)
     UpdateTrayIcon()

RabbitDeployer.ahk

@@ -70,7 +70,7 @@ ExitRabbitDeployer(reason, code) {
 }
 
 CreateFileIfNotExist(filename) {
-    user_data_dir := A_ScriptDir . "\Rime\"
+    user_data_dir := RabbitUserDataPath() . "\"
     if not InStr(DirExist(user_data_dir), "D")
         DirCreate(user_data_dir)
     filepath := user_data_dir . filename