wabbajack/Wabbajack.Common/StatusFileStream.cs

using System;
using System.IO;

namespace Wabbajack.Common
{
    public class StatusFileStream : Stream
    {
        private string _message;
        private Stream _inner;
        private WorkQueue? _queue;
        private DateTime _lastUpdate;
        private TimeSpan _span;

        public StatusFileStream(Stream fs, string message, WorkQueue? queue = null)
        {
            _queue = queue;
            _inner = fs;
            _message = message;
            _lastUpdate = DateTime.UnixEpoch;
            _span = TimeSpan.FromMilliseconds(500);
        }

        public override void Flush()
        {
            _inner.Flush();
        }

        public override long Seek(long offset, SeekOrigin origin)
        {
            return _inner.Seek(offset, origin);
        }

        public override void SetLength(long value)
        {
            _inner.SetLength(value);
        }

        public override int Read(byte[] buffer, int offset, int count)
        {
            UpdateStatus();
            return _inner.Read(buffer, offset, count);
        }

        private void UpdateStatus()
        {
            if (DateTime.Now - _lastUpdate < _span)
            {
                return;
            }
            
            _lastUpdate = DateTime.Now;

            if (_inner.Length == 0)
            {
                return;
            }

            if (_queue != null)
                _queue.Report(_message, Percent.FactoryPutInRange(_inner.Position, _inner.Length));
            else
                Utils.Status(_message, Percent.FactoryPutInRange(_inner.Position, _inner.Length));
        }

        public override void Write(byte[] buffer, int offset, int count)
        {
            UpdateStatus();
            _inner.Write(buffer, offset, count);
        }

        public override bool CanRead => _inner.CanRead;
        public override bool CanSeek => _inner.CanSeek;
        public override bool CanWrite => _inner.CanWrite;
        public override long Length => _inner.Length;

        public override long Position
        {
            get => _inner.Position;
            set => _inner.Position = value;
        }
    }
}