mirror of
https://github.com/TeamOctolings/Octobot.git
synced 2025-05-03 20:49:54 +03:00
Begin guild storage refactor
This commit is contained in:
parent
f0a6c8faff
commit
b79be8a876
8 changed files with 192 additions and 75 deletions
121
Boyfriend/Data/GuildData.cs
Normal file
121
Boyfriend/Data/GuildData.cs
Normal file
|
@ -0,0 +1,121 @@
|
|||
using System.Diagnostics;
|
||||
using System.Diagnostics.CodeAnalysis;
|
||||
using Discord.WebSocket;
|
||||
using Newtonsoft.Json;
|
||||
|
||||
namespace Boyfriend.Data;
|
||||
|
||||
public struct GuildData {
|
||||
public static readonly Dictionary<string, string> DefaultConfiguration = new() {
|
||||
{ "Prefix", "!" },
|
||||
{ "Lang", "en" },
|
||||
{ "ReceiveStartupMessages", "false" },
|
||||
{ "WelcomeMessage", "default" },
|
||||
{ "SendWelcomeMessages", "true" },
|
||||
{ "PublicFeedbackChannel", "0" },
|
||||
{ "PrivateFeedbackChannel", "0" },
|
||||
{ "StarterRole", "0" },
|
||||
{ "MuteRole", "0" },
|
||||
{ "RemoveRolesOnMute", "false" },
|
||||
{ "ReturnRolesOnRejoin", "false" },
|
||||
{ "EventStartedReceivers", "interested,role" },
|
||||
{ "EventNotificationRole", "0" },
|
||||
{ "EventNotificationChannel", "0" },
|
||||
{ "EventEarlyNotificationOffset", "0" }
|
||||
};
|
||||
|
||||
private static readonly Dictionary<ulong, GuildData> GuildDataDictionary = new();
|
||||
|
||||
public readonly Dictionary<string, string> GuildConfiguration;
|
||||
|
||||
public readonly Dictionary<ulong, MemberData> MemberData;
|
||||
|
||||
/*public static Dictionary<string, string> GetGuildConfig(ulong id) {
|
||||
if (GuildConfigDictionary.TryGetValue(id, out var cfg)) return cfg;
|
||||
|
||||
var path = $"config_{id}.json";
|
||||
|
||||
if (!File.Exists(path)) File.Create(path).Dispose();
|
||||
|
||||
var json = File.ReadAllText(path);
|
||||
var config = JsonConvert.DeserializeObject<Dictionary<string, string>>(json)
|
||||
?? new Dictionary<string, string>();
|
||||
|
||||
if (config.Keys.Count < GuildData.DefaultConfiguration.Keys.Count) {
|
||||
// ReSharper disable once ForeachCanBePartlyConvertedToQueryUsingAnotherGetEnumerator
|
||||
// Conversion will result in a lot of memory allocations
|
||||
foreach (var key in GuildData.DefaultConfiguration.Keys)
|
||||
if (!config.ContainsKey(key))
|
||||
config.Add(key, GuildData.DefaultConfiguration[key]);
|
||||
} else if (config.Keys.Count > GuildData.DefaultConfiguration.Keys.Count) {
|
||||
foreach (var key in config.Keys.Where(key => !GuildData.DefaultConfiguration.ContainsKey(key))) config.Remove(key);
|
||||
}
|
||||
|
||||
GuildConfigDictionary.Add(id, config);
|
||||
|
||||
return config;
|
||||
}*/
|
||||
|
||||
/*public static async Task WriteGuildConfigAsync(ulong id) {
|
||||
await File.WriteAllTextAsync($"config_{id}.json",
|
||||
JsonConvert.SerializeObject(GuildConfigDictionary[id], Formatting.Indented));
|
||||
|
||||
if (RemovedRolesDictionary.TryGetValue(id, out var removedRoles))
|
||||
await File.WriteAllTextAsync($"removedroles_{id}.json",
|
||||
JsonConvert.SerializeObject(removedRoles, Formatting.Indented));
|
||||
}*/
|
||||
[SuppressMessage("Performance", "CA1853:Unnecessary call to \'Dictionary.ContainsKey(key)\'")]
|
||||
// https://github.com/dotnet/roslyn-analyzers/issues/6377
|
||||
public GuildData(SocketGuild guild) {
|
||||
var id = guild.Id;
|
||||
if (GuildDataDictionary.TryGetValue(id, out var stored)) {
|
||||
this = stored;
|
||||
return;
|
||||
}
|
||||
|
||||
if (!Directory.Exists($"{id}")) Directory.CreateDirectory($"{id}");
|
||||
if (!Directory.Exists($"{id}/MemberData")) Directory.CreateDirectory($"{id}/MemberData");
|
||||
if (!File.Exists($"{id}/Configuration.json")) File.Create($"{id}/Configuration.json").Dispose();
|
||||
GuildConfiguration = JsonConvert.DeserializeObject<Dictionary<string, string>>($"{id}/Configuration.json") ??
|
||||
new Dictionary<string, string>();
|
||||
|
||||
// ReSharper disable twice ForeachCanBePartlyConvertedToQueryUsingAnotherGetEnumerator
|
||||
if (GuildConfiguration.Keys.Count < DefaultConfiguration.Keys.Count)
|
||||
foreach (var key in DefaultConfiguration.Keys)
|
||||
if (!GuildConfiguration.ContainsKey(key))
|
||||
GuildConfiguration.Add(key, DefaultConfiguration[key]);
|
||||
if (GuildConfiguration.Keys.Count > DefaultConfiguration.Keys.Count)
|
||||
foreach (var key in GuildConfiguration.Keys)
|
||||
if (!DefaultConfiguration.ContainsKey(key))
|
||||
GuildConfiguration.Remove(key);
|
||||
GuildConfiguration.TrimExcess();
|
||||
|
||||
MemberData = new Dictionary<ulong, MemberData>();
|
||||
foreach (var data in Directory.GetFiles($"{id}/MemberData")) {
|
||||
var deserialised = JsonConvert.DeserializeObject<MemberData>($"{id}/MemberData/{data}.json") ??
|
||||
throw new UnreachableException();
|
||||
MemberData.Add(deserialised.Id, deserialised);
|
||||
}
|
||||
|
||||
if (guild.MemberCount > MemberData.Count)
|
||||
foreach (var member in guild.Users) {
|
||||
if (MemberData.TryGetValue(member.Id, out var memberData)) {
|
||||
if (memberData is { IsInGuild: false, BannedUntil: > -1 } &&
|
||||
DateTimeOffset.Now.ToUnixTimeSeconds() - memberData.LeftAt.Last() >
|
||||
60 * 60 * 24 * 30) {
|
||||
File.Delete($"{id}/MemberData/{memberData.Id}.json");
|
||||
MemberData.Remove(memberData.Id);
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
var data = new MemberData(member);
|
||||
MemberData.Add(member.Id, data);
|
||||
File.WriteAllText($"{id}/MemberData/{data.Id}.json",
|
||||
JsonConvert.SerializeObject(data, Formatting.Indented));
|
||||
}
|
||||
|
||||
GuildDataDictionary.Add(id, this);
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue