dotbot/Commands/Polls.cs

64 lines
1.9 KiB
C#

using Discord;
using Discord.Commands;
using dotbot.Services;
using Microsoft.Extensions.Configuration;
using System.Collections.Generic;
using System.Threading.Tasks;
namespace dotbot.Commands
{
[Group("poll")]
public class Polls : ModuleBase<SocketCommandContext>
{
private Dictionary<ulong, Poll> _polls;
private IConfigurationRoot _config;
public Polls(PollService polls, IConfigurationRoot config)
{
_polls = polls.currentPolls;
_config = config;
}
[Command]
[Priority(0)]
[Summary("create a new poll")]
public async Task CreatePoll()
{
var pollId = Context.Channel.Id;
if (_polls.ContainsKey(pollId))
{
await ReplyAsync($"add some more options or start the poll with `{_config["prefix"]}poll start`");
}
else
{
_polls[pollId] = new Poll
{
Owner = Context.User,
IsOpen = false,
};
await ReplyAsync($"you started a new poll. respond with some options and then start the poll with `{_config["prefix"]}poll start`");
}
}
[Command("start")]
[Priority(1)]
[Summary("starts a poll!")]
public async Task StartPoll()
{
var pollId = Context.Channel.Id;
if (_polls.ContainsKey(pollId) && Context.User == _polls[pollId].Owner)
{
foreach (var o in _polls[pollId].Options)
{
var opt = await ReplyAsync($"{o.Text}");
await opt.AddReactionAsync(Emote.Parse(":thumbsup:"));
}
_polls[pollId].IsOpen = true;
}
else await ReplyAsync($"no poll ready to start");
}
}
}