nadekobot/src/NadekoBot/Services/IEmbedBuilderService.cs

using System;
using Discord;
using Discord.Commands;
using NadekoBot.Common.Configs;
using NadekoBot.Extensions;

namespace NadekoBot.Services
{
    public interface IEmbedBuilderService
    {
        IEmbedBuilder Create(ICommandContext ctx = null);
        IEmbedBuilder Create(EmbedBuilder eb);
    }

    public class EmbedBuilderService : IEmbedBuilderService, INService
    {
        private readonly BotConfigService _botConfigService;

        public EmbedBuilderService(BotConfigService botConfigService)
        {
            _botConfigService = botConfigService;
        }
        
        public IEmbedBuilder Create(ICommandContext ctx = null) 
            => new DiscordEmbedBuilderWrapper(_botConfigService.Data);
        
        public IEmbedBuilder Create(EmbedBuilder embed) 
            => new DiscordEmbedBuilderWrapper(_botConfigService.Data, embed);
    }
    
    public sealed class DiscordEmbedBuilderWrapper : IEmbedBuilder
    {
        private readonly BotConfig _botConfig;
        private EmbedBuilder _embed;

        public DiscordEmbedBuilderWrapper(in BotConfig botConfig, EmbedBuilder embed = null)
        {
            _botConfig = botConfig;
            _embed = embed ?? new EmbedBuilder();
        }

        public IEmbedBuilder WithDescription(string desc)
            => Wrap(_embed.WithDescription(desc));

        public IEmbedBuilder WithTitle(string title)
            => Wrap(_embed.WithTitle(title));

        public IEmbedBuilder AddField(string title, object value, bool isInline = false)
            => Wrap(_embed.AddField(title, value, isInline));

        public IEmbedBuilder WithFooter(string text, string iconUrl = null)
            => Wrap(_embed.WithFooter(text, iconUrl));

        public IEmbedBuilder WithAuthor(string name, string iconUrl = null, string url = null)
            => Wrap(_embed.WithAuthor(name, iconUrl, url));

        public IEmbedBuilder WithUrl(string url)
            => Wrap(_embed.WithAuthor(url));

        public IEmbedBuilder WithImageUrl(string url)
            => Wrap(_embed.WithAuthor(url));

        public IEmbedBuilder WithThumbnailUrl(string url)
            => Wrap(_embed.WithThumbnailUrl(url));
        
        public IEmbedBuilder WithColor(EmbedColor color)
            => color switch
            {
                EmbedColor.Ok => Wrap(_embed.WithColor(_botConfig.Color.Ok.ToDiscordColor())),
                EmbedColor.Pending => Wrap(_embed.WithColor(_botConfig.Color.Pending.ToDiscordColor())),
                EmbedColor.Error => Wrap(_embed.WithColor(_botConfig.Color.Error.ToDiscordColor())),
                _ => throw new ArgumentOutOfRangeException(nameof(color), "Unsupported EmbedColor type")
            };

        public Embed Build()
            => _embed.Build();

        private IEmbedBuilder Wrap(EmbedBuilder eb)
        {
            _embed = eb;
            return this;
        }
    }
}