Viewer to Server

Server/Services/AttachmentsService.cs

@@ -0,0 +1,138 @@
+using System.Globalization;
+
+namespace OI.Metrology.Server.Services;
+
+using OI.Metrology.Server.Models;
+using OI.Metrology.Shared.DataModels;
+using OI.Metrology.Shared.Models.Stateless;
+using OI.Metrology.Shared.Services;
+
+public class AttachmentsService : IAttachmentsService
+{
+    private readonly AppSettings _AppSettings;
+    private readonly IMetrologyRepository _MetrologyRepository;
+
+    public AttachmentsService(AppSettings appSettings, IMetrologyRepository metrologyRepository)
+    {
+        _AppSettings = appSettings;
+        _MetrologyRepository = metrologyRepository;
+    }
+
+    protected Stream GetAttachmentStream(string? tableName, Guid attachmentId, string filename)
+    {
+        if (attachmentId.Equals(Guid.Empty))
+            throw new Exception("No attachments found");
+
+        if (tableName is null)
+            throw new NullReferenceException(nameof(tableName));
+
+        DateTime insertDate = Convert.ToDateTime(_MetrologyRepository.GetAttachmentInsertDateByGUID(tableName, attachmentId));
+        int year = insertDate.Year;
+        DateTime d = insertDate;
+        CultureInfo cul = CultureInfo.CurrentCulture;
+        int weekNum = cul.Calendar.GetWeekOfYear(d, CalendarWeekRule.FirstDay, DayOfWeek.Sunday);
+        string workWeek = "WW" + weekNum.ToString("00");
+        string dateDir = year + @"\" + workWeek;
+
+        string fullPath = Path.Combine(_AppSettings.AttachmentPath, tableName + "_", dateDir, attachmentId.ToString(), filename);
+
+        // Check to see if file exists in the "New" directory structure, if not change the path back to the old. and check there
+        if (!File.Exists(fullPath))
+        {
+            fullPath = Path.Combine(_AppSettings.AttachmentPath, tableName, attachmentId.ToString(), filename);
+        }
+
+        if (!File.Exists(fullPath))
+            throw new Exception("File not found");
+
+        return new FileStream(fullPath, FileMode.Open, FileAccess.Read, FileShare.ReadWrite);
+    }
+
+    public Stream GetAttachmentStreamByTitle(ToolType toolType, bool header, string title, string filename)
+    {
+        if (toolType is null)
+            throw new Exception("Invalid tool type");
+        Guid attachmentId;
+        string? tableName;
+        if (header)
+        {
+            tableName = toolType.HeaderTableName;
+            attachmentId = _MetrologyRepository.GetHeaderAttachmentIDByTitle(toolType.ID, title);
+        }
+        else
+        {
+            tableName = toolType.DataTableName;
+            attachmentId = _MetrologyRepository.GetDataAttachmentIDByTitle(toolType.ID, title);
+        }
+        return GetAttachmentStream(tableName, attachmentId, filename);
+    }
+
+    public Stream GetAttachmentStreamByAttachmentId(ToolType toolType, bool header, Guid attachmentId, string filename)
+    {
+        if (toolType is null)
+            throw new Exception("Invalid tool type");
+        string? tableName;
+        if (header)
+            tableName = toolType.HeaderTableName;
+        else
+            tableName = toolType.DataTableName;
+        return GetAttachmentStream(tableName, attachmentId, filename);
+    }
+
+    private void SaveAttachment(ToolType toolType, long headerId, string dataUniqueId, string filename, IFormFile uploadedFile)
+    {
+        if (toolType is null)
+            throw new Exception("Invalid tool type");
+
+        using System.Transactions.TransactionScope trans = _MetrologyRepository.StartTransaction();
+        Guid attachmentId = Guid.Empty;
+        DateTime insertDate = new();
+        string? tableName = "";
+
+        if (string.IsNullOrWhiteSpace(dataUniqueId))
+        {
+            attachmentId = _MetrologyRepository.GetHeaderAttachmentID(toolType.ID, headerId);
+            insertDate = Convert.ToDateTime(_MetrologyRepository.GetHeaderInsertDate(toolType.ID, headerId));
+            tableName = toolType.HeaderTableName;
+
+        }
+        else
+        {
+            attachmentId = _MetrologyRepository.GetDataAttachmentID(toolType.ID, headerId, dataUniqueId);
+            insertDate = Convert.ToDateTime(_MetrologyRepository.GetDataInsertDate(toolType.ID, headerId, dataUniqueId));
+            // Get Date for new directory name
+            tableName = toolType.DataTableName;
+
+        }
+
+        int year = insertDate.Year;
+        DateTime d = insertDate;
+        CultureInfo cul = CultureInfo.CurrentCulture;
+
+        int weekNum = cul.Calendar.GetWeekOfYear(d, CalendarWeekRule.FirstDay, DayOfWeek.Sunday);
+
+        string workWeek = "WW" + weekNum.ToString("00");
+        string dateDir = year + @"\" + workWeek;
+
+        if (Equals(attachmentId, Guid.Empty))
+            throw new Exception("Invalid attachment ID");
+        string directoryPathSecondary = Path.Combine(_AppSettings.AttachmentPath, tableName + "_", dateDir, attachmentId.ToString());
+        if (!Directory.Exists(directoryPathSecondary))
+            _ = Directory.CreateDirectory(directoryPathSecondary);
+
+        string fullPathSecondary = Path.Combine(directoryPathSecondary, filename);
+
+        using (FileStream s = new(fullPathSecondary, FileMode.Create, FileAccess.ReadWrite, FileShare.None))
+        {
+            uploadedFile.CopyTo(s);
+        }
+        trans.Complete();
+    }
+
+    public void SaveAttachment(ToolType toolType, long headerId, string dataUniqueId, string filename, object uploadedFile)
+    {
+        IFormFile formFile = (IFormFile)uploadedFile;
+        SaveAttachment(toolType, headerId, dataUniqueId, filename, formFile);
+    }
+
+}