BirthdayBot/Configuration.cs

using BirthdayBot.Data;
using CommandLine;
using CommandLine.Text;
using Newtonsoft.Json.Linq;
using Npgsql;
using System.Diagnostics.CodeAnalysis;
using System.Reflection;
using System.Text.RegularExpressions;

namespace BirthdayBot;

/// <summary>
/// Loads and holds configuration values.
/// </summary>
class Configuration {
    const string KeySqlHost = "SqlHost";
    const string KeySqlUsername = "SqlUsername";
    const string KeySqlPassword = "SqlPassword";
    const string KeySqlDatabase = "SqlDatabase";
    const string KeyShardRange = "ShardRange";

    public string BotToken { get; }
    public string? DBotsToken { get; }
    public bool QuitOnFails { get; }

    public int ShardStart { get; }
    public int ShardAmount { get; }
    public int ShardTotal { get; }

    public Configuration(string[] args) {
        var cmdline = CmdLineOpts.Parse(args);

        // Looks for configuration file
        var confPath = Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location) + Path.DirectorySeparatorChar;
        confPath += cmdline.Config!;
        if (!File.Exists(confPath)) throw new Exception("Settings file not found in path: " + confPath);

        var jc = JObject.Parse(File.ReadAllText(confPath));

        BotToken = ReadConfKey<string>(jc, nameof(BotToken), true);
        DBotsToken = ReadConfKey<string>(jc, nameof(DBotsToken), false);
        QuitOnFails = ReadConfKey<bool?>(jc, nameof(QuitOnFails), false) ?? false;

        ShardTotal = cmdline.ShardTotal ?? ReadConfKey<int?>(jc, nameof(ShardTotal), false) ?? 1;
        if (ShardTotal < 1) throw new Exception($"'{nameof(ShardTotal)}' must be a positive integer.");

        string shardRangeInput = cmdline.ShardRange ?? ReadConfKey<string>(jc, KeyShardRange, false);
        if (!string.IsNullOrWhiteSpace(shardRangeInput)) {
            Regex srPicker = new(@"(?<low>\d{1,2})[-,]{1}(?<high>\d{1,2})");
            var m = srPicker.Match(shardRangeInput);
            if (m.Success) {
                ShardStart = int.Parse(m.Groups["low"].Value);
                int high = int.Parse(m.Groups["high"].Value);
                ShardAmount = high - (ShardStart - 1);
            } else {
                throw new Exception($"Shard range not properly formatted in '{KeyShardRange}'.");
            }
        } else {
            // Default: this instance handles all shards
            ShardStart = 0;
            ShardAmount = ShardTotal;
        }

        var sqlhost = ReadConfKey<string>(jc, KeySqlHost, false) ?? "localhost"; // Default to localhost
        var sqluser = ReadConfKey<string>(jc, KeySqlUsername, false);
        var sqlpass = ReadConfKey<string>(jc, KeySqlPassword, false);
        if (string.IsNullOrWhiteSpace(sqluser) || string.IsNullOrWhiteSpace(sqlpass))
            throw new Exception("'SqlUsername', 'SqlPassword' must be specified.");
        var csb = new NpgsqlConnectionStringBuilder() {
            Host = sqlhost,
            Username = sqluser,
            Password = sqlpass,
            ApplicationName = $"ClientShard{ShardStart}+{ShardAmount}"
        };
        var sqldb = ReadConfKey<string>(jc, KeySqlDatabase, false);
        if (sqldb != null) csb.Database = sqldb; // Optional database setting
        Database.DBConnectionString = csb.ToString();
    }

    private static T? ReadConfKey<T>(JObject jc, string key, [DoesNotReturnIf(true)] bool failOnEmpty) {
        if (jc.ContainsKey(key)) return jc[key]!.Value<T>();
        if (failOnEmpty) throw new Exception($"'{key}' must be specified.");
        return default;
    }

    private class CmdLineOpts {
        [Option('c', "config", Default = "settings.json",
            HelpText = "Custom path to instance configuration, relative from executable directory.")]
        public string? Config { get; set; }

        [Option("shardtotal",
            HelpText = "Total number of shards online. MUST be the same for all instances.\n"
            + "This value overrides the config file value.")]
        public int? ShardTotal { get; set; }

        [Option("shardrange", HelpText = "Shard range for this instance to handle.\n"
            + "This value overrides the config file value.")]
        public string? ShardRange { get; set; }

        public static CmdLineOpts Parse(string[] args) {
            // Do not automatically print help message
            var clp = new Parser(c => c.HelpWriter = null);

            CmdLineOpts? result = null;
            var r = clp.ParseArguments<CmdLineOpts>(args);
            r.WithParsed(parsed => result = parsed);
            r.WithNotParsed(err => {
                    var ht = HelpText.AutoBuild(r);
                    Console.WriteLine(ht.ToString());
                    Environment.Exit((int)Program.ExitCodes.BadCommand);
                });
            return result!;
        }
    }
}