Bring back delayed start logic

Vin-meido · Dec 25, 2021 · 731aa30 · 731aa30
1 parent 36455bd
commit 731aa30
Showing 1 changed file with 15 additions and 13 deletions.
diff --git a/SugoiOfflineTranslatorEndpoint/SugoiOfflineTranslatorEndpoint.cs b/SugoiOfflineTranslatorEndpoint/SugoiOfflineTranslatorEndpoint.cs
@@ -14,7 +14,7 @@
 
 namespace SugoiOfflineTranslator
 {
-    public class SugoiOfflineTranslatorEndpoint : HttpEndpoint, ITranslateEndpoint, IDisposable, IMonoBehaviour_Update
+    public class SugoiOfflineTranslatorEndpoint : HttpEndpoint, ITranslateEndpoint, IDisposable
     {
         public override string Id => "SugoiOfflineTranslator";
 
@@ -146,18 +146,6 @@ private void StartProcess()
             }
         }
 
-        public void Update()
-        {
-            if (this.isStarted && !this.isDisposing)
-            {
-                if(this.process.HasExited)
-                {
-                    XuaLogger.AutoTranslator.Warn($"Translator server process exited unexpectedly [status {process.ExitCode}]");
-                    this.isStarted = false;
-                }
-            }
-        }
-
         void ServerDataReceivedEventHandler(object sender, DataReceivedEventArgs args)
         {
             if (this.LogServerMessages)
@@ -187,6 +175,20 @@ IEnumerator ITranslateEndpoint.Translate(ITranslationContext context)
 
         public override IEnumerator OnBeforeTranslate(IHttpTranslationContext context)
         {
+            if (this.isStarted && this.process.HasExited)
+            {
+                this.isStarted = false;
+                this.isReady = false;
+
+                XuaLogger.AutoTranslator.Warn($"Translator server process exited unexpectedly [status {process.ExitCode}]");
+            }
+
+            if (!this.isStarted && !this.isDisposing)
+            {
+                XuaLogger.AutoTranslator.Warn($"Translator server process not running. Starting...");
+                this.StartProcess();
+            }
+
             while (!isReady) yield return null;
         }