Merge pull request #1086 from arkuznetsov/feature/get-stdin

feat (#746): Реализована возможность чтения из скрипта данных, переданных через stdin

Author: EvilBeaver

src/ScriptEngine.HostedScript/Library/Binary/GenericStream.cs

@@ -47,7 +47,7 @@ public GenericStream(Stream underlyingStream, bool readOnly)
         }
 
         public bool IsReadOnly => !CanWrite;
-
+        
         /// <summary>
         /// 
         /// Признак доступности записи в поток.

src/ScriptEngine.HostedScript/Library/Binary/MemoryStreamContext.cs

@@ -41,7 +41,7 @@ public class MemoryStreamContext : AutoContext<MemoryStreamContext>, IDisposable
             _underlyingStream = new MemoryStream(capacity);
             _commonImpl = new GenericStreamImpl(_underlyingStream);
         }
-
+        
         /// <summary>
         /// 
         /// Создает поток, в качестве нижележащего хранилища для которого используется заданный байтовый буфер. Ёмкость потока ограничена размером буфера. При выходе за границы буфера будет сгенерировано исключение.

src/ScriptEngine.HostedScript/Library/ConsoleContext.cs

@@ -5,6 +5,8 @@ This Source Code Form is subject to the terms of the
 at http://mozilla.org/MPL/2.0/.
 ----------------------------------------------------------*/
 using System;
+using System.IO;
+using ScriptEngine.HostedScript.Library.Binary;
 using ScriptEngine.Machine;
 using ScriptEngine.Machine.Contexts;
 
@@ -168,6 +170,24 @@ public IValue InputEncoding
                 Console.InputEncoding = TextEncodingEnum.GetEncoding(value);                
             }
         }
+        
+        /// <summary>
+        /// Возвращает или задает кодировку консоли, используемую при чтении входных данных.
+        /// </summary>
+        /// <returns>КодировкаТекста</returns>
+        [ContextProperty("КодировкаВыходногоПотока", "InputEncoding")]
+        public IValue OutputEncoding 
+        {
+            get
+            {
+                var encodingEnum = GlobalsManager.GetEnum<TextEncodingEnum>();
+                return encodingEnum.GetValue(Console.OutputEncoding);
+            }
+            set 
+            {
+                Console.OutputEncoding = TextEncodingEnum.GetEncoding(value);                
+            }
+        }
 
         /// <summary>
         /// Воспроизводит звуковой сигнал.
@@ -178,10 +198,77 @@ public void Beep()
             Console.Beep();
         }
 
+        /// <summary>
+        /// Получает системный поток ввода stdin
+        /// </summary>
+        /// <returns>Поток</returns>
+        [ContextMethod("ОткрытьСтандартныйПотокВвода", "OpenStandardInput")]
+        public GenericStream OpenStandardInput()
+        {
+            var stream = Console.OpenStandardInput();
+            return new GenericStream(stream, true);
+        }
+        
+        /// <summary>
+        /// Получает системный поток вывода ошибок stderr
+        /// </summary>
+        /// <returns>Поток</returns>
+        [ContextMethod("ОткрытьСтандартныйПотокОшибок", "OpenStandardError")]
+        public GenericStream OpenStandardError()
+        {
+            var stream = Console.OpenStandardError();
+            return new GenericStream(stream);
+        }
+        
+        /// <summary>
+        /// Получает системный поток вывода stdout
+        /// </summary>
+        /// <returns>Поток</returns>
+        [ContextMethod("ОткрытьСтандартныйПотокВывода", "OpenStandardOutput")]
+        public GenericStream OpenStandardOutput()
+        {
+            var stream = Console.OpenStandardOutput();
+            return new GenericStream(stream);
+        }
+
+        /// <summary>
+        /// Глобально переопределяет стандартный вывод и направляет в другой поток
+        /// </summary>
+        /// <param name="target">Поток назначения</param>
+        [ContextMethod("УстановитьПотокВывода", "SetOutput")]
+        public void SetOutput(IValue target)
+        {
+            if (!(target.AsObject() is IStreamWrapper stream))
+                throw RuntimeException.InvalidArgumentType(nameof(target));
+            
+            var writer = new StreamWriter(stream.GetUnderlyingStream(), Console.OutputEncoding)
+            {
+                AutoFlush = true,
+            };
+            Console.SetOut(writer);
+        }
+        
+        /// <summary>
+        /// Глобально переопределяет стандартный поток ошибок и направляет в другой поток
+        /// </summary>
+        /// <param name="target">Поток назначения</param>
+        [ContextMethod("УстановитьПотокОшибок", "SetError")]
+        public void SetError(IValue target)
+        {
+            if (!(target.AsObject() is IStreamWrapper stream))
+                throw RuntimeException.InvalidArgumentType(nameof(target));
+            
+            var writer = new StreamWriter(stream.GetUnderlyingStream());
+            Console.SetError(writer);
+        }
+
         [ScriptConstructor]
         public static ConsoleContext Constructor()
         {
-            return new ConsoleContext();
+            var provider = GlobalsManager.GetGlobalContext<ConsoleProvider>();
+            SystemLogger.Write("WARNING: Constructor of Console is obsolete. Use global property Консоль/Console");
+
+            return provider.Console;
         }
     }
 

src/ScriptEngine.HostedScript/Library/ConsoleProvider.cs

@@ -0,0 +1,52 @@
+/*----------------------------------------------------------
+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/.
+----------------------------------------------------------*/
+
+using System;
+using ScriptEngine.Machine;
+using ScriptEngine.Machine.Contexts;
+
+namespace ScriptEngine.HostedScript.Library
+{
+    [GlobalContext(Category = "Работа с консолью")]
+    public class ConsoleProvider : GlobalContextBase<ConsoleProvider>
+    {
+        private readonly ConsoleContext _console = new ConsoleContext();
+
+        private ConsoleProvider()
+        {
+        }
+
+        public override void OnAttach(MachineInstance machine, out IVariable[] variables, out MethodInfo[] methods)
+        {
+            variables = new [] {Variable.CreateContextPropertyReference(this, 0, GetPropName(0))};
+            methods = new MethodInfo[0];
+        }
+
+        [ContextProperty("Консоль", "Console")]
+        public ConsoleContext Console => _console;
+
+        public override bool IsPropWritable(int propNum)
+        {
+            // обратная совместимость. Присваивание Консоль = Новый Консоль не должно ругаться на недоступность записи
+            return true;
+        }
+        
+        public override void SetPropValue(int propNum, IValue newVal)
+        {
+            // обратная совместимость. Присваивание Консоль = Новый Консоль не должно ничего делать
+            if (!ReferenceEquals(newVal.GetRawValue(), _console))
+            {
+                throw new InvalidOperationException("Can't assign to global property Console");
+            }
+        }
+
+        public static ConsoleProvider CreateInstance()
+        {
+            return new ConsoleProvider();
+        }
+    }
+}

src/ScriptEngine.HostedScript/Library/SystemGlobalContext.cs

@@ -38,7 +38,7 @@ public SystemGlobalContext()
             RegisterProperty("ФайловыеПотоки", () => FileStreams);
             RegisterProperty("FileStreams", () => FileStreams);
 
-			RegisterProperty("Символы", () => (IValue)Chars);
+            RegisterProperty("Символы", () => (IValue)Chars);
 			RegisterProperty("Chars", () => (IValue)Chars);
         }
 

src/ScriptEngine.HostedScript/Library/TextReadImpl.cs

@@ -56,7 +56,7 @@ private void OpenStream(IStreamWrapper streamObj, IValue encoding = null, string
             TextReader imReader;
             if (encoding == null)
             {
-                imReader = Environment.FileOpener.OpenReader(streamObj.GetUnderlyingStream());
+                imReader = Environment.FileOpener.OpenReader(streamObj.GetUnderlyingStream(), Encoding.Default);
             }
             else
             {

tests/console.os

@@ -0,0 +1,95 @@
+Перем юТест;
+
+Функция ПолучитьСписокТестов(ЮнитТестирование) Экспорт
+
+	юТест = ЮнитТестирование;
+
+	ВсеТесты = Новый Массив;
+	ВсеТесты.Добавить("ТестДолжен_ПроверитьЧтоСтандартныйПотокВводаЭтоПоток");
+	ВсеТесты.Добавить("ТестДолжен_ПроверитьПередачуДанныхВСкриптЧерезСтандартныйПотокВвода");
+	ВсеТесты.Добавить("ТестДолжен_ПроверитьПеренаправлениеВывода");
+
+	Возврат ВсеТесты;
+
+КонецФункции
+
+Процедура ТестДолжен_ПроверитьЧтоСтандартныйПотокВводаЭтоПоток() Экспорт
+	
+	ПотокВвода = Консоль.ОткрытьСтандартныйПотокВвода();
+	юТест.ПроверитьРавенство(Тип("Поток"), ТипЗнч(ПотокВвода), "Стандартный поток ввода - Поток");
+
+КонецПроцедуры
+
+Процедура ТестДолжен_ПроверитьПередачуДанныхВСкриптЧерезСтандартныйПотокВвода() Экспорт
+
+	СистемнаяИнформация = Новый СистемнаяИнформация;
+
+	ПутьКОскрипт = ОбъединитьПути(КаталогПрограммы(), "oscript.exe");
+	#Если Linux Тогда
+		ПутьКОскрипт = "mono " + ПутьКОскрипт;
+	#КонецЕсли
+
+	КодСкрипта = "Чтение = Новый ЧтениеТекста();
+	             |Чтение.Открыть(Консоль.ОткрытьСтандартныйПотокВвода());
+	             |Сообщить(СокрЛП(Чтение.Прочитать()));
+	             |";
+
+	ТекстСкрипта = Новый ТекстовыйДокумент();
+	ТекстСкрипта.УстановитьТекст(КодСкрипта);
+
+	ВремФайл = ПолучитьИмяВременногоФайла("os");
+
+	ТекстСкрипта.Записать(ВремФайл);
+
+	ТестовыеДанные = "12346";
+
+	ИсполняемаяКоманда = СтрШаблон("echo %1 | %2 %3", ТестовыеДанные, ПутьКОскрипт, ВремФайл);
+
+	#Если Windows Тогда
+		ШаблонЗапуска = "cmd /c ""%1""";
+	#Иначе
+		ШаблонЗапуска = "sh -c '%1'";
+	#КонецЕсли
+
+	ИсполняемаяКоманда = СтрШаблон(ШаблонЗапуска, ИсполняемаяКоманда);
+
+	Процесс = СоздатьПроцесс(ИсполняемаяКоманда, , Истина);
+	Процесс.Запустить();
+
+	Пока НЕ Процесс.Завершен Цикл
+		Приостановить(100);
+	КонецЦикла;
+
+	ВыводКоманды = СокрЛП(Процесс.ПотокВывода.Прочитать());
+	
+	УдалитьФайлы(ВремФайл);
+
+	юТест.ПроверитьРавенство(ВыводКоманды, ТестовыеДанные, "Вывод команды - тестовые данные");
+
+КонецПроцедуры
+
+Процедура ТестДолжен_ПроверитьПеренаправлениеВывода() Экспорт
+	
+	ВФ = ПолучитьИмяВременногоФайла();
+	Поток = ФайловыеПотоки.ОткрытьДляЗаписи(ВФ);
+	Консоль.УстановитьПотокВывода(Поток);
+	Попытка
+		Сообщить("Привет мир!");
+	Исключение
+		// что-то пошло не так
+		Консоль.УстановитьПотокВывода(Консоль.ОткрытьСтандартныйПотокВывода());
+		ВызватьИсключение;
+	КонецПопытки;
+	
+	Поток.Закрыть();
+	Консоль.УстановитьПотокВывода(Консоль.ОткрытьСтандартныйПотокВывода());
+
+	Чтение = Новый ЧтениеТекста(ВФ, Консоль.КодировкаВыходногоПотока);
+	Текст = Чтение.Прочитать();
+	Чтение.Закрыть();
+
+	УдалитьФайлы(ВФ);
+
+	юТест.ПроверитьРавенство("Привет мир!", СокрЛП(Текст));
+
+КонецПроцедуры

tests/process.os

@@ -76,7 +76,6 @@
 
 Процедура ТестДолжен_ЗаписатьВоВходнойПотокПроцесса() Экспорт
 
-	Консоль = Новый Консоль();
 	КодировкаПоУмолчанию = Консоль.КодировкаВходногоПотока; 
 	Если Консоль.КодировкаВходногоПотока = КодировкаТекста.UTF8 Тогда
 		Консоль.КодировкаВходногоПотока = КодировкаТекста.UTF8NoBOM;

tests/reflector.os

@@ -232,7 +232,7 @@
 	МассивОбъектов.Добавить(Новый РегулярноеВыражение("а"));
 	МассивОбъектов.Добавить(Новый ЗаписьXML);
 	МассивОбъектов.Добавить(Новый ЧтениеXML);
-	МассивОбъектов.Добавить(Новый Консоль);
+	МассивОбъектов.Добавить(Консоль);
 	МассивОбъектов.Добавить(Новый Файл("ыв"));
 	МассивОбъектов.Добавить(Новый ГенераторСлучайныхЧисел);
 	МассивОбъектов.Добавить(Новый СистемнаяИнформация);