feat(storage): implements object query

amimart · amimart · commit 5bccebd744d3 · 2023-03-03T10:08:54.000+01:00
diff --git a/contracts/cw-storage/src/contract.rs b/contracts/cw-storage/src/contract.rs
@@ -7,7 +7,7 @@ use cw2::set_contract_version;
 use crate::ContractError::{NotImplemented, ObjectMaxSizeLimitExceeded, MaxObjectPinsLimitExceeded, MaxObjectsLimitExceeded, BucketSizeLimitExceeded};
 
 use crate::error::ContractError;
-use crate::msg::{ExecuteMsg, InstantiateMsg, QueryMsg};
+use crate::msg::{ExecuteMsg, InstantiateMsg, QueryMsg, ObjectId};
 use crate::state::{Bucket, BUCKET, DATA, Object, objects, Pin, pins};
 use crate::crypto::sha256_hash;
 
@@ -113,13 +113,14 @@ pub mod execute {
 pub fn query(deps: Deps, _env: Env, msg: QueryMsg) -> StdResult<Binary> {
     match msg {
         QueryMsg::Bucket {} => to_binary(&query::bucket(deps)?),
+        QueryMsg::Object {id} => to_binary(&query::object(deps, id)?),
         _ => Err(StdError::generic_err("Not implemented")),
     }
 }
 
 pub mod query {
     use super::*;
-    use crate::msg::BucketResponse;
+    use crate::msg::{BucketResponse, ObjectResponse};
 
     pub fn bucket(deps: Deps) -> StdResult<BucketResponse> {
         let bucket = BUCKET.load(deps.storage)?;
@@ -129,6 +130,19 @@ pub mod query {
             limits: bucket.limits.into(),
         })
     }
+
+    pub fn object(deps: Deps, id: ObjectId) -> StdResult<ObjectResponse> {
+        objects().load(deps.storage, id)
+            .map(|object| ObjectResponse{
+                id: object.id,
+                size: object.size.into(),
+                owner: object.owner.into(),
+                is_pinned: pins().idx.object
+                    .keys_raw(deps.storage, None, None, Order::Ascending)
+                    .next()
+                    .is_some(),
+            })
+    }
 }
 
 #[cfg(test)]
diff --git a/contracts/cw-storage/src/msg.rs b/contracts/cw-storage/src/msg.rs
@@ -4,10 +4,10 @@ use cosmwasm_std::Binary;
 use cosmwasm_std::Uint128;
 
 /// ObjectId is the type of identifier of an object in the bucket.
-type ObjectId = String;
+pub type ObjectId = String;
 
 /// Cursor is the opaque type of cursor used for pagination.
-type Cursor = String;
+pub type Cursor = String;
 
 /// Instantiate messages
 #[cw_serde]
