Initial split of the modules

2017-09-30 00:46:33 +02:00
parent cdc2c43913
commit 599245b1ca
499 changed files with 469 additions and 256 deletions
--- a/NadekoBot.Core/Services/Impl/Localization.cs
+++ b/NadekoBot.Core/Services/Impl/Localization.cs
@ -0,0 +1,146 @@
+using System.Collections.Concurrent;
+using System.Collections.Generic;
+using System.Globalization;
+using System.Linq;
+using Discord;
+using NLog;
+using NadekoBot.Services.Database.Models;
+using NadekoBot.Common;
+using Newtonsoft.Json;
+using System.IO;
+
+namespace NadekoBot.Services.Impl
+{
+    public class Localization : ILocalization
+    {
+        private readonly Logger _log;
+        private readonly DbService _db;
+
+        public ConcurrentDictionary<ulong, CultureInfo> GuildCultureInfos { get; }
+        public CultureInfo DefaultCultureInfo { get; private set; } = CultureInfo.CurrentCulture;
+
+        private static readonly Dictionary<string, CommandData> _commandData;
+        
+        static Localization()
+        {
+            _commandData = JsonConvert.DeserializeObject<Dictionary<string, CommandData>>(
+                File.ReadAllText("./data/command_strings.json"));
+        }
+
+        private Localization() { }
+        public Localization(IBotConfigProvider bcp, IEnumerable<GuildConfig> gcs, DbService db)
+        {
+            _log = LogManager.GetCurrentClassLogger();
+
+            var cultureInfoNames = gcs.ToDictionary(x => x.GuildId, x => x.Locale);
+            var defaultCulture = bcp.BotConfig.Locale;
+
+            _db = db;
+
+            if (string.IsNullOrWhiteSpace(defaultCulture))
+                DefaultCultureInfo = new CultureInfo("en-US");
+            else
+            {
+                try
+                {
+                    DefaultCultureInfo = new CultureInfo(defaultCulture);
+                }
+                catch
+                {
+                    _log.Warn("Unable to load default bot's locale/language. Using en-US.");
+                    DefaultCultureInfo = new CultureInfo("en-US");
+                }
+            }
+            GuildCultureInfos = new ConcurrentDictionary<ulong, CultureInfo>(cultureInfoNames.ToDictionary(x => x.Key, x =>
+              {
+                  CultureInfo cultureInfo = null;
+                  try
+                  {
+                      if (x.Value == null)
+                          return null;
+                      cultureInfo = new CultureInfo(x.Value);
+                  }
+                  catch { }
+                  return cultureInfo;
+              }).Where(x => x.Value != null));
+        }
+
+        public void SetGuildCulture(IGuild guild, CultureInfo ci) =>
+            SetGuildCulture(guild.Id, ci);
+
+        public void SetGuildCulture(ulong guildId, CultureInfo ci)
+        {
+            if (ci == DefaultCultureInfo)
+            {
+                RemoveGuildCulture(guildId);
+                return;
+            }
+
+            using (var uow = _db.UnitOfWork)
+            {
+                var gc = uow.GuildConfigs.For(guildId, set => set);
+                gc.Locale = ci.Name;
+                uow.Complete();
+            }
+
+            GuildCultureInfos.AddOrUpdate(guildId, ci, (id, old) => ci);
+        }
+
+        public void RemoveGuildCulture(IGuild guild) => 
+            RemoveGuildCulture(guild.Id);
+
+        public void RemoveGuildCulture(ulong guildId) {
+
+            if (GuildCultureInfos.TryRemove(guildId, out var _))
+            {
+                using (var uow = _db.UnitOfWork)
+                {
+                    var gc = uow.GuildConfigs.For(guildId, set => set);
+                    gc.Locale = null;
+                    uow.Complete();
+                }
+            }
+        }
+
+        public void SetDefaultCulture(CultureInfo ci)
+        {
+            using (var uow = _db.UnitOfWork)
+            {
+                var bc = uow.BotConfig.GetOrCreate();
+                bc.Locale = ci.Name;
+                uow.Complete();
+            }
+            DefaultCultureInfo = ci;
+        }
+
+        public void ResetDefaultCulture() =>
+            SetDefaultCulture(CultureInfo.CurrentCulture);
+
+        public CultureInfo GetCultureInfo(IGuild guild) =>
+            GetCultureInfo(guild?.Id);
+
+        public CultureInfo GetCultureInfo(ulong? guildId)
+        {
+            if (guildId == null)
+                return DefaultCultureInfo;
+            CultureInfo info = null;
+            GuildCultureInfos.TryGetValue(guildId.Value, out info);
+            return info ?? DefaultCultureInfo;
+        }
+
+        public static CommandData LoadCommand(string key)
+        {
+            _commandData.TryGetValue(key, out var toReturn);
+
+            if (toReturn == null)
+                return new CommandData
+                {
+                    Cmd = key,
+                    Desc = key,
+                    Usage = new[] { key },
+                };
+
+            return toReturn;
+        }
+    }
+}