Files
Mitria_Minecraft_Project/CommonLibrary/Packer.cs
2022-03-07 14:35:38 +09:00

148 lines
5.5 KiB
C#

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.IO;
using System.IO.Compression;
namespace CommonLibrary
{
internal class ProgressStream : Stream
{
private readonly Stream insideStream;
private readonly IProgress<int> insideReadProgress;
private readonly IProgress<int> insideWriteProgress;
public ProgressStream(Stream stream, IProgress<int> readProgress, IProgress<int> writeProgress)
{
insideStream = stream;
insideReadProgress = readProgress;
insideWriteProgress = writeProgress;
}
public override bool CanRead { get { return insideStream.CanRead; } }
public override bool CanSeek { get { return insideStream.CanSeek; } }
public override bool CanWrite { get { return insideStream.CanWrite; } }
public override long Length { get { return insideStream.Length; } }
public override long Position
{
get { return insideStream.Position; }
set { insideStream.Position = value; }
}
public override void Flush()
{
insideStream.Flush();
}
public override long Seek(long offset, SeekOrigin origin)
{
return insideStream.Seek(offset, origin);
}
public override void SetLength(long value)
{
insideStream.SetLength(value);
}
public override int Read(byte[] buffer, int offset, int count)
{
int bytesRead = insideStream.Read(buffer, offset, count);
insideReadProgress?.Report(bytesRead);
return bytesRead;
}
public override void Write(byte[] buffer, int offset, int count)
{
insideStream.Write(buffer, offset, count);
insideWriteProgress?.Report(count);
}
}
public class ProgressPacker
{
public void Pack(string sourceDirectoryName, string packFileName, IProgress<double> progress)
{
sourceDirectoryName = Path.GetFullPath(sourceDirectoryName);
FileInfo[] sourceFiles = new DirectoryInfo(sourceDirectoryName).GetFiles("*", SearchOption.AllDirectories);
double totalBytes = sourceFiles.Sum(f => f.Length);
long currentBytes = 0;
using (ZipArchive zipArchive = ZipFile.Open(packFileName, ZipArchiveMode.Create))
{
foreach (FileInfo file in sourceFiles)
{
string entryName = file.FullName.Substring(sourceDirectoryName.Length + 1);
ZipArchiveEntry entry = zipArchive.CreateEntry(entryName);
entry.LastWriteTime = file.LastWriteTime;
using (Stream inputStream = file.OpenRead())
{
using (Stream outputStream = entry.Open())
{
Stream progressStream = new ProgressStream(inputStream,
new BasicProgress<int>(i =>
{
currentBytes += i;
progress.Report(currentBytes / totalBytes);
}), null);
progressStream.CopyTo(outputStream);
}
}
}
}
}
public void UnPack(string packFileName, string destinationDirectoryName, IProgress<double> progress)
{
using (ZipArchive zipArchive = ZipFile.OpenRead(packFileName))
{
double totalBytes = zipArchive.Entries.Sum(f => f.Length);
long currentBytes = 0;
foreach (ZipArchiveEntry entry in zipArchive.Entries)
{
string fileName = Path.Combine(destinationDirectoryName, entry.FullName);
if (entry.Name == "") continue;
Directory.CreateDirectory(Path.GetDirectoryName(fileName));
using (Stream inputStream = entry.Open())
{
using (Stream outputStream = File.OpenWrite(fileName))
{
Stream progressStream = new ProgressStream(outputStream, null,
new BasicProgress<int>(i =>
{
currentBytes += i;
progress.Report(currentBytes / totalBytes);
}));
inputStream.CopyTo(progressStream);
}
File.SetLastWriteTime(fileName, entry.LastWriteTime.LocalDateTime);
}
}
}
}
}
public class BasicProgress<T> : IProgress<T>
{
private readonly Action<T> actionHandler;
public BasicProgress(Action<T> handler)
{
actionHandler = handler;
}
void IProgress<T>.Report(T value)
{
actionHandler(value);
}
}
}