(chocolateyGH-1047) Add "replace" filesystem operation

Allows for the more correct replacement and backup of files on the filesystem.
RichiCoder1 · Dec 20, 2016 · d9d354d · d9d354d
1 parent 5b68b01
commit d9d354d
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/src/chocolatey/infrastructure/filesystem/DotNetFileSystem.cs b/src/chocolatey/infrastructure/filesystem/DotNetFileSystem.cs
@@ -390,6 +390,25 @@ public bool copy_file_unsafe(string sourceFilePath, string destinationFilePath,
             return success != 0;
         }
 
+
+        public void replace_file(string sourceFilePath, string destinationFilePath, string backupFilePath)
+        {
+            this.Log().Debug(ChocolateyLoggers.Verbose, () => "Attempting to replace \"{0}\"{1} with \"{2}\". Backup placed at \"{3}\".".format_with(destinationFilePath, Environment.NewLine, sourceFilePath, backupFilePath));
+
+            allow_retries(
+                () =>
+                {
+                    try
+                    {
+                        File.Replace(sourceFilePath, destinationFilePath, backupFilePath);
+                    }
+                    catch (IOException)
+                    {
+                        Alphaleonis.Win32.Filesystem.File.Replace(sourceFilePath, destinationFilePath, backupFilePath);
+                    }
+                });
+        }
+
         // ReSharper disable InconsistentNaming
 
         // http://msdn.microsoft.com/en-us/library/windows/desktop/aa363851.aspx

diff --git a/src/chocolatey/infrastructure/filesystem/IFileSystem.cs b/src/chocolatey/infrastructure/filesystem/IFileSystem.cs
@@ -204,6 +204,14 @@ public interface IFileSystem
         /// <returns>true if copy was successful, otherwise false</returns>
         bool copy_file_unsafe(string sourceFilePath, string destinationFilePath, bool overwriteExisting);
 
+        /// <summary>
+        ///   Replace an existing file.
+        /// </summary>
+        /// <param name="sourceFilePath">Where is the file now?</param>
+        /// <param name="destinationFilePath">Where would you like it to go?</param>
+        /// <param name="backupFilePath">Where should the existing file be placed? Null if nowhere.</param>
+        void replace_file(string sourceFilePath, string destinationFilePath, string backupFilePath);
+
         /// <summary>
         ///   Deletes the specified file.
         /// </summary>