feat: getExtractions method

ilbertt · May 29, 2024 · 85d96f2 · 85d96f2
1 parent f41faee
commit 85d96f2
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -92,6 +92,14 @@ To fetch the extraction for a principal, use the following command:
 dfx canister call fortune-wheel-booth-backend getExtraction '(principal "<the-principal-you-want-to-fetch-the-extraction-for>")'
 ```
 
+#### `getExtractions`
+
+To fetch all extractions, use the following command:
+
+```bash
+dfx canister call fortune-wheel-booth-backend getExtractions
+```
+
 #### `manualTransfer`
 
 To manually send tokens to a principal, use the following command:

diff --git a/src/fortune-wheel-booth-backend/main.mo b/src/fortune-wheel-booth-backend/main.mo
@@ -297,6 +297,14 @@ shared ({ caller = initialController }) actor class Main() {
     sorted_entries.next();
   };
 
+  public shared query ({ caller }) func getExtractions() : async [(Principal, Extraction)] {
+    if (not isAdmin(caller)) {
+      throw Error.reject("Only admins can read extractions");
+    };
+
+    Iter.toArray(extractedPrincipals.entries());
+  };
+
   type ManualSendTokens = {
     #icp : Nat;
     #ckBtc : Nat;