@stake-dao/reader 0.4.26 → 0.4.27

package/src/votemarket/fetchCampaigns.ts

@@ -1,11 +1,15 @@
-import { tokenWithAddress, tokens } from '@stake-dao/constants'
-import { chunk, range, remove, uniq } from 'lodash-es'
+import { ONE_WEEK, tokenWithAddress, tokens } from '@stake-dao/constants'
+import { chunk, filter, range, remove, uniq } from 'lodash-es'
 import { decodeAbiParameters, encodeAbiParameters, formatUnits, parseAbiParameters } from 'viem'
 import { batchTokenData } from '../bytecodes/index.js'
 import batchCampaigns from '../bytecodes/votemarket/batchCampaigns.js'
+import { getPrices } from '../prices.js'
 import { batchJsonRpc, concatBytecode, equalTlc, rpcCall } from '../utils.js'
+import type { Campaign } from './types.js'
 
 export const CAMPAIGNS_CHUNK_SIZE = 10
+export const CLAIM_WINDOW_LENGTH = 24 // weeks
+export const CLOSE_WINDOW_LENGTH = 4 // weeks
 
 interface FetchCampaignsProps {
   platform: string
@@ -13,16 +17,7 @@ interface FetchCampaignsProps {
   rpc: string
 }
 
-export type Gauge = {
-  gauge: string
-  inflationRate: string
-  relativeWeight: string
-  futureRelativeWeight: string
-}
-
-export type GaugesData = { totalGaugesWeight: string; gauges: Gauge[] }
-
-export const fetchCampaigns = async ({ platform, chainId, rpc }: FetchCampaignsProps): Promise<any> => {
+export const fetchCampaigns = async ({ platform, chainId, rpc }: FetchCampaignsProps): Promise<Campaign[]> => {
   const nCampaignsRequest = await rpcCall(rpc, [{ to: platform, data: '0x7274e30d' }]) // campaignCount
   const nCampaigns = Number(
     decodeAbiParameters([{ type: 'uint256', name: 'campaignCount' }], nCampaignsRequest[0].result),
@@ -47,10 +42,8 @@ export const fetchCampaigns = async ({ platform, chainId, rpc }: FetchCampaignsP
     callsKey: `votemarket/fetchCampaigns.ts: rawCampaigns - chainId ${chainId} - rpc ${rpc}`,
   })
 
-  const tokensToFetch: string[] = remove(
-    uniq(rawCampaigns.map((rc) => rc.campaign.rewardToken)),
-    (t) => !tokenWithAddress(t, chainId),
-  )
+  const campaignsTokens: string[] = uniq(rawCampaigns.map((rc) => rc.campaign.rewardToken))
+  const tokensToFetch: string[] = remove(campaignsTokens, (t) => !tokenWithAddress(t, chainId))
 
   let rawTokensData: any[] = []
   if (tokensToFetch.length > 0) {
@@ -69,8 +62,8 @@ export const fetchCampaigns = async ({ platform, chainId, rpc }: FetchCampaignsP
     })
   }
 
-  const fetchedTokens = [
-    ...tokens,
+  const fetchedTokens: any = [
+    ...filter(tokens, (t) => campaignsTokens.find((ct) => equalTlc(ct, t.address))),
     ...rawTokensData.map((t) => ({
       name: t.name,
       symbol: t.symbol,
@@ -80,22 +73,32 @@ export const fetchCampaigns = async ({ platform, chainId, rpc }: FetchCampaignsP
     })),
   ]
 
+  const prices = await getPrices(fetchedTokens as any, chainId)
+
+  const currentTimestamp = Math.floor(Date.now() / 1000)
+
   const campaigns = rawCampaigns.map((rc) => {
     const rewardToken = fetchedTokens.find((t) => equalTlc(t.address, rc.campaign.rewardToken) && t.chainId === chainId)
     const decimals = rewardToken ? rewardToken.decimals : 18
 
+    const rewardTokenPrice = prices.find((p) => equalTlc(rc.campaign.rewardToken, p.address))?.usdPrice || 0
+
+    const startTimestamp = Number(rc.campaign.startTimestamp)
+    const endTimestamp = Number(rc.campaign.endTimestamp)
+
     return {
       id: Number(rc.id),
       chainId: Number(rc.campaign.chainId),
       gauge: rc.campaign.gauge,
       manager: rc.campaign.manager,
       rewardToken,
+      rewardTokenPrice,
       numberOfPeriods: Number(rc.campaign.numberOfPeriods),
       maxRewardPerVote: formatUnits(rc.campaign.maxRewardPerVote, decimals),
       totalRewardAmount: formatUnits(rc.campaign.totalRewardAmount, decimals),
       totalDistributed: formatUnits(rc.campaign.totalDistributed, decimals),
-      startTimestamp: Number(rc.campaign.startTimestamp),
-      endTimestamp: Number(rc.campaign.endTimestamp),
+      startTimestamp,
+      endTimestamp,
       hook: rc.campaign.hook,
       isKilled: rc.isKilled,
       addresses: rc.addresses,
@@ -107,7 +110,15 @@ export const fetchCampaigns = async ({ platform, chainId, rpc }: FetchCampaignsP
         leftover: formatUnits(rc.currentPeriod.leftover, decimals),
         updated: rc.currentPeriod.updated,
       },
+      totalPeriods: (endTimestamp - startTimestamp) / ONE_WEEK,
       periodLeft: Number(rc.periodLeft),
+      status: {
+        voteOpen: endTimestamp - ONE_WEEK > currentTimestamp,
+        voteClosed: currentTimestamp > endTimestamp - ONE_WEEK,
+        claimOpen: currentTimestamp > startTimestamp,
+        claimClosed: currentTimestamp > endTimestamp + CLAIM_WINDOW_LENGTH * ONE_WEEK,
+        expired: currentTimestamp > endTimestamp + (CLAIM_WINDOW_LENGTH + CLOSE_WINDOW_LENGTH) * ONE_WEEK,
+      },
     }
   })
 

package/src/votemarket/types.ts

@@ -0,0 +1,39 @@
+export type Campaign = {
+  id: number
+  chainId: number
+  gauge: string
+  manager: string
+  rewardToken: {
+    name: string
+    symbol: string
+    address: string
+    decimals: number
+    chainId: number
+  }
+  numberOfPeriods: number
+  maxRewardPerVote: string
+  totalRewardAmount: string
+  totalDistributed: string
+  startTimestamp: number
+  endTimestamp: number
+  hook: string
+  isKilled: boolean
+  addresses: string[]
+  isWhitelist: boolean
+  isBlacklist: boolean
+  currentPeriod: {
+    rewardPerPeriod: string
+    rewardPerVote: string
+    leftover: string
+    updated: boolean
+  }
+  totalPeriods: number
+  periodLeft: number
+  status: {
+    voteOpen: boolean
+    voteClosed: boolean
+    claimOpen: boolean
+    claimClosed: boolean
+    expired: boolean
+  }
+}