6to workshop: Deposito y retiro en los contratos

Author: andreMD287

contracts/rent-a-car/src/contract.rs

@@ -1,8 +1,10 @@
 use crate::{
     interfaces::contract::RentACarContractTrait,
+    methods::token::token::token_transfer,
     storage::{
         admin::{read_admin, write_admin},
         car::{read_car, remove_car, write_car},
+        contract_balance::{read_contract_balance, write_contract_balance},
         rental::write_rental,
         structs::car::Car,
         structs::rental::Rental,
@@ -31,6 +33,7 @@ impl RentACarContractTrait for RentACarContract {
         let car = Car {
             price_per_day,
             car_status: CarStatus::Available,
+            available_to_withdraw: 0,
         };
 
         write_car(env, &owner, &car);
@@ -43,17 +46,41 @@ impl RentACarContractTrait for RentACarContract {
     }
 
     fn rental(env: &Env, renter: Address, owner: Address, total_days_to_rent: u32, amount: i128) {
+        renter.require_auth();
+
         let mut car = read_car(env, &owner);
 
         car.car_status = CarStatus::Rented;
+        car.available_to_withdraw += amount;
 
         let rental = Rental {
             total_days_to_rent,
             amount,
         };
 
+        let mut contract_balance = read_contract_balance(&env);
+        contract_balance += amount;
+
+        write_contract_balance(&env, &contract_balance);
         write_car(env, &owner, &car);
         write_rental(env, &renter, &owner, &rental);
+
+        token_transfer(&env, &renter, &env.current_contract_address(), &amount);
+    }
+
+    fn payout_owner(env: &Env, owner: Address, amount: i128) {
+        owner.require_auth();
+
+        let mut car = read_car(&env, &owner);
+        let mut contract_balance = read_contract_balance(&env);
+
+        car.available_to_withdraw -= amount;
+        contract_balance -= amount;
+
+        write_car(&env, &owner, &car);
+        write_contract_balance(&env, &contract_balance);
+
+        token_transfer(&env, &env.current_contract_address(), &owner, &amount);
     }
 
     fn remove_car(env: &Env, owner: Address) {

contracts/rent-a-car/src/interfaces/contract.rs

@@ -9,4 +9,5 @@ pub trait RentACarContractTrait {
     fn get_car_status(env: &Env, owner: Address) -> CarStatus;
     fn rental(env: &Env, renter: Address, owner: Address, total_days_to_rent: u32, amount: i128);
     fn remove_car(env: &Env, owner: Address);
+    fn payout_owner(env: &Env, owner: Address, amount: i128);
 }

contracts/rent-a-car/src/lib.rs

@@ -2,6 +2,7 @@
 
 mod contract;
 mod interfaces;
+mod methods;
 mod storage;
 mod tests;
 

contracts/rent-a-car/src/methods/mod.rs

@@ -0,0 +1 @@
+pub mod token;

contracts/rent-a-car/src/methods/token/mod.rs

@@ -0,0 +1 @@
+pub mod token;

contracts/rent-a-car/src/methods/token/token.rs

@@ -0,0 +1,12 @@
+use soroban_sdk::{
+    token::{self},
+    Address, Env,
+};
+
+use crate::storage::token::read_token;
+
+pub fn token_transfer(env: &Env, from: &Address, to: &Address, amount: &i128) {
+    let token_address = read_token(env);
+    let token = token::TokenClient::new(env, &token_address);
+    token.transfer(from, to, amount);
+}

contracts/rent-a-car/src/storage/contract_balance.rs

@@ -0,0 +1,16 @@
+use soroban_sdk::Env;
+
+use crate::storage::types::storage::DataKey;
+
+pub fn read_contract_balance(env: &Env) -> i128 {
+    env.storage()
+        .persistent()
+        .get(&DataKey::ContractBalance)
+        .unwrap_or(0)
+}
+
+pub fn write_contract_balance(env: &Env, amount: &i128) {
+    env.storage()
+        .persistent()
+        .set(&DataKey::ContractBalance, amount);
+}

contracts/rent-a-car/src/storage/mod.rs

@@ -1,5 +1,6 @@
 pub mod admin;
 pub mod car;
+pub mod contract_balance;
 pub mod rental;
 pub mod structs;
 pub mod token;

contracts/rent-a-car/src/storage/structs/car.rs

@@ -7,4 +7,5 @@ use crate::storage::types::car_status::CarStatus;
 pub struct Car {
     pub price_per_day: i128,
     pub car_status: CarStatus,
+    pub available_to_withdraw: i128,
 }

contracts/rent-a-car/src/storage/types/storage.rs

@@ -5,6 +5,7 @@ use soroban_sdk::{contracttype, Address};
 pub enum DataKey {
     Admin,
     Token,
+    ContractBalance,
     Car(Address),
     Rental(Address, Address),
 }