From 6e9e4aaf841f7fd91ac6685466179882233cf0b9 Mon Sep 17 00:00:00 2001
From: FaDeOkno <logkedr18@gmail.com>
Date: Tue, 6 Aug 2024 21:19:31 +0400
Subject: [PATCH] =?UTF-8?q?=D0=9F=D0=B5=D1=80=D0=B5=D0=B2=D0=BE=D0=B4?=
 =?UTF-8?q?=D1=87=D0=B8=D0=BA=D0=B8=20=D0=B7=D0=B0=D1=80=D0=B0=D0=B1=D0=BE?=
 =?UTF-8?q?=D1=82=D0=B0=D0=BB=D0=B8=20=D0=BA=D0=B0=D0=BA=20=D0=BD=D0=B0?=
 =?UTF-8?q?=D0=B4=D0=BE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../Language/UI/LanguageMenuWindow.xaml.cs    | 39 ++++++++-----------
 .../Languages/LanguageMenuUIController.cs     |  1 -
 .../ADT/Language/TranslatorSystem.cs          | 21 ++++++++--
 .../Components/HandheldTranslatorComponent.cs |  1 +
 .../Language/Systems/SharedLanguageSystem.cs  | 18 ++++++++-
 .../ADT/prototypes/Actions/language-menu.ftl  |  4 +-
 .../Entities/Objects/Device/translators.ftl   |  4 +-
 7 files changed, 56 insertions(+), 32 deletions(-)

diff --git a/Content.Client/ADT/Language/UI/LanguageMenuWindow.xaml.cs b/Content.Client/ADT/Language/UI/LanguageMenuWindow.xaml.cs
index 087a8a93a3d..095e5148ff9 100644
--- a/Content.Client/ADT/Language/UI/LanguageMenuWindow.xaml.cs
+++ b/Content.Client/ADT/Language/UI/LanguageMenuWindow.xaml.cs
@@ -41,20 +41,19 @@ public void UpdateState(LanguageMenuStateMessage state)
     {
         CurrentLanguageLabel.Text = Loc.GetString("language-menu-current-language", ("language", _language.GetLanguage(state.CurrentLanguage).LocalizedName));
 
-        //OptionsList.RemoveAllChildren();
         List<string> options = new();
         List<Option> optionList = _optionLists;
 
         options.AddRange(state.Options);
-        options.AddRange(state.TranslatorOptions);
 
-        List<string> translatorOptions = new();
-        translatorOptions.AddRange(state.TranslatorOptions);
+        List<string> translatorOptions = state.TranslatorOptions;
+        List<EntryState> entries = _entries.ToList();
+        
         foreach (var lng in state.Options)
         {
             translatorOptions.Remove(lng);
         }
-        foreach (var entry in _entries)
+        foreach (var entry in entries)
         {
             if (state.Options.Contains(entry.Language))
             {
@@ -63,7 +62,7 @@ public void UpdateState(LanguageMenuStateMessage state)
             }
             else if (state.TranslatorOptions.Contains(entry.Language))
             {
-                options.Remove(entry.Language);
+                translatorOptions.Remove(entry.Language);
                 continue;
             }
             else
@@ -85,7 +84,14 @@ public void UpdateState(LanguageMenuStateMessage state)
         {
             foreach (var language in options)
             {
-                AddLanguageEntry(language, translatorOptions.Contains(language));
+                AddLanguageEntry(language);
+            }
+        }
+        if (translatorOptions.Count > 0)
+        {
+            foreach (var language in translatorOptions)
+            {
+                AddLanguageEntry(language, true);
             }
         }
 
@@ -99,14 +105,6 @@ public void UpdateState(LanguageMenuStateMessage state)
                     entry.Button.Text = Loc.GetString("language-choose-button-chosen");
                 if (!_language.CanSpeak(_entManager.GetEntity(state.ComponentOwner), _language.GetLanguage(entry.Language)))
                     entry.Button.Text = Loc.GetString("language-choose-button-cannot");
-
-                if (translatorOptions.Contains(entry.Language))
-                {
-                    entry.Button.Text = "П | " + entry.Button.Text;
-                    entry.Button.ToolTip = Loc.GetString("language-choose-button-tooltip-translator");
-                }
-                else
-                    entry.Button.ToolTip = Loc.GetString("language-choose-button-tooltip-known");
             }
         }
     }
@@ -137,13 +135,10 @@ private void AddLanguageEntry(string language, bool translator = false)
             var button = new Button();
             button.Text = Loc.GetString("language-choose-button");
             button.OnPressed += _ => OnLanguageChosen(language);
-            if (translator)
-            {
-                //button.Text = "П | " + button.Text;
-                button.ToolTip = Loc.GetString("language-choose-button-tooltip-translator");
-            }
-            else
-                button.ToolTip = Loc.GetString("language-choose-button-tooltip-known");
+            button.ToolTip =
+                translator ?
+                Loc.GetString("language-choose-button-tooltip-translator") :
+                Loc.GetString("language-choose-button-tooltip-known");
 
             state.Button = button;
 
diff --git a/Content.Client/UserInterface/Systems/Languages/LanguageMenuUIController.cs b/Content.Client/UserInterface/Systems/Languages/LanguageMenuUIController.cs
index 2367dce4b8a..b50648d72a4 100644
--- a/Content.Client/UserInterface/Systems/Languages/LanguageMenuUIController.cs
+++ b/Content.Client/UserInterface/Systems/Languages/LanguageMenuUIController.cs
@@ -50,7 +50,6 @@ private void ToggleLanguagesMenu()
             _menu = UIManager.CreateWindow<LanguageMenuWindow>();
             _menu.OnClose += OnWindowClosed;
             _menu.OnOpen += OnWindowOpen;
-            //_menu.OnPlayEmote += OnPlayEmote;
 
             if (LanguagesButton != null)
                 LanguagesButton.SetClickPressed(true);
diff --git a/Content.Server/ADT/Language/TranslatorSystem.cs b/Content.Server/ADT/Language/TranslatorSystem.cs
index 97df66e6203..4eb8708b1d3 100644
--- a/Content.Server/ADT/Language/TranslatorSystem.cs
+++ b/Content.Server/ADT/Language/TranslatorSystem.cs
@@ -22,8 +22,8 @@ public override void Initialize()
         SubscribeLocalEvent<HandheldTranslatorComponent, ActivateInWorldEvent>(OnTranslatorActivateInWorld);
         SubscribeLocalEvent<HandheldTranslatorComponent, UseInHandEvent>(OnTranslatorUseInHand);
 
-        SubscribeLocalEvent<HandheldTranslatorComponent, EquippedHandEvent>(OnPickUp);
-        SubscribeLocalEvent<HandheldTranslatorComponent, DroppedEvent>(OnDrop);
+        SubscribeLocalEvent<HandheldTranslatorComponent, GotEquippedHandEvent>(OnPickUp);
+        SubscribeLocalEvent<HandheldTranslatorComponent, GotUnequippedHandEvent>(OnDrop);
 
         SubscribeLocalEvent<HandheldTranslatorComponent, PowerCellSlotEmptyEvent>(OnPowerCellSlotEmpty);
     }
@@ -49,6 +49,8 @@ private void OnTranslatorUseInHand(EntityUid translator, HandheldTranslatorCompo
         Dirty(translator, component);
 
         ToggleTranslator(translator);
+        component.User = component.Enabled ? args.User : null;
+
         if (_language.GetLanguages(args.User, out var understood, out _, out var translatorUnderstood, out _, out var current))
         {
             var ev = new LanguageMenuStateMessage(GetNetEntity(args.User), current, understood, translatorUnderstood);
@@ -56,10 +58,12 @@ private void OnTranslatorUseInHand(EntityUid translator, HandheldTranslatorCompo
         }
     }
 
-    private void OnPickUp(EntityUid translator, HandheldTranslatorComponent component, EquippedHandEvent args)
+    private void OnPickUp(EntityUid translator, HandheldTranslatorComponent component, GotEquippedHandEvent args)
     {
         Dirty(translator, component);
 
+        component.User = args.User;
+
         if (_language.GetLanguages(args.User, out var understood, out _, out var translatorUnderstood, out _, out var current))
         {
             var ev = new LanguageMenuStateMessage(GetNetEntity(args.User), current, understood, translatorUnderstood);
@@ -67,10 +71,15 @@ private void OnPickUp(EntityUid translator, HandheldTranslatorComponent componen
         }
     }
 
-    private void OnDrop(EntityUid translator, HandheldTranslatorComponent component, DroppedEvent args)
+    private void OnDrop(EntityUid translator, HandheldTranslatorComponent component, GotUnequippedHandEvent args)
     {
         Dirty(translator, component);
 
+        if (component.User.HasValue)
+            _language.SelectDefaultLanguage(component.User.Value);
+
+        component.User = null;
+
         if (_language.GetLanguages(args.User, out var understood, out _, out var translatorUnderstood, out _, out var current))
         {
             var ev = new LanguageMenuStateMessage(GetNetEntity(args.User), current, understood, translatorUnderstood);
@@ -90,6 +99,8 @@ private void ToggleTranslator(EntityUid uid, HandheldTranslatorComponent? compon
             component.Enabled = !component.Enabled;
             var popupMessage = Loc.GetString(component.Enabled ? "translator-component-turnon" : "translator-component-shutoff", ("translator", component.Owner));
             _popup.PopupEntity(popupMessage, component.Owner);
+            if (!component.Enabled && component.User.HasValue)
+                _language.SelectDefaultLanguage(component.User.Value);
         }
 
         Dirty(uid, component);
@@ -99,6 +110,8 @@ private void OnPowerCellSlotEmpty(EntityUid translator, HandheldTranslatorCompon
     {
         component.Enabled = false;
 
+        component.User = null;
+
         Dirty(translator, component);
         OnAppearanceChange(translator, component);
     }
diff --git a/Content.Shared/ADT/Language/Components/HandheldTranslatorComponent.cs b/Content.Shared/ADT/Language/Components/HandheldTranslatorComponent.cs
index fa8f2ee7f95..354f924692a 100644
--- a/Content.Shared/ADT/Language/Components/HandheldTranslatorComponent.cs
+++ b/Content.Shared/ADT/Language/Components/HandheldTranslatorComponent.cs
@@ -32,4 +32,5 @@ public sealed partial class HandheldTranslatorComponent : Component
     [DataField]
     public bool Enabled = false;
 
+    public EntityUid? User;
 }
diff --git a/Content.Shared/ADT/Language/Systems/SharedLanguageSystem.cs b/Content.Shared/ADT/Language/Systems/SharedLanguageSystem.cs
index d0f3cffa67e..7e5748008d2 100644
--- a/Content.Shared/ADT/Language/Systems/SharedLanguageSystem.cs
+++ b/Content.Shared/ADT/Language/Systems/SharedLanguageSystem.cs
@@ -150,6 +150,22 @@ public void SelectLanguage(NetEntity ent, string language, LanguageSpeakerCompon
         RaiseNetworkEvent(new LanguageChosenMessage(ent, language));
     }
 
+    public void SelectDefaultLanguage(EntityUid uid, LanguageSpeakerComponent? component = null)
+    {
+        if (!Resolve(uid, ref component))
+            return;
+
+        var language = component.SpokenLanguages.FirstOrDefault("Universal");
+
+        GetLanguages(uid, out _, out _, out var translator, out _, out _);
+        component.CurrentLanguage = language;
+
+        Dirty(uid, component);
+
+        RaiseNetworkEvent(new LanguageMenuStateMessage(GetNetEntity(uid), language, component.UnderstoodLanguages, translator));
+        RaiseNetworkEvent(new LanguageChosenMessage(GetNetEntity(uid), language));
+    }
+
     public bool GetLanguages(
         EntityUid? player,
         out List<string> understood,
@@ -178,7 +194,7 @@ public bool GetLanguages(
 
         foreach (var item in _hands.EnumerateHeld(uid))
         {
-            if (TryComp<HandheldTranslatorComponent>(item, out var translator))
+            if (TryComp<HandheldTranslatorComponent>(item, out var translator) && translator.Enabled)
             {
                 foreach (var lang in translator.Required)
                 {
diff --git a/Resources/Locale/ru-RU/ADT/prototypes/Actions/language-menu.ftl b/Resources/Locale/ru-RU/ADT/prototypes/Actions/language-menu.ftl
index d4265c8e797..f6dd08b8c28 100644
--- a/Resources/Locale/ru-RU/ADT/prototypes/Actions/language-menu.ftl
+++ b/Resources/Locale/ru-RU/ADT/prototypes/Actions/language-menu.ftl
@@ -4,7 +4,7 @@ language-menu-description-header = Описание
 language-choose-button = Выбрать
 language-choose-button-chosen = Выбрано
 language-choose-button-cannot = Только понимание
-language-menu-action = Меню языков
-language-menu-action-desc = Открыть меню выбора языка.
+language-choose-button-tooltip-translator = Вы используете переводчик для пользования данным языком.
+language-choose-button-tooltip-known = Вы полностью владеете данным языком.
 
 game-hud-open-languages-menu-button-tooltip = Открыть меню выбора языка
diff --git a/Resources/Locale/ru-RU/ADT/prototypes/Entities/Objects/Device/translators.ftl b/Resources/Locale/ru-RU/ADT/prototypes/Entities/Objects/Device/translators.ftl
index 4a6a944c6c4..24d0c528470 100644
--- a/Resources/Locale/ru-RU/ADT/prototypes/Entities/Objects/Device/translators.ftl
+++ b/Resources/Locale/ru-RU/ADT/prototypes/Entities/Objects/Device/translators.ftl
@@ -1,5 +1,5 @@
-translator-component-shutoff = {$translator} выключается.
-translator-component-turnon = The {$translator} включается.
+translator-component-shutoff = {CAPITALIZE($translator)} выключается.
+translator-component-turnon = {CAPITALIZE($translator)} включается.
 translator-enabled = Оно включено.
 translator-disabled = Оно выключено.