Masuit.Tools/Masuit.Tools.Abstractions/Extensions/BaseType/StreamExtensions.cs

using Masuit.Tools.Systems;
using System;
using System.Collections.Generic;
using System.IO;
using System.Text;
using System.Threading;
using System.Threading.Tasks;

#if NET5_0_OR_GREATER
using System;
using System.Buffers;
using System.Runtime.InteropServices;
#endif

namespace Masuit.Tools;

public static class StreamExtensions
{
    /// <summary>
    /// 将流转换为内存流
    /// </summary>
    /// <param name="stream"></param>
    /// <returns></returns>
    public static PooledMemoryStream SaveAsMemoryStream(this Stream stream)
    {
        if (stream is PooledMemoryStream pooledMemoryStream)
        {
            return pooledMemoryStream;
        }

        stream.Position = 0;
        var ms = new PooledMemoryStream();
        stream.CopyTo(ms);
        stream.Position = 0;
        return ms;
    }

    /// <summary>
    ///
    /// </summary>
    /// <param name="stream"></param>
    /// <returns></returns>
    public static byte[] ToArray(this Stream stream)
    {
        stream.Position = 0;
        byte[] bytes = new byte[stream.Length];
        stream.Read(bytes, 0, bytes.Length);

        // 设置当前流的位置为流的开始
        stream.Seek(0, SeekOrigin.Begin);
        return bytes;
    }

    /// <summary>
    /// 流洗码，在流的末端随即增加几个空字节，重要数据请谨慎使用，可能造成流损坏
    /// </summary>
    /// <param name="stream"></param>
    public static void ShuffleCode(this Stream stream)
    {
        if (stream.CanWrite && stream.CanSeek)
        {
            var position = stream.Position;
            stream.Position = stream.Length;
            for (int i = 0; i < new Random().Next(1, 20); i++)
            {
                stream.WriteByte(0);
            }
            stream.Flush();
            stream.Position = position;
        }
    }

    /// <summary>
    /// 读取所有行
    /// </summary>
    /// <param name="stream"></param>
    /// <param name="closeAfter">读取完毕后关闭流</param>
    /// <returns></returns>
    public static List<string> ReadAllLines(this StreamReader stream, bool closeAfter = true)
    {
        var stringList = new List<string>();
        while (stream.ReadLine() is { } str)
        {
            stringList.Add(str);
        }

        if (closeAfter)
        {
            stream.Close();
            stream.Dispose();
        }
        return stringList;
    }

    /// <summary>
    /// 读取所有行
    /// </summary>
    /// <param name="stream"></param>
    /// <param name="encoding"></param>
    /// <param name="closeAfter">读取完毕后关闭流</param>
    /// <returns></returns>
    public static List<string> ReadAllLines(this FileStream stream, Encoding encoding, bool closeAfter = true)
    {
        var stringList = new List<string>();
        var sr = new StreamReader(stream, encoding);
        while (sr.ReadLine() is { } str)
        {
            stringList.Add(str);
        }

        if (closeAfter)
        {
            sr.Close();
            sr.Dispose();
            stream.Close();
            stream.Dispose();
        }

        return stringList;
    }

    /// <summary>
    /// 读取所有文本
    /// </summary>
    /// <param name="stream"></param>
    /// <param name="encoding"></param>
    /// <param name="closeAfter">读取完毕后关闭流</param>
    /// <returns></returns>
    public static string ReadAllText(this FileStream stream, Encoding encoding, bool closeAfter = true)
    {
        var sr = new StreamReader(stream, encoding);
        var text = sr.ReadToEnd();
        if (closeAfter)
        {
            sr.Close();
            sr.Dispose();
            stream.Close();
            stream.Dispose();
        }

        return text;
    }

    /// <summary>
    /// 写入所有文本
    /// </summary>
    /// <param name="stream"></param>
    /// <param name="content"></param>
    /// <param name="encoding"></param>
    /// <param name="closeAfter">读取完毕后关闭流</param>
    /// <returns></returns>
    public static void WriteAllText(this FileStream stream, string content, Encoding encoding, bool closeAfter = true)
    {
        var sw = new StreamWriter(stream, encoding);
        stream.SetLength(0);
        sw.Write(content);
        if (closeAfter)
        {
            sw.Close();
            sw.Dispose();
            stream.Close();
            stream.Dispose();
        }
    }

    /// <summary>
    /// 写入所有文本行
    /// </summary>
    /// <param name="stream"></param>
    /// <param name="lines"></param>
    /// <param name="encoding"></param>
    /// <param name="closeAfter">读取完毕后关闭流</param>
    /// <returns></returns>
    public static void WriteAllLines(this FileStream stream, IEnumerable<string> lines, Encoding encoding, bool closeAfter = true)
    {
        var sw = new StreamWriter(stream, encoding);
        stream.SetLength(0);
        foreach (var line in lines)
        {
            sw.WriteLine(line);
        }

        sw.Flush();
        if (closeAfter)
        {
            sw.Close();
            sw.Dispose();
            stream.Close();
            stream.Dispose();
        }
    }

    /// <summary>
    /// 共享读写打开文件
    /// </summary>
    /// <param name="file"></param>
    /// <returns></returns>
    public static FileStream ShareReadWrite(this FileInfo file)
    {
        return file.Open(FileMode.OpenOrCreate, FileAccess.ReadWrite, FileShare.ReadWrite);
    }

    /// <summary>
    /// 读取所有行
    /// </summary>
    /// <param name="stream"></param>
    /// <param name="closeAfter">读取完毕后关闭流</param>
    /// <returns></returns>
    public static async Task<List<string>> ReadAllLinesAsync(this StreamReader stream, bool closeAfter = true)
    {
        var stringList = new List<string>();
        while (await stream.ReadLineAsync().ConfigureAwait(false) is { } str)
        {
            stringList.Add(str);
        }

        if (closeAfter)
        {
            stream.Close();
            stream.Dispose();
        }
        return stringList;
    }

    /// <summary>
    /// 读取所有行
    /// </summary>
    /// <param name="stream"></param>
    /// <param name="encoding"></param>
    /// <param name="closeAfter">读取完毕后关闭流</param>
    /// <returns></returns>
    public static async Task<List<string>> ReadAllLinesAsync(this FileStream stream, Encoding encoding, bool closeAfter = true)
    {
        var stringList = new List<string>();
        var sr = new StreamReader(stream, encoding);
        while (await sr.ReadLineAsync().ConfigureAwait(false) is { } str)
        {
            stringList.Add(str);
        }

        if (closeAfter)
        {
            sr.Close();
            sr.Dispose();
            stream.Close();
#if NET5_0_OR_GREATER
                await stream.DisposeAsync().ConfigureAwait(false);
#else
            stream.Dispose();
#endif
        }

        return stringList;
    }

    /// <summary>
    /// 读取所有文本
    /// </summary>
    /// <param name="stream"></param>
    /// <param name="encoding"></param>
    /// <param name="closeAfter">读取完毕后关闭流</param>
    /// <returns></returns>
    public static async Task<string> ReadAllTextAsync(this FileStream stream, Encoding encoding, bool closeAfter = true)
    {
        var sr = new StreamReader(stream, encoding);
        var text = await sr.ReadToEndAsync().ConfigureAwait(false);
        if (closeAfter)
        {
            sr.Close();
            sr.Dispose();
            stream.Close();
#if NET5_0_OR_GREATER
                await stream.DisposeAsync().ConfigureAwait(false);
#else
            stream.Dispose();
#endif
        }

        return text;
    }

    /// <summary>
    /// 写入所有文本
    /// </summary>
    /// <param name="stream"></param>
    /// <param name="content"></param>
    /// <param name="encoding"></param>
    /// <param name="closeAfter">读取完毕后关闭流</param>
    /// <returns></returns>
    public static async Task WriteAllTextAsync(this FileStream stream, string content, Encoding encoding, bool closeAfter = true)
    {
        var sw = new StreamWriter(stream, encoding);
        stream.SetLength(0);
        await sw.WriteAsync(content).ConfigureAwait(false);
        await sw.FlushAsync().ConfigureAwait(false);
        if (closeAfter)
        {
            sw.Close();
            stream.Close();
#if NET5_0_OR_GREATER
                await sw.DisposeAsync().ConfigureAwait(false);
                await stream.DisposeAsync().ConfigureAwait(false);
#else
            sw.Dispose();
            stream.Dispose();
#endif
        }
    }

    /// <summary>
    /// 写入所有文本行
    /// </summary>
    /// <param name="stream"></param>
    /// <param name="lines"></param>
    /// <param name="encoding"></param>
    /// <param name="closeAfter">读取完毕后关闭流</param>
    /// <returns></returns>
    public static async Task WriteAllLinesAsync(this FileStream stream, IEnumerable<string> lines, Encoding encoding, bool closeAfter = true)
    {
        var sw = new StreamWriter(stream, encoding);
        stream.SetLength(0);
        foreach (var line in lines)
        {
            await sw.WriteLineAsync(line).ConfigureAwait(false);
        }

        await sw.FlushAsync().ConfigureAwait(false);
        if (closeAfter)
        {
            sw.Close();
            stream.Close();
#if NET5_0_OR_GREATER
                await sw.DisposeAsync().ConfigureAwait(false);
                await stream.DisposeAsync().ConfigureAwait(false);
#else
            sw.Dispose();
            stream.Dispose();
#endif
        }
    }

#if NET5_0_OR_GREATER

        /// <summary>
        ///
        /// </summary>
        /// <param name="stream"></param>
        /// <param name="cancellationToken"></param>
        /// <returns></returns>
        public static async Task<byte[]> ToArrayAsync(this Stream stream, CancellationToken cancellationToken = default)
        {
            stream.Position = 0;
            byte[] bytes = new byte[stream.Length];
            await stream.ReadAsync(bytes, cancellationToken);
            stream.Seek(0, SeekOrigin.Begin);
            return bytes;
        }
#endif
}