feat(namada): allow querying epoch at height

hackbg · Jul 22, 2024 · bd9e4e2 · bd9e4e2
1 parent 86b00e1
commit bd9e4e2
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git a/packages/namada/NamadaChain.ts b/packages/namada/NamadaChain.ts
@@ -142,8 +142,8 @@ export default class NamadaChain extends CW.Chain {
   fetchProposalInfo (id: number|bigint) {
     return this.getConnection().fetchProposalInfoImpl(id)
   }
-  fetchEpoch () {
-    return this.getConnection().fetchEpochImpl()
+  fetchEpoch (...args: Parameters<NamadaConnection["fetchEpochImpl"]>) {
+    return this.getConnection().fetchEpochImpl(...args)
   }
   fetchEpochFirstBlock () {
     return this.getConnection().fetchEpochFirstBlockImpl()

diff --git a/packages/namada/NamadaConnection.ts b/packages/namada/NamadaConnection.ts
@@ -39,8 +39,8 @@ export default class NamadaConnection extends CW.Connection {
     return fetchProtocolParameters(this)
   }
 
-  fetchEpochImpl () {
-    return Epoch.fetchEpoch(this)
+  fetchEpochImpl (options?: { height?: number|bigint }) {
+    return Epoch.fetchEpoch(this, options?.height)
   }
   fetchEpochFirstBlockImpl () {
     return Epoch.fetchEpochFirstBlock(this)

diff --git a/packages/namada/NamadaEpoch.ts b/packages/namada/NamadaEpoch.ts
@@ -3,9 +3,14 @@ import type { Decode } from './NamadaDecode'
 import type NamadaConnection from './NamadaConnection'
 
 export async function fetchEpoch (
-  connection: Pick<NamadaConnection, 'abciQuery'>
+  connection: Pick<NamadaConnection, 'abciQuery'>,
+  height?: number|bigint
 ) {
-  return decode(u64, await connection.abciQuery("/shell/epoch"))
+  if (height !== undefined) {
+    return decode(u64, await connection.abciQuery(`/shell/epoch_at_height/${height}`))
+  } else {
+    return decode(u64, await connection.abciQuery("/shell/epoch"))
+  }
 }
 
 export async function fetchEpochFirstBlock (