Update referenda tooling (#51)

Moonsong-Labs · Jul 2, 2024 · 21eb46e · 21eb46e
1 parent 08f69e9
commit 21eb46e
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 19 deletions.
diff --git a/src/utils/monitoring.ts b/src/utils/monitoring.ts
@@ -183,7 +183,7 @@ export const getAccountIdentity = async (
       : [null, null];
     identityCache[account] = {
       lastUpdate: Date.now(),
-      identity: identity && ("info" in identity ? identity : identity[0]),
+      identity: identity[0],
       superOf:
         superOfIdentity && ("info" in superOfIdentity ? superOfIdentity : superOfIdentity[0]),
     };

diff --git a/src/utils/referenda.ts b/src/utils/referenda.ts
@@ -69,7 +69,7 @@ export function isConvictionTally(
   );
 }
 
-function curveDelay(curve: PalletReferendaCurve, input: BN, div: BN): BN {
+export function curveDelay(curve: PalletReferendaCurve, input: BN, div: BN): BN {
   // if divisor is zero, we return the max
   if (div.isZero()) {
     return BN_BILLION;
@@ -181,12 +181,8 @@ function parseImage(
   }
 
   const [proposer, balance] = status.isUnrequested
-    ? "deposit" in status.asUnrequested
-      ? status.asUnrequested["deposit"]
-      : status.asUnrequested.ticket
-    : ("deposit" in status.asRequested
-        ? status.asRequested["deposit"]
-        : status.asRequested.maybeTicket
+    ? status.asUnrequested.ticket
+    : (status.asRequested.maybeTicket
       ).unwrapOrDefault();
   let proposal: Call | undefined;
 
@@ -202,10 +198,7 @@ function parseImage(
 }
 
 async function getImageProposal(api: ApiPromise | ApiDecoration<"promise">, hash: string) {
-  const optStatus =
-    "requestStatusFor" in api.query.preimage
-      ? await api.query.preimage["requestStatusFor"](hash)
-      : await api.query.preimage.statusFor(hash);
+  const optStatus = await api.query.preimage.requestStatusFor(hash);
   const status = optStatus.unwrapOr(null) as PalletPreimageRequestStatus;
   if (!status) {
     return null;

diff --git a/src/utils/transactions.ts b/src/utils/transactions.ts
@@ -158,23 +158,19 @@ export async function callInterpreter(
         subCalls: subCallsData,
       };
     }
-    const imageStatusFn =
-      "requestStatusFor" in api.query.preimage
-        ? api.query.preimage["requestStatusFor"]
-        : api.query.preimage.statusFor;
     const callData = nested.inlined
       ? call.args[nested.argumentPosition]
-      : await imageStatusFn(call.args[nested.argumentPosition].toHex()).then((optStatus) => {
+      : await api.query.preimage.requestStatusFor(call.args[nested.argumentPosition].toHex()).then((optStatus) => {
           if (optStatus.isNone) {
             return null;
           }
           const status = optStatus.unwrap();
           const len = status.isRequested
-            ? status.asRequested.len.unwrapOr(0)
+            ? status.asRequested.maybeLen.unwrapOr(0)
             : status.asUnrequested.len || 0;
           return api.query.preimage
             .preimageFor([call.args[nested.argumentPosition].toHex(), len])
-            .then((preimage: Option<PalletPreimageRequestStatus>) => preimage.unwrap().toHex());
+            .then((preimage) => preimage.unwrap().toHex());
         });
     if (callData) {
       const subCall = await api.registry.createType("Call", callData);