From 4196d35ce58cb74aa29d523599df75ff63024742 Mon Sep 17 00:00:00 2001
From: fxliang <fx.liang@outlook.com>
Date: Fri, 19 Jul 2024 12:39:29 +0800
Subject: [PATCH] feat: add testing update channel

---
 WeaselServer/WeaselServer.rc   | Bin 21798 -> 22800 bytes
 WeaselServer/WeaselServerApp.h |   6 ++++++
 WeaselSetup/WeaselSetup.cpp    |   8 ++++++++
 3 files changed, 14 insertions(+)

diff --git a/WeaselServer/WeaselServer.rc b/WeaselServer/WeaselServer.rc
index bfc74ed91f7e81b80fea7cbb7ed006411580b96f..1234e97d6cfec60f29bb30dc67e09ec01d498431 100644
GIT binary patch
delta 116
zcmZ3sigCgw#tj0zydey(48aT`44w>r4DOTlx#jtJ8MqjnfTHeTF@?#o+_NV$ahXjv
k;7I|}LX$UeWuWO2=XPhrs&2BE+M>w<oLrj^@TR!~0PwgQ`~Uy|

delta 18
acmbQRiE-H~#tj0zli#Rq*t|+G%^d(sy9d$$

diff --git a/WeaselServer/WeaselServerApp.h b/WeaselServer/WeaselServerApp.h
index 9fbc38302..d5e32e006 100644
--- a/WeaselServer/WeaselServerApp.h
+++ b/WeaselServer/WeaselServerApp.h
@@ -36,6 +36,12 @@ class WeaselServerApp {
   static bool check_update() {
     // when checked manually, show testing versions too
     std::string feed_url = GetCustomResource("ManualUpdateFeedURL", "APPCAST");
+    std::wstring channel{};
+    auto ret = RegGetStringValue(HKEY_CURRENT_USER, L"Software\\Rime\\Weasel",
+                                 L"UpdateChannel", channel);
+    if (!ret && channel == L"testing") {
+      feed_url = GetCustomResource("TestingManualUpdateFeedURL", "APPCAST");
+    }
     if (!feed_url.empty()) {
       win_sparkle_set_appcast_url(feed_url.c_str());
     }
diff --git a/WeaselSetup/WeaselSetup.cpp b/WeaselSetup/WeaselSetup.cpp
index d9fe9861d..bef216a7c 100644
--- a/WeaselSetup/WeaselSetup.cpp
+++ b/WeaselSetup/WeaselSetup.cpp
@@ -161,6 +161,14 @@ static int Run(LPTSTR lpCmdLine) {
                           L"CheckForUpdates", L"0", REG_SZ);
   }
 
+  if (!wcscmp(L"/testing", lpCmdLine)) {
+    return SetRegKeyValue(HKEY_CURRENT_USER, L"Software\\Rime\\weasel",
+                          L"UpdateChannel", L"testing", REG_SZ);
+  }
+  if (!wcscmp(L"/release", lpCmdLine)) {
+    return SetRegKeyValue(HKEY_CURRENT_USER, L"Software\\Rime\\weasel",
+                          L"UpdateChannel", L"release", REG_SZ);
+  }
   bool hans = !wcscmp(L"/s", lpCmdLine);
   if (hans)
     return install(false, silent, old_ime_support);