From 468dd6cdfc70ed6f5737d3030cd630fa8d169f63 Mon Sep 17 00:00:00 2001
From: Anairkoen Schno <aaron.cirr.com@gmail.com>
Date: Sat, 1 Sep 2018 21:53:48 -0500
Subject: [PATCH] Re-added BackupUnit.cs because it keeps getting ignored

---
 IPA.Loader/Updating/Backup/BackupUnit.cs | 144 +++++++++++++++++++++++
 1 file changed, 144 insertions(+)
 create mode 100644 IPA.Loader/Updating/Backup/BackupUnit.cs

diff --git a/IPA.Loader/Updating/Backup/BackupUnit.cs b/IPA.Loader/Updating/Backup/BackupUnit.cs
new file mode 100644
index 00000000..0796d27e
--- /dev/null
+++ b/IPA.Loader/Updating/Backup/BackupUnit.cs
@@ -0,0 +1,144 @@
+using IPA.Logging;
+using IPA.Utilities;
+using System;
+using System.Collections.Generic;
+using System.Collections.Specialized;
+using System.IO;
+using System.Linq;
+using System.Text;
+
+namespace IPA.Updating.Backup
+{
+    /// <summary>
+    /// A unit for backup. WIP.
+    /// </summary>
+    internal class BackupUnit
+    {
+        public string Name { get; private set; }
+
+        private DirectoryInfo _BackupPath;
+        private List<string> _Files = new List<string>();
+        private FileInfo _ManifestFile;
+        private static string _ManifestFileName = "$manifest$.txt";
+
+        public BackupUnit(string path) : this(path, DateTime.Now.ToString("yyyy-MM-dd_h-mm-ss"))
+        {
+        }
+
+        internal BackupUnit(string path, string name)
+        {
+            Name = name;
+            _BackupPath = new DirectoryInfo(Path.Combine(path, Name));
+            _ManifestFile = new FileInfo(Path.Combine(_BackupPath.FullName, _ManifestFileName));
+        }
+
+        public static BackupUnit FromDirectory(DirectoryInfo directory, string backupPath)
+        {
+            var unit = new BackupUnit(backupPath, 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)
+        {
+            var relativePath = LoneFunctions.GetRelativePath(Environment.CurrentDirectory, file.FullName);
+            var backupPath = new FileInfo(Path.Combine(_BackupPath.FullName, relativePath));
+
+            if (_Files.Contains(relativePath))
+            {
+                Logger.log.Debug($"Skipping backup of {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)
+            {
+                Logger.log.Debug($"Restoring {relativePath}");
+                // Original version
+                var backupFile = new FileInfo(Path.Combine(_BackupPath.FullName, relativePath));
+                var target = new FileInfo(Path.Combine(Environment.CurrentDirectory, relativePath));
+
+                if (backupFile.Exists)
+                {
+                    if (backupFile.Length > 0)
+                    {
+                        Logger.log.Debug($"  {backupFile.FullName} => {target.FullName}");
+                        target.Directory.Create();
+                        backupFile.CopyTo(target.FullName, true);
+                    }
+                    else
+                    {
+                        Logger.log.Debug($"  x {target.FullName}");
+                        if (target.Exists)
+                        {
+                            target.Delete();
+                        }
+                    }
+                }
+                else
+                {
+                    Logger.log.Error("Backup not found!");
+                }
+            }
+        }
+
+
+    }
+}