Skip to content

Commit

Permalink
added formula to get all voters like other voting modules
Browse files Browse the repository at this point in the history
  • Loading branch information
NoahSaso committed Jul 9, 2024
1 parent 11184d3 commit 84467ff
Show file tree
Hide file tree
Showing 2 changed files with 60 additions and 38 deletions.
47 changes: 9 additions & 38 deletions src/formulas/formulas/contract/daoCore/members.ts
Original file line number Diff line number Diff line change
Expand Up @@ -9,11 +9,7 @@ import { groupContract } from '../voting/daoVotingCw4'
import { topStakers as topCw721Stakers } from '../voting/daoVotingCw721Staked'
import { topStakers as topNativeStakers } from '../voting/daoVotingNativeStaked'
import { topStakers as topOnftStakers } from '../voting/daoVotingOnftStaked'
import {
listVoters as listSgCommunityNftVoters,
totalPowerAtHeight as sgCommunityNftTotalPower,
votingPowerAtHeight as sgCommunityNftVotingPower,
} from '../voting/daoVotingSgCommunityNft'
import { allVotersWithVotingPower as sgCommunityAllVoters } from '../voting/daoVotingSgCommunityNft'
import { topStakers as topTokenStakers } from '../voting/daoVotingTokenStaked'
import { config, votingModule } from './base'
import { getUniqueSubDaosInTree } from './utils'
Expand Down Expand Up @@ -217,40 +213,15 @@ export const listMembers: ContractFormula<DaoMember[]> = {
'dao-voting-sg-community-nft'
)
) {
const [{ voters }, { power: totalPower }] = await Promise.all([
listSgCommunityNftVoters.compute({
...env,
contractAddress: votingModuleAddress,
}),
sgCommunityNftTotalPower.compute({
...env,
contractAddress: votingModuleAddress,
}),
])

const totalPowerNum = Number(totalPower)
const voters = await sgCommunityAllVoters.compute({
...env,
contractAddress: votingModuleAddress,
})

return await Promise.all(
voters.map(async (address) => ({
address,
votingPowerPercent:
totalPowerNum === 0
? 0
: (Number(
(
await sgCommunityNftVotingPower.compute({
...env,
contractAddress: votingModuleAddress,
args: {
address,
},
})
).power
) /
totalPowerNum) *
100,
}))
)
return voters.map(({ address, votingPowerPercent }) => ({
address,
votingPowerPercent,
}))
}

throw new Error('voting module passthrough not supported for this DAO')
Expand Down
51 changes: 51 additions & 0 deletions src/formulas/formulas/contract/voting/daoVotingSgCommunityNft.ts
Original file line number Diff line number Diff line change
Expand Up @@ -131,3 +131,54 @@ export const listVoters: ContractFormula<
}
},
}

type Voter = {
address: string
weight: number
votingPowerPercent: number
}

export const allVotersWithVotingPower: ContractFormula<
Voter[],
{
limit?: string
}
> = {
filter: {
codeIdsKeys: CODE_IDS_KEYS,
},
compute: async (env) => {
const {
contractAddress,
getTransformationMap,
args: { limit },
} = env
const limitNum = limit ? Math.max(0, Number(limit)) : Infinity

const voterMap =
(await getTransformationMap<string, string>(contractAddress, 'vp')) ?? {}
const voterEntries = Object.entries(voterMap)
// Ascending by voter address.
.sort(([a], [b]) => a.localeCompare(b))
.slice(0, limitNum)

// Get total power.
const totalVotingPower = Number(
(await totalPowerAtHeight.compute(env)).power
)

// Compute voting power percent for each voter.
const voters = voterEntries.map(
([address, weight]): Voter => ({
address,
weight: Number(weight),
votingPowerPercent:
totalVotingPower === 0
? 0
: (Number(weight) / totalVotingPower) * 100,
})
)

return voters
},
}

0 comments on commit 84467ff

Please sign in to comment.