Merge pull request #27 from SolarRepublic/fix/token-permissions-owner

Fix token permissions owner
scrtlabs · Feb 22, 2025 · 0fadf51 · 0fadf51
2 parents 5ed9ecf + ca1602a
commit 0fadf51
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/contract.rs b/src/contract.rs
@@ -455,7 +455,8 @@ fn permit_queries(
     // Permit validated! We can now execute the query.
     match query {
         QueryWithPermit::Balance {} => {
-            if !permit.check_permission(&TokenPermissions::Balance) {
+            if !permit.check_permission(&TokenPermissions::Balance)
+                && !permit.check_permission(&TokenPermissions::Owner) {
                 return Err(StdError::generic_err(format!(
                     "No permission to query balance, got permissions {:?}",
                     permit.params.permissions
@@ -468,7 +469,8 @@ fn permit_queries(
             Err(StdError::generic_err(TRANSFER_HISTORY_UNSUPPORTED_MSG))
         }
         QueryWithPermit::TransactionHistory { page, page_size } => {
-            if !permit.check_permission(&TokenPermissions::History) {
+            if !permit.check_permission(&TokenPermissions::History)
+                && !permit.check_permission(&TokenPermissions::Owner) {
                 return Err(StdError::generic_err(format!(
                     "No permission to query history, got permissions {:?}",
                     permit.params.permissions
@@ -478,7 +480,8 @@ fn permit_queries(
             query::query_transactions(deps, account, page.unwrap_or(0), page_size)
         }
         QueryWithPermit::Allowance { owner, spender } => {
-            if !permit.check_permission(&TokenPermissions::Allowance) {
+            if !permit.check_permission(&TokenPermissions::Allowance)
+                && !permit.check_permission(&TokenPermissions::Owner) {
                 return Err(StdError::generic_err(format!(
                     "No permission to query allowance, got permissions {:?}",
                     permit.params.permissions