Move request/response models (#1754)

2025-07-01 08:02:49 -05:00 · 2021-12-14 15:05:07 +00:00
parent 3ae573bd8d
commit 63f6dd9a24
206 changed files with 641 additions and 516 deletions
--- a/src/Api/Models/Request/SendRequestModel.cs
+++ b/src/Api/Models/Request/SendRequestModel.cs
@ -0,0 +1,144 @@
+using System;
+using System.ComponentModel.DataAnnotations;
+using Bit.Core.Enums;
+using Bit.Core.Exceptions;
+using Bit.Core.Models.Data;
+using Bit.Core.Models.Table;
+using Bit.Core.Services;
+using Bit.Core.Utilities;
+using Newtonsoft.Json;
+
+namespace Bit.Api.Models.Request
+{
+    public class SendRequestModel
+    {
+        public SendType Type { get; set; }
+        public long? FileLength { get; set; } = null;
+        [EncryptedString]
+        [EncryptedStringLength(1000)]
+        public string Name { get; set; }
+        [EncryptedString]
+        [EncryptedStringLength(1000)]
+        public string Notes { get; set; }
+        [Required]
+        [EncryptedString]
+        [EncryptedStringLength(1000)]
+        public string Key { get; set; }
+        [Range(1, int.MaxValue)]
+        public int? MaxAccessCount { get; set; }
+        public DateTime? ExpirationDate { get; set; }
+        [Required]
+        public DateTime? DeletionDate { get; set; }
+        public SendFileModel File { get; set; }
+        public SendTextModel Text { get; set; }
+        [StringLength(1000)]
+        public string Password { get; set; }
+        [Required]
+        public bool? Disabled { get; set; }
+        public bool? HideEmail { get; set; }
+
+        public Send ToSend(Guid userId, ISendService sendService)
+        {
+            var send = new Send
+            {
+                Type = Type,
+                UserId = (Guid?)userId
+            };
+            ToSend(send, sendService);
+            return send;
+        }
+
+        public (Send, SendFileData) ToSend(Guid userId, string fileName, ISendService sendService)
+        {
+            var send = ToSendBase(new Send
+            {
+                Type = Type,
+                UserId = (Guid?)userId
+            }, sendService);
+            var data = new SendFileData(Name, Notes, fileName);
+            return (send, data);
+        }
+
+        public Send ToSend(Send existingSend, ISendService sendService)
+        {
+            existingSend = ToSendBase(existingSend, sendService);
+            switch (existingSend.Type)
+            {
+                case SendType.File:
+                    var fileData = JsonConvert.DeserializeObject<SendFileData>(existingSend.Data);
+                    fileData.Name = Name;
+                    fileData.Notes = Notes;
+                    existingSend.Data = JsonConvert.SerializeObject(fileData,
+                        new JsonSerializerSettings { NullValueHandling = NullValueHandling.Ignore });
+                    break;
+                case SendType.Text:
+                    existingSend.Data = JsonConvert.SerializeObject(ToSendData(),
+                        new JsonSerializerSettings { NullValueHandling = NullValueHandling.Ignore });
+                    break;
+                default:
+                    throw new ArgumentException("Unsupported type: " + nameof(Type) + ".");
+            }
+            return existingSend;
+        }
+
+        public void ValidateCreation()
+        {
+            var now = DateTime.UtcNow;
+            // Add 1 minute for a sane buffer and client clock float
+            var nowPlus1Minute = now.AddMinutes(1);
+            if (ExpirationDate.HasValue && ExpirationDate.Value <= nowPlus1Minute)
+            {
+                throw new BadRequestException("You cannot create a Send that is already expired. " +
+                    "Adjust the expiration date and try again.");
+            }
+            ValidateEdit();
+        }
+
+        public void ValidateEdit()
+        {
+            var now = DateTime.UtcNow;
+            // Add 1 minute for a sane buffer and client clock float
+            var nowPlus1Minute = now.AddMinutes(1);
+            if (DeletionDate.HasValue)
+            {
+                if (DeletionDate.Value <= nowPlus1Minute)
+                {
+                    throw new BadRequestException("You cannot have a Send with a deletion date in the past. " +
+                        "Adjust the deletion date and try again.");
+                }
+                if (DeletionDate.Value > now.AddDays(31))
+                {
+                    throw new BadRequestException("You cannot have a Send with a deletion date that far " +
+                        "into the future. Adjust the Deletion Date to a value less than 31 days from now " +
+                        "and try again.");
+                }
+            }
+        }
+
+        private Send ToSendBase(Send existingSend, ISendService sendService)
+        {
+            existingSend.Key = Key;
+            existingSend.ExpirationDate = ExpirationDate;
+            existingSend.DeletionDate = DeletionDate.Value;
+            existingSend.MaxAccessCount = MaxAccessCount;
+            if (!string.IsNullOrWhiteSpace(Password))
+            {
+                existingSend.Password = sendService.HashPassword(Password);
+            }
+            existingSend.Disabled = Disabled.GetValueOrDefault();
+            existingSend.HideEmail = HideEmail.GetValueOrDefault();
+            return existingSend;
+        }
+
+        private SendData ToSendData()
+        {
+            return new SendTextData(Name, Notes, Text.Text, Text.Hidden);
+        }
+    }
+
+    public class SendWithIdRequestModel : SendRequestModel
+    {
+        [Required]
+        public Guid? Id { get; set; }
+    }
+}