feat: add get referendum by id

polkadot-api · Jan 17, 2025 · c6c9230 · c6c9230
1 parent a2bd1a0
commit c6c9230
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/packages/sdk-governance/src/referenda/referenda-sdk.ts b/packages/sdk-governance/src/referenda/referenda-sdk.ts
@@ -136,6 +136,14 @@ export function createReferendaSdk(
       })
       .filter((v) => !!v)
   }
+  async function getOngoingReferendum(id: number) {
+    const referendum =
+      await typedApi.query.Referenda.ReferendumInfoFor.getValue(id)
+    if (referendum?.type === "Ongoing") {
+      return enhanceOngoingReferendum(id, referendum.value)
+    }
+    return null
+  }
 
   const [rawReferendumById$, referendaKeyChange$] = partitionEntries(
     typedApi.query.Referenda.ReferendumInfoFor.watchEntries().pipe(
@@ -288,6 +296,7 @@ export function createReferendaSdk(
       ongoingReferendaIds$,
     },
     getOngoingReferenda,
+    getOngoingReferendum,
     getSpenderTrack,
     getTrack,
     createReferenda,

diff --git a/packages/sdk-governance/src/referenda/sdk-types.ts b/packages/sdk-governance/src/referenda/sdk-types.ts
@@ -61,6 +61,7 @@ export type ReferendaTrack = Omit<
 
 export interface ReferendaSdk {
   getOngoingReferenda(): Promise<OngoingReferendum[]>
+  getOngoingReferendum(id: number): Promise<OngoingReferendum | null>
   watch: {
     ongoingReferenda$: Observable<Map<number, OngoingReferendum>>
     ongoingReferendaIds$: Observable<number[]>