Заготовка для #860

EvilBeaver · Jul 24, 2021 · e89cbad · e89cbad
1 parent 0e259aa
commit e89cbad
Show file tree

Hide file tree

Showing 4 changed files with 81 additions and 1 deletion.
diff --git a/src/ScriptEngine.HostedScript/DefaultEventProcessor.cs b/src/ScriptEngine.HostedScript/DefaultEventProcessor.cs
@@ -76,6 +76,10 @@ public void AddHandler(
             }
 
             handlers[eventName].Add(handlerScript, handlerMethod);
+            if (eventSource is IEventSourceNotify notify)
+            {
+                notify.OnSubscribe(eventName, handlerScript);
+            }
         }
 
         public void RemoveHandler(
@@ -90,6 +94,10 @@ public void RemoveHandler(
             if (_registeredHandlers.TryGetValue(eventSource, out var handlers))
             {
                 handlers[eventName].Remove(handlerScript, handlerMethod);
+                if (eventSource is IEventSourceNotify notify)
+                {
+                    notify.OnUnsubscribe(eventName, handlerScript);
+                }
             }
         }
 

diff --git a/src/ScriptEngine.HostedScript/Library/ConsoleContext.cs b/src/ScriptEngine.HostedScript/Library/ConsoleContext.cs
@@ -17,8 +17,11 @@ namespace ScriptEngine.HostedScript.Library
     /// Предназначен для низкоуровнего манипулирования выводом в консоль.
     /// </summary>
     [ContextClass("Консоль", "Console")]
-    public class ConsoleContext : AutoContext<ConsoleContext>
+    public class ConsoleContext : AutoContext<ConsoleContext>, IEventSourceNotify
     {
+        private int _cancelKeyHandlers;
+        private IEventProcessor _eventProcessor;
+
         [ContextProperty("НажатаКлавиша", "KeyPressed")]
         public bool HasKey
         {
@@ -270,6 +273,45 @@ public static ConsoleContext Constructor()
 
             return provider.Console;
         }
+
+        public void OnSubscribe(string eventName, IRuntimeContextInstance target)
+        {
+            if (_cancelKeyHandlers == 0)
+            {
+                Console.CancelKeyPress += CancelKeyHandler;
+                _eventProcessor = MachineInstance.Current.EventProcessor;
+            }
+
+            ++_cancelKeyHandlers;
+        }
+
+        public void OnUnsubscribe(string eventName, IRuntimeContextInstance target)
+        {
+            if(_cancelKeyHandlers == 0)
+                return;
+
+            if (--_cancelKeyHandlers == 0)
+            {
+                Console.CancelKeyPress -= CancelKeyHandler;
+                _eventProcessor = null;
+            }
+        }
+
+        private void CancelKeyHandler(object sender, ConsoleCancelEventArgs args)
+        {
+            if(_eventProcessor == null)
+                return;
+
+            var cancelVariable = Variable.Create(ValueFactory.Create(false), "Отказ"); 
+            var scriptArgs = new IValue[] { cancelVariable };
+
+            _eventProcessor.HandleEvent(this, "CancelKeyPress", scriptArgs);
+
+            if (cancelVariable.DataType == DataType.Boolean)
+            {
+                args.Cancel = cancelVariable.AsBoolean();
+            }
+        }
     }
 
 

diff --git a/src/ScriptEngine/Machine/IEventSourceNotify.cs b/src/ScriptEngine/Machine/IEventSourceNotify.cs
@@ -0,0 +1,16 @@
+/*----------------------------------------------------------
+This Source Code Form is subject to the terms of the
+Mozilla Public License, v.2.0. If a copy of the MPL
+was not distributed with this file, You can obtain one
+at http://mozilla.org/MPL/2.0/.
+----------------------------------------------------------*/
+
+namespace ScriptEngine.Machine
+{
+    public interface IEventSourceNotify
+    {
+        void OnSubscribe(string eventName, IRuntimeContextInstance target);
+
+        void OnUnsubscribe(string eventName, IRuntimeContextInstance target);
+    }
+}
diff --git a/tests/console.os b/tests/console.os
@@ -70,6 +70,15 @@
 
 Процедура ТестДолжен_ПроверитьПеренаправлениеВывода() Экспорт
 
+	ДобавитьОбработчик Консоль.CancelKeyPress, ЭтотОбъект.Обработчик;
+
+	Сч = 0;
+	Пока Сч < 20 Цикл
+		Приостановить(1000);
+		Сообщить(Сч);
+		Сч = Сч + 1;
+	КонецЦикла;
+
 	ВФ = ПолучитьИмяВременногоФайла();
 	Поток = ФайловыеПотоки.ОткрытьДляЗаписи(ВФ);
 	Консоль.УстановитьПотокВывода(Поток);
@@ -92,4 +101,9 @@
 
 	юТест.ПроверитьРавенство("Привет мир!", СокрЛП(Текст));
 
+КонецПроцедуры
+
+Процедура Обработчик(Отказ) Экспорт
+	Сообщить("Обработчик вызван");
+	Отказ = Истина;
 КонецПроцедуры