123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820 |
- using System;
- using System.Collections;
- using System.Collections.Generic;
- using System.IO;
- using System.Linq;
- using System.Net;
- using System.Reflection;
- using System.Text;
- using System.Text.RegularExpressions;
- using System.Threading;
- using ExIni;
- using UnityEngine;
- using UnityEngine.UI;
- using System.Globalization;
- using XUnity.AutoTranslator.Plugin.Core.Extensions;
- using UnityEngine.EventSystems;
- using XUnity.AutoTranslator.Plugin.Core.Configuration;
- using XUnity.AutoTranslator.Plugin.Core.Utilities;
- using XUnity.AutoTranslator.Plugin.Core.Web;
- using XUnity.AutoTranslator.Plugin.Core.Hooks;
- using XUnity.AutoTranslator.Plugin.Core.Hooks.TextMeshPro;
- using XUnity.AutoTranslator.Plugin.Core.Hooks.UGUI;
- using XUnity.AutoTranslator.Plugin.Core.Hooks.NGUI;
- using UnityEngine.SceneManagement;
- using XUnity.AutoTranslator.Plugin.Core.Constants;
- using XUnity.AutoTranslator.Plugin.Core.Debugging;
- using Harmony;
- using XUnity.AutoTranslator.Plugin.Core.Parsing;
- using System.Diagnostics;
- using XUnity.AutoTranslator.Plugin.Core.UI;
- using XUnity.AutoTranslator.Plugin.Core.Endpoints;
- using XUnity.AutoTranslator.Plugin.Core.Web.Internal;
- namespace XUnity.AutoTranslator.Plugin.Core
- {
- /// <summary>
- /// Main plugin class for the AutoTranslator.
- /// </summary>
- public class AutoTranslationPlugin : MonoBehaviour
- {
- private static readonly char[][] TranslationSplitters = new char[][] { new char[] { '\t' }, new char[] { '=' } };
- /// <summary>
- /// Allow the instance to be accessed statically, as only one will exist.
- /// </summary>
- internal static AutoTranslationPlugin Current;
- private XuaWindow _window;
- /// <summary>
- /// These are the currently running translation jobs (being translated by an http request).
- /// </summary>
- private List<TranslationJob> _completedJobs = new List<TranslationJob>();
- private Dictionary<string, TranslationJob> _unstartedJobs = new Dictionary<string, TranslationJob>();
- private Dictionary<string, TranslationJob> _ongoingJobs = new Dictionary<string, TranslationJob>();
- /// <summary>
- /// All the translations are stored in this dictionary.
- /// </summary>
- private Dictionary<string, string> _staticTranslations = new Dictionary<string, string>();
- private Dictionary<string, string> _translations = new Dictionary<string, string>();
- private Dictionary<string, string> _reverseTranslations = new Dictionary<string, string>();
- /// <summary>
- /// These are the new translations that has not yet been persisted to the file system.
- /// </summary>
- private object _writeToFileSync = new object();
- private Dictionary<string, string> _newTranslations = new Dictionary<string, string>();
- private HashSet<string> _newUntranslated = new HashSet<string>();
- /// <summary>
- /// Keeps track of things to copy to clipboard.
- /// </summary>
- private List<string> _textsToCopyToClipboardOrdered = new List<string>();
- private HashSet<string> _textsToCopyToClipboard = new HashSet<string>();
- private float _clipboardUpdated = 0.0f;
- /// <summary>
- /// The number of http translation errors that has occurred up until now.
- /// </summary>
- private int _consecutiveErrors = 0;
- /// <summary>
- /// This is a hash set that contains all Text components that is currently being worked on by
- /// the translation plugin.
- /// </summary>
- private HashSet<object> _ongoingOperations = new HashSet<object>();
- /// <summary>
- /// This function will check if there are symbols of a given language contained in a string.
- /// </summary>
- private Func<string, bool> _symbolCheck;
- /// <summary>
- /// Texts currently being scheduled for translation by 'immediate' components.
- /// </summary>
- private HashSet<string> _immediatelyTranslating = new HashSet<string>();
- private Dictionary<string, byte[]> _translatedImages = new Dictionary<string, byte[]>( StringComparer.InvariantCultureIgnoreCase );
- private HashSet<string> _untranslatedImages = new HashSet<string>();
- private Component _advEngine;
- private float? _nextAdvUpdate;
- private HttpSecurity _httpSecurity;
- private List<ConfiguredEndpoint> _configuredEndpoints;
- private ConfiguredEndpoint _endpoint;
- private int[] _currentTranslationsQueuedPerSecondRollingWindow = new int[ Settings.TranslationQueueWatchWindow ];
- private float? _timeExceededThreshold;
- private float _translationsQueuedPerSecond;
- private bool _isInTranslatedMode = true;
- private bool _textHooksEnabled = true;
- private bool _imageHooksEnabled = true;
- private bool _batchLogicHasFailed = false;
- private int _availableBatchOperations = Settings.MaxAvailableBatchOperations;
- private float _batchOperationSecondCounter = 0;
- private string[] _previouslyQueuedText = new string[ Settings.PreviousTextStaggerCount ];
- private int _staggerTextCursor = 0;
- private int _concurrentStaggers = 0;
- private int _frameForLastQueuedTranslation = -1;
- private int _consecutiveFramesTranslated = 0;
- private int _secondForQueuedTranslation = -1;
- private int _consecutiveSecondsTranslated = 0;
- private bool _hasOverrideFont = false;
- private bool _overrideFont = false;
- private bool _initialized = false;
- private bool _temporarilyDisabled = false;
- private string _requireSpriteRendererCheckCausedBy = null;
- private int _lastSpriteUpdateFrame = -1;
- private bool _isCalledFromSceneManager = false;
- /// <summary>
- /// Initialized the plugin.
- /// </summary>
- public void Initialize()
- {
- // Setup 'singleton'
- Current = this;
- // Setup logger, if it was not already initialized by a plugin-version
- if( XuaLogger.Current == null )
- {
- XuaLogger.Current = new ConsoleLogger();
- }
- // Setup configuration
- try
- {
- Settings.Configure();
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An error occurred during configuration. Shutting plugin down." );
- Settings.IsShutdown = true;
- Settings.IsShutdownFatal = true;
- return;
- }
- // Setup console, if enabled
- if( Settings.EnableConsole )
- {
- DebugConsole.Enable();
- }
- // Setup hooks
- HooksSetup.InstallTextHooks();
- HooksSetup.InstallImageHooks();
- HooksSetup.InstallTextGetterCompatHooks();
- _httpSecurity = new HttpSecurity();
- try
- {
- var context = new InitializationContext( _httpSecurity, Settings.FromLanguage, Settings.Language );
- _configuredEndpoints = KnownTranslateEndpoints.CreateEndpoints( gameObject, context )
- .OrderBy( x => x.Error != null )
- .ThenBy( x => x.Endpoint.FriendlyName )
- .ToList();
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An error occurred while constructing endpoints. Shutting plugin down." );
- Settings.IsShutdown = true;
- Settings.IsShutdownFatal = true;
- return;
- }
- try
- {
- var primaryEndpoint = _configuredEndpoints.FirstOrDefault( x => x.Endpoint.Id == Settings.ServiceEndpoint );
- if( primaryEndpoint == null ) throw new Exception( "The primary endpoint was not properly configured." );
- if( primaryEndpoint.Error != null ) throw new Exception( "The primary endpoint was not properly configured.", primaryEndpoint.Error );
- _endpoint = primaryEndpoint;
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An unexpected error occurred during initialization of endpoint." );
- }
- // TODO: Perhaps some bleeding edge check to see if this is required?
- var callback = _httpSecurity.GetCertificateValidationCheck();
- if( callback != null )
- {
- ServicePointManager.ServerCertificateValidationCallback += callback;
- }
- // Save again because configuration may be modified by endpoints
- try
- {
- PluginEnvironment.Current.SaveConfig();
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An error occurred during while saving configuration." );
- }
- if( !LanguageHelper.IsFromLanguageSupported( Settings.FromLanguage ) )
- {
- XuaLogger.Current.Error( $"The plugin has been configured to use the 'FromLanguage={Settings.FromLanguage}'. This language is not supported. Shutting plugin down." );
- _endpoint = null;
- Settings.IsShutdown = true;
- Settings.IsShutdownFatal = true;
- }
- _symbolCheck = LanguageHelper.GetSymbolCheck( Settings.FromLanguage );
- if( !string.IsNullOrEmpty( Settings.OverrideFont ) )
- {
- var available = Font.GetOSInstalledFontNames();
- if( !available.Contains( Settings.OverrideFont ) )
- {
- XuaLogger.Current.Error( $"The specified override font is not available. Available fonts: " + string.Join( ", ", available ) );
- Settings.OverrideFont = null;
- }
- else
- {
- _hasOverrideFont = true;
- }
- _overrideFont = _hasOverrideFont;
- }
- try
- {
- EnableSceneLoadScan();
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An error occurred while settings up texture scene-load scans." );
- }
- LoadTranslations();
- LoadStaticTranslations();
- try
- {
- _window = new XuaWindow(
- new List<ToggleViewModel>
- {
- new ToggleViewModel(
- " Translated",
- "<b>TRANSLATED</b>\nThe plugin currently displays translated texts. Disabling this does not mean the plugin will no longer perform translations, just that they will not be displayed.",
- "<b>NOT TRANSLATED</b>\nThe plugin currently displays untranslated texts.",
- ToggleTranslation, () => _isInTranslatedMode )
- },
- _configuredEndpoints.Select( x =>
- new TranslatorDropdownOptionViewModel( () => x == _endpoint, x, OnEndpointSelected ) ).ToList(),
- new List<ButtonViewModel>
- {
- new ButtonViewModel( "Reboot", "<b>REBOOT PLUGIN</b>\nReboots the plugin if it has been shutdown. This only works if the plugin was shut down due to consequtive errors towards the translation endpoint.", RebootPlugin, () => Settings.IsShutdown && !Settings.IsShutdownFatal ),
- new ButtonViewModel( "Reload", "<b>RELOAD TRANSLATION</b>\nReloads all translation text files and texture files from disk.", ReloadTranslations, null ),
- new ButtonViewModel( "Hook", "<b>MANUAL HOOK</b>\nTraverses the unity object tree for looking for anything that can be translated. Performs a translation if something is found.", ManualHook, null )
- },
- new List<LabelViewModel>
- {
- new LabelViewModel( "Version: ", () => PluginData.Version ),
- new LabelViewModel( "Status: ", () => Settings.IsShutdown ? "Shutdown" : "Running" ),
- new LabelViewModel( "Served translations: ", () => $"{Settings.TranslationCount} / {Settings.MaxTranslationsBeforeShutdown}" ),
- new LabelViewModel( "Queued translations: ", () => $"{(_unstartedJobs.Count + _ongoingJobs.Count)} / {Settings.MaxUnstartedJobs}" ),
- new LabelViewModel( "Error'ed translations: ", () => $"{_consecutiveErrors} / {Settings.MaxErrors}" ),
- } );
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An error occurred while setting up UI." );
- }
- UnityTextParsers.Initialize( text => IsTranslatable( text ) && IsBelowMaxLength( text ) );
- // start a thread that will periodically removed unused references
- var t1 = new Thread( MaintenanceLoop );
- t1.IsBackground = true;
- t1.Start();
- // start a thread that will periodically save new translations
- var t2 = new Thread( SaveTranslationsLoop );
- t2.IsBackground = true;
- t2.Start();
- }
- private void OnEndpointSelected( ConfiguredEndpoint endpoint )
- {
- if( _endpoint != endpoint )
- {
- _endpoint = endpoint;
- if( Settings.IsShutdown && !Settings.IsShutdownFatal )
- {
- RebootPlugin();
- ManualHook();
- }
- Settings.SetEndpoint( _endpoint.Endpoint.Id );
- }
- }
- private IEnumerable<string> GetTranslationFiles()
- {
- return Directory.GetFiles( Path.Combine( PluginEnvironment.Current.DataPath, Settings.TranslationDirectory ).Parameterize(), $"*.txt", SearchOption.AllDirectories )
- .Select( x => x.Replace( "/", "\\" ) );
- }
- private IEnumerable<string> GetTextureFiles()
- {
- return Directory.GetFiles( Path.Combine( PluginEnvironment.Current.DataPath, Settings.TextureDirectory ).Parameterize(), $"*.png", SearchOption.AllDirectories )
- .Select( x => x.Replace( "/", "\\" ) );
- }
- private void MaintenanceLoop( object state )
- {
- while( true )
- {
- try
- {
- ObjectReferenceMapper.Cull();
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An unexpected error occurred while removing GC'ed resources." );
- }
- Thread.Sleep( 1000 * 60 );
- }
- }
- private void SaveTranslationsLoop( object state )
- {
- try
- {
- while( true )
- {
- if( _newTranslations.Count > 0 )
- {
- lock( _writeToFileSync )
- {
- if( _newTranslations.Count > 0 )
- {
- using( var stream = File.Open( Settings.AutoTranslationsFilePath, FileMode.Append, FileAccess.Write ) )
- using( var writer = new StreamWriter( stream, Encoding.UTF8 ) )
- {
- foreach( var kvp in _newTranslations )
- {
- writer.WriteLine( TextHelper.Encode( kvp.Key ) + '=' + TextHelper.Encode( kvp.Value ) );
- }
- writer.Flush();
- }
- _newTranslations.Clear();
- }
- }
- }
- else
- {
- Thread.Sleep( 5000 );
- }
- }
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An error occurred while saving translations to disk." );
- }
- }
- private void EnableSceneLoadScan()
- {
- XuaLogger.Current.Info( "Probing whether OnLevelWasLoaded or SceneManager is supported in this version of Unity. Any warnings related to OnLevelWasLoaded coming from Unity can safely be ignored." );
- if( Features.SupportsSceneManager )
- {
- XuaLogger.Current.Info( "SceneManager is supported in this version of Unity." );
- EnableSceneLoadScanInternal();
- }
- else
- {
- XuaLogger.Current.Info( "SceneManager is not supported in this version of Unity. Falling back to OnLevelWasLoaded and Application level API." );
- }
- }
- private void EnableSceneLoadScanInternal()
- {
- // do this in a different class to avoid having an anonymous method with references to the "Scene" class
- SceneManagerLoader.EnableSceneLoadScanInternal( this );
- }
- internal void OnLevelWasLoadedFromSceneManager( int id )
- {
- try
- {
- _isCalledFromSceneManager = true;
- OnLevelWasLoaded( id );
- }
- finally
- {
- _isCalledFromSceneManager = false;
- }
- }
- private void OnLevelWasLoaded( int id )
- {
- if( !Features.SupportsSceneManager || ( Features.SupportsSceneManager && _isCalledFromSceneManager ) )
- {
- if( Settings.EnableTextureScanOnSceneLoad && ( Settings.EnableTextureDumping || Settings.EnableTextureTranslation ) )
- {
- XuaLogger.Current.Info( "Performing texture lookup during scene load..." );
- var startTime = Time.realtimeSinceStartup;
- ManualHookForTextures();
- var endTime = Time.realtimeSinceStartup;
- XuaLogger.Current.Info( $"Finished texture lookup (took {Math.Round( endTime - startTime, 2 )} seconds)" );
- }
- }
- }
- /// <summary>
- /// Loads the translations found in Translation.{lang}.txt
- /// </summary>
- private void LoadTranslations()
- {
- try
- {
- lock( _writeToFileSync )
- {
- Directory.CreateDirectory( Path.Combine( PluginEnvironment.Current.DataPath, Settings.TranslationDirectory ).Parameterize() );
- Directory.CreateDirectory( Path.GetDirectoryName( Settings.AutoTranslationsFilePath ) );
- var mainTranslationFile = Settings.AutoTranslationsFilePath;
- LoadTranslationsInFile( mainTranslationFile );
- foreach( var fullFileName in GetTranslationFiles().Reverse().Except( new[] { mainTranslationFile } ) )
- {
- LoadTranslationsInFile( fullFileName );
- }
- }
- if( Settings.EnableTextureTranslation || Settings.EnableTextureDumping )
- {
- _translatedImages.Clear();
- _untranslatedImages.Clear();
- Directory.CreateDirectory( Path.Combine( PluginEnvironment.Current.DataPath, Settings.TextureDirectory ).Parameterize() );
- foreach( var fullFileName in GetTextureFiles() )
- {
- RegisterImageFromFile( fullFileName );
- }
- }
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An error occurred while loading translations." );
- }
- }
- private void RegisterImageFromFile( string fullFileName )
- {
- var fileName = Path.GetFileNameWithoutExtension( fullFileName );
- var startHash = fileName.LastIndexOf( "[" );
- var endHash = fileName.LastIndexOf( "]" );
- if( endHash > -1 && startHash > -1 && endHash > startHash )
- {
- var takeFrom = startHash + 1;
- // load based on whether or not the key is image hashed
- var parts = fileName.Substring( takeFrom, endHash - takeFrom ).Split( '-' );
- string key;
- string originalHash;
- if( parts.Length == 1 )
- {
- key = parts[ 0 ];
- originalHash = parts[ 0 ];
- }
- else if( parts.Length == 2 )
- {
- key = parts[ 0 ];
- originalHash = parts[ 1 ];
- }
- else
- {
- XuaLogger.Current.Warn( $"Image not loaded (unknown hash): {fullFileName}." );
- return;
- }
- var data = File.ReadAllBytes( fullFileName );
- var currentHash = HashHelper.Compute( data );
- var isModified = StringComparer.InvariantCultureIgnoreCase.Compare( originalHash, currentHash ) != 0;
- // only load images that someone has modified!
- if( Settings.LoadUnmodifiedTextures || isModified )
- {
- RegisterTranslatedImage( key, data );
- XuaLogger.Current.Debug( $"Image loaded: {fullFileName}." );
- }
- else
- {
- RegisterUntranslatedImage( key );
- XuaLogger.Current.Warn( $"Image not loaded (unmodified): {fullFileName}." );
- }
- //if( Settings.DeleteUnmodifiedTextures && !isModified )
- //{
- // try
- // {
- // File.Delete( fullFileName );
- // Logger.Current.Warn( $"Image deleted (unmodified): {fullFileName}." );
- // }
- // catch( Exception e )
- // {
- // Logger.Current.Warn( e, $"An error occurred while trying to delete unmodified image: {fullFileName}." );
- // }
- //}
- }
- else
- {
- XuaLogger.Current.Warn( $"Image not loaded (no hash): {fullFileName}." );
- }
- }
- private void RegisterImageFromData( string textureName, string key, byte[] data )
- {
- var name = textureName.SanitizeForFileSystem();
- var root = Path.Combine( PluginEnvironment.Current.DataPath, Settings.TextureDirectory ).Parameterize();
- var originalHash = HashHelper.Compute( data );
- // allow hash and key to be the same; only store one of them then!
- string fileName;
- if( key == originalHash )
- {
- fileName = name + " [" + key + "].png";
- }
- else
- {
- fileName = name + " [" + key + "-" + originalHash + "].png";
- }
- var fullName = Path.Combine( root, fileName );
- File.WriteAllBytes( fullName, data );
- XuaLogger.Current.Info( "Dumped texture file: " + fileName );
- if( Settings.LoadUnmodifiedTextures )
- {
- RegisterTranslatedImage( key, data );
- }
- else
- {
- RegisterUntranslatedImage( key );
- }
- }
- private void RegisterTranslatedImage( string key, byte[] data )
- {
- _translatedImages[ key ] = data;
- }
- private void RegisterUntranslatedImage( string key )
- {
- _untranslatedImages.Add( key );
- }
- private void LoadTranslationsInFile( string fullFileName )
- {
- if( File.Exists( fullFileName ) )
- {
- XuaLogger.Current.Debug( $"Loading texts: {fullFileName}." );
- string[] translations = File.ReadAllLines( fullFileName, Encoding.UTF8 );
- foreach( string translation in translations )
- {
- for( int i = 0 ; i < TranslationSplitters.Length ; i++ )
- {
- var splitter = TranslationSplitters[ i ];
- string[] kvp = translation.Split( splitter, StringSplitOptions.None );
- if( kvp.Length == 2 )
- {
- string key = TextHelper.Decode( kvp[ 0 ].TrimIfConfigured() );
- string value = TextHelper.Decode( kvp[ 1 ] );
- if( !string.IsNullOrEmpty( key ) && !string.IsNullOrEmpty( value ) && IsTranslatable( key ) )
- {
- AddTranslation( key, value );
- break;
- }
- }
- }
- }
- }
- }
- private void LoadStaticTranslations()
- {
- if( Settings.UseStaticTranslations && Settings.FromLanguage == Settings.DefaultFromLanguage && Settings.Language == Settings.DefaultLanguage )
- {
- var tab = new char[] { '\t' };
- var equals = new char[] { '=' };
- var splitters = new char[][] { tab, equals };
- // load static translations from previous titles
- string[] translations = Properties.Resources.StaticTranslations.Split( new string[] { "\r\n" }, StringSplitOptions.RemoveEmptyEntries );
- foreach( string translation in translations )
- {
- for( int i = 0 ; i < splitters.Length ; i++ )
- {
- var splitter = splitters[ i ];
- string[] kvp = translation.Split( splitter, StringSplitOptions.None );
- if( kvp.Length >= 2 )
- {
- string key = TextHelper.Decode( kvp[ 0 ].TrimIfConfigured() );
- string value = TextHelper.Decode( kvp[ 1 ].TrimIfConfigured() );
- if( !string.IsNullOrEmpty( key ) && !string.IsNullOrEmpty( value ) )
- {
- _staticTranslations[ key ] = value;
- break;
- }
- }
- }
- }
- }
- }
- private TranslationJob GetOrCreateTranslationJobFor( object ui, TranslationKey key, TranslationContext context )
- {
- var lookupKey = key.GetDictionaryLookupKey();
- if( _unstartedJobs.TryGetValue( lookupKey, out TranslationJob unstartedJob ) )
- {
- unstartedJob.Associate( context );
- return unstartedJob;
- }
- if( _ongoingJobs.TryGetValue( lookupKey, out TranslationJob ongoingJob ) )
- {
- ongoingJob.Associate( context );
- return ongoingJob;
- }
- foreach( var completedJob in _completedJobs )
- {
- if( completedJob.Key.GetDictionaryLookupKey() == lookupKey )
- {
- completedJob.Associate( context );
- return completedJob;
- }
- }
- XuaLogger.Current.Debug( "Queued translation for: " + lookupKey );
- ongoingJob = new TranslationJob( key );
- if( ui != null )
- {
- ongoingJob.OriginalSources.Add( ui );
- }
- ongoingJob.Associate( context );
- _unstartedJobs.Add( lookupKey, ongoingJob );
- CheckStaggerText( lookupKey );
- CheckConsecutiveFrames();
- CheckConsecutiveSeconds();
- CheckThresholds();
- return ongoingJob;
- }
- private void CheckConsecutiveSeconds()
- {
- var currentSecond = (int)Time.time;
- var lastSecond = currentSecond - 1;
- if( lastSecond == _secondForQueuedTranslation )
- {
- // we also queued something last frame, lets increment our counter
- _consecutiveSecondsTranslated++;
- if( _consecutiveSecondsTranslated > Settings.MaximumConsecutiveSecondsTranslated )
- {
- // Shutdown, this wont be tolerated!!!
- _unstartedJobs.Clear();
- _completedJobs.Clear();
- _ongoingJobs.Clear();
- Settings.IsShutdown = true;
- Settings.IsShutdownFatal = true;
- XuaLogger.Current.Error( $"SPAM DETECTED: Translations were queued every second for more than {Settings.MaximumConsecutiveSecondsTranslated} consecutive seconds. Shutting down plugin." );
- }
- }
- else if( currentSecond == _secondForQueuedTranslation )
- {
- // do nothing, there may be multiple translations per frame, that wont increase this counter
- }
- else
- {
- // but if multiple Update frames has passed, we will reset the counter
- _consecutiveSecondsTranslated = 0;
- }
- _secondForQueuedTranslation = currentSecond;
- }
- private void CheckConsecutiveFrames()
- {
- var currentFrame = Time.frameCount;
- var lastFrame = currentFrame - 1;
- if( lastFrame == _frameForLastQueuedTranslation )
- {
- // we also queued something last frame, lets increment our counter
- _consecutiveFramesTranslated++;
- if( _consecutiveFramesTranslated > Settings.MaximumConsecutiveFramesTranslated )
- {
- // Shutdown, this wont be tolerated!!!
- _unstartedJobs.Clear();
- _completedJobs.Clear();
- _ongoingJobs.Clear();
- Settings.IsShutdown = true;
- Settings.IsShutdownFatal = true;
- XuaLogger.Current.Error( $"SPAM DETECTED: Translations were queued every frame for more than {Settings.MaximumConsecutiveFramesTranslated} consecutive frames. Shutting down plugin." );
- }
- }
- else if( currentFrame == _frameForLastQueuedTranslation )
- {
- // do nothing, there may be multiple translations per frame, that wont increase this counter
- }
- else if( _consecutiveFramesTranslated > 0 )
- {
- // but if multiple Update frames has passed, we will reset the counter
- _consecutiveFramesTranslated--;
- }
- _frameForLastQueuedTranslation = currentFrame;
- }
- private void PeriodicResetFrameCheck()
- {
- var currentSecond = (int)Time.time;
- if( currentSecond % 100 == 0 )
- {
- _consecutiveFramesTranslated = 0;
- }
- }
- private void CheckStaggerText( string untranslatedText )
- {
- bool wasProblematic = false;
- for( int i = 0 ; i < _previouslyQueuedText.Length ; i++ )
- {
- var previouslyQueuedText = _previouslyQueuedText[ i ];
- if( previouslyQueuedText != null )
- {
- if( untranslatedText.RemindsOf( previouslyQueuedText ) )
- {
- wasProblematic = true;
- break;
- }
- }
- }
- if( wasProblematic )
- {
- _concurrentStaggers++;
- if( _concurrentStaggers > Settings.MaximumStaggers )
- {
- _unstartedJobs.Clear();
- _completedJobs.Clear();
- _ongoingJobs.Clear();
- Settings.IsShutdown = true;
- Settings.IsShutdownFatal = true;
- XuaLogger.Current.Error( $"SPAM DETECTED: Text that is 'scrolling in' is being translated. Disable that feature. Shutting down plugin." );
- }
- }
- else
- {
- _concurrentStaggers = 0;
- }
- _previouslyQueuedText[ _staggerTextCursor % _previouslyQueuedText.Length ] = untranslatedText;
- _staggerTextCursor++;
- }
- private void CheckThresholds()
- {
- if( _unstartedJobs.Count > Settings.MaxUnstartedJobs )
- {
- _unstartedJobs.Clear();
- _completedJobs.Clear();
- _ongoingJobs.Clear();
- Settings.IsShutdown = true;
- Settings.IsShutdownFatal = true;
- XuaLogger.Current.Error( $"SPAM DETECTED: More than {Settings.MaxUnstartedJobs} queued for translations due to unknown reasons. Shutting down plugin." );
- }
- var previousIdx = ( (int)( Time.time - Time.deltaTime ) ) % Settings.TranslationQueueWatchWindow;
- var newIdx = ( (int)Time.time ) % Settings.TranslationQueueWatchWindow;
- if( previousIdx != newIdx )
- {
- _currentTranslationsQueuedPerSecondRollingWindow[ newIdx ] = 0;
- }
- _currentTranslationsQueuedPerSecondRollingWindow[ newIdx ]++;
- var translationsInWindow = _currentTranslationsQueuedPerSecondRollingWindow.Sum();
- _translationsQueuedPerSecond = (float)translationsInWindow / Settings.TranslationQueueWatchWindow;
- if( _translationsQueuedPerSecond > Settings.MaxTranslationsQueuedPerSecond )
- {
- if( !_timeExceededThreshold.HasValue )
- {
- _timeExceededThreshold = Time.time;
- }
- if( Time.time - _timeExceededThreshold.Value > Settings.MaxSecondsAboveTranslationThreshold )
- {
- _unstartedJobs.Clear();
- _completedJobs.Clear();
- _ongoingJobs.Clear();
- Settings.IsShutdown = true;
- Settings.IsShutdownFatal = true;
- XuaLogger.Current.Error( $"SPAM DETECTED: More than {Settings.MaxTranslationsQueuedPerSecond} translations per seconds queued for a {Settings.MaxSecondsAboveTranslationThreshold} second period. Shutting down plugin." );
- }
- }
- else
- {
- _timeExceededThreshold = null;
- }
- }
- private void IncrementBatchOperations()
- {
- _batchOperationSecondCounter += Time.deltaTime;
- if( _batchOperationSecondCounter > Settings.IncreaseBatchOperationsEvery )
- {
- if( _availableBatchOperations < Settings.MaxAvailableBatchOperations )
- {
- _availableBatchOperations++;
- }
- _batchOperationSecondCounter = 0;
- }
- }
- private void ResetThresholdTimerIfRequired()
- {
- var previousIdx = ( (int)( Time.time - Time.deltaTime ) ) % Settings.TranslationQueueWatchWindow;
- var newIdx = ( (int)Time.time ) % Settings.TranslationQueueWatchWindow;
- if( previousIdx != newIdx )
- {
- _currentTranslationsQueuedPerSecondRollingWindow[ newIdx ] = 0;
- }
- var translationsInWindow = _currentTranslationsQueuedPerSecondRollingWindow.Sum();
- _translationsQueuedPerSecond = (float)translationsInWindow / Settings.TranslationQueueWatchWindow;
- if( _translationsQueuedPerSecond <= Settings.MaxTranslationsQueuedPerSecond )
- {
- _timeExceededThreshold = null;
- }
- }
- private bool IsImageRegistered( string key )
- {
- return _translatedImages.ContainsKey( key ) || _untranslatedImages.Contains( key );
- }
- private bool TryGetTranslatedImage( string key, out byte[] data )
- {
- return _translatedImages.TryGetValue( key, out data );
- }
- private void AddTranslation( string key, string value )
- {
- _translations[ key ] = value;
- _reverseTranslations[ value ] = key;
- }
- private void AddTranslation( TranslationKey key, string value )
- {
- var lookup = key.GetDictionaryLookupKey();
- _translations[ lookup ] = value;
- _reverseTranslations[ value ] = lookup;
- }
- private void UpdateSpriteRenderers()
- {
- if( Settings.EnableSpriteRendererHooking && ( Settings.EnableTextureTranslation || Settings.EnableTextureDumping ) )
- {
- if( _requireSpriteRendererCheckCausedBy != null )
- {
- try
- {
- var start = Time.realtimeSinceStartup;
- var spriteRenderers = GameObject.FindObjectsOfType<SpriteRenderer>();
- foreach( var sr in spriteRenderers )
- {
- // simulate a hook
- Hook_ImageChangedOnComponent( sr, null, false, false );
- }
- var end = Time.realtimeSinceStartup;
- var delta = Math.Round( end - start, 2 );
- XuaLogger.Current.Debug( $"Update SpriteRenderers caused by {_requireSpriteRendererCheckCausedBy} component (took " + delta + " seconds)" );
- }
- finally
- {
- _requireSpriteRendererCheckCausedBy = null;
- }
- }
- }
- }
- private void QueueNewUntranslatedForClipboard( TranslationKey key )
- {
- if( Settings.CopyToClipboard && Features.SupportsClipboard )
- {
- if( !_textsToCopyToClipboard.Contains( key.RelevantText ) )
- {
- _textsToCopyToClipboard.Add( key.RelevantText );
- _textsToCopyToClipboardOrdered.Add( key.RelevantText );
- _clipboardUpdated = Time.realtimeSinceStartup;
- }
- }
- }
- private void QueueNewUntranslatedForDisk( TranslationKey key )
- {
- _newUntranslated.Add( key.GetDictionaryLookupKey() );
- }
- private void QueueNewTranslationForDisk( TranslationKey key, string value )
- {
- lock( _writeToFileSync )
- {
- _newTranslations[ key.GetDictionaryLookupKey() ] = value;
- }
- }
- private void QueueNewTranslationForDisk( string key, string value )
- {
- lock( _writeToFileSync )
- {
- _newTranslations[ key ] = value;
- }
- }
- private bool TryGetTranslation( TranslationKey key, out string value )
- {
- return TryGetTranslation( key.GetDictionaryLookupKey(), out value );
- }
- private bool TryGetTranslation( string key, out string value )
- {
- var result = _translations.TryGetValue( key, out value );
- if( result )
- {
- return result;
- }
- else if( _staticTranslations.Count > 0 )
- {
- if( _staticTranslations.TryGetValue( key, out value ) )
- {
- QueueNewTranslationForDisk( key, value );
- AddTranslation( key, value );
- return true;
- }
- }
- return result;
- }
- internal bool TryGetReverseTranslation( string value, out string key )
- {
- return _reverseTranslations.TryGetValue( value, out key );
- }
- internal string Hook_TextChanged_WithResult( object ui, string text )
- {
- if( !ui.IsKnownTextType() ) return null;
- if( _textHooksEnabled && !_temporarilyDisabled )
- {
- return TranslateOrQueueWebJob( ui, text, false );
- }
- return null;
- }
- internal string ExternalHook_TextChanged_WithResult( object ui, string text )
- {
- if( !ui.IsKnownTextType() ) return null;
- if( _textHooksEnabled && !_temporarilyDisabled )
- {
- return TranslateOrQueueWebJob( ui, text, true );
- }
- return null;
- }
- internal void Hook_TextChanged( object ui, bool onEnable )
- {
- if( _textHooksEnabled && !_temporarilyDisabled )
- {
- TranslateOrQueueWebJob( ui, null, false );
- }
- if( onEnable )
- {
- CheckSpriteRenderer( ui );
- }
- }
- internal void Hook_ImageChangedOnComponent( object source, Texture2D texture, bool isPrefixHooked, bool onEnable )
- {
- if( !_imageHooksEnabled ) return;
- if( !source.IsKnownImageType() ) return;
- HandleImage( source, texture, isPrefixHooked );
- if( onEnable )
- {
- CheckSpriteRenderer( source );
- }
- }
- internal void Hook_ImageChanged( Texture2D texture, bool isPrefixHooked )
- {
- if( !_imageHooksEnabled ) return;
- if( texture == null ) return;
- HandleImage( null, texture, isPrefixHooked );
- }
- private void SetTranslatedText( object ui, string translatedText, TextTranslationInfo info )
- {
- info?.SetTranslatedText( translatedText );
- if( _isInTranslatedMode )
- {
- SetText( ui, translatedText, true, info );
- }
- }
- internal void Hook_HandleComponent( object ui )
- {
- if( _hasOverrideFont )
- {
- var info = ui.GetOrCreateTextTranslationInfo();
- if( _overrideFont )
- {
- info?.ChangeFont( ui );
- }
- else
- {
- info?.UnchangeFont( ui );
- }
- }
- if( Settings.ForceUIResizing )
- {
- var info = ui.GetOrCreateTextTranslationInfo();
- if( info?.IsCurrentlySettingText == false )
- {
- // force UI resizing is highly problematic for NGUI because text should somehow
- // be set after changing "resize" properties... brilliant stuff
- if( ui.GetType() != ClrTypes.UILabel )
- {
- info?.ResizeUI( ui );
- }
- }
- }
- }
- private void CheckSpriteRenderer( object ui )
- {
- if( Settings.EnableSpriteRendererHooking )
- {
- var currentFrame = Time.frameCount;
- var lastFrame = currentFrame - 1;
- if( lastFrame != _lastSpriteUpdateFrame && currentFrame != _lastSpriteUpdateFrame )
- {
- _requireSpriteRendererCheckCausedBy = ui?.GetType().Name;
- }
- _lastSpriteUpdateFrame = currentFrame;
- }
- }
- /// <summary>
- /// Sets the text of a UI text, while ensuring this will not fire a text changed event.
- /// </summary>
- private void SetText( object ui, string text, bool isTranslated, TextTranslationInfo info )
- {
- if( !info?.IsCurrentlySettingText ?? true )
- {
- try
- {
- _textHooksEnabled = false;
- if( info != null )
- {
- info.IsCurrentlySettingText = true;
- }
- if( Settings.EnableUIResizing || Settings.ForceUIResizing )
- {
- if( isTranslated || Settings.ForceUIResizing )
- {
- info?.ResizeUI( ui );
- }
- else
- {
- info?.UnresizeUI( ui );
- }
- }
- // NGUI only behaves if you set the text after the resize behaviour
- ui.SetText( text );
- info?.ResetScrollIn( ui );
- }
- catch( TargetInvocationException )
- {
- // might happen with NGUI
- }
- catch( NullReferenceException )
- {
- // This is likely happened due to a scene change.
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An error occurred while setting text on a component." );
- }
- finally
- {
- _textHooksEnabled = true;
- if( info != null )
- {
- info.IsCurrentlySettingText = false;
- }
- }
- }
- }
- /// <summary>
- /// Determines if a text should be translated.
- /// </summary>
- private bool IsTranslatable( string str )
- {
- return _symbolCheck( str )
- //&& str.Length <= Settings.MaxCharactersPerTranslation
- && !_reverseTranslations.ContainsKey( str )
- && !Settings.IgnoreTextStartingWith.Any( x => str.StartsWithStrict( x ) );
- }
- private bool IsBelowMaxLength( string str )
- {
- return str.Length <= Settings.MaxCharactersPerTranslation;
- }
- private bool IsBelowMaxLengthStrict( string str )
- {
- return str.Length <= ( Settings.MaxCharactersPerTranslation / 2 );
- }
- private bool ShouldTranslateImageComponent( object ui )
- {
- var component = ui as Component;
- if( component != null )
- {
- // dummy check
- var go = component.gameObject;
- var ignore = go.HasIgnoredName();
- if( ignore )
- {
- return false;
- }
- var behaviour = component as Behaviour;
- if( behaviour?.isActiveAndEnabled == false )
- {
- return false;
- }
- }
- return true;
- }
- private bool ShouldTranslateTextComponent( object ui, bool ignoreComponentState )
- {
- var component = ui as Component;
- if( component != null )
- {
- // dummy check
- var go = component.gameObject;
- var ignore = go.HasIgnoredName();
- if( ignore )
- {
- return false;
- }
- if( !ignoreComponentState )
- {
- var behaviour = component as Behaviour;
- if( behaviour?.isActiveAndEnabled == false )
- {
- return false;
- }
- }
- var inputField = component.gameObject.GetFirstComponentInSelfOrAncestor( ClrTypes.InputField )
- ?? component.gameObject.GetFirstComponentInSelfOrAncestor( ClrTypes.TMP_InputField );
- return inputField == null;
- }
- return true;
- }
- private string TranslateOrQueueWebJob( object ui, string text, bool ignoreComponentState )
- {
- var info = ui.GetOrCreateTextTranslationInfo();
- if( _ongoingOperations.Contains( ui ) )
- {
- return TranslateImmediate( ui, text, info, ignoreComponentState );
- }
- var supportsStabilization = ui.SupportsStabilization();
- if( Settings.Delay == 0 || !supportsStabilization )
- {
- return TranslateOrQueueWebJobImmediate( ui, text, info, supportsStabilization, ignoreComponentState );
- }
- else
- {
- StartCoroutine(
- DelayForSeconds( Settings.Delay, () =>
- {
- TranslateOrQueueWebJobImmediate( ui, text, info, supportsStabilization, ignoreComponentState );
- } ) );
- }
- return null;
- }
- private static bool IsCurrentlySetting( TextTranslationInfo info )
- {
- if( info == null ) return false;
- return info.IsCurrentlySettingText;
- }
- private void HandleImage( object source, Texture2D texture, bool isPrefixHooked )
- {
- if( Settings.EnableTextureDumping )
- {
- try
- {
- DumpTexture( source, texture );
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An error occurred while dumping texture." );
- }
- }
- if( Settings.EnableTextureTranslation )
- {
- try
- {
- TranslateTexture( source, texture, isPrefixHooked, null );
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An error occurred while translating texture." );
- }
- }
- }
- private void TranslateTexture( object ui, TextureReloadContext context )
- {
- if( ui is Texture2D texture2d )
- {
- TranslateTexture( null, texture2d, false, context );
- }
- else
- {
- TranslateTexture( ui, null, false, context );
- }
- }
- private void TranslateTexture( object source, Texture2D texture, bool isPrefixHooked, TextureReloadContext context )
- {
- try
- {
- _imageHooksEnabled = false;
- texture = texture ?? source.GetTexture();
- if( texture == null ) return;
- var tti = texture.GetOrCreateTextureTranslationInfo();
- var iti = source.GetOrCreateImageTranslationInfo();
- var key = tti.GetKey( texture );
- if( string.IsNullOrEmpty( key ) ) return;
- bool hasContext = context != null;
- bool forceReload = false;
- if( hasContext )
- {
- forceReload = context.RegisterTextureInContextAndDetermineWhetherToReload( texture );
- }
- if( TryGetTranslatedImage( key, out var newData ) )
- {
- if( _isInTranslatedMode )
- {
- // handle texture
- if( !tti.IsTranslated || forceReload )
- {
- try
- {
- texture.LoadImageEx( newData, tti.IsNonReadable( texture ) );
- }
- finally
- {
- tti.IsTranslated = true;
- }
- }
- // handle containing component
- if( iti != null )
- {
- if( !iti.IsTranslated || hasContext )
- {
- try
- {
- if( !isPrefixHooked )
- {
- source.SetAllDirtyEx();
- }
- }
- finally
- {
- iti.IsTranslated = true;
- }
- }
- }
- }
- }
- else
- {
- // if we cannot find the texture, and the texture is considered translated... hmmm someone has removed a file
- // handle texture
- var originalData = tti.GetOriginalData( texture );
- if( originalData != null )
- {
- if( tti.IsTranslated )
- {
- try
- {
- texture.LoadImageEx( originalData, tti.IsNonReadable( texture ) );
- }
- finally
- {
- tti.IsTranslated = true;
- }
- }
- // handle containing component
- if( iti != null )
- {
- if( iti.IsTranslated )
- {
- try
- {
- if( !isPrefixHooked )
- {
- source.SetAllDirtyEx();
- }
- }
- finally
- {
- iti.IsTranslated = true;
- }
- }
- }
- }
- }
- if( !_isInTranslatedMode )
- {
- var originalData = tti.GetOriginalData( texture );
- if( originalData != null )
- {
- // handle texture
- if( tti.IsTranslated )
- {
- try
- {
- texture.LoadImageEx( originalData, tti.IsNonReadable( texture ) );
- }
- finally
- {
- tti.IsTranslated = false;
- }
- }
- // handle containing component
- if( iti != null )
- {
- if( iti.IsTranslated )
- {
- try
- {
- if( !isPrefixHooked )
- {
- source.SetAllDirtyEx();
- }
- }
- finally
- {
- iti.IsTranslated = false;
- }
- }
- }
- }
- }
- if( forceReload )
- {
- XuaLogger.Current.Info( $"Reloaded texture: {texture.name} ({key})." );
- }
- }
- finally
- {
- _imageHooksEnabled = true;
- }
- }
- private void DumpTexture( object source, Texture2D texture )
- {
- try
- {
- _imageHooksEnabled = false;
- texture = texture ?? source.GetTexture();
- if( texture == null ) return;
- var info = texture.GetOrCreateTextureTranslationInfo();
- if( info.HasDumpedAlternativeTexture ) return;
- try
- {
- if( ShouldTranslate( texture ) )
- {
- var key = info.GetKey( texture );
- if( string.IsNullOrEmpty( key ) ) return;
- if( !IsImageRegistered( key ) )
- {
- var name = texture.GetTextureName();
- //var format = "[" + texture.format.ToString() + "] ";
- var originalData = info.GetOrCreateOriginalData( texture );
- RegisterImageFromData( name, key, originalData );
- }
- }
- }
- finally
- {
- info.HasDumpedAlternativeTexture = true;
- }
- }
- finally
- {
- _imageHooksEnabled = true;
- }
- }
- private bool ShouldTranslate( Texture2D texture )
- {
- // convert to int so engine versions that does not have specific enums still work
- var format = (int)texture.format;
- // 1 = Alpha8
- // 9 = R16
- // 63 = R8
- return format != 1
- && format != 9
- && format != 63;
- }
- private string TranslateImmediate( object ui, string text, TextTranslationInfo info, bool ignoreComponentState )
- {
- // Get the trimmed text
- text = ( text ?? ui.GetText() ).TrimIfConfigured();
- if( !string.IsNullOrEmpty( text ) && IsTranslatable( text ) && ShouldTranslateTextComponent( ui, ignoreComponentState ) && !IsCurrentlySetting( info ) )
- {
- info?.Reset( text );
- var textKey = new TranslationKey( ui, text, ui.IsSpammingComponent(), false );
- // if we already have translation loaded in our _translatios dictionary, simply load it and set text
- string translation;
- if( TryGetTranslation( textKey, out translation ) )
- {
- if( !string.IsNullOrEmpty( translation ) )
- {
- SetTranslatedText( ui, textKey.Untemplate( translation ), info );
- return translation;
- }
- }
- else
- {
- if( UnityTextParsers.GameLogTextParser.CanApply( ui ) )
- {
- var result = UnityTextParsers.GameLogTextParser.Parse( text );
- if( result.Succeeded )
- {
- translation = TranslateOrQueueWebJobImmediateByParserResult( ui, result, false );
- if( translation != null )
- {
- SetTranslatedText( ui, translation, info );
- return translation;
- }
- }
- }
- }
- }
- return null;
- }
- /// <summary>
- /// Translates the string of a UI text or queues it up to be translated
- /// by the HTTP translation service.
- /// </summary>
- private string TranslateOrQueueWebJobImmediate( object ui, string text, TextTranslationInfo info, bool supportsStabilization, bool ignoreComponentState, TranslationContext context = null )
- {
- text = text ?? ui.GetText();
- // make sure text exists
- var originalText = text;
- if( context == null )
- {
- // Get the trimmed text
- text = text.TrimIfConfigured();
- }
- // Ensure that we actually want to translate this text and its owning UI element.
- if( !string.IsNullOrEmpty( text ) && IsTranslatable( text ) && ShouldTranslateTextComponent( ui, ignoreComponentState ) && !IsCurrentlySetting( info ) )
- {
- //Logger.Current.Debug( "START: " + ui.GetType().Name + ": " + text );
- info?.Reset( originalText );
- var isSpammer = ui.IsSpammingComponent();
- var textKey = new TranslationKey( ui, text, isSpammer, context != null );
- // if we already have translation loaded in our _translatios dictionary, simply load it and set text
- string translation;
- if( TryGetTranslation( textKey, out translation ) )
- {
- QueueNewUntranslatedForClipboard( textKey );
- if( !string.IsNullOrEmpty( translation ) )
- {
- if( context == null ) // never set text if operation is contextualized (only a part translation)
- {
- SetTranslatedText( ui, textKey.Untemplate( translation ), info );
- }
- return translation;
- }
- }
- else
- {
- if( context == null )
- {
- if( UnityTextParsers.GameLogTextParser.CanApply( ui ) )
- {
- var result = UnityTextParsers.GameLogTextParser.Parse( text );
- if( result.Succeeded )
- {
- translation = TranslateOrQueueWebJobImmediateByParserResult( ui, result, false );
- if( translation != null )
- {
- SetTranslatedText( ui, translation, info );
- return translation;
- }
- }
- }
- else if( UnityTextParsers.RichTextParser.CanApply( ui ) && IsBelowMaxLength( text ) )
- {
- var result = UnityTextParsers.RichTextParser.Parse( text );
- if( result.Succeeded )
- {
- var isWhitelisted = ui.IsWhitelistedForImmediateRichTextTranslation();
- translation = TranslateOrQueueWebJobImmediateByParserResult( ui, result, isWhitelisted );
- if( translation != null )
- {
- SetTranslatedText( ui, translation, info );
- return translation;
- }
- else if( isWhitelisted )
- {
- return null;
- }
- }
- }
- }
- if( supportsStabilization && context == null ) // never stabilize a text that is contextualized or that does not support stabilization
- {
- // if we dont know what text to translate it to, we need to figure it out.
- // this might take a while, so add the UI text component to the ongoing operations
- // list, so we dont start multiple operations for it, as its text might be constantly
- // changing.
- _ongoingOperations.Add( ui );
- // start a coroutine, that will execute once the string of the UI text has stopped
- // changing. For all texts except 'story' texts, this will add a delay for exactly
- // 0.5s to the translation. This is barely noticable.
- //
- // on the other hand, for 'story' texts, this will take the time that it takes
- // for the text to stop 'scrolling' in.
- try
- {
- StartCoroutine(
- WaitForTextStablization(
- ui: ui,
- delay: 1.0f, // 1 second to prevent '1 second tickers' from getting translated
- maxTries: 60, // 50 tries, about 1 minute
- currentTries: 0,
- onMaxTriesExceeded: () =>
- {
- _ongoingOperations.Remove( ui );
- },
- onTextStabilized: stabilizedText =>
- {
- _ongoingOperations.Remove( ui );
- originalText = stabilizedText;
- stabilizedText = stabilizedText.TrimIfConfigured();
- if( !string.IsNullOrEmpty( stabilizedText ) && IsTranslatable( stabilizedText ) )
- {
- var stabilizedTextKey = new TranslationKey( ui, stabilizedText, false );
- QueueNewUntranslatedForClipboard( stabilizedTextKey );
- info?.Reset( originalText );
- // once the text has stabilized, attempt to look it up
- if( TryGetTranslation( stabilizedTextKey, out translation ) )
- {
- if( !string.IsNullOrEmpty( translation ) )
- {
- // stabilized, no need to untemplate
- SetTranslatedText( ui, translation, info );
- }
- }
- else
- {
- if( context == null )
- {
- if( UnityTextParsers.GameLogTextParser.CanApply( ui ) )
- {
- var result = UnityTextParsers.GameLogTextParser.Parse( stabilizedText );
- if( result.Succeeded )
- {
- var translatedText = TranslateOrQueueWebJobImmediateByParserResult( ui, result, true );
- if( translatedText != null )
- {
- // stabilized, no need to untemplate
- SetTranslatedText( ui, translatedText, info );
- }
- return;
- }
- }
- else if( UnityTextParsers.RichTextParser.CanApply( ui ) && IsBelowMaxLength( stabilizedText ) )
- {
- var result = UnityTextParsers.RichTextParser.Parse( stabilizedText );
- if( result.Succeeded )
- {
- var translatedText = TranslateOrQueueWebJobImmediateByParserResult( ui, result, true );
- if( translatedText != null )
- {
- // stabilized, no need to untemplate
- SetTranslatedText( ui, translatedText, info );
- }
- return;
- }
- }
- }
- // Lets try not to spam a service that might not be there...
- if( _endpoint != null )
- {
- if( IsBelowMaxLength( stabilizedText ) )
- {
- if( !Settings.IsShutdown )
- {
- var job = GetOrCreateTranslationJobFor( ui, stabilizedTextKey, context );
- job.Components.Add( ui );
- }
- }
- }
- else
- {
- QueueNewUntranslatedForDisk( stabilizedTextKey );
- }
- }
- }
- } ) );
- }
- catch( Exception )
- {
- _ongoingOperations.Remove( ui );
- }
- }
- else if( !isSpammer || ( isSpammer && IsBelowMaxLengthStrict( text ) ) )
- {
- if( context != null )
- {
- // if there is a context, this is a part-translation, which means it is not a candidate for scrolling-in text
- if( _endpoint != null )
- {
- if( !Settings.IsShutdown )
- {
- // once the text has stabilized, attempt to look it up
- var job = GetOrCreateTranslationJobFor( ui, textKey, context );
- }
- }
- else
- {
- QueueNewUntranslatedForDisk( textKey );
- }
- }
- else
- {
- StartCoroutine(
- WaitForTextStablization(
- textKey: textKey,
- delay: 1.0f,
- onTextStabilized: () =>
- {
- // Lets try not to spam a service that might not be there...
- if( _endpoint != null )
- {
- // once the text has stabilized, attempt to look it up
- if( !Settings.IsShutdown )
- {
- if( !TryGetTranslation( textKey, out translation ) )
- {
- var job = GetOrCreateTranslationJobFor( ui, textKey, context );
- }
- }
- }
- else
- {
- QueueNewUntranslatedForDisk( textKey );
- }
- } ) );
- }
- }
- }
- }
- return null;
- }
- private string TranslateOrQueueWebJobImmediateByParserResult( object ui, ParserResult result, bool allowStartJob )
- {
- Dictionary<string, string> translations = new Dictionary<string, string>();
- // attempt to lookup ALL strings immediately; return result if possible; queue operations
- foreach( var kvp in result.Arguments )
- {
- var key = kvp.Key;
- var value = kvp.Value.TrimIfConfigured();
- if( !string.IsNullOrEmpty( value ) && IsTranslatable( value ) && IsBelowMaxLength( value ) )
- {
- string partTranslation;
- if( TryGetTranslation( value, out partTranslation ) )
- {
- translations.Add( key, partTranslation );
- }
- else if( allowStartJob )
- {
- // incomplete, must start job
- var context = new TranslationContext( ui, result );
- TranslateOrQueueWebJobImmediate( ui, value, null, false, true, context );
- }
- }
- else
- {
- // the value will do
- translations.Add( key, value );
- }
- }
- if( result.Arguments.Count == translations.Count )
- {
- return result.Untemplate( translations );
- }
- else
- {
- return null; // could not perform complete translation
- }
- }
- /// <summary>
- /// Utility method that allows me to wait to call an action, until
- /// the text has stopped changing. This is important for 'story'
- /// mode text, which 'scrolls' into place slowly.
- /// </summary>
- private IEnumerator WaitForTextStablization( object ui, float delay, int maxTries, int currentTries, Action<string> onTextStabilized, Action onMaxTriesExceeded )
- {
- yield return 0; // wait a single frame to allow any external plugins to complete their hooking logic
- bool succeeded = false;
- while( currentTries < maxTries ) // shortcircuit
- {
- var beforeText = ui.GetText();
- yield return new WaitForSeconds( delay );
- var afterText = ui.GetText();
- //Logger.Current.Debug( "WAITING: " + ui.GetType().Name + ": " + afterText );
- if( beforeText == afterText )
- {
- onTextStabilized( afterText );
- succeeded = true;
- break;
- }
- currentTries++;
- }
- if( !succeeded )
- {
- onMaxTriesExceeded();
- }
- }
- /// <summary>
- /// Utility method that allows me to wait to call an action, until
- /// the text has stopped changing. This is important for 'story'
- /// mode text, which 'scrolls' into place slowly. This version is
- /// for global text, where the component cannot tell us if the text
- /// has changed itself.
- /// </summary>
- private IEnumerator WaitForTextStablization( TranslationKey textKey, float delay, Action onTextStabilized, Action onFailed = null )
- {
- var text = textKey.GetDictionaryLookupKey();
- if( !_immediatelyTranslating.Contains( text ) )
- {
- _immediatelyTranslating.Add( text );
- try
- {
- yield return new WaitForSeconds( delay );
- bool succeeded = true;
- foreach( var otherImmediatelyTranslating in _immediatelyTranslating )
- {
- if( text != otherImmediatelyTranslating )
- {
- if( text.RemindsOf( otherImmediatelyTranslating ) )
- {
- succeeded = false;
- break;
- }
- }
- }
- if( succeeded )
- {
- onTextStabilized();
- }
- else
- {
- onFailed?.Invoke();
- }
- }
- finally
- {
- _immediatelyTranslating.Remove( text );
- }
- }
- }
- private IEnumerator DelayForSeconds( float delay, Action onContinue )
- {
- yield return new WaitForSeconds( delay );
- onContinue();
- }
- void Awake()
- {
- if( !_initialized )
- {
- _initialized = true;
- try
- {
- Initialize();
- ManualHook();
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An unexpected error occurred during plugin initialization." );
- }
- }
- }
- void Start()
- {
- try
- {
- HooksSetup.InstallOverrideTextHooks();
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An unexpected error occurred during plugin start." );
- }
- }
- void Update()
- {
- try
- {
- // perform this check every 100 frames!
- if( Time.frameCount % 100 == 0 )
- {
- ConnectionTrackingWebClient.CheckServicePoints();
- }
- if( Features.SupportsClipboard )
- {
- CopyToClipboard();
- }
- if( !Settings.IsShutdown )
- {
- UpdateSpriteRenderers();
- PeriodicResetFrameCheck();
- IncrementBatchOperations();
- ResetThresholdTimerIfRequired();
- KickoffTranslations();
- FinishTranslations();
- if( ClrTypes.AdvEngine != null && _nextAdvUpdate.HasValue && Time.time > _nextAdvUpdate )
- {
- _nextAdvUpdate = null;
- UpdateUtageText();
- }
- }
- if( Input.anyKey )
- {
- var isAltPressed = Input.GetKey( KeyCode.LeftAlt ) || Input.GetKey( KeyCode.RightAlt );
- if( Settings.EnablePrintHierarchy && isAltPressed && Input.GetKeyDown( KeyCode.Y ) )
- {
- PrintObjects();
- }
- else if( isAltPressed && Input.GetKeyDown( KeyCode.T ) )
- {
- ToggleTranslation();
- }
- else if( isAltPressed && Input.GetKeyDown( KeyCode.F ) )
- {
- ToggleFont();
- }
- else if( isAltPressed && Input.GetKeyDown( KeyCode.D ) )
- {
- DumpUntranslated();
- }
- else if( isAltPressed && Input.GetKeyDown( KeyCode.R ) )
- {
- ReloadTranslations();
- }
- else if( isAltPressed && Input.GetKeyDown( KeyCode.U ) )
- {
- ManualHook();
- }
- else if( isAltPressed && Input.GetKeyDown( KeyCode.Q ) )
- {
- RebootPlugin();
- }
- else if( isAltPressed && ( Input.GetKeyDown( KeyCode.Alpha0 ) || Input.GetKeyDown( KeyCode.Keypad0 ) ) )
- {
- if( _window != null )
- {
- _window.IsShown = !_window.IsShown;
- }
- }
- }
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An error occurred in Update callback. " );
- }
- }
- void OnGUI()
- {
- if( _window != null )
- {
- try
- {
- DisableAutoTranslator();
- if( _window.IsShown ) _window.OnGUI();
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An error occurred in XUnity.AutoTranslator UI. Disabling the UI." );
- _window = null;
- }
- finally
- {
- EnableAutoTranslator();
- }
- }
- }
- private void RebootPlugin()
- {
- if( Settings.IsShutdown )
- {
- if( !Settings.IsShutdownFatal )
- {
- _consecutiveErrors = 0;
- Settings.IsShutdown = false;
- XuaLogger.Current.Info( "Rebooted Auto Translator." );
- }
- else
- {
- XuaLogger.Current.Info( "Cannot reboot Auto Translator because the error that caused the shutdown is bad behaviour by the game." );
- }
- }
- else
- {
- XuaLogger.Current.Info( "Cannot reboot Auto Translator because it has not been shut down." );
- }
- }
- private void KickoffTranslations()
- {
- if( _endpoint == null ) return;
- if( Settings.EnableBatching && _endpoint.Endpoint.MaxTranslationsPerRequest > 1 && !_batchLogicHasFailed && _unstartedJobs.Count > 1 && _availableBatchOperations > 0 )
- {
- while( _unstartedJobs.Count > 0 && _availableBatchOperations > 0 )
- {
- if( _endpoint.IsBusy ) break;
- var kvps = _unstartedJobs.Take( _endpoint.Endpoint.MaxTranslationsPerRequest ).ToList();
- //var batch = new TranslationBatch();
- var untranslatedTexts = new List<string>();
- var jobs = new List<TranslationJob>();
- foreach( var kvp in kvps )
- {
- var key = kvp.Key;
- var job = kvp.Value;
- _unstartedJobs.Remove( key );
- if( !job.AnyComponentsStillHasOriginalUntranslatedTextOrContextual() ) continue;
- jobs.Add( job );
- untranslatedTexts.Add( job.Key.GetDictionaryLookupKey() );
- _ongoingJobs[ key ] = job;
- }
- if( jobs.Count > 0 )
- {
- _availableBatchOperations--;
- var jobsArray = jobs.ToArray();
- foreach( var untranslatedText in untranslatedTexts )
- {
- XuaLogger.Current.Debug( "Starting translation for: " + untranslatedText );
- }
- StartCoroutine(
- _endpoint.Translate(
- untranslatedTexts.ToArray(),
- Settings.FromLanguage,
- Settings.Language,
- translatedText => OnBatchTranslationCompleted( jobsArray, translatedText ),
- ( msg, e ) => OnTranslationFailed( jobsArray, msg, e ) ) );
- }
- }
- }
- else
- {
- while( _unstartedJobs.Count > 0 )
- {
- if( _endpoint.IsBusy ) break;
- // ERROR ERROR ERROR --- MEGA BUG MEGA BUG, MUST REMOVE FROM _unstartedJobs INSIDE LOOP!!!!
- var kvp = _unstartedJobs.FirstOrDefault();
- var key = kvp.Key;
- var job = kvp.Value;
- _unstartedJobs.Remove( key );
- // lets see if the text should still be translated before kicking anything off
- if( !job.AnyComponentsStillHasOriginalUntranslatedTextOrContextual() ) continue;
- _ongoingJobs[ key ] = job;
- var untranslatedText = job.Key.GetDictionaryLookupKey();
- XuaLogger.Current.Debug( "Starting translation for: " + untranslatedText );
- StartCoroutine(
- _endpoint.Translate(
- new[] { untranslatedText },
- Settings.FromLanguage,
- Settings.Language,
- translatedText => OnSingleTranslationCompleted( job, translatedText ),
- ( msg, e ) => OnTranslationFailed( new[] { job }, msg, e ) ) );
- }
- }
- }
- private void OnBatchTranslationCompleted( TranslationJob[] jobs, string[] translatedTexts )
- {
- _consecutiveErrors = 0;
- var succeeded = jobs.Length == translatedTexts.Length;
- if( succeeded )
- {
- for( int i = 0 ; i < jobs.Length ; i++ )
- {
- Settings.TranslationCount++;
- var job = jobs[ i ];
- var translatedText = translatedTexts[ i ];
- if( !string.IsNullOrEmpty( translatedText ) )
- {
- if( Settings.ForceSplitTextAfterCharacters > 0 )
- {
- translatedText = translatedText.SplitToLines( Settings.ForceSplitTextAfterCharacters, '\n', ' ', ' ' );
- }
- job.TranslatedText = job.Key.RepairTemplate( translatedText );
- QueueNewTranslationForDisk( job.Key, translatedText );
- _completedJobs.Add( job );
- }
- AddTranslation( job.Key, job.TranslatedText );
- job.State = TranslationJobState.Succeeded;
- _ongoingJobs.Remove( job.Key.GetDictionaryLookupKey() );
- XuaLogger.Current.Debug( $"Translation for '{job.Key.GetDictionaryLookupKey()}' succeded. Result: {translatedText}" );
- }
- }
- else
- {
- // might as well re-add all translation jobs, and never do this again!
- _batchLogicHasFailed = true;
- for( int i = 0 ; i < jobs.Length ; i++ )
- {
- Settings.TranslationCount++;
- var job = jobs[ i ];
- var key = job.Key.GetDictionaryLookupKey();
- if( !_unstartedJobs.ContainsKey( key ) )
- {
- _unstartedJobs[ key ] = job;
- }
- _ongoingJobs.Remove( key );
- }
- XuaLogger.Current.Error( "A batch operation failed. Disabling batching and restarting failed jobs." );
- }
- if( !Settings.IsShutdown )
- {
- if( Settings.TranslationCount > Settings.MaxTranslationsBeforeShutdown )
- {
- Settings.IsShutdown = true;
- Settings.IsShutdownFatal = true;
- XuaLogger.Current.Error( $"Maximum translations ({Settings.MaxTranslationsBeforeShutdown}) per session reached. Shutting plugin down." );
- _unstartedJobs.Clear();
- _completedJobs.Clear();
- _ongoingJobs.Clear();
- }
- }
- }
- private void OnSingleTranslationCompleted( TranslationJob job, string[] translatedTextArray )
- {
- var translatedText = translatedTextArray[ 0 ];
- Settings.TranslationCount++;
- XuaLogger.Current.Debug( $"Translation for '{job.Key.GetDictionaryLookupKey()}' succeded. Result: {translatedText}" );
- _consecutiveErrors = 0;
- if( !string.IsNullOrEmpty( translatedText ) )
- {
- if( Settings.ForceSplitTextAfterCharacters > 0 )
- {
- translatedText = translatedText.SplitToLines( Settings.ForceSplitTextAfterCharacters, '\n', ' ', ' ' );
- }
- job.TranslatedText = job.Key.RepairTemplate( translatedText );
- QueueNewTranslationForDisk( job.Key, translatedText );
- _completedJobs.Add( job );
- }
- AddTranslation( job.Key, job.TranslatedText );
- job.State = TranslationJobState.Succeeded;
- _ongoingJobs.Remove( job.Key.GetDictionaryLookupKey() );
- if( !Settings.IsShutdown )
- {
- if( Settings.TranslationCount > Settings.MaxTranslationsBeforeShutdown )
- {
- Settings.IsShutdown = true;
- Settings.IsShutdownFatal = true;
- XuaLogger.Current.Error( $"Maximum translations ({Settings.MaxTranslationsBeforeShutdown}) per session reached. Shutting plugin down." );
- _unstartedJobs.Clear();
- _completedJobs.Clear();
- _ongoingJobs.Clear();
- }
- }
- }
- private void OnTranslationFailed( TranslationJob[] jobs, string error, Exception e )
- {
- if( e == null )
- {
- XuaLogger.Current.Error( error );
- }
- else
- {
- XuaLogger.Current.Error( e, error );
- }
- _consecutiveErrors++;
- if( jobs.Length == 1 )
- {
- Settings.TranslationCount++; // counts as a translation
- foreach( var job in jobs )
- {
- job.State = TranslationJobState.Failed;
- _ongoingJobs.Remove( job.Key.GetDictionaryLookupKey() );
- }
- }
- else
- {
- _batchLogicHasFailed = true;
- for( int i = 0 ; i < jobs.Length ; i++ )
- {
- Settings.TranslationCount++;
- var job = jobs[ i ];
- var key = job.Key.GetDictionaryLookupKey();
- if( !_unstartedJobs.ContainsKey( key ) )
- {
- _unstartedJobs[ key ] = job;
- }
- _ongoingJobs.Remove( key );
- }
- XuaLogger.Current.Error( "A batch operation failed. Disabling batching and restarting failed jobs." );
- }
- if( !Settings.IsShutdown )
- {
- if( _consecutiveErrors >= Settings.MaxErrors )
- {
- Settings.IsShutdown = true;
- XuaLogger.Current.Error( $"{Settings.MaxErrors} or more consecutive errors occurred. Shutting down plugin." );
- _unstartedJobs.Clear();
- _completedJobs.Clear();
- _ongoingJobs.Clear();
- }
- }
- }
- private void FinishTranslations()
- {
- if( _completedJobs.Count > 0 )
- {
- for( int i = _completedJobs.Count - 1 ; i >= 0 ; i-- )
- {
- var job = _completedJobs[ i ];
- _completedJobs.RemoveAt( i );
- foreach( var component in job.Components )
- {
- // update the original text, but only if it has not been chaanged already for some reason (could be other translator plugin or game itself)
- try
- {
- var text = component.GetText().TrimIfConfigured();
- if( text == job.Key.OriginalText )
- {
- var info = component.GetOrCreateTextTranslationInfo();
- SetTranslatedText( component, job.TranslatedText, info );
- }
- }
- catch( NullReferenceException )
- {
- // might fail if compoent is no longer associated to game
- }
- }
- // handle each context
- foreach( var context in job.Contexts )
- {
- // are all jobs within this context completed? If so, we can set the text
- if( context.Jobs.All( x => x.State == TranslationJobState.Succeeded ) )
- {
- try
- {
- var text = context.Component.GetText().TrimIfConfigured();
- var result = context.Result;
- Dictionary<string, string> translations = new Dictionary<string, string>();
- var translatedText = TranslateOrQueueWebJobImmediateByParserResult( context.Component, result, false );
- if( !string.IsNullOrEmpty( translatedText ) )
- {
- if( result.PersistCombinedResult && !_translations.ContainsKey( context.Result.OriginalText ) )
- {
- AddTranslation( context.Result.OriginalText, translatedText );
- QueueNewTranslationForDisk( context.Result.OriginalText, translatedText );
- }
- if( text == result.OriginalText )
- {
- if( translatedText != null )
- {
- var info = context.Component.GetOrCreateTextTranslationInfo();
- SetTranslatedText( context.Component, translatedText, info );
- }
- }
- }
- }
- catch( NullReferenceException )
- {
- }
- }
- }
- // Utage support
- if( ClrTypes.AdvEngine != null
- && job.OriginalSources.Any( x => ClrTypes.AdvCommand.IsAssignableFrom( x.GetType() ) ) )
- {
- _nextAdvUpdate = Time.time + 0.5f;
- }
- }
- }
- }
- private void UpdateUtageText()
- {
- // After an object is destroyed, an equality check with null will return true. The variable does not go to null, you can still call GetInstanceID() on it, but the "==" operator is overloaded and behaves as expected.
- if( _advEngine == null || _advEngine?.gameObject == null )
- {
- _advEngine = (Component)GameObject.FindObjectOfType( Constants.ClrTypes.AdvEngine );
- }
- if( _advEngine != null )
- {
- AccessTools.Method( Constants.ClrTypes.AdvEngine, "ChangeLanguage" )?.Invoke( _advEngine, new object[ 0 ] );
- }
- }
- private void ReloadTranslations()
- {
- LoadTranslations();
- var context = new TextureReloadContext();
- foreach( var kvp in ObjectReferenceMapper.GetAllRegisteredObjects() )
- {
- var ui = kvp.Key;
- try
- {
- if( ui is Component component )
- {
- if( component.gameObject?.activeSelf ?? false )
- {
- var tti = kvp.Value as TextTranslationInfo;
- if( tti != null && !string.IsNullOrEmpty( tti.OriginalText ) )
- {
- var key = new TranslationKey( kvp.Key, tti.OriginalText, false );
- if( TryGetTranslation( key, out string translatedText ) && !string.IsNullOrEmpty( translatedText ) )
- {
- SetTranslatedText( kvp.Key, translatedText, tti ); // no need to untemplatize the translated text
- }
- }
- }
- }
- if( Settings.EnableTextureTranslation )
- {
- TranslateTexture( ui, context );
- }
- }
- catch( Exception )
- {
- // not super pretty, no...
- ObjectReferenceMapper.Remove( ui );
- }
- }
- }
- private string CalculateDumpFileName()
- {
- int idx = 0;
- string fileName = null;
- do
- {
- idx++;
- fileName = $"UntranslatedDump{idx}.txt";
- }
- while( File.Exists( fileName ) );
- return fileName;
- }
- private void DumpUntranslated()
- {
- if( _newUntranslated.Count > 0 )
- {
- using( var stream = File.Open( CalculateDumpFileName(), FileMode.Append, FileAccess.Write ) )
- using( var writer = new StreamWriter( stream, Encoding.UTF8 ) )
- {
- foreach( var untranslated in _newUntranslated )
- {
- writer.WriteLine( TextHelper.Encode( untranslated ) + '=' );
- }
- writer.Flush();
- }
- _newUntranslated.Clear();
- }
- }
- private void ToggleFont()
- {
- if( _hasOverrideFont )
- {
- _overrideFont = !_overrideFont;
- var objects = ObjectReferenceMapper.GetAllRegisteredObjects();
- XuaLogger.Current.Info( $"Toggling fonts of {objects.Count} objects." );
- if( _overrideFont )
- {
- // make sure we use the translated version of all texts
- foreach( var kvp in objects )
- {
- var tti = kvp.Value as TextTranslationInfo;
- if( tti != null )
- {
- var ui = kvp.Key;
- try
- {
- if( ( ui as Component )?.gameObject?.activeSelf ?? false )
- {
- tti?.ChangeFont( ui );
- }
- }
- catch( Exception )
- {
- // not super pretty, no...
- ObjectReferenceMapper.Remove( ui );
- }
- }
- }
- }
- else
- {
- // make sure we use the original version of all texts
- foreach( var kvp in objects )
- {
- var tti = kvp.Value as TextTranslationInfo;
- var ui = kvp.Key;
- try
- {
- if( ( ui as Component )?.gameObject?.activeSelf ?? false )
- {
- tti?.UnchangeFont( ui );
- }
- }
- catch( Exception )
- {
- // not super pretty, no...
- ObjectReferenceMapper.Remove( ui );
- }
- }
- }
- }
- }
- private void ToggleTranslation()
- {
- _isInTranslatedMode = !_isInTranslatedMode;
- var objects = ObjectReferenceMapper.GetAllRegisteredObjects();
- XuaLogger.Current.Info( $"Toggling translations of {objects.Count} objects." );
- if( _isInTranslatedMode )
- {
- // make sure we use the translated version of all texts
- foreach( var kvp in objects )
- {
- var ui = kvp.Key;
- try
- {
- if( ui is Component component )
- {
- if( component.gameObject?.activeSelf ?? false )
- {
- var tti = kvp.Value as TextTranslationInfo;
- if( tti != null && tti.IsTranslated )
- {
- SetText( ui, tti.TranslatedText, true, tti );
- }
- }
- }
- if( Settings.EnableTextureTranslation && Settings.EnableTextureToggling )
- {
- TranslateTexture( ui, null );
- }
- }
- catch( Exception )
- {
- // not super pretty, no...
- ObjectReferenceMapper.Remove( ui );
- }
- }
- }
- else
- {
- // make sure we use the original version of all texts
- foreach( var kvp in objects )
- {
- var ui = kvp.Key;
- try
- {
- if( ui is Component component )
- {
- if( component.gameObject?.activeSelf ?? false )
- {
- var tti = kvp.Value as TextTranslationInfo;
- if( tti != null && tti.IsTranslated )
- {
- SetText( ui, tti.OriginalText, true, tti );
- }
- }
- }
- if( Settings.EnableTextureTranslation && Settings.EnableTextureToggling )
- {
- TranslateTexture( ui, null );
- }
- }
- catch( Exception )
- {
- // not super pretty, no...
- ObjectReferenceMapper.Remove( ui );
- }
- }
- }
- }
- private void CopyToClipboard()
- {
- if( Settings.CopyToClipboard
- && _textsToCopyToClipboardOrdered.Count > 0
- && Time.realtimeSinceStartup - _clipboardUpdated > Settings.ClipboardDebounceTime )
- {
- try
- {
- var builder = new StringBuilder();
- foreach( var text in _textsToCopyToClipboardOrdered )
- {
- if( text.Length + builder.Length > Settings.MaxClipboardCopyCharacters ) break;
- builder.AppendLine( text );
- }
- TextEditor editor = (TextEditor)GUIUtility.GetStateObject( typeof( TextEditor ), GUIUtility.keyboardControl );
- editor.text = builder.ToString();
- editor.SelectAll();
- editor.Copy();
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An error while copying text to clipboard." );
- }
- finally
- {
- _textsToCopyToClipboard.Clear();
- _textsToCopyToClipboardOrdered.Clear();
- }
- }
- }
- private void PrintObjects()
- {
- using( var stream = File.Open( Path.Combine( Environment.CurrentDirectory, "hierarchy.txt" ), FileMode.Create ) )
- using( var writer = new StreamWriter( stream ) )
- {
- foreach( var root in GetAllRoots() )
- {
- TraverseChildren( writer, root, "" );
- }
- writer.Flush();
- }
- }
- private void ManualHook()
- {
- ManualHookForComponents();
- ManualHookForTextures();
- }
- private void ManualHookForComponents()
- {
- foreach( var root in GetAllRoots() )
- {
- TraverseChildrenManualHook( root );
- }
- }
- private void ManualHookForTextures()
- {
- if( Settings.EnableTextureScanOnSceneLoad && ( Settings.EnableTextureTranslation || Settings.EnableTextureDumping ) )
- {
- // scan all textures and update
- var textures = Resources.FindObjectsOfTypeAll<Texture2D>();
- foreach( var texture in textures )
- {
- Hook_ImageChanged( texture, false );
- }
- //// scan all components and set dirty
- //var components = GameObject.FindObjectsOfType<Component>();
- //foreach( var component in components )
- //{
- // component.SetAllDirtyEx();
- //}
- }
- }
- private IEnumerable<GameObject> GetAllRoots()
- {
- var objects = GameObject.FindObjectsOfType<GameObject>();
- foreach( var obj in objects )
- {
- if( obj.transform != null && obj.transform.parent == null )
- {
- yield return obj;
- }
- }
- }
- private void TraverseChildren( StreamWriter writer, GameObject obj, string identation )
- {
- if( obj != null )
- {
- var layer = LayerMask.LayerToName( obj.layer );
- var components = string.Join( ", ", obj.GetComponents<Component>().Select( x => x?.GetType()?.Name ).Where( x => x != null ).ToArray() );
- var line = string.Format( "{0,-50} {1,100}",
- identation + obj.name + " [" + layer + "]",
- components );
- writer.WriteLine( line );
- if( obj.transform != null )
- {
- for( int i = 0 ; i < obj.transform.childCount ; i++ )
- {
- var child = obj.transform.GetChild( i );
- TraverseChildren( writer, child.gameObject, identation + " " );
- }
- }
- }
- }
- private void TraverseChildrenManualHook( GameObject obj )
- {
- if( obj != null )
- {
- var components = obj.GetComponents<Component>();
- foreach( var component in components )
- {
- if( component.IsKnownTextType() )
- {
- Hook_TextChanged( component, false );
- }
- if( Settings.EnableTextureTranslation || Settings.EnableTextureDumping )
- {
- if( component.IsKnownImageType() )
- {
- Hook_ImageChangedOnComponent( component, null, false, false );
- }
- }
- }
- if( obj.transform != null )
- {
- for( int i = 0 ; i < obj.transform.childCount ; i++ )
- {
- var child = obj.transform.GetChild( i );
- TraverseChildrenManualHook( child.gameObject );
- }
- }
- }
- }
- /// <summary>
- /// Temporarily disables the Auto Translator functionality.
- /// </summary>
- public void DisableAutoTranslator()
- {
- _temporarilyDisabled = true;
- }
- /// <summary>
- /// Re-enables the Auto Translator functionality after
- /// having disabled it thorugh DisableAutoTranslator
- /// </summary>
- public void EnableAutoTranslator()
- {
- _temporarilyDisabled = false;
- }
- void OnApplicationQuit()
- {
- if( _configuredEndpoints == null ) return;
- foreach( var ce in _configuredEndpoints )
- {
- try
- {
- if( ce.Endpoint is IDisposable disposable ) disposable.Dispose();
- }
- catch( Exception e )
- {
- XuaLogger.Current.Error( e, "An error occurred while disposing endpoint." );
- }
- }
- }
- }
- internal static class SceneManagerLoader
- {
- public static void EnableSceneLoadScanInternal( AutoTranslationPlugin plugin )
- {
- // specified in own method, because of chance that this has changed through Unity lifetime
- SceneManager.sceneLoaded += ( arg1, arg2 ) => plugin.OnLevelWasLoadedFromSceneManager( arg1.buildIndex );
- }
- }
- }
|