NadekoBot/NadekoBot.Core/Modules/Xp/Club.cs

349 lines
13 KiB
C#
Raw Normal View History

using Discord;
using Discord.Commands;
using Discord.WebSocket;
using NadekoBot.Common.Attributes;
using NadekoBot.Extensions;
using NadekoBot.Modules.Xp.Common;
using NadekoBot.Modules.Xp.Services;
using NadekoBot.Core.Services.Database.Models;
using System;
using System.Linq;
using System.Threading.Tasks;
namespace NadekoBot.Modules.Xp
{
public partial class Xp
{
[Group]
public class Club : NadekoSubmodule<ClubService>
{
private readonly XpService _xps;
private readonly DiscordSocketClient _client;
public Club(XpService xps, DiscordSocketClient client)
{
_xps = xps;
_client = client;
}
[NadekoCommand, Usage, Description, Aliases]
public async Task ClubAdmin([Remainder]IUser toAdmin)
{
bool admin;
try
{
admin = _service.ToggleAdmin(Context.User, toAdmin);
}
catch (InvalidOperationException)
{
await ReplyErrorLocalized("club_admin_error").ConfigureAwait(false);
return;
}
if(admin)
await ReplyConfirmLocalized("club_admin_add", Format.Bold(toAdmin.ToString())).ConfigureAwait(false);
else
await ReplyConfirmLocalized("club_admin_remove", Format.Bold(toAdmin.ToString())).ConfigureAwait(false);
}
[NadekoCommand, Usage, Description, Aliases]
public async Task ClubCreate([Remainder]string clubName)
{
if (string.IsNullOrWhiteSpace(clubName) || clubName.Length > 20)
return;
if (!_service.CreateClub(Context.User, clubName, out ClubInfo club))
{
await ReplyErrorLocalized("club_create_error").ConfigureAwait(false);
return;
}
await ReplyConfirmLocalized("club_created", Format.Bold(club.ToString())).ConfigureAwait(false);
}
[NadekoCommand, Usage, Description, Aliases]
public Task ClubIcon([Remainder]string url = null)
{
if ((!Uri.IsWellFormedUriString(url, UriKind.Absolute) && url != null)
|| !_service.SetClubIcon(Context.User.Id, url))
{
return ReplyErrorLocalized("club_icon_error");
}
return ReplyConfirmLocalized("club_icon_set");
}
[NadekoCommand, Usage, Description, Aliases]
[Priority(1)]
public async Task ClubInformation(IUser user = null)
{
user = user ?? Context.User;
var club = _service.GetClubByMember(user);
if (club == null)
{
await ReplyErrorLocalized("club_not_exists").ConfigureAwait(false);
return;
}
await ClubInformation(club.ToString()).ConfigureAwait(false);
}
[NadekoCommand, Usage, Description, Aliases]
[Priority(0)]
public async Task ClubInformation([Remainder]string clubName = null)
{
if (string.IsNullOrWhiteSpace(clubName))
{
await ClubInformation(Context.User).ConfigureAwait(false);
return;
}
if (!_service.GetClubByName(clubName, out ClubInfo club))
{
await ReplyErrorLocalized("club_not_exists").ConfigureAwait(false);
return;
}
var lvl = new LevelStats(club.Xp);
await Context.Channel.SendPaginatedConfirmAsync(_client, 0, (page) =>
{
var embed = new EmbedBuilder()
.WithOkColor()
.WithTitle($"{club.ToString()}")
.WithDescription(GetText("level_x", lvl.Level) + $" ({club.Xp} xp)")
.AddField("Owner", club.Owner.ToString(), true)
.AddField("Level Req.", club.MinimumLevelReq.ToString(), true)
.AddField("Members", string.Join("\n", club.Users
.OrderByDescending(x => {
if (club.OwnerId == x.Id)
return 2;
else if (x.IsClubAdmin)
return 1;
else
return 0;
})
.Skip(page * 10)
.Take(10)
.Select(x =>
{
if (club.OwnerId == x.Id)
return x.ToString() + "🌟";
else if (x.IsClubAdmin)
return x.ToString() + "⭐";
return x.ToString();
})), false);
if (Uri.IsWellFormedUriString(club.ImageUrl, UriKind.Absolute))
return embed.WithThumbnailUrl(club.ImageUrl);
return embed;
}, club.Users.Count / 10);
}
[NadekoCommand, Usage, Description, Aliases]
public Task ClubBans(int page = 1)
{
if (--page < 0)
return Task.CompletedTask;
var club = _service.GetClubWithBansAndApplications(Context.User.Id);
if (club == null)
return ReplyErrorLocalized("club_not_exists_owner");
var bans = club
.Bans
.Select(x => x.User)
.ToArray();
return Context.Channel.SendPaginatedConfirmAsync(_client, page,
curPage =>
{
var toShow = string.Join("\n", bans
.Skip(page * 10)
.Take(10)
.Select(x => x.ToString()));
return new EmbedBuilder()
.WithTitle(GetText("club_bans_for", club.ToString()))
.WithDescription(toShow)
.WithOkColor();
}, bans.Length / 10);
}
[NadekoCommand, Usage, Description, Aliases]
public Task ClubApps(int page = 1)
{
if (--page < 0)
return Task.CompletedTask;
var club = _service.GetClubWithBansAndApplications(Context.User.Id);
if (club == null)
return ReplyErrorLocalized("club_not_exists_owner");
var apps = club
.Applicants
.Select(x => x.User)
.ToArray();
return Context.Channel.SendPaginatedConfirmAsync(_client, page,
curPage =>
{
var toShow = string.Join("\n", apps
.Skip(page * 10)
.Take(10)
.Select(x => x.ToString()));
return new EmbedBuilder()
.WithTitle(GetText("club_apps_for", club.ToString()))
.WithDescription(toShow)
.WithOkColor();
}, apps.Length / 10);
}
[NadekoCommand, Usage, Description, Aliases]
public async Task ClubApply([Remainder]string clubName)
{
if (string.IsNullOrWhiteSpace(clubName))
return;
if (!_service.GetClubByName(clubName, out ClubInfo club))
{
await ReplyErrorLocalized("club_not_exists").ConfigureAwait(false);
return;
}
if (_service.ApplyToClub(Context.User, club))
{
await ReplyConfirmLocalized("club_applied", Format.Bold(club.ToString())).ConfigureAwait(false);
}
else
{
await ReplyErrorLocalized("club_apply_error").ConfigureAwait(false);
}
}
[NadekoCommand, Usage, Description, Aliases]
[Priority(1)]
public Task ClubAccept(IUser user)
=> ClubAccept(user.ToString());
[NadekoCommand, Usage, Description, Aliases]
[Priority(0)]
public async Task ClubAccept([Remainder]string userName)
{
if (_service.AcceptApplication(Context.User.Id, userName, out var discordUser))
{
await ReplyConfirmLocalized("club_accepted", Format.Bold(discordUser.ToString())).ConfigureAwait(false);
}
else
await ReplyErrorLocalized("club_accept_error").ConfigureAwait(false);
}
[NadekoCommand, Usage, Description, Aliases]
public async Task Clubleave()
{
if (_service.LeaveClub(Context.User))
await ReplyConfirmLocalized("club_left").ConfigureAwait(false);
else
await ReplyErrorLocalized("club_not_in_club").ConfigureAwait(false);
}
[NadekoCommand, Usage, Description, Aliases]
[Priority(1)]
public Task ClubKick([Remainder]IUser user)
=> ClubKick(user.ToString());
[NadekoCommand, Usage, Description, Aliases]
[Priority(0)]
public Task ClubKick([Remainder]string userName)
{
if (_service.Kick(Context.User.Id, userName, out var club))
return ReplyConfirmLocalized("club_user_kick", Format.Bold(userName), Format.Bold(club.ToString()));
else
return ReplyErrorLocalized("club_user_kick_fail");
}
[NadekoCommand, Usage, Description, Aliases]
[Priority(1)]
public Task ClubBan([Remainder]IUser user)
=> ClubBan(user.ToString());
[NadekoCommand, Usage, Description, Aliases]
[Priority(0)]
public Task ClubBan([Remainder]string userName)
{
if (_service.Ban(Context.User.Id, userName, out var club))
return ReplyConfirmLocalized("club_user_banned", Format.Bold(userName), Format.Bold(club.ToString()));
else
return ReplyErrorLocalized("club_user_ban_fail");
}
[NadekoCommand, Usage, Description, Aliases]
[Priority(1)]
public Task ClubUnBan([Remainder]IUser user)
=> ClubUnBan(user.ToString());
[NadekoCommand, Usage, Description, Aliases]
[Priority(0)]
public Task ClubUnBan([Remainder]string userName)
{
if (_service.UnBan(Context.User.Id, userName, out var club))
return ReplyConfirmLocalized("club_user_unbanned", Format.Bold(userName), Format.Bold(club.ToString()));
else
return ReplyErrorLocalized("club_user_unban_fail");
}
[NadekoCommand, Usage, Description, Aliases]
public async Task ClubLevelReq(int level)
{
if (_service.ChangeClubLevelReq(Context.User.Id, level))
{
await ReplyConfirmLocalized("club_level_req_changed", Format.Bold(level.ToString())).ConfigureAwait(false);
}
else
{
await ReplyErrorLocalized("club_level_req_change_error").ConfigureAwait(false);
}
}
[NadekoCommand, Usage, Description, Aliases]
public async Task ClubDisband()
{
if (_service.Disband(Context.User.Id, out ClubInfo club))
{
await ReplyConfirmLocalized("club_disbanded", Format.Bold(club.ToString())).ConfigureAwait(false);
}
else
{
await ReplyErrorLocalized("club_disband_error").ConfigureAwait(false);
}
}
[NadekoCommand, Usage, Description, Aliases]
public Task ClubLeaderboard(int page = 1)
{
if (--page < 0)
return Task.CompletedTask;
var clubs = _service.GetClubLeaderboardPage(page);
var embed = new EmbedBuilder()
.WithTitle(GetText("club_leaderboard", page + 1))
.WithOkColor();
var i = page * 9;
foreach (var club in clubs)
{
embed.AddField($"#{++i} " + club.ToString(), club.Xp.ToString() + " xp", false);
}
return Context.Channel.EmbedAsync(embed);
}
}
}
}