every single file changed lulw

This commit is contained in:
l1ttleO 2022-01-30 13:43:15 +05:00
parent f30485dd71
commit 4d838e5af3
Signed by: Octol1ttle
GPG key ID: B77C34313AEE1FFF
14 changed files with 825 additions and 511 deletions

View file

@ -1,5 +1,5 @@
using System.Globalization; using System.Globalization;
using System.Text.Json; using Newtonsoft.Json;
using Discord; using Discord;
using Discord.WebSocket; using Discord.WebSocket;
@ -26,7 +26,7 @@ public static class Boyfriend {
await Client.LoginAsync(TokenType.Bot, token); await Client.LoginAsync(TokenType.Bot, token);
await Client.StartAsync(); await Client.StartAsync();
await Client.SetActivityAsync(new Game("Retrospecter - Chiller", ActivityType.Listening)); await Client.SetActivityAsync(new Game("Retrospecter - Expurgation", ActivityType.Listening));
new EventHandler().InitEvents(); new EventHandler().InitEvents();
@ -41,27 +41,33 @@ public static class Boyfriend {
public static async Task SetupGuildConfigs() { public static async Task SetupGuildConfigs() {
foreach (var guild in Client.Guilds) { foreach (var guild in Client.Guilds) {
var path = "config_" + guild.Id + ".json"; var path = "config_" + guild.Id + ".json";
var openStream = !File.Exists(path) ? File.Create(path) : File.OpenRead(path); if (!File.Exists(path)) File.Create(path);
GuildConfig config; var config = JsonConvert.DeserializeObject<GuildConfig>(await File.ReadAllTextAsync(path));
try { if (config == null) {
config = await JsonSerializer.DeserializeAsync<GuildConfig>(openStream) ?? throw new Exception();
} catch (JsonException) {
Messages.Culture = new CultureInfo("ru"); Messages.Culture = new CultureInfo("ru");
config = new GuildConfig(guild.Id, "ru", "!", false, true, config = new GuildConfig(guild.Id);
true, Messages.DefaultWelcomeMessage, 0, 0, 0, 0);
} }
config.Validate();
GuildConfigDictionary.Add(config.Id.GetValueOrDefault(0), config);
}
}
public static void ResetGuildConfig(IGuild guild) {
GuildConfigDictionary.Remove(guild.Id);
var config = new GuildConfig(guild.Id);
config.Validate();
GuildConfigDictionary.Add(guild.Id, config); GuildConfigDictionary.Add(guild.Id, config);
} }
}
public static GuildConfig GetGuildConfig(IGuild guild) { public static GuildConfig GetGuildConfig(IGuild guild) {
Messages.Culture = new CultureInfo("ru"); Messages.Culture = new CultureInfo("ru");
var toReturn = GuildConfigDictionary.ContainsKey(guild.Id) ? GuildConfigDictionary[guild.Id] var config = GuildConfigDictionary.ContainsKey(guild.Id) ? GuildConfigDictionary[guild.Id] :
: new GuildConfig(guild.Id, "ru", "!", false, true, true, Messages.DefaultWelcomeMessage, 0, 0, 0, 0); new GuildConfig(guild.Id);
config.Validate();
if (toReturn.Id != guild.Id) throw new Exception(); return config;
return toReturn;
} }
public static IGuild FindGuild(IMessageChannel channel) { public static IGuild FindGuild(IMessageChannel channel) {

View file

@ -15,7 +15,8 @@
</PropertyGroup> </PropertyGroup>
<ItemGroup> <ItemGroup>
<PackageReference Include="Discord.Net" Version="3.1.0"/> <PackageReference Include="Discord.Net" Version="3.2.0" />
<PackageReference Include="Newtonsoft.Json" Version="13.0.1" />
</ItemGroup> </ItemGroup>
<ItemGroup> <ItemGroup>

View file

@ -9,14 +9,15 @@ namespace Boyfriend;
public static class CommandHandler { public static class CommandHandler {
public static readonly Command[] Commands = { public static readonly Command[] Commands = {
new BanCommand(), new ClearCommand(), new HelpCommand(), new BanCommand(), new ClearCommand(), new HelpCommand(),
new KickCommand(), new MuteCommand() new KickCommand(), new MuteCommand(), new PingCommand(),
new SettingsCommand(), new UnbanCommand(), new UnmuteCommand()
}; };
public static async Task HandleCommand(SocketUserMessage message) { public static async Task HandleCommand(SocketUserMessage message) {
var context = new SocketCommandContext(Boyfriend.Client, message); var context = new SocketCommandContext(Boyfriend.Client, message);
foreach (var command in Commands) { foreach (var command in Commands) {
var regex = new Regex(Regex.Escape(Boyfriend.GetGuildConfig(context.Guild).Prefix)); var regex = new Regex(Regex.Escape(Boyfriend.GetGuildConfig(context.Guild).Prefix!));
if (!command.GetAliases().Contains(regex.Replace(message.Content, "", 1).Split()[0])) continue; if (!command.GetAliases().Contains(regex.Replace(message.Content, "", 1).Split()[0])) continue;
var args = message.Content.Split().Skip(1).ToArray(); var args = message.Content.Split().Skip(1).ToArray();
@ -35,6 +36,7 @@ public static class CommandHandler {
await context.Channel.SendMessageAsync($"{signature} `{e.Message}`"); await context.Channel.SendMessageAsync($"{signature} `{e.Message}`");
if (e.StackTrace != null && e is not ApplicationException or UnauthorizedAccessException) if (e.StackTrace != null && e is not ApplicationException or UnauthorizedAccessException)
await context.Channel.SendMessageAsync(Utils.Wrap(e.StackTrace)); await context.Channel.SendMessageAsync(Utils.Wrap(e.StackTrace));
throw;
} }
break; break;

View file

@ -28,15 +28,17 @@ public class MuteCommand : Command {
toMute.TimedOutUntil != null && toMute.TimedOutUntil.Value.ToUnixTimeMilliseconds() toMute.TimedOutUntil != null && toMute.TimedOutUntil.Value.ToUnixTimeMilliseconds()
> DateTimeOffset.Now.ToUnixTimeMilliseconds()) > DateTimeOffset.Now.ToUnixTimeMilliseconds())
throw new ApplicationException(Messages.MemberAlreadyMuted); throw new ApplicationException(Messages.MemberAlreadyMuted);
var rolesRemoved = Boyfriend.GetGuildConfig(context.Guild).RolesRemovedOnMute; var config = Boyfriend.GetGuildConfig(context.Guild);
var rolesRemoved = config.RolesRemovedOnMute!;
if (rolesRemoved.ContainsKey(toMute.Id)) { if (rolesRemoved.ContainsKey(toMute.Id)) {
foreach (var roleId in rolesRemoved[toMute.Id]) await toMute.AddRoleAsync(roleId); foreach (var roleId in rolesRemoved[toMute.Id]) await toMute.AddRoleAsync(roleId);
rolesRemoved.Remove(toMute.Id); rolesRemoved.Remove(toMute.Id);
await config.Save();
await Warn(context.Channel, Messages.RolesReturned); await Warn(context.Channel, Messages.RolesReturned);
return; return;
} }
await CommandHandler.CheckPermissions(author, GuildPermission.ManageMessages, GuildPermission.ManageRoles); await CommandHandler.CheckPermissions(author, GuildPermission.ModerateMembers, GuildPermission.ManageRoles);
await CommandHandler.CheckInteractions(author, toMute); await CommandHandler.CheckInteractions(author, toMute);
MuteMember(context.Guild, context.Channel as ITextChannel, context.Guild.GetUser(context.User.Id), toMute, MuteMember(context.Guild, context.Channel as ITextChannel, context.Guild.GetUser(context.User.Id), toMute,
duration, reason); duration, reason);
@ -48,7 +50,7 @@ public class MuteCommand : Command {
var authorMention = author.Mention; var authorMention = author.Mention;
var role = Utils.GetMuteRole(guild); var role = Utils.GetMuteRole(guild);
var config = Boyfriend.GetGuildConfig(guild); var config = Boyfriend.GetGuildConfig(guild);
if (config.RemoveRolesOnMute && role != null) { if (config.RemoveRolesOnMute.GetValueOrDefault(false) && role != null) {
var rolesRemoved = new List<ulong>(); var rolesRemoved = new List<ulong>();
try { try {
foreach (var roleId in toMute.RoleIds) { foreach (var roleId in toMute.RoleIds) {
@ -59,7 +61,7 @@ public class MuteCommand : Command {
} }
catch (NullReferenceException) { } catch (NullReferenceException) { }
config.RolesRemovedOnMute.Add(toMute.Id, rolesRemoved); config.RolesRemovedOnMute!.Add(toMute.Id, rolesRemoved);
await config.Save(); await config.Save();
} }

View file

@ -7,7 +7,7 @@ namespace Boyfriend.Commands;
public class PingCommand : Command { public class PingCommand : Command {
public override async Task Run(SocketCommandContext context, string[] args) { public override async Task Run(SocketCommandContext context, string[] args) {
await context.Channel.SendMessageAsync($"{Utils.GetBeep(Boyfriend.GetGuildConfig(context.Guild).Lang)}" + await context.Channel.SendMessageAsync($"{Utils.GetBeep(Boyfriend.GetGuildConfig(context.Guild).Lang!)}" +
$"{Boyfriend.Client.Latency}{Messages.Milliseconds}"); $"{Boyfriend.Client.Latency}{Messages.Milliseconds}");
} }

View file

@ -14,71 +14,89 @@ public class SettingsCommand : Command {
var guild = context.Guild; var guild = context.Guild;
if (args.Length == 0) { if (args.Length == 0) {
var nl = Environment.NewLine; var nl = Environment.NewLine;
var adminLogChannel = guild.GetTextChannel(config.AdminLogChannel); var adminLogChannel = guild.GetTextChannel(config.AdminLogChannel.GetValueOrDefault(0));
var admin = adminLogChannel == null ? Messages.ChannelNotSpecified : adminLogChannel.Mention; var admin = adminLogChannel == null ? Messages.ChannelNotSpecified : adminLogChannel.Mention;
var botLogChannel = guild.GetTextChannel(config.BotLogChannel); var botLogChannel = guild.GetTextChannel(config.BotLogChannel.GetValueOrDefault(0));
var bot = botLogChannel == null ? Messages.ChannelNotSpecified : botLogChannel.Mention; var bot = botLogChannel == null ? Messages.ChannelNotSpecified : botLogChannel.Mention;
var muteRole = guild.GetRole(config.MuteRole); var muteRole = guild.GetRole(config.MuteRole.GetValueOrDefault(0));
var mute = muteRole == null ? Messages.RoleNotSpecified : muteRole.Mention; var mute = muteRole == null ? Messages.RoleNotSpecified : muteRole.Mention;
var defaultRole = guild.GetRole(config.DefaultRole); var defaultRole = guild.GetRole(config.DefaultRole.GetValueOrDefault(0));
var defaultr = muteRole == null ? Messages.RoleNotSpecified : defaultRole.Mention; var defaultr = defaultRole == null ? Messages.RoleNotSpecified : defaultRole.Mention;
var toSend = string.Format(Messages.CurrentSettings, nl) + var toSend = string.Format(Messages.CurrentSettings, nl) +
string.Format(Messages.CurrentSettingsLang, config.Lang, nl) + string.Format(Messages.CurrentSettingsLang, config.Lang, nl) +
string.Format(Messages.CurrentSettingsPrefix, config.Prefix, nl) + string.Format(Messages.CurrentSettingsPrefix, config.Prefix, nl) +
string.Format(Messages.CurrentSettingsRemoveRoles, YesOrNo(config.RemoveRolesOnMute), nl) + string.Format(Messages.CurrentSettingsRemoveRoles, YesOrNo(
string.Format(Messages.CurrentSettingsUseSystemChannel, YesOrNo(config.UseSystemChannel), nl) + config.RemoveRolesOnMute.GetValueOrDefault(false)), nl) +
string.Format(Messages.CurrentSettingsSendWelcomeMessages, YesOrNo(config.UseSystemChannel), string.Format(Messages.CurrentSettingsUseSystemChannel, YesOrNo(
nl) + config.UseSystemChannel.GetValueOrDefault(true)), nl) +
string.Format(Messages.CurrentSettingsSendWelcomeMessages, YesOrNo(
config.SendWelcomeMessages.GetValueOrDefault(true)), nl) +
string.Format(Messages.CurrentSettingsReceiveStartupMessages, YesOrNo(
config.ReceiveStartupMessages.GetValueOrDefault(true)), nl) +
string.Format(Messages.CurrentSettingsWelcomeMessage, config.WelcomeMessage, nl) + string.Format(Messages.CurrentSettingsWelcomeMessage, config.WelcomeMessage, nl) +
string.Format(Messages.CurrentSettingsDefaultRole, defaultr, nl) + string.Format(Messages.CurrentSettingsDefaultRole, defaultr, nl) +
string.Format(Messages.CurrentSettingsMuteRole, mute, nl) + string.Format(Messages.CurrentSettingsMuteRole, mute, nl) +
string.Format(Messages.CurrentSettingsAdminLogChannel, admin, nl) + string.Format(Messages.CurrentSettingsAdminLogChannel, admin, nl) +
string.Format(Messages.CurrentSettingsBotLogChannel, bot); string.Format(Messages.CurrentSettingsBotLogChannel, bot);
await Utils.SilentSendAsync(context.Channel as ITextChannel ?? throw new Exception(), toSend); await Utils.SilentSendAsync(context.Channel as ITextChannel ?? throw new ApplicationException(), toSend);
return; return;
} }
var setting = args[0].ToLower(); var setting = args[0].ToLower();
var value = args[1].ToLower(); var value = "";
var shouldDefault = false;
if (args.Length >= 2)
value = args[1].ToLower();
else
shouldDefault = true;
var boolValue = ParseBool(args[1]); var boolValue = ParseBool(value);
var channel = await Utils.ParseChannelNullable(value) as IGuildChannel; var channel = await Utils.ParseChannelNullable(value) as IGuildChannel;
var role = Utils.ParseRoleNullable(guild, value); var role = Utils.ParseRoleNullable(guild, value);
switch (setting) { switch (setting) {
case "reset":
Boyfriend.ResetGuildConfig(guild);
break;
case "lang" when value is not ("ru" or "en"): case "lang" when value is not ("ru" or "en"):
throw new Exception(Messages.LanguageNotSupported); throw new ApplicationException(Messages.LanguageNotSupported);
case "lang": case "lang":
config.Lang = value; config.Lang = shouldDefault ? "ru" : value;
Messages.Culture = new CultureInfo(value); Messages.Culture = new CultureInfo(shouldDefault ? "ru" : value);
break; break;
case "prefix": case "prefix":
config.Prefix = value; config.Prefix = shouldDefault ? "!" : value;
break; break;
case "removerolesonmute": case "removerolesonmute":
config.RemoveRolesOnMute = GetBoolValue(boolValue); config.RemoveRolesOnMute = !shouldDefault && GetBoolValue(boolValue);
break; break;
case "usesystemchannel": case "usesystemchannel":
config.UseSystemChannel = GetBoolValue(boolValue); config.UseSystemChannel = shouldDefault || GetBoolValue(boolValue);
break; break;
case "sendwelcomemessages": case "sendwelcomemessages":
config.SendWelcomeMessages = GetBoolValue(boolValue); config.SendWelcomeMessages = shouldDefault || GetBoolValue(boolValue);
break;
case "receivestartupmessages":
config.ReceiveStartupMessages = shouldDefault || GetBoolValue(boolValue);
break; break;
case "welcomemessage": case "welcomemessage":
config.WelcomeMessage = value; config.WelcomeMessage = shouldDefault ? Messages.DefaultWelcomeMessage : value;
break; break;
case "defaultrole": case "defaultrole":
config.DefaultRole = GetRoleId(role); config.DefaultRole = shouldDefault ? 0 : GetRoleId(role);
break; break;
case "muterole": case "muterole":
config.MuteRole = GetRoleId(role); config.MuteRole = shouldDefault ? 0 : GetRoleId(role);
break; break;
case "adminlogchannel": case "adminlogchannel":
config.AdminLogChannel = GetChannelId(channel); config.AdminLogChannel = shouldDefault ? 0 : GetChannelId(channel);
break; break;
case "botlogchannel": case "botlogchannel":
config.BotLogChannel = GetChannelId(channel); config.BotLogChannel = shouldDefault ? 0 : GetChannelId(channel);
break; break;
default:
await context.Channel.SendMessageAsync(Messages.SettingDoesntExist);
return;
} }
await config.Save(); await config.Save();
@ -89,22 +107,21 @@ public class SettingsCommand : Command {
private static bool? ParseBool(string toParse) { private static bool? ParseBool(string toParse) {
try { try {
return bool.Parse(toParse.ToLower()); return bool.Parse(toParse.ToLower());
} } catch (FormatException) {
catch (FormatException) {
return null; return null;
} }
} }
private static bool GetBoolValue(bool? from) { private static bool GetBoolValue(bool? from) {
return from ?? throw new Exception(Messages.InvalidBoolean); return from ?? throw new ApplicationException(Messages.InvalidBoolean);
} }
private static ulong GetRoleId(IRole? role) { private static ulong GetRoleId(IRole? role) {
return (role ?? throw new Exception(Messages.InvalidRoleSpecified)).Id; return (role ?? throw new ApplicationException(Messages.InvalidRoleSpecified)).Id;
} }
private static ulong GetChannelId(IGuildChannel? channel) { private static ulong GetChannelId(IGuildChannel? channel) {
return (channel ?? throw new Exception(Messages.InvalidChannelSpecified)).Id; return (channel ?? throw new ApplicationException(Messages.InvalidChannelSpecified)).Id;
} }
private static string YesOrNo(bool isYes) { private static string YesOrNo(bool isYes) {

View file

@ -11,7 +11,7 @@ public class UnbanCommand : Command {
public override async Task Run(SocketCommandContext context, string[] args) { public override async Task Run(SocketCommandContext context, string[] args) {
var toUnban = await Utils.ParseUser(args[0]); var toUnban = await Utils.ParseUser(args[0]);
if (context.Guild.GetBanAsync(toUnban.Id) == null) if (context.Guild.GetBanAsync(toUnban.Id) == null)
throw new Exception(Messages.UserNotBanned); throw new ApplicationException(Messages.UserNotBanned);
UnbanUser(context.Guild, context.Channel as ITextChannel, context.Guild.GetUser(context.User.Id), toUnban, UnbanUser(context.Guild, context.Channel as ITextChannel, context.Guild.GetUser(context.User.Id), toUnban,
Utils.JoinString(args, 1)); Utils.JoinString(args, 1));
} }

View file

@ -14,14 +14,21 @@ public class UnmuteCommand : Command {
await CommandHandler.CheckPermissions(author, GuildPermission.ManageMessages, GuildPermission.ManageRoles); await CommandHandler.CheckPermissions(author, GuildPermission.ManageMessages, GuildPermission.ManageRoles);
await CommandHandler.CheckInteractions(author, toUnmute); await CommandHandler.CheckInteractions(author, toUnmute);
var role = Utils.GetMuteRole(context.Guild); var role = Utils.GetMuteRole(context.Guild);
if (role != null) if (role != null) {
if (toUnmute.RoleIds.All(x => x != role.Id)) { if (toUnmute.RoleIds.All(x => x != role.Id)) {
var rolesRemoved = Boyfriend.GetGuildConfig(context.Guild).RolesRemovedOnMute; var config = Boyfriend.GetGuildConfig(context.Guild);
var rolesRemoved = config.RolesRemovedOnMute;
foreach (var roleId in rolesRemoved[toUnmute.Id]) await toUnmute.AddRoleAsync(roleId); foreach (var roleId in rolesRemoved![toUnmute.Id]) await toUnmute.AddRoleAsync(roleId);
rolesRemoved.Remove(toUnmute.Id); rolesRemoved.Remove(toUnmute.Id);
await config.Save();
throw new ApplicationException(Messages.RolesReturned); throw new ApplicationException(Messages.RolesReturned);
} }
}
if (role != null && toUnmute.RoleIds.All(x => x != role.Id) ||
toUnmute.TimedOutUntil == null || toUnmute.TimedOutUntil.Value.ToUnixTimeMilliseconds()
< DateTimeOffset.Now.ToUnixTimeMilliseconds())
throw new ApplicationException(Messages.MemberNotMuted);
UnmuteMember(context.Guild, context.Channel as ITextChannel, context.Guild.GetUser(context.User.Id), UnmuteMember(context.Guild, context.Channel as ITextChannel, context.Guild.GetUser(context.User.Id),
toUnmute, Utils.JoinString(args, 1)); toUnmute, Utils.JoinString(args, 1));
@ -33,12 +40,19 @@ public class UnmuteCommand : Command {
var authorMention = author.Mention; var authorMention = author.Mention;
var notification = string.Format(Messages.MemberUnmuted, authorMention, toUnmute.Mention, var notification = string.Format(Messages.MemberUnmuted, authorMention, toUnmute.Mention,
Utils.WrapInline(reason)); Utils.WrapInline(reason));
await toUnmute.RemoveRoleAsync(Utils.GetMuteRole(guild)); var role = Utils.GetMuteRole(guild);
if (role != null) {
await toUnmute.RemoveRoleAsync(role);
var config = Boyfriend.GetGuildConfig(guild); var config = Boyfriend.GetGuildConfig(guild);
if (config.RolesRemovedOnMute.ContainsKey(toUnmute.Id)) { if (config.RolesRemovedOnMute!.ContainsKey(toUnmute.Id)) {
foreach (var roleId in config.RolesRemovedOnMute[toUnmute.Id]) await toUnmute.AddRoleAsync(roleId); foreach (var roleId in config.RolesRemovedOnMute[toUnmute.Id]) await toUnmute.AddRoleAsync(roleId);
config.RolesRemovedOnMute.Remove(toUnmute.Id); config.RolesRemovedOnMute.Remove(toUnmute.Id);
await config.Save();
}
} else {
await toUnmute.RemoveTimeOutAsync();
} }
await Utils.SilentSendAsync(channel, string.Format(Messages.UnmuteResponse, toUnmute.Mention, await Utils.SilentSendAsync(channel, string.Format(Messages.UnmuteResponse, toUnmute.Mention,

View file

@ -23,10 +23,10 @@ public class EventHandler {
var i = new Random().Next(3); var i = new Random().Next(3);
foreach (var guild in Boyfriend.Client.Guilds) { foreach (var guild in Boyfriend.Client.Guilds) {
var config = Boyfriend.GetGuildConfig(guild); var config = Boyfriend.GetGuildConfig(guild);
Messages.Culture = new CultureInfo(config.Lang); Messages.Culture = new CultureInfo(config.Lang!);
var channel = guild.GetTextChannel(config.BotLogChannel); var channel = guild.GetTextChannel(config.BotLogChannel.GetValueOrDefault(0));
if (channel == null) continue; if (!config.ReceiveStartupMessages.GetValueOrDefault(true) || channel == null) continue;
await channel.SendMessageAsync(string.Format(Messages.Ready, Utils.GetBeep(config.Lang, i))); await channel.SendMessageAsync(string.Format(Messages.Ready, Utils.GetBeep(config.Lang!, i)));
} }
} }
@ -48,7 +48,7 @@ public class EventHandler {
var argPos = 0; var argPos = 0;
var guildConfig = Boyfriend.GetGuildConfig(guild); var guildConfig = Boyfriend.GetGuildConfig(guild);
Messages.Culture = new CultureInfo(guildConfig.Lang); Messages.Culture = new CultureInfo(guildConfig.Lang!);
if ((message.MentionedUsers.Count > 3 || message.MentionedRoles.Count > 2) if ((message.MentionedUsers.Count > 3 || message.MentionedRoles.Count > 2)
&& !user.GuildPermissions.MentionEveryone) && !user.GuildPermissions.MentionEveryone)
await BanCommand.BanUser(guild, null, await guild.GetCurrentUserAsync(), user, await BanCommand.BanUser(guild, null, await guild.GetCurrentUserAsync(), user,
@ -67,7 +67,7 @@ public class EventHandler {
if (!(message.HasStringPrefix(guildConfig.Prefix, ref argPos) if (!(message.HasStringPrefix(guildConfig.Prefix, ref argPos)
|| message.HasMentionPrefix(Boyfriend.Client.CurrentUser, ref argPos)) || message.HasMentionPrefix(Boyfriend.Client.CurrentUser, ref argPos))
|| user == await guild.GetCurrentUserAsync() || user == await guild.GetCurrentUserAsync()
|| user.IsBot && message.Content.Contains(prev) || message.Content.Contains(prevFailsafe)) || user.IsBot && (message.Content.Contains(prev) || message.Content.Contains(prevFailsafe)))
return; return;
await CommandHandler.HandleCommand(message); await CommandHandler.HandleCommand(message);
@ -91,10 +91,10 @@ public class EventHandler {
private static async Task UserJoinedEvent(SocketGuildUser user) { private static async Task UserJoinedEvent(SocketGuildUser user) {
var guild = user.Guild; var guild = user.Guild;
var config = Boyfriend.GetGuildConfig(guild); var config = Boyfriend.GetGuildConfig(guild);
if (config.SendWelcomeMessages) if (config.SendWelcomeMessages.GetValueOrDefault(true))
await Utils.SilentSendAsync(guild.SystemChannel, string.Format(config.WelcomeMessage, user.Mention, await Utils.SilentSendAsync(guild.SystemChannel, string.Format(config.WelcomeMessage!, user.Mention,
guild.Name)); guild.Name));
if (config.DefaultRole != 0) if (config.DefaultRole != 0)
await user.AddRoleAsync(Utils.ParseRole(guild, config.DefaultRole.ToString())); await user.AddRoleAsync(Utils.ParseRole(guild, config.DefaultRole.ToString()!));
} }
} }

View file

@ -1,40 +1,48 @@
using System.Text.Json; using System.Globalization;
using Newtonsoft.Json;
namespace Boyfriend; namespace Boyfriend;
public class GuildConfig { public class GuildConfig {
public ulong Id { get; } public ulong? Id { get; }
public string Lang { get; set; } public string? Lang { get; set; }
public string Prefix { get; set; } public string? Prefix { get; set; }
public bool RemoveRolesOnMute { get; set; } public bool? RemoveRolesOnMute { get; set; }
public bool UseSystemChannel { get; set; } public bool? UseSystemChannel { get; set; }
public bool SendWelcomeMessages { get; set; } public bool? SendWelcomeMessages { get; set; }
public string WelcomeMessage { get; set; } public bool? ReceiveStartupMessages { get; set; }
public ulong DefaultRole { get; set; } public string? WelcomeMessage { get; set; }
public ulong MuteRole { get; set; } public ulong? DefaultRole { get; set; }
public ulong AdminLogChannel { get; set; } public ulong? MuteRole { get; set; }
public ulong BotLogChannel { get; set; } public ulong? AdminLogChannel { get; set; }
public Dictionary<ulong, List<ulong>> RolesRemovedOnMute { get; set; } public ulong? BotLogChannel { get; set; }
public Dictionary<ulong, List<ulong>>? RolesRemovedOnMute { get; private set; }
public GuildConfig(ulong id, string lang, string prefix, bool removeRolesOnMute, bool useSystemChannel, public GuildConfig(ulong id) {
bool sendWelcomeMessages, string welcomeMessage, ulong defaultRole, ulong muteRole, ulong adminLogChannel,
ulong botLogChannel) {
Id = id; Id = id;
Lang = lang; Validate();
Prefix = prefix; }
RemoveRolesOnMute = removeRolesOnMute;
UseSystemChannel = useSystemChannel; public void Validate() {
SendWelcomeMessages = sendWelcomeMessages; if (Id == null) throw new Exception("Something went horribly, horribly wrong");
WelcomeMessage = welcomeMessage; Lang ??= "ru";
DefaultRole = defaultRole; Messages.Culture = new CultureInfo(Lang);
MuteRole = muteRole; Prefix ??= "!";
AdminLogChannel = adminLogChannel; RemoveRolesOnMute ??= false;
BotLogChannel = botLogChannel; UseSystemChannel ??= true;
RolesRemovedOnMute = new Dictionary<ulong, List<ulong>>(); SendWelcomeMessages ??= true;
ReceiveStartupMessages ??= true;
WelcomeMessage ??= Messages.DefaultWelcomeMessage;
DefaultRole ??= 0;
MuteRole ??= 0;
AdminLogChannel ??= 0;
BotLogChannel ??= 0;
RolesRemovedOnMute ??= new Dictionary<ulong, List<ulong>>();
} }
public async Task Save() { public async Task Save() {
await using var stream = File.OpenWrite("config_" + Id + ".json"); Validate();
await JsonSerializer.SerializeAsync(stream, this); RolesRemovedOnMute!.TrimExcess();
await File.WriteAllTextAsync("config_" + Id + ".json", JsonConvert.SerializeObject(this));
} }
} }

File diff suppressed because it is too large Load diff

View file

@ -243,4 +243,10 @@
<data name="UnmuteResponse" xml:space="preserve"> <data name="UnmuteResponse" xml:space="preserve">
<value>:white_check_mark: Successfully unmuted {0} for {1}</value> <value>:white_check_mark: Successfully unmuted {0} for {1}</value>
</data> </data>
<data name="SettingDoesntExist" xml:space="preserve">
<value>That setting doesn't exist!</value>
</data>
<data name="CurrentSettingsReceiveStartupMessages" xml:space="preserve">
<value>Receive startup messages (`receiveStartupMessages`): {0}{1}</value>
</data>
</root> </root>

View file

@ -234,4 +234,10 @@
<data name="UnmuteResponse" xml:space="preserve"> <data name="UnmuteResponse" xml:space="preserve">
<value>:white_check_mark: Успешно возвращён из мута {0} за {1}</value> <value>:white_check_mark: Успешно возвращён из мута {0} за {1}</value>
</data> </data>
<data name="SettingDoesntExist" xml:space="preserve">
<value>Такая настройка не существует!</value>
</data>
<data name="CurrentSettingsReceiveStartupMessages" xml:space="preserve">
<value>Получать сообщения о запуске (`receiveStartupMessages`): {0}{1}</value>
</data>
</root> </root>

View file

@ -21,7 +21,7 @@ public static class Utils {
} }
public static async Task<ITextChannel?> GetAdminLogChannel(IGuild guild) { public static async Task<ITextChannel?> GetAdminLogChannel(IGuild guild) {
var adminLogChannel = await ParseChannelNullable(Boyfriend.GetGuildConfig(guild).AdminLogChannel.ToString()); var adminLogChannel = await ParseChannelNullable(Boyfriend.GetGuildConfig(guild).AdminLogChannel.ToString()!);
return adminLogChannel as ITextChannel; return adminLogChannel as ITextChannel;
} }