5to workshop: Almacenamiento en los contratos

Author: andreMD287

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

@@ -1,23 +1,62 @@
-use crate::interfaces::contract::RentACarContractTrait;
-use soroban_sdk::{contract, contractimpl, symbol_short, Address, Env, Symbol};
+use crate::{
+    interfaces::contract::RentACarContractTrait,
+    storage::{
+        admin::{read_admin, write_admin},
+        car::{read_car, remove_car, write_car},
+        rental::write_rental,
+        structs::car::Car,
+        structs::rental::Rental,
+        token::write_token,
+        types::car_status::CarStatus,
+    },
+};
+
+use soroban_sdk::{contract, contractimpl, Address, Env};
 
 #[contract]
 pub struct RentACarContract;
 
-pub const ADMIN_KEY: &Symbol = &symbol_short!("ADMIN");
-pub const TOKEN_KEY: &Symbol = &symbol_short!("TOKEN");
-
 #[contractimpl]
 impl RentACarContractTrait for RentACarContract {
     fn __constructor(env: &Env, admin: Address, token: Address) {
-        env.storage().instance().set(ADMIN_KEY, &admin);
-        env.storage().instance().set(TOKEN_KEY, &token);
+        write_admin(env, &admin);
+        write_token(env, &token);
     }
 
     fn get_admin(env: &Env) -> Address {
-        env.storage()
-            .instance()
-            .get::<Symbol, Address>(ADMIN_KEY)
-            .unwrap()
+        read_admin(env)
+    }
+
+    fn add_car(env: &Env, owner: Address, price_per_day: i128) {
+        let car = Car {
+            price_per_day,
+            car_status: CarStatus::Available,
+        };
+
+        write_car(env, &owner, &car);
+    }
+
+    fn get_car_status(env: &Env, owner: Address) -> CarStatus {
+        let car = read_car(env, &owner);
+
+        car.car_status
+    }
+
+    fn rental(env: &Env, renter: Address, owner: Address, total_days_to_rent: u32, amount: i128) {
+        let mut car = read_car(env, &owner);
+
+        car.car_status = CarStatus::Rented;
+
+        let rental = Rental {
+            total_days_to_rent,
+            amount,
+        };
+
+        write_car(env, &owner, &car);
+        write_rental(env, &renter, &owner, &rental);
+    }
+
+    fn remove_car(env: &Env, owner: Address) {
+        remove_car(env, &owner);
     }
 }

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

@@ -1,6 +1,12 @@
 use soroban_sdk::{Address, Env};
 
+use crate::storage::types::car_status::CarStatus;
+
 pub trait RentACarContractTrait {
     fn __constructor(env: &Env, admin: Address, token: Address);
     fn get_admin(env: &Env) -> Address;
+    fn add_car(env: &Env, owner: Address, price_per_day: i128);
+    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);
 }

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

@@ -1,6 +1,8 @@
 #![no_std]
+
 mod contract;
-pub mod interfaces;
+mod interfaces;
+mod storage;
 mod tests;
 
 pub use contract::RentACarContract;

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

@@ -0,0 +1,21 @@
+use soroban_sdk::{Address, Env};
+
+use super::types::storage::DataKey;
+
+pub(crate) fn has_admin(env: &Env) -> bool {
+    let key = DataKey::Admin;
+
+    env.storage().instance().has(&key)
+}
+
+pub(crate) fn read_admin(env: &Env) -> Address {
+    let key = DataKey::Admin;
+
+    env.storage().instance().get(&key).unwrap()
+}
+
+pub(crate) fn write_admin(env: &Env, admin: &Address) {
+    let key = DataKey::Admin;
+
+    env.storage().instance().set(&key, admin);
+}

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

@@ -0,0 +1,26 @@
+use soroban_sdk::{Address, Env};
+
+use crate::storage::{structs::car::Car, types::storage::DataKey};
+
+pub(crate) fn has_car(env: &Env, owner: &Address) -> bool {
+    env.storage().instance().has(&DataKey::Car(owner.clone()))
+}
+
+pub(crate) fn read_car(env: &Env, owner: &Address) -> Car {
+    env.storage()
+        .instance()
+        .get(&DataKey::Car(owner.clone()))
+        .unwrap()
+}
+
+pub(crate) fn write_car(env: &Env, owner: &Address, car: &Car) {
+    env.storage()
+        .instance()
+        .set(&DataKey::Car(owner.clone()), car);
+}
+
+pub(crate) fn remove_car(env: &Env, owner: &Address) {
+    env.storage()
+        .instance()
+        .remove(&DataKey::Car(owner.clone()));
+}

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

@@ -0,0 +1,6 @@
+pub mod admin;
+pub mod car;
+pub mod rental;
+pub mod structs;
+pub mod token;
+pub mod types;

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

@@ -0,0 +1,28 @@
+use soroban_sdk::{Address, Env};
+
+use crate::storage::{structs::rental::Rental, types::storage::DataKey};
+
+pub(crate) fn has_rental(env: &Env, renter: &Address, car_owner: &Address) -> bool {
+    let key: DataKey = DataKey::Rental(renter.clone(), car_owner.clone());
+
+    env.storage().instance().has(&key)
+}
+
+pub(crate) fn write_rental(env: &Env, renter: &Address, car_owner: &Address, rental: &Rental) {
+    env.storage()
+        .instance()
+        .set(&DataKey::Rental(renter.clone(), car_owner.clone()), rental);
+}
+
+pub(crate) fn read_rental(env: &Env, renter: &Address, car_owner: &Address) -> Rental {
+    env.storage()
+        .instance()
+        .get(&DataKey::Rental(renter.clone(), car_owner.clone()))
+        .unwrap()
+}
+
+pub(crate) fn remove_rental(env: &Env, renter: &Address, car_owner: &Address) {
+    env.storage()
+        .instance()
+        .remove(&DataKey::Rental(renter.clone(), car_owner.clone()));
+}

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

@@ -0,0 +1,10 @@
+use soroban_sdk::contracttype;
+
+use crate::storage::types::car_status::CarStatus;
+
+#[derive(Clone)]
+#[contracttype]
+pub struct Car {
+    pub price_per_day: i128,
+    pub car_status: CarStatus,
+}

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

@@ -0,0 +1,2 @@
+pub mod car;
+pub mod rental;

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

@@ -0,0 +1,8 @@
+use soroban_sdk::contracttype;
+
+#[derive(Clone)]
+#[contracttype]
+pub struct Rental {
+    pub total_days_to_rent: u32,
+    pub amount: i128,
+}