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

src/ScriptEngine/Environment/CodeSources.cs
108 строк3 KB
Andrei Ovsiankin
Возможно исправлена ошибка #1016
01 окт 2020, 11:20
01 окт 2020, 11:20dc15852
/*----------------------------------------------------------
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.Text;
using System.IO;

namespace ScriptEngine.Environment
{
    class StringBasedSource : ICodeSource
    {
        readonly string _src;

        public StringBasedSource(string src)
        {
            _src = src;
        }

        #region ICodeSource Members

        string ICodeSource.Code
        {
            get
            {
                return _src;
            }
        }

        string ICodeSource.SourceDescription => $"<string {_src.GetHashCode().ToString("X8")}>" ;

        #endregion

    }

    class FileBasedSource : ICodeSource
    {
        readonly string _path;
        string _code;

        readonly Encoding _noBomEncoding;

        public FileBasedSource(string path, Encoding defaultEncoding)
        {
            _path = System.IO.Path.GetFullPath(path);
            _noBomEncoding = defaultEncoding;
        }

        private string GetCodeString()
        {
            if (_code == null)
            {
                using (var fStream = new FileStream(_path, FileMode.Open, FileAccess.Read))
                {
                    var buf = new byte[2];
                    fStream.Read(buf, 0, 2);
                    Encoding enc = null;
                    bool skipShebang = false;
                    if (IsLinuxScript(buf))
                    {
                        enc = Encoding.UTF8; // скрипты с shebang считать в формате UTF-8
                        skipShebang = true;
                    }
                    else
                    {
                        fStream.Position = 0;
                        enc = FileOpener.AssumeEncoding(fStream, _noBomEncoding);
                    }

                    using (var reader = new StreamReader(fStream, enc))
                    {
                        if (skipShebang)
                            reader.ReadLine();

                        _code = reader.ReadToEnd();
                    }

                }
            }

            return _code;
        }

        private static bool IsLinuxScript(byte[] buf)
        {
            return buf[0] == '#' && buf[1] == '!';
        }

        #region ICodeSource Members

        string ICodeSource.Code
        {
            get
            {
                return GetCodeString();
            }
        }

        string ICodeSource.SourceDescription
        {
            get { return _path[0].ToString().ToUpperInvariant() + _path.Substring(1); }
        }

        #endregion
    }
}