Parcourir la source

Added Watson translate support

Vimek il y a 6 ans
Parent
commit
8a88500b9f

+ 7 - 0
src/XUnity.AutoTranslator.Plugin.Core/Configuration/Settings.cs

@@ -45,6 +45,9 @@ namespace XUnity.AutoTranslator.Plugin.Core.Configuration
       public static string BaiduAppId;
       public static string BaiduAppSecret;
       public static string YandexAPIKey;
+      public static string WatsonAPIUrl;
+      public static string WatsonAPIUsername;
+      public static string WatsonAPIPassword;
       public static int ForceSplitTextAfterCharacters;
 
       public static bool CopyToClipboard;
@@ -98,6 +101,10 @@ namespace XUnity.AutoTranslator.Plugin.Core.Configuration
 
          YandexAPIKey = Config.Current.Preferences["Yandex"]["YandexAPIKey"].GetOrDefault("");
 
+         WatsonAPIUrl = Config.Current.Preferences["Watson"]["WatsonAPIUrl"].GetOrDefault("");
+         WatsonAPIUsername = Config.Current.Preferences["Watson"]["WatsonAPIUsername"].GetOrDefault("");
+         WatsonAPIPassword = Config.Current.Preferences["Watson"]["WatsonAPIPassword"].GetOrDefault("");
+
          EnablePrintHierarchy = Config.Current.Preferences[ "Debug" ][ "EnablePrintHierarchy" ].GetOrDefault( false );
 
          AutoTranslationsFilePath = Path.Combine( Config.Current.DataPath, OutputFile.Replace( "{lang}", Language ) );

+ 2 - 0
src/XUnity.AutoTranslator.Plugin.Core/Constants/KnownEndpointNames.cs

@@ -14,5 +14,7 @@ namespace XUnity.AutoTranslator.Plugin.Core.Constants
       public const string BaiduTranslate = "BaiduTranslate";
 
       public const string YandexTranslate = "YandexTranslate";
+
+      public const string WatsonTranslate = "WatsonTranslate";
    }
 }

+ 3 - 0
src/XUnity.AutoTranslator.Plugin.Core/Web/KnownEndpoints.cs

@@ -12,6 +12,7 @@ namespace XUnity.AutoTranslator.Plugin.Core.Web
       public static readonly KnownEndpoint GoogleTranslateHack = new GoogleTranslateHackEndpoint();
       public static readonly KnownEndpoint BaiduTranslate = new BaiduTranslateEndpoint();
       public static readonly KnownEndpoint YandexTranslate = new YandexTranslateEndpoint();
+      public static readonly KnownEndpoint WatsonTranslate = new WatsonTranslateEndpoint();
 
       public static KnownEndpoint FindEndpoint( string identifier )
       {
@@ -27,6 +28,8 @@ namespace XUnity.AutoTranslator.Plugin.Core.Web
                return BaiduTranslate;
             case KnownEndpointNames.YandexTranslate:
                return YandexTranslate;
+            case KnownEndpointNames.WatsonTranslate:
+                    return WatsonTranslate;
             default:
                return new DefaultEndpoint( identifier );
          }

+ 84 - 0
src/XUnity.AutoTranslator.Plugin.Core/Web/WatsonTranslateEndpoint.cs

@@ -0,0 +1,84 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Net;
+using System.Text;
+using SimpleJSON;
+using UnityEngine;
+using XUnity.AutoTranslator.Plugin.Core.Configuration;
+using XUnity.AutoTranslator.Plugin.Core.Constants;
+using XUnity.AutoTranslator.Plugin.Core.Extensions;
+
+namespace XUnity.AutoTranslator.Plugin.Core.Web
+{
+    public class WatsonTranslateEndpoint : KnownEndpoint
+    {
+        private static ServicePoint ServicePoint;
+
+        private static readonly string HttpsServicePointTemplateUrl = Settings.WatsonAPIUrl.TrimEnd('/')+ "/v2/translate?model_id={0}-{1}&text={2}";
+
+        public WatsonTranslateEndpoint()
+           : base(KnownEndpointNames.WatsonTranslate)
+        {
+
+        }
+
+        public override void ApplyHeaders(Dictionary<string, string> headers)
+        {
+            headers["User-Agent"] = "curl/7.55.1";
+            headers["Accept"] = "application/json";
+            headers["Accept-Charset"] = "UTF-8";
+            headers["Authorization"] = "Basic "+System.Convert.ToBase64String(System.Text.Encoding.ASCII.GetBytes(Settings.WatsonAPIUsername+":"+Settings.WatsonAPIPassword));
+        }
+
+        public override void ApplyHeaders(WebHeaderCollection headers)
+        {
+            headers[HttpRequestHeader.UserAgent] = "curl/7.55.1";
+            headers[HttpRequestHeader.Accept] = "application/json";
+            headers[HttpRequestHeader.AcceptCharset] = "UTF-8";
+            headers[HttpRequestHeader.Authorization] = "Basic " + System.Convert.ToBase64String(System.Text.Encoding.ASCII.GetBytes(Settings.WatsonAPIUsername + ":" + Settings.WatsonAPIPassword));
+        }
+
+        public override void ConfigureServicePointManager()
+        {
+            try
+            {
+                ServicePoint = ServicePointManager.FindServicePoint(new Uri(Settings.WatsonAPIUrl));
+                ServicePoint.ConnectionLimit = Settings.MaxConcurrentTranslations;
+            }
+            catch
+            {
+            }
+        }
+
+        public override bool TryExtractTranslated(string result, out string translated)
+        {
+            try
+            {
+                var obj = JSON.Parse(result);
+                var lineBuilder = new StringBuilder(result.Length);
+                
+                foreach (JSONNode entry in obj.AsObject["translations"].AsArray) {
+                    var token = entry.AsObject["translation"].ToString();
+                    token = token.Substring(1, token.Length - 2).UnescapeJson();
+
+                    if (!lineBuilder.EndsWithWhitespaceOrNewline()) lineBuilder.Append("\n");
+
+                    lineBuilder.Append(token);
+                }
+                translated = lineBuilder.ToString();               
+                return true;
+            }
+            catch (Exception)
+            {
+                translated = null;
+                return false;
+            }
+        }
+
+        public override string GetServiceUrl(string untranslatedText, string from, string to)
+        {            
+            return string.Format(HttpsServicePointTemplateUrl, from, to, WWW.EscapeURL(untranslatedText));
+        }
+    }
+}