EvilBeaver/OneScript: Исполняющая среда скриптов на языке 1С

src/ScriptEngine.HostedScript/Library/TextReadImpl.cs
253 строки10 KB
Andrey Ovsiankin
Заготовка по #746
24 июл 2021, 20:46
24 июл 2021, 20:46f40bb11
/*----------------------------------------------------------
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 System.IO;
using System.Text;
using ScriptEngine.HostedScript.Library.Binary;
using ScriptEngine.Machine;
using ScriptEngine.Machine.Contexts;

namespace ScriptEngine.HostedScript.Library
{
    /// <summary>
    /// Предназначен для последовательного чтения файлов, в том числе большого размера.
    /// </summary>
    [ContextClass("ЧтениеТекста", "TextReader")]
    public class TextReadImpl : AutoContext<TextReadImpl>, IDisposable
    {
        // TextReader _reader;
        CustomLineFeedStreamReader _reader;
        string _lineDelimiter = "\n";

        public TextReadImpl ()
        {
            AnalyzeDefaultLineFeed = true;
        }

        /// <summary>
        /// Открывает текстовый файл для чтения. Ранее открытый файл закрывается.
        /// </summary>
        /// <param name="input">Путь к файлу или поток</param>
        /// <param name="encoding">Кодировка</param>
        /// <param name="lineDelimiter">Раздедитель строк</param>
        /// <param name="eolDelimiter">Разделитель строк в файле</param>
        /// <param name="monopoly">Открывать монопольно</param>
        [ContextMethod("Открыть", "Open")]
        public void Open(IValue input, IValue encoding = null, string lineDelimiter = "\n", string eolDelimiter = null,
            bool? monopoly = null)
        {
            Close();
            if(IsStream(input, out var wrapper))
            {
                OpenStream(wrapper, encoding, lineDelimiter, eolDelimiter);
            }
            else
            {
                OpenFile(input.AsString(), encoding, lineDelimiter, eolDelimiter, monopoly);
            }
        }

        private void OpenStream(IStreamWrapper streamObj, IValue encoding = null, string lineDelimiter = "\n", string eolDelimiter = null)
        {
            TextReader imReader;
            if (encoding == null)
            {
                imReader = Environment.FileOpener.OpenReader(streamObj.GetUnderlyingStream(), Encoding.Default);
            }
            else
            {
                var enc = TextEncodingEnum.GetEncoding(encoding);
                imReader = Environment.FileOpener.OpenReader(streamObj.GetUnderlyingStream(), enc);
            }
            _reader = GetCustomLineFeedReader(imReader, lineDelimiter, eolDelimiter, AnalyzeDefaultLineFeed);
        }

        private void OpenFile(string path, IValue encoding = null, string lineDelimiter = "\n", string eolDelimiter = null,
            bool? monopoly = null)
        {
            TextReader imReader;
            var shareMode = (monopoly ?? true) ? FileShare.None : FileShare.ReadWrite;
            if (encoding == null)
            {
                imReader = Environment.FileOpener.OpenReader(path, shareMode);
            }
            else
            {
                var enc = TextEncodingEnum.GetEncoding(encoding);
                imReader = Environment.FileOpener.OpenReader(path, shareMode, enc);
            }
            _reader = GetCustomLineFeedReader(imReader, lineDelimiter, eolDelimiter, AnalyzeDefaultLineFeed);
        }

        private bool AnalyzeDefaultLineFeed { get; set; }

        private int ReadNext()
        {
            return _reader.Read ();
        }

        /// <summary>
        /// Считывает строку указанной длины или до конца файла.
        /// </summary>
        /// <param name="size">Размер строки. Если не задан, текст считывается до конца файла</param>
        /// <returns>Строка - считанная строка, Неопределено - в файле больше нет данных</returns>
        [ContextMethod("Прочитать", "Read")]
        public IValue ReadAll(int size = 0)
        {
            RequireOpen();

            var sb = new StringBuilder();
            var read = 0;
            do {
                var ic = ReadNext();
                if (ic == -1)
                    break;
                sb.Append((char)ic);
                ++read;
            } while (size == 0 || read < size);

            if (sb.Length == 0)
                return ValueFactory.Create ();

            return ValueFactory.Create(sb.ToString());
        }

        /// <summary>
        /// Считывает очередную строку текстового файла.
        /// </summary>
        /// <param name="overridenLineDelimiter">Подстрока, считающаяся концом строки. Переопределяет РазделительСтрок,
        /// переданный в конструктор или в метод Открыть</param>
        /// <returns>Строка - в случае успешного чтения, Неопределено - больше нет данных</returns>
        [ContextMethod("ПрочитатьСтроку", "ReadLine")]
        public IValue ReadLine(string overridenLineDelimiter = null)
        {
            RequireOpen();
            string l = _reader.ReadLine (overridenLineDelimiter ?? _lineDelimiter);

            if (l == null)
                return ValueFactory.Create();

            return ValueFactory.Create(l);
        }

        /// <summary>
        /// Закрывает открытый текстовый файл. Если файл был открыт монопольно, то после закрытия он становится доступен.
        /// </summary>
        [ContextMethod("Закрыть", "Close")]
        public void Close()
        {
            Dispose();
        }

        private void RequireOpen()
        {
            if (_reader == null)
            {
                throw new RuntimeException("Файл не открыт");
            }
        }

        /// <summary>
        /// Открывает текстовый файл для чтения.
        /// </summary>
        /// <param name="input">Путь к файлу или поток</param>
        /// <returns>ЧтениеТекста</returns>
        [ScriptConstructor(Name = "На основании файла или потока без кодировки")]
        public static TextReadImpl Constructor (IValue input)
        {
            var reader = new TextReadImpl ();
            reader.AnalyzeDefaultLineFeed = false;
            reader.Open (input, null, "\n", "\r\n");
            return reader;
        }

        /// <summary>
        /// Открывает текстовый файл или поток для чтения. Работает аналогично методу Открыть.
        /// </summary>
        /// <param name="input">Путь к файлу или поток</param>
        /// <param name="encoding">Кодировка</param>
        /// <param name="lineDelimiter">Разделитель строк</param>
        /// <param name="eolDelimiter">Разделитель строк в файле</param>
        /// <param name="monopoly">Открывать файл монопольно</param>
        /// <returns>ЧтениеТекста</returns>
        [ScriptConstructor(Name = "На основании потока или файла")]
        public static TextReadImpl Constructor(IValue input, IValue encoding = null,
            IValue lineDelimiter = null, IValue eolDelimiter = null, IValue monopoly = null)
        {
            var reader = new TextReadImpl();
            if (lineDelimiter != null)
                reader.AnalyzeDefaultLineFeed = false;

            if(IsStream(input, out var wrapper))
            {
                reader.OpenStream(wrapper, encoding,
                        lineDelimiter?.GetRawValue().AsString() ?? "\n",
                        eolDelimiter?.GetRawValue().AsString());
            }
            else
            {
                reader.OpenFile(input.AsString(), encoding,
                    lineDelimiter?.GetRawValue().AsString() ?? "\n",
                    eolDelimiter?.GetRawValue().AsString(),
                    monopoly?.AsBoolean() ?? true);
            }

            return reader;
        }

        /// <summary>
        /// Создаёт неинициализированный объект. Для инициализации необходимо открыть файл методом Открыть.
        /// </summary>
        /// <returns>ЧтениеТекста</returns>
        [ScriptConstructor(Name = "Формирование неинициализированного объекта")]
        public static TextReadImpl Constructor()
        {
            var reader = new TextReadImpl();
            reader.AnalyzeDefaultLineFeed = false;
            return reader;
        }

        private static bool IsStream(IValue input, out IStreamWrapper wrapper)
        {
            wrapper = null;
            if (input.DataType == DataType.Object)
            {
                var obj = input.AsObject();
                if (obj is IStreamWrapper wrap)
                {
                    wrapper = wrap;
                    return true;
                }
            }
            return false;
        }

        private CustomLineFeedStreamReader GetCustomLineFeedReader(TextReader imReader, string lineDelimiter,
            string eolDelimiter, bool AnalyzeDefaultLineFeed)
        {
            _lineDelimiter = lineDelimiter ?? "\n";
            if (eolDelimiter != null)
                return new CustomLineFeedStreamReader(imReader, eolDelimiter, AnalyzeDefaultLineFeed);
            else
                return new CustomLineFeedStreamReader(imReader, "\r\n", AnalyzeDefaultLineFeed);
        }


        #region IDisposable Members

        public void Dispose()
        {
            if (_reader != null)
            {
                _reader.Dispose();
                _reader = null;
            }
        }

        #endregion
    }
}