Dividing admincmd into owner, admin and mod commands, add username history command
This commit is contained in:
parent
14dfbca389
commit
7167ea3ebc
3 changed files with 160 additions and 89 deletions
|
@ -11,24 +11,20 @@ using StackExchange.Redis;
|
||||||
namespace Geekbot.net.Commands
|
namespace Geekbot.net.Commands
|
||||||
{
|
{
|
||||||
[Group("admin")]
|
[Group("admin")]
|
||||||
public class AdminCmd : ModuleBase
|
[RequireUserPermission(GuildPermission.Administrator)]
|
||||||
|
public class Admin : ModuleBase
|
||||||
{
|
{
|
||||||
private readonly IDatabase _redis;
|
private readonly IDatabase _redis;
|
||||||
private readonly DiscordSocketClient _client;
|
private readonly DiscordSocketClient _client;
|
||||||
private readonly ILogger _logger;
|
|
||||||
private readonly IUserRepository _userRepository;
|
|
||||||
private readonly IErrorHandler _errorHandler;
|
private readonly IErrorHandler _errorHandler;
|
||||||
|
|
||||||
public AdminCmd(IDatabase redis, DiscordSocketClient client, ILogger logger, IUserRepository userRepositry, IErrorHandler errorHandler)
|
public Admin(IDatabase redis, DiscordSocketClient client, IErrorHandler errorHandler)
|
||||||
{
|
{
|
||||||
_redis = redis;
|
_redis = redis;
|
||||||
_client = client;
|
_client = client;
|
||||||
_logger = logger;
|
|
||||||
_userRepository = userRepositry;
|
|
||||||
_errorHandler = errorHandler;
|
_errorHandler = errorHandler;
|
||||||
}
|
}
|
||||||
|
|
||||||
[RequireUserPermission(GuildPermission.Administrator)]
|
|
||||||
[Command("welcome", RunMode = RunMode.Async)]
|
[Command("welcome", RunMode = RunMode.Async)]
|
||||||
[Remarks(CommandCategories.Admin)]
|
[Remarks(CommandCategories.Admin)]
|
||||||
[Summary("Set a Welcome Message (use '$user' to mention the new joined user).")]
|
[Summary("Set a Welcome Message (use '$user' to mention the new joined user).")]
|
||||||
|
@ -40,86 +36,6 @@ namespace Geekbot.net.Commands
|
||||||
formatedMessage);
|
formatedMessage);
|
||||||
}
|
}
|
||||||
|
|
||||||
[Command("youtubekey", RunMode = RunMode.Async)]
|
|
||||||
[Remarks(CommandCategories.Admin)]
|
|
||||||
[Summary("Set the youtube api key")]
|
|
||||||
public async Task SetYoutubeKey([Summary("API Key")] string key)
|
|
||||||
{
|
|
||||||
var botOwner = Context.Guild.GetUserAsync(ulong.Parse(_redis.StringGet("botOwner"))).Result;
|
|
||||||
if (!Context.User.Id.ToString().Equals(botOwner.Id.ToString()))
|
|
||||||
{
|
|
||||||
await ReplyAsync($"Sorry, only the botowner can do this ({botOwner.Username}#{botOwner.Discriminator})");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
_redis.StringSet("youtubeKey", key);
|
|
||||||
await ReplyAsync("Apikey has been set");
|
|
||||||
}
|
|
||||||
|
|
||||||
[Command("game", RunMode = RunMode.Async)]
|
|
||||||
[Remarks(CommandCategories.Admin)]
|
|
||||||
[Summary("Set the game that the bot is playing")]
|
|
||||||
public async Task SetGame([Remainder] [Summary("Game")] string key)
|
|
||||||
{
|
|
||||||
var botOwner = Context.Guild.GetUserAsync(ulong.Parse(_redis.StringGet("botOwner"))).Result;
|
|
||||||
if (!Context.User.Id.ToString().Equals(botOwner.Id.ToString()))
|
|
||||||
{
|
|
||||||
await ReplyAsync($"Sorry, only the botowner can do this ({botOwner.Username}#{botOwner.Discriminator})");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
_redis.StringSet("Game", key);
|
|
||||||
await _client.SetGameAsync(key);
|
|
||||||
_logger.Information($"[Geekbot] Changed game to {key}");
|
|
||||||
await ReplyAsync($"Now Playing {key}");
|
|
||||||
}
|
|
||||||
|
|
||||||
[Command("popuserrepo", RunMode = RunMode.Async)]
|
|
||||||
[Remarks(CommandCategories.Admin)]
|
|
||||||
[Summary("Populate user cache")]
|
|
||||||
public async Task popUserRepoCommand()
|
|
||||||
{
|
|
||||||
try
|
|
||||||
{
|
|
||||||
var botOwner = Context.Guild.GetUserAsync(ulong.Parse(_redis.StringGet("botOwner"))).Result;
|
|
||||||
if (!Context.User.Id.ToString().Equals(botOwner.Id.ToString()))
|
|
||||||
{
|
|
||||||
await ReplyAsync(
|
|
||||||
$"Sorry, only the botowner can do this ({botOwner.Username}#{botOwner.Discriminator})");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
catch (Exception)
|
|
||||||
{
|
|
||||||
await ReplyAsync(
|
|
||||||
$"Sorry, only the botowner can do this");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
var success = 0;
|
|
||||||
var failed = 0;
|
|
||||||
try
|
|
||||||
{
|
|
||||||
_logger.Warning("[UserRepository] Populating User Repositry");
|
|
||||||
await ReplyAsync("Starting Population of User Repository");
|
|
||||||
foreach (var guild in _client.Guilds)
|
|
||||||
{
|
|
||||||
_logger.Information($"[UserRepository] Populating users from {guild.Name}");
|
|
||||||
foreach (var user in guild.Users)
|
|
||||||
{
|
|
||||||
var succeded = await _userRepository.Update(user);
|
|
||||||
var inc = succeded ? success++ : failed++;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_logger.Warning("[UserRepository] Finished Updating User Repositry");
|
|
||||||
await ReplyAsync($"Successfully Populated User Repository with {success} Users in {_client.Guilds.Count} Guilds (Failed: {failed})");
|
|
||||||
}
|
|
||||||
catch (Exception e)
|
|
||||||
{
|
|
||||||
_errorHandler.HandleCommandException(e, Context, "Couldn't complete User Repository, see console for more info");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
[RequireUserPermission(GuildPermission.Administrator)]
|
|
||||||
[Command("modchannel", RunMode = RunMode.Async)]
|
[Command("modchannel", RunMode = RunMode.Async)]
|
||||||
[Remarks(CommandCategories.Admin)]
|
[Remarks(CommandCategories.Admin)]
|
||||||
[Summary("Set a channel for moderation purposes")]
|
[Summary("Set a channel for moderation purposes")]
|
||||||
|
@ -140,7 +56,6 @@ namespace Geekbot.net.Commands
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
[RequireUserPermission(GuildPermission.Administrator)]
|
|
||||||
[Command("showleave", RunMode = RunMode.Async)]
|
[Command("showleave", RunMode = RunMode.Async)]
|
||||||
[Remarks(CommandCategories.Admin)]
|
[Remarks(CommandCategories.Admin)]
|
||||||
[Summary("Notify modchannel when someone leaves")]
|
[Summary("Notify modchannel when someone leaves")]
|
||||||
|
@ -167,7 +82,6 @@ namespace Geekbot.net.Commands
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
[RequireUserPermission(GuildPermission.Administrator)]
|
|
||||||
[Command("showdel", RunMode = RunMode.Async)]
|
[Command("showdel", RunMode = RunMode.Async)]
|
||||||
[Remarks(CommandCategories.Admin)]
|
[Remarks(CommandCategories.Admin)]
|
||||||
[Summary("Notify modchannel when someone deletes a message")]
|
[Summary("Notify modchannel when someone deletes a message")]
|
47
Geekbot.net/Commands/Mod.cs
Normal file
47
Geekbot.net/Commands/Mod.cs
Normal file
|
@ -0,0 +1,47 @@
|
||||||
|
using System;
|
||||||
|
using System.Text;
|
||||||
|
using System.Threading.Tasks;
|
||||||
|
using Discord;
|
||||||
|
using Discord.Commands;
|
||||||
|
using Geekbot.net.Lib;
|
||||||
|
|
||||||
|
namespace Geekbot.net.Commands
|
||||||
|
{
|
||||||
|
[Group("mod")]
|
||||||
|
[RequireUserPermission(GuildPermission.KickMembers)]
|
||||||
|
[RequireUserPermission(GuildPermission.ManageMessages)]
|
||||||
|
[RequireUserPermission(GuildPermission.ManageRoles)]
|
||||||
|
public class Mod : ModuleBase
|
||||||
|
{
|
||||||
|
private readonly IUserRepository _userRepository;
|
||||||
|
private readonly IErrorHandler _errorHandler;
|
||||||
|
|
||||||
|
public Mod(IUserRepository userRepositry, IErrorHandler errorHandler)
|
||||||
|
{
|
||||||
|
_userRepository = userRepositry;
|
||||||
|
_errorHandler = errorHandler;
|
||||||
|
}
|
||||||
|
|
||||||
|
[Command("namehistory", RunMode = RunMode.Async)]
|
||||||
|
[Remarks(CommandCategories.Admin)]
|
||||||
|
[Summary("See past usernames of an user")]
|
||||||
|
public async Task usernameHistory([Summary("@user")] IUser user)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
var userRepo = _userRepository.Get(user.Id);
|
||||||
|
var sb = new StringBuilder();
|
||||||
|
sb.AppendLine($":bust_in_silhouette: {user.Username} has been known as:");
|
||||||
|
foreach (var name in userRepo.UsedNames)
|
||||||
|
{
|
||||||
|
sb.AppendLine($"- `{name}`");
|
||||||
|
}
|
||||||
|
await ReplyAsync(sb.ToString());
|
||||||
|
}
|
||||||
|
catch (Exception e)
|
||||||
|
{
|
||||||
|
_errorHandler.HandleCommandException(e, Context, "Modchannel doesn't seem to exist, please set one with `!admin modchannel [channelId]`");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
110
Geekbot.net/Commands/Owner.cs
Normal file
110
Geekbot.net/Commands/Owner.cs
Normal file
|
@ -0,0 +1,110 @@
|
||||||
|
using System;
|
||||||
|
using System.Threading.Tasks;
|
||||||
|
using Discord;
|
||||||
|
using Discord.Commands;
|
||||||
|
using Discord.WebSocket;
|
||||||
|
using Geekbot.net.Lib;
|
||||||
|
using Serilog;
|
||||||
|
using StackExchange.Redis;
|
||||||
|
|
||||||
|
namespace Geekbot.net.Commands
|
||||||
|
{
|
||||||
|
[Group("owner")]
|
||||||
|
[RequireUserPermission(GuildPermission.Administrator)]
|
||||||
|
public class Owner : ModuleBase
|
||||||
|
{
|
||||||
|
private readonly IDatabase _redis;
|
||||||
|
private readonly DiscordSocketClient _client;
|
||||||
|
private readonly ILogger _logger;
|
||||||
|
private readonly IUserRepository _userRepository;
|
||||||
|
private readonly IErrorHandler _errorHandler;
|
||||||
|
|
||||||
|
public Owner(IDatabase redis, DiscordSocketClient client, ILogger logger, IUserRepository userRepositry, IErrorHandler errorHandler)
|
||||||
|
{
|
||||||
|
_redis = redis;
|
||||||
|
_client = client;
|
||||||
|
_logger = logger;
|
||||||
|
_userRepository = userRepositry;
|
||||||
|
_errorHandler = errorHandler;
|
||||||
|
}
|
||||||
|
|
||||||
|
[Command("youtubekey", RunMode = RunMode.Async)]
|
||||||
|
[Remarks(CommandCategories.Admin)]
|
||||||
|
[Summary("Set the youtube api key")]
|
||||||
|
public async Task SetYoutubeKey([Summary("API Key")] string key)
|
||||||
|
{
|
||||||
|
var botOwner = Context.Guild.GetUserAsync(ulong.Parse(_redis.StringGet("botOwner"))).Result;
|
||||||
|
if (!Context.User.Id.ToString().Equals(botOwner.Id.ToString()))
|
||||||
|
{
|
||||||
|
await ReplyAsync($"Sorry, only the botowner can do this ({botOwner.Username}#{botOwner.Discriminator})");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
_redis.StringSet("youtubeKey", key);
|
||||||
|
await ReplyAsync("Apikey has been set");
|
||||||
|
}
|
||||||
|
|
||||||
|
[Command("game", RunMode = RunMode.Async)]
|
||||||
|
[Remarks(CommandCategories.Admin)]
|
||||||
|
[Summary("Set the game that the bot is playing")]
|
||||||
|
public async Task SetGame([Remainder] [Summary("Game")] string key)
|
||||||
|
{
|
||||||
|
var botOwner = Context.Guild.GetUserAsync(ulong.Parse(_redis.StringGet("botOwner"))).Result;
|
||||||
|
if (!Context.User.Id.ToString().Equals(botOwner.Id.ToString()))
|
||||||
|
{
|
||||||
|
await ReplyAsync($"Sorry, only the botowner can do this ({botOwner.Username}#{botOwner.Discriminator})");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
_redis.StringSet("Game", key);
|
||||||
|
await _client.SetGameAsync(key);
|
||||||
|
_logger.Information($"[Geekbot] Changed game to {key}");
|
||||||
|
await ReplyAsync($"Now Playing {key}");
|
||||||
|
}
|
||||||
|
|
||||||
|
[Command("popuserrepo", RunMode = RunMode.Async)]
|
||||||
|
[Remarks(CommandCategories.Admin)]
|
||||||
|
[Summary("Populate user cache")]
|
||||||
|
public async Task popUserRepoCommand()
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
var botOwner = Context.Guild.GetUserAsync(ulong.Parse(_redis.StringGet("botOwner"))).Result;
|
||||||
|
if (!Context.User.Id.ToString().Equals(botOwner.Id.ToString()))
|
||||||
|
{
|
||||||
|
await ReplyAsync(
|
||||||
|
$"Sorry, only the botowner can do this ({botOwner.Username}#{botOwner.Discriminator})");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (Exception)
|
||||||
|
{
|
||||||
|
await ReplyAsync(
|
||||||
|
$"Sorry, only the botowner can do this");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
var success = 0;
|
||||||
|
var failed = 0;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
_logger.Warning("[UserRepository] Populating User Repositry");
|
||||||
|
await ReplyAsync("Starting Population of User Repository");
|
||||||
|
foreach (var guild in _client.Guilds)
|
||||||
|
{
|
||||||
|
_logger.Information($"[UserRepository] Populating users from {guild.Name}");
|
||||||
|
foreach (var user in guild.Users)
|
||||||
|
{
|
||||||
|
var succeded = await _userRepository.Update(user);
|
||||||
|
var inc = succeded ? success++ : failed++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_logger.Warning("[UserRepository] Finished Updating User Repositry");
|
||||||
|
await ReplyAsync($"Successfully Populated User Repository with {success} Users in {_client.Guilds.Count} Guilds (Failed: {failed})");
|
||||||
|
}
|
||||||
|
catch (Exception e)
|
||||||
|
{
|
||||||
|
_errorHandler.HandleCommandException(e, Context, "Couldn't complete User Repository, see console for more info");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue