From 7d1763246f1111458d3d708d208639837eb1b9cd Mon Sep 17 00:00:00 2001
From: nsnail <taokeu@gmail.com>
Date: Fri, 30 Dec 2022 10:32:21 +0800
Subject: [PATCH] =?UTF-8?q?=E7=BF=BB=E8=AF=91=E5=B7=A5=E5=85=B7.=E7=8A=B6?=
 =?UTF-8?q?=E6=80=81=E6=96=87=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/Tran/Main.cs    |  7 +++++-
 src/Tran/WinMain.cs | 54 ++++++++++++++++++++++++++++++++-------------
 2 files changed, 45 insertions(+), 16 deletions(-)

diff --git a/src/Tran/Main.cs b/src/Tran/Main.cs
index 87b0c56..66ca7f8 100644
--- a/src/Tran/Main.cs
+++ b/src/Tran/Main.cs
@@ -21,7 +21,12 @@ internal sealed class Main : ToolBase<Option>
             AppDomain.CurrentDomain.UnhandledException += UnhandledException;
             Application.ThreadException                += UIThreadException;
             using var frm = new WinMain();
-            Application.Run();
+            try {
+                Application.Run();
+            }
+            catch (Exception ex) {
+                Log(ex.Json());
+            }
         });
         th.SetApartmentState(ApartmentState.STA);
         th.Start();
diff --git a/src/Tran/WinMain.cs b/src/Tran/WinMain.cs
index 970e151..1babd76 100644
--- a/src/Tran/WinMain.cs
+++ b/src/Tran/WinMain.cs
@@ -15,17 +15,19 @@ namespace Dot.Tran;
 internal sealed partial class WinMain : Form
 {
     private const    int          _RETRY_WAIT_MIL_SEC = 1000;
-    private const    string       _TRANSLATE_API_URL = $"{_TRANSLATE_HOME_URL}/v2transapi";
+    private const    string       _TRANSLATE_API_URL  = $"{_TRANSLATE_HOME_URL}/v2transapi";
     private const    string       _TRANSLATE_HOME_URL = "https://fanyi.baidu.com";
-    private readonly HttpClient   _httpClient = new();
-    private readonly KeyboardHook _keyboardHook = new();
-    private readonly Label        _labelDest = new();
-    private readonly MouseHook    _mouseHook = new();
-    private readonly Size         _mouseMargin = new(10, 10); //        窗体与鼠标指针的距离
-    private          bool         _capsLockPressed; //                               大写键按下进行翻译，区分ctrl+c
+    private readonly HttpClient   _httpClient         = new(new HttpClientHandler { UseProxy = false });
+    private readonly KeyboardHook _keyboardHook       = new();
+    private readonly Label        _labelDest          = new();
+    private readonly MouseHook    _mouseHook          = new();
+    private readonly Size         _mouseMargin        = new(10, 10);
+    private readonly string       _stateFile          = Path.Combine(Path.GetTempPath(), "dot-tran-state.tmp");
+    private          bool         _capsLockPressed;
+    private volatile string       _cookie;
     private          bool         _disposed;
-    private          nint         _nextClipMonitor; //                               下一个剪贴板监视链对象句柄
-    private          string       _token = "ae72ebad4113270fd26ada5125301268"; //    翻译api token
+    private          nint         _nextClipMonitor;
+    private volatile string       _token;
 
     public WinMain()
     {
@@ -33,6 +35,7 @@ internal sealed partial class WinMain : Form
         InitHook();
         InitLabelDest();
         InitHttpClient();
+        InitTokenCookie();
         InitClipMonitor();
     }
 
@@ -167,6 +170,19 @@ internal sealed partial class WinMain : Form
         Controls.Add(_labelDest);
     }
 
+    private void InitTokenCookie()
+    {
+        if (File.Exists(_stateFile)) {
+            var lines = File.ReadLines(_stateFile).ToArray();
+            _token  = lines[0];
+            _cookie = lines[1];
+            _httpClient.DefaultRequestHeaders.Add(nameof(Cookie), _cookie);
+        }
+        else {
+            Task.Run(UpdateStateFile);
+        }
+    }
+
     private void TranslateAndShow()
     {
         var clipText = ClipboardService.GetText();
@@ -214,13 +230,21 @@ internal sealed partial class WinMain : Form
 
             //cookie or token invalid
             Task.Delay(_RETRY_WAIT_MIL_SEC).Wait();
-            var cookie = string.Join(
-                ';', rsp.Headers.First(x => x.Key == "Set-Cookie").Value.Select(x => x.Split(';').First()));
-            _httpClient.DefaultRequestHeaders.Remove(nameof(Cookie));
-            _httpClient.DefaultRequestHeaders.Add(nameof(Cookie), cookie);
-            var html = _httpClient.GetStringAsync(_TRANSLATE_HOME_URL).Result;
-            _token = TokenRegex().Match(html).Groups[1].Value;
+            UpdateStateFile();
         }
     }
+
+    private void UpdateStateFile()
+    {
+        var rsp = _httpClient.SendAsync(new HttpRequestMessage(HttpMethod.Head, _TRANSLATE_HOME_URL)).Result;
+        _cookie = string.Join(
+            ';', rsp.Headers.First(x => x.Key == "Set-Cookie").Value.Select(x => x.Split(';').First()));
+        _httpClient.DefaultRequestHeaders.Remove(nameof(Cookie));
+        _httpClient.DefaultRequestHeaders.Add(nameof(Cookie), _cookie);
+        var html = _httpClient.GetStringAsync(_TRANSLATE_HOME_URL).Result;
+        _token = TokenRegex().Match(html).Groups[1].Value;
+
+        File.WriteAllLines(_stateFile, new[] { _token, _cookie });
+    }
 }
 #endif
\ No newline at end of file