Store all repositories if storage is empty

Author: Simon Langbehn

Control/AzureStorage.cs

@@ -3,7 +3,8 @@
 using System;
 using System.IO;
 using DataOne.BitUp.Properties;
-
+using System.Linq;
+
 namespace DataOne.BitUp
 {
     public class AzureStorage : IStorage
@@ -12,23 +13,13 @@ public void SaveFile(string filePath)
         {
             string fileName = Path.GetFileName(filePath);
 
-            string accountName = Settings.Default.AzureAccountName;
-            string accountKey = Settings.Default.AzureAccountKey;
             string storageContainer = Settings.Default.AzureStorageContainer;
-            string defaultEndpointsProtocol = Resources.AzureEndpointsProtocol;
+            string accountName = Settings.Default.AzureAccountName;
             string fileLocation = Resources.AzureFileLocation;
             string fileUri = Resources.AzureFileUri;
 
-            string connectionKey = string.Concat(defaultEndpointsProtocol, "AccountName=", accountName, ";AccountKey=", accountKey, ";");
-
-            CloudStorageAccount storageAccount = CloudStorageAccount.Parse(connectionKey);
-            CloudFileClient fileClient = storageAccount.CreateCloudFileClient();
-
-            CloudFileShare share = fileClient.GetShareReference(storageContainer);
-            share.CreateIfNotExists();
-
-            CloudFileDirectory rootDir = share.GetRootDirectoryReference();
-            rootDir.CreateIfNotExists();
+            var fileClient = GetFileClient();
+            var rootDir = EnsureRootDirectory(fileClient);
 
             var cloudFileUrl = string.Concat(fileLocation, accountName, fileUri, storageContainer, "/", fileName);
             var uriToFile = new Uri(cloudFileUrl);
@@ -42,5 +33,37 @@ public string GetName()
             return "Azure File Storage";
         }
 
+        public bool IsEmpty()
+        {
+            var fileClient = GetFileClient();
+            var rootDir = EnsureRootDirectory(fileClient);
+            var content = rootDir.ListFilesAndDirectories().ToList();
+            return content.Count > 0;
+        }
+
+        private CloudFileClient GetFileClient()
+        {
+            string defaultEndpointsProtocol = Resources.AzureEndpointsProtocol;
+            string accountName = Settings.Default.AzureAccountName;
+            string accountKey = Settings.Default.AzureAccountKey;
+
+            string connectionKey = string.Concat(defaultEndpointsProtocol, "AccountName=", accountName, ";AccountKey=", accountKey, ";");
+
+            CloudStorageAccount storageAccount = CloudStorageAccount.Parse(connectionKey);
+            CloudFileClient fileClient = storageAccount.CreateCloudFileClient();
+
+            return fileClient;
+        }
+
+        private CloudFileDirectory EnsureRootDirectory(CloudFileClient fileClient)
+        {
+            string storageContainer = Settings.Default.AzureStorageContainer;
+            CloudFileShare share = fileClient.GetShareReference(storageContainer);
+            share.CreateIfNotExists();
+
+            CloudFileDirectory rootDir = share.GetRootDirectoryReference();
+            rootDir.CreateIfNotExists();
+            return rootDir;
+        }
     }
 }

Control/LocalStorage.cs

@@ -4,21 +4,30 @@ namespace DataOne.BitUp
 {
     public class LocalStorage : IStorage
     {
+        private const string RepositoryFolderName = @".\DataOne.BitUp.RepositoryBackups";
+
         public string GetName()
         {
             return "local file system";
-        }
-
+        }
+
         public void SaveFile(string filePath)
         {
             try
             {
-                Directory.CreateDirectory(@".\DataOne.BitUp.RepositoryBackups");
+                Directory.CreateDirectory(RepositoryFolderName);
             }
             finally
             {
-                File.Copy(filePath, @".\DataOne.BitUp.RepositoryBackups\" + Path.GetFileName(filePath), true);
+                File.Copy(filePath, RepositoryFolderName + @"\" + Path.GetFileName(filePath), true);
             }
         }
+
+        public bool IsEmpty()
+        {
+            var subDirs = Directory.GetDirectories(RepositoryFolderName);
+            var files = Directory.GetFiles(RepositoryFolderName);
+            return subDirs.Length > 0 && files.Length > 0;
+        }
     }
 }

Interfaces/IStorage.cs

@@ -14,5 +14,11 @@ interface IStorage
         /// </summary>
         /// <returns>The name of the location where files are saved, e.g. Azure, local file system...</returns>
         string GetName();
+
+        /// <summary>
+        /// Get the information whether the storage contents any files or directories
+        /// </summary>
+        /// <returns>Boolean value whether the storage is empty</returns>
+        bool IsEmpty();
     }
 }

Program.cs

@@ -30,7 +30,7 @@ public static void Main()
             }
             var bitbucketTeams = Settings.Default.BitbucketTeams;
             System.Net.ServicePointManager.SecurityProtocol = System.Net.SecurityProtocolType.Tls12;
-            DateTime lastBackup = DateTime.Now.AddDays(-1.0);
+            DateTime lastBackup = storage.IsEmpty() ? DateTime.MinValue : DateTime.Now.AddDays(-1.0);
 
             string workingDirectory = string.Concat(Directory.GetCurrentDirectory(), @"\DataOne.BitUp.TempRepositoryBackups");
             ForceDeleteDirectory(workingDirectory);