nadekobot/NadekoBot.Core/Services/Impl/BotCredentials.cs

using Discord;
using Microsoft.Extensions.Configuration;
using NadekoBot.Common;
using Newtonsoft.Json;
using System;
using System.Collections.Immutable;
using System.IO;
using System.Linq;
using NadekoBot.Core.Common;
using Serilog;

namespace NadekoBot.Core.Services.Impl
{
    public class BotCredentials : IBotCredentials
    {
        public string GoogleApiKey { get; }
        public string MashapeKey { get; }
        public string Token { get; }

        public ImmutableArray<ulong> OwnerIds { get; }

        public string OsuApiKey { get; }
        public string CleverbotApiKey { get; }
        public RestartConfig RestartCommand { get; }
        public DBConfig Db { get; }
        public int TotalShards { get; }
        public string CarbonKey { get; }

        private readonly string _credsFileName = Path.Combine(Directory.GetCurrentDirectory(), "credentials.json");
        public string PatreonAccessToken { get; }
        public string ShardRunCommand { get; }
        public string ShardRunArguments { get; }
        public int ShardRunPort { get; }

        public string PatreonCampaignId { get; }

        public string TwitchClientId { get; }

        public string VotesUrl { get; }
        public string VotesToken { get; }
        public string BotListToken { get; }
        public string RedisOptions { get; }
        public string LocationIqApiKey { get; }
        public string TimezoneDbApiKey { get; }
        public string CoinmarketcapApiKey { get; }

        public BotCredentials()
        {
            try
            {
                File.WriteAllText("./credentials_example.json",
                    JsonConvert.SerializeObject(new CredentialsModel(), Formatting.Indented));
            }
            catch
            {
            }

            if (!File.Exists(_credsFileName))
                Log.Warning(
                    $"credentials.json is missing. Attempting to load creds from environment variables prefixed with 'NadekoBot_'. Example is in {Path.GetFullPath("./credentials_example.json")}");
            try
            {
                var configBuilder = new ConfigurationBuilder();
                configBuilder.AddJsonFile(_credsFileName, true)
                    .AddEnvironmentVariables("NadekoBot_");

                var data = configBuilder.Build();

                Token = data[nameof(Token)];
                if (string.IsNullOrWhiteSpace(Token))
                {
                    Log.Error(
                        "Token is missing from credentials.json or Environment variables. Add it and restart the program.");
                    Helpers.ReadErrorAndExit(5);
                }

                OwnerIds = data.GetSection("OwnerIds").GetChildren().Select(c => ulong.Parse(c.Value))
                    .ToImmutableArray();
                GoogleApiKey = data[nameof(GoogleApiKey)];
                MashapeKey = data[nameof(MashapeKey)];
                OsuApiKey = data[nameof(OsuApiKey)];
                PatreonAccessToken = data[nameof(PatreonAccessToken)];
                PatreonCampaignId = data[nameof(PatreonCampaignId)] ?? "334038";
                ShardRunCommand = data[nameof(ShardRunCommand)];
                ShardRunArguments = data[nameof(ShardRunArguments)];
                CleverbotApiKey = data[nameof(CleverbotApiKey)];
                LocationIqApiKey = data[nameof(LocationIqApiKey)];
                TimezoneDbApiKey = data[nameof(TimezoneDbApiKey)];
                CoinmarketcapApiKey = data[nameof(CoinmarketcapApiKey)];
                if (string.IsNullOrWhiteSpace(CoinmarketcapApiKey))
                {
                    CoinmarketcapApiKey = "e79ec505-0913-439d-ae07-069e296a6079";
                }

                if (!string.IsNullOrWhiteSpace(data[nameof(RedisOptions)]))
                    RedisOptions = data[nameof(RedisOptions)];
                else
                    RedisOptions = "127.0.0.1,syncTimeout=3000";

                VotesToken = data[nameof(VotesToken)];
                VotesUrl = data[nameof(VotesUrl)];
                BotListToken = data[nameof(BotListToken)];

                var restartSection = data.GetSection(nameof(RestartCommand));
                var cmd = restartSection["cmd"];
                var args = restartSection["args"];
                if (!string.IsNullOrWhiteSpace(cmd))
                    RestartCommand = new RestartConfig(cmd, args);

                if (Environment.OSVersion.Platform == PlatformID.Unix)
                {
                    if (string.IsNullOrWhiteSpace(ShardRunCommand))
                        ShardRunCommand = "dotnet";
                    if (string.IsNullOrWhiteSpace(ShardRunArguments))
                        ShardRunArguments = "run -c Release --no-build -- {0} {1}";
                }
                else //windows
                {
                    if (string.IsNullOrWhiteSpace(ShardRunCommand))
                        ShardRunCommand = "NadekoBot.exe";
                    if (string.IsNullOrWhiteSpace(ShardRunArguments))
                        ShardRunArguments = "{0} {1}";
                }

                var portStr = data[nameof(ShardRunPort)];
                if (string.IsNullOrWhiteSpace(portStr))
                    ShardRunPort = new NadekoRandom().Next(5000, 6000);
                else
                    ShardRunPort = int.Parse(portStr);

                if (!int.TryParse(data[nameof(TotalShards)], out var ts))
                    ts = 0;
                TotalShards = ts < 1 ? 1 : ts;

                CarbonKey = data[nameof(CarbonKey)];
                var dbSection = data.GetSection("db");
                Db = new DBConfig(string.IsNullOrWhiteSpace(dbSection["Type"])
                        ? "sqlite"
                        : dbSection["Type"],
                    string.IsNullOrWhiteSpace(dbSection["ConnectionString"])
                        ? "Data Source=data/NadekoBot.db"
                        : dbSection["ConnectionString"]);

                TwitchClientId = data[nameof(TwitchClientId)];
                if (string.IsNullOrWhiteSpace(TwitchClientId))
                {
                    TwitchClientId = "67w6z9i09xv2uoojdm9l0wsyph4hxo6";
                }
            }
            catch (Exception ex)
            {
                Log.Error("JSON serialization has failed. Fix your credentials file and restart the bot.");
                Log.Fatal(ex.ToString());
                Helpers.ReadErrorAndExit(6);
            }
        }

        /// <summary>
        /// No idea why this thing exists
        /// </summary>
        private class CredentialsModel : IBotCredentials
        {
            public string Token { get; set; } = "";

            public ulong[] OwnerIds { get; set; } = new ulong[]
            {
                105635576866156544
            };

            public string GoogleApiKey { get; set; } = "";
            public string MashapeKey { get; set; } = "";
            public string OsuApiKey { get; set; } = "";
            public string SoundCloudClientId { get; set; } = "";
            public string CleverbotApiKey { get; } = "";
            public string CarbonKey { get; set; } = "";
            public DBConfig Db { get; set; } = new DBConfig("sqlite", "Data Source=data/NadekoBot.db");
            public int TotalShards { get; set; } = 1;
            public string PatreonAccessToken { get; set; } = "";
            public string PatreonCampaignId { get; set; } = "334038";
            public string RestartCommand { get; set; } = null;

            public string ShardRunCommand { get; set; } = "";
            public string ShardRunArguments { get; set; } = "";
            public int? ShardRunPort { get; set; } = null;

            public string BotListToken { get; set; }
            public string TwitchClientId { get; set; }
            public string VotesToken { get; set; }
            public string VotesUrl { get; set; }
            public string RedisOptions { get; set; }
            public string LocationIqApiKey { get; set; }
            public string TimezoneDbApiKey { get; set; }
            public string CoinmarketcapApiKey { get; set; }

            [JsonIgnore] ImmutableArray<ulong> IBotCredentials.OwnerIds => throw new NotImplementedException();

            [JsonIgnore] RestartConfig IBotCredentials.RestartCommand => throw new NotImplementedException();

            public bool IsOwner(IUser u)
            {
                throw new NotImplementedException();
            }
        }

        public bool IsOwner(IUser u) => OwnerIds.Contains(u.Id);
    }
}