withdraw requests API

Author: mx2s

App/AL/Controller/Withdraw/WithdrawalsCrudController.cs

@@ -16,20 +16,30 @@ public class WithdrawalsCrudController : BaseController {
         protected override IMiddleware[] Middleware() => new IMiddleware[] { new JwtMiddleware() };
 
         public WithdrawalsCrudController() {
-            Post("/api/v1/me/withdrawals/new", _ => {
+            Post("/api/v1/me/withdrawal/new", _ => {
                 var errors = ValidationProcessor.Process(Request, new IValidatorRule[] {
-                    new ShouldHaveParameters(new[] {"amount", "currency_type"}),
+                    new ShouldHaveParameters(new[] {"amount", "currency_type", "address"}),
+                    new MinLength("address", 4),
                     new ShouldBeCorrectEnumValue("currency_type", typeof(CurrencyType)),
                 }, true);
                 if (errors.Count > 0) return HttpResponse.Errors(errors);
 
                 var me = UserRepository.Find(CurrentRequest.UserId);
 
                 var currencyType = (CurrencyType) GetRequestEnum("currency_type", typeof(CurrencyType));
+                
+                decimal amount = System.Convert.ToDecimal(GetRequestStr("amount"));
 
-                var amount = System.Convert.ToDecimal(GetRequestStr("amount"));
+                if (amount < 0.01M)
+                    return HttpResponse.Error(HttpStatusCode.Forbidden, "Amount cannot be less than 0.01");
+                
+                var userBalance = UserBalanceRepository.Find(me, currencyType);
+                if (userBalance == null || amount > userBalance.balance) 
+                    return HttpResponse.Error(HttpStatusCode.Forbidden, "You cannot withdraw more that you currently have");
+                
+                var address = GetRequestStr("address");
 
-                var withdrawalRequest = WithdrawalRequestRepository.Create(me, currencyType, amount);
+                var withdrawalRequest = WithdrawalRequestRepository.Create(me, currencyType, amount, address);
 
                 return HttpResponse.Item(
                     "withdraw_request", new WithdrawalRequestTransformer().Transform(withdrawalRequest),
@@ -39,10 +49,10 @@ public WithdrawalsCrudController() {
 
             Get("/api/v1/me/withdrawals/get", _ => {
                 var me = UserRepository.Find(CurrentRequest.UserId);
-                var withdrawalRequest = WithdrawalRequestRepository.Get(me);
+                var withdrawalRequests = WithdrawalRequestRepository.Get(me);
 
                 return HttpResponse.Item(
-                    "withdraw_requests", new WithdrawalRequestTransformer().Many(withdrawalRequest)
+                    "withdraw_requests", new WithdrawalRequestTransformer().Many(withdrawalRequests)
                 );
             });
         }

App/DL/Model/User/UserBalance.cs

@@ -30,6 +30,12 @@ public static UserBalance Find(int id)
                 "SELECT * FROM user_balances WHERE id = @id LIMIT 1",
                 new {id}
             ).FirstOrDefault();
+        
+        public static UserBalance Find(User user, CurrencyType currencyType)
+            => Connection().Query<UserBalance>(
+                $"SELECT * FROM user_balances WHERE user_id = @user_id AND currency_type = '{currencyType.ToString()}' LIMIT 1",
+                new {user_id = user.id}
+            ).FirstOrDefault();
 
         public static UserBalance FindByGuid(string guid)
             => Connection().Query<UserBalance>(

App/DL/Model/Withdraw/WithdrawalRequest.cs

@@ -14,6 +14,8 @@ public class WithdrawalRequest : Micron.DL.Model.Model {
 
         public decimal amount;
 
+        public string address;
+
         public bool paid;
 
         public CurrencyType currency_type;
@@ -31,13 +33,13 @@ public static WithdrawalRequest[] Get(User.User user)
                 new {user_id = user.id}
             ).ToArray();
 
-        public static int Create(User.User user, CurrencyType currencyType, decimal amount) {
+        public static int Create(User.User user, CurrencyType currencyType, decimal amount, string address) {
             return ExecuteScalarInt(
-                $@"INSERT INTO withdrawal_requests(guid, user_id, currency_type, amount)
-                           VALUES (@guid, @user_id, '{currencyType.ToString()}', @amount);
+                $@"INSERT INTO withdrawal_requests(guid, user_id, currency_type, amount, address)
+                           VALUES (@guid, @user_id, '{currencyType.ToString()}', @amount, @address);
                            SELECT currval('withdrawal_requests_id_seq');"
                 , new {
-                    guid = Guid.NewGuid().ToString(), user_id = user.id, amount
+                    guid = Guid.NewGuid().ToString(), user_id = user.id, amount, address
                 }
             );
         }

App/DL/Repository/User/UserBalanceRepository.cs

@@ -12,6 +12,8 @@ public static UserBalance Find(int id) {
         public static UserBalance Find(UserModel user) {
             return UserBalance.FindByUserId(user.id);
         }
+        
+        public static UserBalance Find(UserModel user, CurrencyType currencyType) => UserBalance.Find(user, currencyType);
 
         public static UserBalance FindOrCreate(Invoice invoice) {
             return FindOrCreate(

App/DL/Repository/Withdrawal/WithdrawalRequestRepository.cs

@@ -7,7 +7,7 @@ public static class WithdrawalRequestRepository {
 
         public static WithdrawalRequest[] Get(Model.User.User user) => WithdrawalRequest.Get(user);
 
-        public static WithdrawalRequest Create(Model.User.User user, CurrencyType currencyType, decimal amount)
-            => Find(WithdrawalRequest.Create(user, currencyType, amount));
+        public static WithdrawalRequest Create(Model.User.User user, CurrencyType currencyType, decimal amount, string address)
+            => Find(WithdrawalRequest.Create(user, currencyType, amount, address));
     }
 }

App/PL/Transformer/Withdraw/WithdrawalRequestTransformer.cs

@@ -11,6 +11,7 @@ public override JObject Transform(object obj) {
                 ["guid"] = item.guid,
                 ["user_guid"] = item.User().guid,
                 ["amount"] = item.amount,
+                ["address"] = item.address,
                 ["paid"] = item.paid,
                 ["currency_type"] = item.currency_type.ToString(),
                 ["created_at"] = item.created_at.ToString(CultureInfo.InvariantCulture),

migrations/db/migrations/20191228144719_create_withdrawal_requests_table.php

@@ -16,6 +16,7 @@ public function change()
                 'null' => false,
                 'scale' => 8
             ])
+            ->addColumn('address', 'string')
             ->addColumn('paid', 'boolean', ['default' => false])
             ->addTimestamps()