src/OneScript.StandardLibrary/Http/HttpRequestBodyBinary.cs - develop - smwwiki/OneScript_fork

src/OneScript.StandardLibrary/Http/HttpRequestBodyBinary.cs

74 строки2 KB

EvilBeaver

Перевел тело Http запроса на новый поток

20 апр 2024, 23:21

20 апр 2024, 23:21240bfff

/*----------------------------------------------------------
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.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using OneScript.StandardLibrary.Binary;
using OneScript.StandardLibrary.Text;
using ScriptEngine.Machine;

namespace OneScript.StandardLibrary.Http
{
    class HttpRequestBodyBinary : IHttpRequestBody
    {
        private readonly FileBackingStream _storage = new FileBackingStream();

        public HttpRequestBodyBinary()
        {
        }

        public HttpRequestBodyBinary(BinaryDataContext data)
        {
            data.CopyTo(_storage);
        }

        public HttpRequestBodyBinary(string body, IValue encoding = null,
            ByteOrderMarkUsageEnum bomUsage = ByteOrderMarkUsageEnum.Auto)
        {
            var utfs = new List<string> {"utf-16", "utf-32"};
            var addBom = utfs.Contains(encoding?.AsString(), StringComparer.OrdinalIgnoreCase) &&
                         bomUsage == ByteOrderMarkUsageEnum.Auto || bomUsage == ByteOrderMarkUsageEnum.Use;

            var encoder = encoding == null ? new UTF8Encoding(addBom) : TextEncodingEnum.GetEncoding(encoding, addBom);

            var byteArray = encoder.GetBytes(body);
            _storage.Write(byteArray, 0, byteArray.Length);
        }

        public IValue GetAsString()
        {
            _storage.Seek(0, SeekOrigin.Begin);
            var reader = new StreamReader(_storage);
            return ValueFactory.Create(reader.ReadToEnd());
        }

        public IValue GetAsBinary()
        {
            _storage.Seek(0, SeekOrigin.Begin);
            return new BinaryDataContext(_storage);
        }

        public IValue GetAsFilename()
        {
            return ValueFactory.Create();
        }

        public Stream GetDataStream()
        {
            _storage.Seek(0, SeekOrigin.Begin);
            return _storage;
        }

        public void Dispose()
        {
            _storage.Close();
        }
    }
}