108 lines
4.1 KiB
C#
108 lines
4.1 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Data;
|
|
using ChaosBot.Database.Entity;
|
|
using Microsoft.Extensions.Configuration;
|
|
using NLog;
|
|
|
|
namespace ChaosBot.Database.Repository
|
|
{
|
|
public static class ConfigurationRepository
|
|
{
|
|
public static IConfiguration AppSettingsHandler { get; set; }
|
|
|
|
private static readonly string ServersTable = "ServerConfigurationFlags";
|
|
|
|
/// <summary>
|
|
/// Get global configuration option
|
|
/// </summary>
|
|
/// <param name="configurationFlag"></param>
|
|
/// <typeparam name="T"></typeparam>
|
|
/// <returns></returns>
|
|
public static T GetValue<T>(string configurationFlag)
|
|
{
|
|
return AppSettingsHandler.GetValue<T>(configurationFlag);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Get guild specific configuration option
|
|
/// </summary>
|
|
/// <param name="configurationFlag"></param>
|
|
/// <param name="guildId"></param>
|
|
/// <typeparam name="T"></typeparam>
|
|
/// <returns></returns>
|
|
public static T GetValue<T>(string configurationFlag, ulong guildId)
|
|
{
|
|
if (Program._logger == null) return AppSettingsHandler.GetValue<T>($"Servers:{guildId}:{configurationFlag}");
|
|
|
|
string valueSerialized = GetValueRaw(configurationFlag, Convert.ToInt64(guildId));
|
|
if (valueSerialized != null)
|
|
return ServerConfigurationFlag<T>.Deserialize(valueSerialized);
|
|
return AppSettingsHandler.GetValue<T>($"Servers:{guildId}:{configurationFlag}");
|
|
}
|
|
|
|
public static string GetValueRaw(string configurationFlag, long guildId)
|
|
{
|
|
Dictionary<string, object> filterColumns = new Dictionary<string, object>();
|
|
filterColumns.Add("key", configurationFlag);
|
|
filterColumns.Add("guildId", guildId);
|
|
DataTable valueTable = Controller.SelectQuery(ServersTable, filterColumns: filterColumns);
|
|
if (valueTable.Rows.Count == 1)
|
|
{
|
|
return valueTable.Rows[0]["serializedValue"].ToString();
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Grab configuration from appsettings.json, method does not look at the database
|
|
/// </summary>
|
|
/// <param name="configurationFlag"></param>
|
|
/// <returns></returns>
|
|
public static IConfigurationSection GetSection(string configurationFlag)
|
|
{
|
|
return AppSettingsHandler.GetSection(configurationFlag);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Get guild specific configuration option
|
|
/// </summary>
|
|
/// <param name="configurationFlag"></param>
|
|
/// <param name="guildId"></param>
|
|
/// <typeparam name="T"></typeparam>
|
|
/// <returns></returns>
|
|
public static void SetValue<T>(string configurationFlag, ulong guildId, T value)
|
|
{
|
|
SetValueRaw(configurationFlag, Convert.ToInt64(guildId), ServerConfigurationFlag<T>.Serialize(value));
|
|
}
|
|
|
|
public static void SetValueRaw(string configurationFlag, long guildId, string serializedValue)
|
|
{
|
|
try
|
|
{
|
|
Dictionary<string, object> filterColumns = new Dictionary<string, object>();
|
|
filterColumns.Add("key", configurationFlag);
|
|
filterColumns.Add("guildId", guildId);
|
|
|
|
Dictionary<string, object> newValue = new Dictionary<string, object>();
|
|
newValue.Add("key", configurationFlag);
|
|
newValue.Add("guildId", guildId);
|
|
newValue.Add("serializedValue", serializedValue);
|
|
|
|
DataTable valueTable = Controller.SelectQuery(ServersTable, "COUNT(*)", filterColumns);
|
|
if ((long) valueTable.Rows[0]["COUNT(*)"] == 1)
|
|
{
|
|
Controller.UpdateQuery(ServersTable, newValue, filterColumns);
|
|
return;
|
|
}
|
|
|
|
Controller.InsertQuery(ServersTable, newValue);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Program._logger.Fatal($"ConfigurationRepository.SetValueRaw: Exception [{ex}] thrown, <[{ex.Message}]>.");
|
|
}
|
|
}
|
|
}
|
|
} |