Message forwarding with appropriate commands

2016-10-10 10:06:06 +02:00
parent 6f0a78613e
commit e92db040be
5 changed files with 199 additions and 4 deletions
--- a/src/NadekoBot/Modules/Administration/Commands/DMForwardCommands.cs
+++ b/src/NadekoBot/Modules/Administration/Commands/DMForwardCommands.cs
@@ -0,0 +1,89 @@
+using Discord;
+using Discord.Commands;
+using NadekoBot.Attributes;
+using NadekoBot.Services;
+using NadekoBot.Services.Database;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace NadekoBot.Modules.Administration
+{
+    public partial class Administration
+    {
+        [Group]
+        public class DMForwardCommands
+        {
+            private static bool ForwardDMs { get; set; }
+            private static bool ForwardDMsToAllOwners { get; set; }
+            
+            static DMForwardCommands()
+            {
+                using (var uow = DbHandler.UnitOfWork())
+                {
+                    var config = uow.BotConfig.GetOrCreate();
+                    ForwardDMs = config.ForwardMessages;
+                    ForwardDMsToAllOwners = config.ForwardToAllOwners;
+                }
+            }
+
+            [NadekoCommand, Usage, Description, Aliases]
+            [OwnerOnly]
+            public async Task ForwardMessages(IUserMessage imsg)
+            {
+                var channel = imsg.Channel;
+
+                using (var uow = DbHandler.UnitOfWork())
+                {
+                    var config = uow.BotConfig.GetOrCreate();
+                    ForwardDMs = config.ForwardMessages = !config.ForwardMessages;
+                    uow.Complete();
+                }
+                if (ForwardDMs)
+                    await channel.SendMessageAsync("`I will forward DMs from now on.`").ConfigureAwait(false);
+                else
+                    await channel.SendMessageAsync("`I will stop forwarding DMs.`").ConfigureAwait(false);
+            }
+
+            [NadekoCommand, Usage, Description, Aliases]
+            [OwnerOnly]
+            public async Task ForwardToAll(IUserMessage imsg)
+            {
+                var channel = imsg.Channel;
+
+                using (var uow = DbHandler.UnitOfWork())
+                {
+                    var config = uow.BotConfig.GetOrCreate();
+                    ForwardDMsToAllOwners = config.ForwardToAllOwners = !config.ForwardToAllOwners;
+                    uow.Complete();
+                }
+                if (ForwardDMsToAllOwners)
+                    await channel.SendMessageAsync("`I will forward DMs to all owners.`").ConfigureAwait(false);
+                else
+                    await channel.SendMessageAsync("`I will forward DMs only to the first owner.`").ConfigureAwait(false);
+
+            }
+
+            public static async Task HandleDMForwarding(IMessage msg, List<IDMChannel> ownerChannels)
+            {
+                if (ForwardDMs && ownerChannels.Any())
+                {
+                    var toSend = $"`I received a message from {msg.Author} ({msg.Author.Id})`: {msg.Content}";
+                    if (ForwardDMsToAllOwners)
+                    {
+                        var msgs = await Task.WhenAll(ownerChannels.Where(ch => ch.Recipient.Id != msg.Author.Id)
+                                                                   .Select(ch => ch.SendMessageAsync(toSend))).ConfigureAwait(false);
+                    }
+                    else
+                    {
+                        var firstOwnerChannel = ownerChannels.First();
+                        if (firstOwnerChannel.Recipient.Id != msg.Author.Id)
+                            try { await firstOwnerChannel.SendMessageAsync(msg.Content).ConfigureAwait(false); } catch { }
+                    }
+                }
+            }
+        }
+    }
+}