You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
149 lines
4.9 KiB
149 lines
4.9 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using System.Collections.Specialized;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Text;
|
|
|
|
namespace IPA.Patcher
|
|
{
|
|
/// <summary>
|
|
/// A unit for backup. WIP.
|
|
/// </summary>
|
|
public class BackupUnit
|
|
{
|
|
public string Name { get; private set; }
|
|
|
|
private DirectoryInfo _BackupPath;
|
|
private PatchContext _Context;
|
|
private List<string> _Files = new List<string>();
|
|
private FileInfo _ManifestFile;
|
|
private static string _ManifestFileName = "$manifest$.txt";
|
|
|
|
|
|
public BackupUnit(PatchContext context) : this(context, DateTime.Now.ToString("yyyy-MM-dd_h-mm-ss"))
|
|
{
|
|
}
|
|
|
|
private BackupUnit(PatchContext context, string name)
|
|
{
|
|
Name = name;
|
|
_Context = context;
|
|
_BackupPath = new DirectoryInfo(Path.Combine(_Context.BackupPath, Name));
|
|
_ManifestFile = new FileInfo(Path.Combine(_BackupPath.FullName, _ManifestFileName));
|
|
}
|
|
|
|
public static BackupUnit FromDirectory(DirectoryInfo directory, PatchContext context)
|
|
{
|
|
var unit = new BackupUnit(context, directory.Name);
|
|
|
|
// Read Manifest
|
|
if (unit._ManifestFile.Exists)
|
|
{
|
|
string manifest = File.ReadAllText(unit._ManifestFile.FullName);
|
|
foreach (var line in manifest.Split(new string[] { Environment.NewLine }, StringSplitOptions.RemoveEmptyEntries))
|
|
unit._Files.Add(line);
|
|
}
|
|
else
|
|
{
|
|
foreach (var file in directory.GetFiles("*", SearchOption.AllDirectories))
|
|
{
|
|
if (file.Name == _ManifestFileName) continue;
|
|
var relativePath = file.FullName.Substring(directory.FullName.Length + 1);
|
|
unit._Files.Add(relativePath);
|
|
}
|
|
}
|
|
|
|
return unit;
|
|
}
|
|
|
|
public void Add(string file)
|
|
{
|
|
Add(new FileInfo(file));
|
|
}
|
|
|
|
internal void Delete()
|
|
{
|
|
_BackupPath.Delete(true);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Adds a file to the list of changed files and backups it.
|
|
/// </summary>
|
|
/// <param name="path"></param>
|
|
public void Add(FileInfo file)
|
|
{
|
|
if(!file.FullName.StartsWith(_Context.ProjectRoot))
|
|
{
|
|
Console.Error.WriteLine("Invalid file path for backup! {0}", file);
|
|
return;
|
|
}
|
|
|
|
var relativePath = file.FullName.Substring(_Context.ProjectRoot.Length + 1);
|
|
var backupPath = new FileInfo(Path.Combine(_BackupPath.FullName, relativePath));
|
|
|
|
if(_Files.Contains(relativePath))
|
|
{
|
|
Console.WriteLine("Skipping backup of {0}", relativePath);
|
|
return;
|
|
}
|
|
|
|
// Copy over
|
|
backupPath.Directory.Create();
|
|
if (file.Exists)
|
|
{
|
|
file.CopyTo(backupPath.FullName);
|
|
}
|
|
else
|
|
{
|
|
// Make empty file
|
|
backupPath.Create().Close();
|
|
}
|
|
|
|
if (!File.Exists(_ManifestFile.FullName))
|
|
_ManifestFile.Create().Close();
|
|
var stream = _ManifestFile.AppendText();
|
|
stream.WriteLine(relativePath);
|
|
stream.Close();
|
|
|
|
// Add to list
|
|
_Files.Add(relativePath);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Reverts the changes made in this unit.
|
|
/// </summary>
|
|
public void Restore()
|
|
{
|
|
foreach(var relativePath in _Files)
|
|
{
|
|
Console.WriteLine("Restoring {0}", relativePath);
|
|
// Original version
|
|
var backupFile = new FileInfo(Path.Combine(_BackupPath.FullName, relativePath));
|
|
var target = new FileInfo(Path.Combine(_Context.ProjectRoot, relativePath));
|
|
|
|
if (backupFile.Exists)
|
|
{
|
|
if (backupFile.Length > 0)
|
|
{
|
|
Console.WriteLine(" {0} => {1}", backupFile.FullName, target.FullName);
|
|
target.Directory.Create();
|
|
backupFile.CopyTo(target.FullName, true);
|
|
}
|
|
else
|
|
{
|
|
Console.WriteLine(" x {0}", target.FullName);
|
|
if(target.Exists)
|
|
{
|
|
target.Delete();
|
|
}
|
|
}
|
|
} else
|
|
{
|
|
Console.Error.WriteLine("Backup not found!");
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|