@stake-dao/reader 0.5.2 → 0.5.4

package/src/votemarket/yb/fetchYbUserVotes.ts

@@ -0,0 +1,61 @@
+import { contracts } from '@stake-dao/constants'
+import { encodeAbiParameters, formatUnits, parseAbiParameters, zeroAddress } from 'viem'
+import { mainnet } from 'viem/chains'
+import batchYbUserVotes from '../../bytecodes/votemarket/batchYbUserVotes.js'
+import { concatBytecode } from '../../index.js'
+import { batchJsonRpc, equalTlc } from '../../utils.js'
+
+export type Vote = {
+  gauge: string
+  slope: string
+  relativeWeight: number
+  weight: number
+  start: number
+  end: number
+}
+
+export type GaugesUserData = { veBalance: string; unusedVeBalance: string; unusedRelativeWeight: number; votes: Vote[] }
+
+export const fetchYbUserVotes = async (rpc: string, user: string): Promise<GaugesUserData> => {
+  const calls = [
+    concatBytecode(
+      batchYbUserVotes.bytecode[mainnet.id]!,
+      encodeAbiParameters(parseAbiParameters(batchYbUserVotes.inputType[mainnet.id]!) as any, [
+        user,
+        contracts.ybGaugeController![1]!,
+        contracts.veYB![1]!,
+      ]),
+    ),
+  ]
+
+  const rawVotes = await batchJsonRpc({
+    rpc,
+    calls,
+    outputTypeAbi: batchYbUserVotes.outputType![mainnet.id],
+    parse: false,
+    callsKey: `votemarket/yb/fetchYbUserVotes.ts: rawGauges - chainId ${mainnet.id} - rpc ${rpc}`,
+  })
+
+  const veBalance = rawVotes[0].veBalance
+
+  const votes = rawVotes[0].votes
+    .filter((rv) => !equalTlc(rv.gauge, zeroAddress))
+    .map((v) => ({
+      gauge: v.gauge,
+      slope: formatUnits(v.slope, 0),
+      relativeWeight: Number(v.power) / 100,
+      weight: formatUnits((veBalance * BigInt(v.power)) / BigInt(10000), 18),
+      start: Number(v.start),
+      end: Number(v.end),
+    }))
+
+  const unusedRelativeWeight = 10000 - votes.reduce((total, x) => total + x.relativeWeight * 100, 0)
+  const unusedVeBalance = (veBalance * BigInt(unusedRelativeWeight)) / BigInt(10000)
+
+  return {
+    veBalance: formatUnits(veBalance, 18),
+    unusedVeBalance: formatUnits(unusedVeBalance, 18),
+    unusedRelativeWeight: unusedRelativeWeight / 100,
+    votes,
+  }
+}