-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #42 from iluxonchik/feature/govbot-0.0.12
govbot 0.0.12 - Vote Counting In #admin
- Loading branch information
Showing
8 changed files
with
1,117 additions
and
814 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,93 @@ | ||
| import { Action, TrackedInteraction, Screen } from '../../../core/BaseClasses'; | ||
| import { ActionRowBuilder, ButtonBuilder, ButtonStyle, EmbedBuilder, StringSelectMenuBuilder } from 'discord.js'; | ||
| import { ArgumentOracle, CustomIDOracle } from '../../../CustomIDOracle'; | ||
| import { AllFundingRoundsPaginator, FundingRoundPaginator } from '../../../components/FundingRoundPaginator'; | ||
| import { VoteCountingLogic } from '../../../logic/VoteCountingLogic'; | ||
| import { EndUserError } from '../../../Errors'; | ||
| import { AnyModalMessageComponent } from '../../../types/common'; | ||
| import { DiscordStatus } from '../../DiscordStatus'; | ||
|
|
||
| export class CountVotesAction extends Action { | ||
| public allSubActions(): Action[] { | ||
| return []; | ||
| } | ||
| getComponent(...args: any[]): AnyModalMessageComponent { | ||
| throw new Error('Method not implemented.'); | ||
| } | ||
| public static readonly ID = 'countVotes'; | ||
|
|
||
| private fundingRoundPaginator: FundingRoundPaginator; | ||
|
|
||
| constructor(screen: Screen, id: string) { | ||
| super(screen, id); | ||
| this.fundingRoundPaginator = new AllFundingRoundsPaginator(screen, this, 'selectPhase', 'fundingRoundPaginator'); | ||
| } | ||
|
|
||
| protected async handleOperation(interaction: TrackedInteraction, operationId: string): Promise<void> { | ||
| switch (operationId) { | ||
| case 'selectFundingRound': | ||
| await this.handleSelectFundingRound(interaction); | ||
| break; | ||
| case 'selectPhase': | ||
| await this.handleSelectPhase(interaction); | ||
| break; | ||
| case 'countVotes': | ||
| await this.handleCountVotes(interaction); | ||
| break; | ||
| default: | ||
| throw new EndUserError('Invalid operation'); | ||
| } | ||
| } | ||
|
|
||
| private async handleSelectFundingRound(interaction: TrackedInteraction): Promise<void> { | ||
| await this.fundingRoundPaginator.handlePagination(interaction); | ||
| } | ||
|
|
||
| private async handleSelectPhase(interaction: TrackedInteraction): Promise<void> { | ||
| const fundingRoundId = ArgumentOracle.getNamedArgument(interaction, ArgumentOracle.COMMON_ARGS.FUNDING_ROUND_ID, 0); | ||
|
|
||
| const embed = new EmbedBuilder() | ||
| .setColor('#0099ff') | ||
| .setTitle('Select Funding Round Phase') | ||
| .setDescription('Choose the phase for which you want to count votes:'); | ||
|
|
||
| const row = new ActionRowBuilder<StringSelectMenuBuilder>().addComponents( | ||
| new StringSelectMenuBuilder() | ||
| .setCustomId(CustomIDOracle.addArgumentsToAction(this, 'countVotes', ArgumentOracle.COMMON_ARGS.FUNDING_ROUND_ID, fundingRoundId.toString())) | ||
| .setPlaceholder('Select a phase') | ||
| .addOptions([ | ||
| { label: 'Consideration Phase', value: 'consideration' }, | ||
| { label: 'Deliberation Phase', value: 'deliberation' }, | ||
| { label: 'Voting Phase', value: 'voting' }, | ||
| ]), | ||
| ); | ||
|
|
||
| await interaction.update({ embeds: [embed], components: [row] }); | ||
| } | ||
|
|
||
| private async handleCountVotes(interaction: TrackedInteraction): Promise<void> { | ||
| const fundingRoundId = ArgumentOracle.getNamedArgument(interaction, ArgumentOracle.COMMON_ARGS.FUNDING_ROUND_ID); | ||
| const phase = ArgumentOracle.getNamedArgument(interaction, ArgumentOracle.COMMON_ARGS.PHASE, 0); | ||
|
|
||
| const voteResults = await VoteCountingLogic.countVotes(parseInt(fundingRoundId), phase); | ||
|
|
||
| const embed = new EmbedBuilder() | ||
| .setColor('#0099ff') | ||
| .setTitle(`Vote Count Results - ${phase.charAt(0).toUpperCase() + phase.slice(1)} Phase`) | ||
| .setDescription('Here are the vote counts for each project:'); | ||
|
|
||
| voteResults.forEach((result, index) => { | ||
| let voteInfo = `Yes Votes: ${result.yesVotes}\nNo Votes: ${result.noVotes}`; | ||
| if (phase === 'deliberation' && result.approvedModifiedVotes !== undefined) { | ||
| voteInfo += `\nApproved Modified Votes: ${result.approvedModifiedVotes}`; | ||
| } | ||
|
|
||
| embed.addFields({ | ||
| name: `${index + 1}. ${result.projectName} (ID: ${result.projectId})`, | ||
| value: `Proposer: ${result.proposerDuid}\n${voteInfo}`, | ||
| }); | ||
| }); | ||
|
|
||
| await interaction.update({ embeds: [embed], components: [] }); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,40 @@ | ||
| import { Screen, Dashboard, Permission, Action, TrackedInteraction } from '../../../core/BaseClasses'; | ||
| import { ActionRowBuilder, ButtonBuilder, ButtonStyle, EmbedBuilder } from 'discord.js'; | ||
| import { CountVotesAction } from '../actions/CountVotesAction'; | ||
| import { CustomIDOracle } from '../../../CustomIDOracle'; | ||
|
|
||
| export class CountVotesScreen extends Screen { | ||
| public static readonly ID = 'countVotes'; | ||
| protected permissions: Permission[] = []; | ||
| private countVotesAction: CountVotesAction; | ||
|
|
||
| constructor(dashboard: Dashboard, id: string) { | ||
| super(dashboard, id); | ||
| this.countVotesAction = new CountVotesAction(this, CountVotesAction.ID); | ||
| } | ||
|
|
||
| async getResponse(interaction: TrackedInteraction): Promise<any> { | ||
| const embed = new EmbedBuilder().setColor('#0099ff').setTitle('Count Votes').setDescription('Select a funding round to count votes for:'); | ||
|
|
||
| const row = new ActionRowBuilder<ButtonBuilder>().addComponents( | ||
| new ButtonBuilder() | ||
| .setCustomId(CustomIDOracle.addArgumentsToAction(this.countVotesAction, 'selectFundingRound')) | ||
| .setLabel('Select Funding Round') | ||
| .setStyle(ButtonStyle.Primary), | ||
| ); | ||
|
|
||
| return { | ||
| embeds: [embed], | ||
| components: [row], | ||
| ephemeral: true, | ||
| }; | ||
| } | ||
|
|
||
| protected allSubScreens(): Screen[] { | ||
| return []; | ||
| } | ||
|
|
||
| protected allActions(): Action[] { | ||
| return [this.countVotesAction]; | ||
| } | ||
| } |
Oops, something went wrong.