382 lines
15 KiB
C#
382 lines
15 KiB
C#
using Discord;
|
|
using Discord.Commands;
|
|
using NadekoBot.Extensions;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using NadekoBot.Core.Services;
|
|
using NadekoBot.Core.Services.Database.Models;
|
|
using System.Collections.Generic;
|
|
using NadekoBot.Common;
|
|
using NadekoBot.Common.Attributes;
|
|
using System;
|
|
|
|
namespace NadekoBot.Modules.Gambling
|
|
{
|
|
public partial class Gambling : NadekoTopLevelModule
|
|
{
|
|
private readonly IBotConfigProvider _bc;
|
|
private readonly DbService _db;
|
|
private readonly CurrencyService _cs;
|
|
private readonly IDataCache _cache;
|
|
|
|
private string CurrencyName => _bc.BotConfig.CurrencyName;
|
|
private string CurrencyPluralName => _bc.BotConfig.CurrencyPluralName;
|
|
private string CurrencySign => _bc.BotConfig.CurrencySign;
|
|
|
|
public Gambling(IBotConfigProvider bc, DbService db, CurrencyService currency,
|
|
IDataCache cache)
|
|
{
|
|
_bc = bc;
|
|
_db = db;
|
|
_cs = currency;
|
|
_cache = cache;
|
|
}
|
|
|
|
public long GetCurrency(ulong id)
|
|
{
|
|
using (var uow = _db.UnitOfWork)
|
|
{
|
|
return uow.Currency.GetUserCurrency(id);
|
|
}
|
|
}
|
|
|
|
[NadekoCommand, Usage, Description, Aliases]
|
|
public async Task Timely()
|
|
{
|
|
var val = _bc.BotConfig.TimelyCurrency;
|
|
var period = _bc.BotConfig.TimelyCurrencyPeriod;
|
|
if (val <= 0 || period <= 0)
|
|
{
|
|
await ReplyErrorLocalized("timely_none").ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
TimeSpan? rem;
|
|
if ((rem = _cache.AddTimelyClaim(Context.User.Id, period)) != null)
|
|
{
|
|
await ReplyErrorLocalized("timely_already_claimed", rem?.ToString(@"dd\d\ hh\h\ mm\m\ ss\s")).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
await _cs.AddAsync(Context.User.Id, "Timely claim", val).ConfigureAwait(false);
|
|
|
|
await ReplyConfirmLocalized("timely", val + _bc.BotConfig.CurrencySign, period).ConfigureAwait(false);
|
|
}
|
|
|
|
[NadekoCommand, Usage, Description, Aliases]
|
|
[OwnerOnly]
|
|
public async Task TimelyReset()
|
|
{
|
|
_cache.RemoveAllTimelyClaims();
|
|
await ReplyConfirmLocalized("timely_reset").ConfigureAwait(false);
|
|
}
|
|
|
|
[NadekoCommand, Usage, Description, Aliases]
|
|
[OwnerOnly]
|
|
public async Task TimelySet(int num, int period = 24)
|
|
{
|
|
if (num < 0 || period < 0)
|
|
return;
|
|
using (var uow = _db.UnitOfWork)
|
|
{
|
|
var bc = uow.BotConfig.GetOrCreate(set => set);
|
|
bc.TimelyCurrency = num;
|
|
bc.TimelyCurrencyPeriod = period;
|
|
uow.Complete();
|
|
}
|
|
_bc.Reload();
|
|
if(num == 0)
|
|
await ReplyConfirmLocalized("timely_set_none").ConfigureAwait(false);
|
|
else
|
|
await ReplyConfirmLocalized("timely_set", Format.Bold(num + _bc.BotConfig.CurrencySign), Format.Bold(period.ToString())).ConfigureAwait(false);
|
|
}
|
|
|
|
[NadekoCommand, Usage, Description, Aliases]
|
|
[RequireContext(ContextType.Guild)]
|
|
public async Task Raffle([Remainder] IRole role = null)
|
|
{
|
|
role = role ?? Context.Guild.EveryoneRole;
|
|
|
|
var members = (await role.GetMembersAsync()).Where(u => u.Status != UserStatus.Offline);
|
|
var membersArray = members as IUser[] ?? members.ToArray();
|
|
if (membersArray.Length == 0)
|
|
{
|
|
return;
|
|
}
|
|
var usr = membersArray[new NadekoRandom().Next(0, membersArray.Length)];
|
|
await Context.Channel.SendConfirmAsync("🎟 "+ GetText("raffled_user"), $"**{usr.Username}#{usr.Discriminator}**", footer: $"ID: {usr.Id}").ConfigureAwait(false);
|
|
}
|
|
|
|
[NadekoCommand, Usage, Description, Aliases]
|
|
[RequireContext(ContextType.Guild)]
|
|
public async Task RaffleAny([Remainder] IRole role = null)
|
|
{
|
|
role = role ?? Context.Guild.EveryoneRole;
|
|
|
|
var members = (await role.GetMembersAsync());
|
|
var membersArray = members as IUser[] ?? members.ToArray();
|
|
if (membersArray.Length == 0)
|
|
{
|
|
return;
|
|
}
|
|
var usr = membersArray[new NadekoRandom().Next(0, membersArray.Length)];
|
|
await Context.Channel.SendConfirmAsync("🎟 " + GetText("raffled_user"), $"**{usr.Username}#{usr.Discriminator}**", footer: $"ID: {usr.Id}").ConfigureAwait(false);
|
|
}
|
|
|
|
[NadekoCommand, Usage, Description, Aliases]
|
|
[Priority(1)]
|
|
public async Task Cash([Remainder] IUser user = null)
|
|
{
|
|
if(user == null)
|
|
await ConfirmLocalized("has", Format.Bold(Context.User.ToString()), $"{GetCurrency(Context.User.Id)} {CurrencySign}").ConfigureAwait(false);
|
|
else
|
|
await ReplyConfirmLocalized("has", Format.Bold(user.ToString()), $"{GetCurrency(user.Id)} {CurrencySign}").ConfigureAwait(false);
|
|
}
|
|
|
|
[NadekoCommand, Usage, Description, Aliases]
|
|
[Priority(0)]
|
|
public async Task Cash(ulong userId)
|
|
{
|
|
await ReplyConfirmLocalized("has", Format.Code(userId.ToString()), $"{GetCurrency(userId)} {CurrencySign}").ConfigureAwait(false);
|
|
}
|
|
|
|
[NadekoCommand, Usage, Description, Aliases]
|
|
[RequireContext(ContextType.Guild)]
|
|
public async Task Give(long amount, [Remainder] IGuildUser receiver)
|
|
{
|
|
if (amount <= 0 || Context.User.Id == receiver.Id)
|
|
return;
|
|
var success = await _cs.RemoveAsync((IGuildUser)Context.User, $"Gift to {receiver.Username} ({receiver.Id}).", amount, false).ConfigureAwait(false);
|
|
if (!success)
|
|
{
|
|
await ReplyErrorLocalized("not_enough", CurrencyPluralName).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
await _cs.AddAsync(receiver, $"Gift from {Context.User.Username} ({Context.User.Id}).", amount, true).ConfigureAwait(false);
|
|
await ReplyConfirmLocalized("gifted", amount + CurrencySign, Format.Bold(receiver.ToString()))
|
|
.ConfigureAwait(false);
|
|
}
|
|
|
|
[NadekoCommand, Usage, Description, Aliases]
|
|
[RequireContext(ContextType.Guild)]
|
|
[OwnerOnly]
|
|
[Priority(0)]
|
|
public Task Award(int amount, [Remainder] IGuildUser usr) =>
|
|
Award(amount, usr.Id);
|
|
|
|
[NadekoCommand, Usage, Description, Aliases]
|
|
[OwnerOnly]
|
|
[Priority(1)]
|
|
public async Task Award(int amount, ulong usrId)
|
|
{
|
|
if (amount <= 0)
|
|
return;
|
|
|
|
await _cs.AddAsync(usrId, $"Awarded by bot owner. ({Context.User.Username}/{Context.User.Id})", amount).ConfigureAwait(false);
|
|
await ReplyConfirmLocalized("awarded", amount + CurrencySign, $"<@{usrId}>").ConfigureAwait(false);
|
|
}
|
|
|
|
[NadekoCommand, Usage, Description, Aliases]
|
|
[RequireContext(ContextType.Guild)]
|
|
[OwnerOnly]
|
|
[Priority(2)]
|
|
public async Task Award(int amount, [Remainder] IRole role)
|
|
{
|
|
var users = (await Context.Guild.GetUsersAsync())
|
|
.Where(u => u.GetRoles().Contains(role))
|
|
.ToList();
|
|
await Task.WhenAll(users.Select(u => _cs.AddAsync(u.Id,
|
|
$"Awarded by bot owner to **{role.Name}** role. ({Context.User.Username}/{Context.User.Id})",
|
|
amount)))
|
|
.ConfigureAwait(false);
|
|
|
|
await ReplyConfirmLocalized("mass_award",
|
|
amount + CurrencySign,
|
|
Format.Bold(users.Count.ToString()),
|
|
Format.Bold(role.Name)).ConfigureAwait(false);
|
|
}
|
|
|
|
[NadekoCommand, Usage, Description, Aliases]
|
|
[RequireContext(ContextType.Guild)]
|
|
[OwnerOnly]
|
|
public async Task Take(long amount, [Remainder] IGuildUser user)
|
|
{
|
|
if (amount <= 0)
|
|
return;
|
|
|
|
if (await _cs.RemoveAsync(user, $"Taken by bot owner.({Context.User.Username}/{Context.User.Id})", amount, true).ConfigureAwait(false))
|
|
await ReplyConfirmLocalized("take", amount+CurrencySign, Format.Bold(user.ToString())).ConfigureAwait(false);
|
|
else
|
|
await ReplyErrorLocalized("take_fail", amount + CurrencySign, Format.Bold(user.ToString()), CurrencyPluralName).ConfigureAwait(false);
|
|
}
|
|
|
|
|
|
[NadekoCommand, Usage, Description, Aliases]
|
|
[OwnerOnly]
|
|
public async Task Take(long amount, [Remainder] ulong usrId)
|
|
{
|
|
if (amount <= 0)
|
|
return;
|
|
|
|
if (await _cs.RemoveAsync(usrId, $"Taken by bot owner.({Context.User.Username}/{Context.User.Id})", amount).ConfigureAwait(false))
|
|
await ReplyConfirmLocalized("take", amount + CurrencySign, $"<@{usrId}>").ConfigureAwait(false);
|
|
else
|
|
await ReplyErrorLocalized("take_fail", amount + CurrencySign, Format.Code(usrId.ToString()), CurrencyPluralName).ConfigureAwait(false);
|
|
}
|
|
|
|
//[NadekoCommand, Usage, Description, Aliases]
|
|
//[OwnerOnly]
|
|
//public Task BrTest(int tests = 1000)
|
|
//{
|
|
// var t = Task.Run(async () =>
|
|
// {
|
|
// if (tests <= 0)
|
|
// return;
|
|
// //multi vs how many times it occured
|
|
// var dict = new Dictionary<int, int>();
|
|
// var generator = new NadekoRandom();
|
|
// for (int i = 0; i < tests; i++)
|
|
// {
|
|
// var rng = generator.Next(0, 101);
|
|
// var mult = 0;
|
|
// if (rng < 67)
|
|
// {
|
|
// mult = 0;
|
|
// }
|
|
// else if (rng < 91)
|
|
// {
|
|
// mult = 2;
|
|
// }
|
|
// else if (rng < 100)
|
|
// {
|
|
// mult = 4;
|
|
// }
|
|
// else
|
|
// mult = 10;
|
|
|
|
// if (dict.ContainsKey(mult))
|
|
// dict[mult] += 1;
|
|
// else
|
|
// dict.Add(mult, 1);
|
|
// }
|
|
|
|
// var sb = new StringBuilder();
|
|
// const int bet = 1;
|
|
// int payout = 0;
|
|
// foreach (var key in dict.Keys.OrderByDescending(x => x))
|
|
// {
|
|
// sb.AppendLine($"x{key} occured {dict[key]} times. {dict[key] * 1.0f / tests * 100}%");
|
|
// payout += key * dict[key];
|
|
// }
|
|
// try
|
|
// {
|
|
// await Context.Channel.SendConfirmAsync("BetRoll Test Results", sb.ToString(),
|
|
// footer: $"Total Bet: {tests * bet} | Payout: {payout * bet} | {payout * 1.0f / tests * 100}%");
|
|
// }
|
|
// catch { }
|
|
|
|
// });
|
|
// return Task.CompletedTask;
|
|
//}
|
|
public enum Allin { Allin = int.MinValue, All = int.MinValue }
|
|
|
|
private async Task InternallBetroll(long amount)
|
|
{
|
|
if (amount < 1)
|
|
return;
|
|
|
|
if (!await _cs.RemoveAsync(Context.User, "Betroll Gamble", amount, false).ConfigureAwait(false))
|
|
{
|
|
await ReplyErrorLocalized("not_enough", CurrencyPluralName).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
var rnd = new NadekoRandom().Next(0, 101);
|
|
var str = Context.User.Mention + Format.Code(GetText("roll", rnd));
|
|
if (rnd < 67)
|
|
{
|
|
str += GetText("better_luck");
|
|
}
|
|
else
|
|
{
|
|
if (rnd < 91)
|
|
{
|
|
str += GetText("br_win", (amount * _bc.BotConfig.Betroll67Multiplier) + CurrencySign, 66);
|
|
await _cs.AddAsync(Context.User, "Betroll Gamble",
|
|
(int)(amount * _bc.BotConfig.Betroll67Multiplier), false).ConfigureAwait(false);
|
|
}
|
|
else if (rnd < 100)
|
|
{
|
|
str += GetText("br_win", (amount * _bc.BotConfig.Betroll91Multiplier) + CurrencySign, 90);
|
|
await _cs.AddAsync(Context.User, "Betroll Gamble",
|
|
(int)(amount * _bc.BotConfig.Betroll91Multiplier), false).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
str += GetText("br_win", (amount * _bc.BotConfig.Betroll100Multiplier) + CurrencySign, 99) + " 👑";
|
|
await _cs.AddAsync(Context.User, "Betroll Gamble",
|
|
(int)(amount * _bc.BotConfig.Betroll100Multiplier), false).ConfigureAwait(false);
|
|
}
|
|
}
|
|
await Context.Channel.SendConfirmAsync(str).ConfigureAwait(false);
|
|
}
|
|
|
|
[NadekoCommand, Usage, Description, Aliases]
|
|
public Task BetRoll(long amount)
|
|
=> InternallBetroll(amount);
|
|
|
|
[NadekoCommand, Usage, Description, Aliases]
|
|
public Task BetRoll(Allin _)
|
|
{
|
|
long cur;
|
|
using (var uow = _db.UnitOfWork)
|
|
{
|
|
cur = uow.Currency.GetUserCurrency(Context.User.Id);
|
|
}
|
|
return InternallBetroll(cur);
|
|
}
|
|
|
|
[NadekoCommand, Usage, Description, Aliases]
|
|
public async Task Leaderboard(int page = 1)
|
|
{
|
|
if (page < 1)
|
|
return;
|
|
|
|
List<Currency> richest;
|
|
using (var uow = _db.UnitOfWork)
|
|
{
|
|
richest = uow.Currency.GetTopRichest(9, 9 * (page - 1)).ToList();
|
|
}
|
|
|
|
var embed = new EmbedBuilder()
|
|
.WithOkColor()
|
|
.WithTitle(CurrencySign +
|
|
" " + GetText("leaderboard"))
|
|
.WithFooter(efb => efb.WithText(GetText("page", page)));
|
|
|
|
if (!richest.Any())
|
|
{
|
|
embed.WithDescription(GetText("no_users_found"));
|
|
await Context.Channel.EmbedAsync(embed).ConfigureAwait(false);
|
|
return;
|
|
}
|
|
|
|
for (var i = 0; i < richest.Count; i++)
|
|
{
|
|
var x = richest[i];
|
|
var usr = await Context.Guild.GetUserAsync(x.UserId).ConfigureAwait(false);
|
|
var usrStr = usr == null
|
|
? x.UserId.ToString()
|
|
: usr.Username?.TrimTo(20, true);
|
|
|
|
var j = i;
|
|
embed.AddField(efb => efb.WithName("#" + (9 * (page - 1) + j + 1) + " " + usrStr)
|
|
.WithValue(x.Amount.ToString() + " " + CurrencySign)
|
|
.WithIsInline(true));
|
|
}
|
|
|
|
await Context.Channel.EmbedAsync(embed).ConfigureAwait(false);
|
|
}
|
|
}
|
|
}
|