Merge pull request #586 from hargata/Hargata/cleanup.api

Add API Controller method to clean up temp files and unlinked thumbnails
hargata · Aug 22, 2024 · 6918138 · 6918138
2 parents 522322e + 0defb0f
commit 6918138
Show file tree

Hide file tree

Showing 2 changed files with 57 additions and 0 deletions.
diff --git a/Controllers/APIController.cs b/Controllers/APIController.cs
@@ -562,6 +562,28 @@ public IActionResult MakeBackup()
         }
         [Authorize(Roles = nameof(UserData.IsRootUser))]
         [HttpGet]
+        [Route("/api/cleanup")]
+        public IActionResult CleanUp(bool deepClean = false)
+        {
+            var jsonResponse = new Dictionary<string, string>();
+            //Clear out temp folder
+            var tempFilesDeleted = _fileHelper.ClearTempFolder();
+            jsonResponse.Add("temp_files_deleted", tempFilesDeleted.ToString());
+            if (deepClean)
+            {
+                //clear out unused vehicle thumbnails
+                var vehicles = _dataAccess.GetVehicles();
+                var vehicleImages = vehicles.Select(x => x.ImageLocation).Where(x => x.StartsWith("/images/")).Select(x=>Path.GetFileName(x)).ToList();
+                if (vehicleImages.Any())
+                {
+                    var thumbnailsDeleted = _fileHelper.ClearUnlinkedThumbnails(vehicleImages);
+                    jsonResponse.Add("unlinked_thumbnails_deleted", thumbnailsDeleted.ToString());
+                }
+            }
+            return Json(jsonResponse);
+        }
+        [Authorize(Roles = nameof(UserData.IsRootUser))]
+        [HttpGet]
         [Route("/api/demo/restore")]
         public IActionResult RestoreDemo()
         {

diff --git a/Helper/FileHelper.cs b/Helper/FileHelper.cs
@@ -13,6 +13,8 @@ public interface IFileHelper
         bool RestoreBackup(string fileName, bool clearExisting = false);
         string MakeAttachmentsExport(List<GenericReportModel> exportData);
         List<string> GetLanguages();
+        int ClearTempFolder();
+        int ClearUnlinkedThumbnails(List<string> linkedImages);
     }
     public class FileHelper : IFileHelper
     {
@@ -314,5 +316,38 @@ public bool DeleteFile(string currentFilePath)
                 return false;
             }
         }
+        public int ClearTempFolder()
+        {
+            int filesDeleted = 0;
+            var tempPath = GetFullFilePath("temp", false);
+            if (Directory.Exists(tempPath))
+            {
+                var files = Directory.GetFiles(tempPath);
+                foreach (var file in files)
+                {
+                    File.Delete(file);
+                    filesDeleted++;
+                }
+            }
+            return filesDeleted;
+        }
+        public int ClearUnlinkedThumbnails(List<string> linkedImages)
+        {
+            int filesDeleted = 0;
+            var imagePath = GetFullFilePath("images", false);
+            if (Directory.Exists(imagePath))
+            {
+                var files = Directory.GetFiles(imagePath);
+                foreach(var file in files)
+                {
+                    if (!linkedImages.Contains(Path.GetFileName(file)))
+                    {
+                        File.Delete(file);
+                        filesDeleted++;
+                    }
+                }
+            }
+            return filesDeleted;
+        }
     }
 }