From c56ccc19ef9e9ad075cda4803a1654d6bc8b7fac Mon Sep 17 00:00:00 2001
From: Mayuki Sawatari <mayuki+github@misuzilla.org>
Date: Thu, 28 Nov 2019 12:02:33 +0900
Subject: [PATCH] feat: Add Extensions to create a broadcaster from IGroup.

---
 sandbox/Sandbox.Hosting/Program.cs  | 30 +++++++++++--
 src/MagicOnion/Server/Hubs/Group.cs | 67 +++++++++++++++++++++++++++++
 2 files changed, 93 insertions(+), 4 deletions(-)

diff --git a/sandbox/Sandbox.Hosting/Program.cs b/sandbox/Sandbox.Hosting/Program.cs
index e094f2985..58c75454a 100644
--- a/sandbox/Sandbox.Hosting/Program.cs
+++ b/sandbox/Sandbox.Hosting/Program.cs
@@ -60,11 +60,19 @@ static async Task Main(string[] args)
             var result = await clientMyService.HelloAsync();
             var result2 = await clientManagementService.FooBarAsync();
 
-            var clientHub = StreamingHubClient.Connect<IMyHub, IMyHubReceiver>(new Channel("localhost", 12345, creds), null);
+            var clientHub = StreamingHubClient.Connect<IMyHub, IMyHubReceiver>(new Channel("localhost", 12345, creds), new MyHubReceiver());
             var result3 = await clientHub.HelloAsync();
 
             await hostTask;
         }
+
+        class MyHubReceiver : IMyHubReceiver
+        {
+            public void OnNantoka(string value)
+            {
+                Console.WriteLine(value);
+            }
+        }
     }
 
     public class MyStreamingHubFilterAttribute : StreamingHubFilterAttribute
@@ -144,14 +152,28 @@ public interface IMyHub : IStreamingHub<IMyHub, IMyHubReceiver>
     {
         Task<string> HelloAsync();
     }
+
     public interface IMyHubReceiver
-    { }
+    {
+        void OnNantoka(string value);
+    }
 
     public class MyHub : StreamingHubBase<IMyHub, IMyHubReceiver>, IMyHub
     {
-        public Task<string> HelloAsync()
+        public async Task<string> HelloAsync()
         {
-            return Task.FromResult("Konnnichiwa!");
+            var group = await this.Group.AddAsync("Nantoka");
+            group.CreateBroadcaster<IMyHubReceiver>().OnNantoka("BroadcastAll");
+            group.CreateBroadcasterTo<IMyHubReceiver>(Context.ContextId).OnNantoka("BroadcastTo(Self)");
+            group.CreateBroadcasterTo<IMyHubReceiver>(Guid.NewGuid()).OnNantoka("BroadcastTo(Non-self)");
+            group.CreateBroadcasterTo<IMyHubReceiver>(new[] { Guid.NewGuid(), Guid.NewGuid() }).OnNantoka("BroadcastTo(Non-self, Non-self)");
+            group.CreateBroadcasterTo<IMyHubReceiver>(new[] { Context.ContextId, Guid.NewGuid() }).OnNantoka("BroadcastTo(Self, Non-self)");
+            group.CreateBroadcasterExcept<IMyHubReceiver>(Context.ContextId).OnNantoka("BroadcastExcept(Self)");
+            group.CreateBroadcasterExcept<IMyHubReceiver>(Guid.NewGuid()).OnNantoka("BroadcastExcept(Non-self)");
+            group.CreateBroadcasterExcept<IMyHubReceiver>(new[] { Guid.NewGuid(), Guid.NewGuid() }).OnNantoka("BroadcastExcept(Non-self, Non-self)");
+            group.CreateBroadcasterExcept<IMyHubReceiver>(new[] { Context.ContextId, Guid.NewGuid() }).OnNantoka("BroadcastExcept(Self, Non-self)");
+
+            return "Konnnichiwa!";
         }
     }
 }
diff --git a/src/MagicOnion/Server/Hubs/Group.cs b/src/MagicOnion/Server/Hubs/Group.cs
index 1e1327fb8..b09429f34 100644
--- a/src/MagicOnion/Server/Hubs/Group.cs
+++ b/src/MagicOnion/Server/Hubs/Group.cs
@@ -207,4 +207,71 @@ public void Remove(Guid id)
             storage.TryRemove(id, out _);
         }
     }
+
+    public static class GroupBroadcastExtensions
+    {
+        /// <summary>
+        /// Create a receiver proxy from the group. Can be use to broadcast messages to all clients.
+        /// </summary>
+        /// <typeparam name="TReceiver"></typeparam>
+        /// <param name="group"></param>
+        /// <returns></returns>
+        public static TReceiver CreateBroadcaster<TReceiver>(this IGroup group)
+        {
+            var type = DynamicBroadcasterBuilder<TReceiver>.BroadcasterType;
+            return (TReceiver) Activator.CreateInstance(type, group);
+        }
+
+        /// <summary>
+        /// Create a receiver proxy from the group. Can be use to broadcast messages to all clients excepts one.
+        /// </summary>
+        /// <typeparam name="TReceiver"></typeparam>
+        /// <param name="group"></param>
+        /// <param name="except"></param>
+        /// <returns></returns>
+        public static TReceiver CreateBroadcasterExcept<TReceiver>(this IGroup group, Guid except)
+        {
+            var type = DynamicBroadcasterBuilder<TReceiver>.BroadcasterType_ExceptOne;
+            return (TReceiver) Activator.CreateInstance(type, new object[] {group, except});
+        }
+
+        /// <summary>
+        /// Create a receiver proxy from the group. Can be use to broadcast messages to all clients excepts some clients.
+        /// </summary>
+        /// <typeparam name="TReceiver"></typeparam>
+        /// <param name="group"></param>
+        /// <param name="excepts"></param>
+        /// <returns></returns>
+        public static TReceiver CreateBroadcasterExcept<TReceiver>(this IGroup group, Guid[] excepts)
+        {
+            var type = DynamicBroadcasterBuilder<TReceiver>.BroadcasterType_ExceptMany;
+            return (TReceiver) Activator.CreateInstance(type, new object[] {group, excepts});
+        }
+
+        /// <summary>
+        /// Create a receiver proxy from the group. Can be use to broadcast messages to one client.
+        /// </summary>
+        /// <typeparam name="TReceiver"></typeparam>
+        /// <param name="group"></param>
+        /// <param name="toConnectionId"></param>
+        /// <returns></returns>
+        public static TReceiver CreateBroadcasterTo<TReceiver>(this IGroup group, Guid toConnectionId)
+        {
+            var type = DynamicBroadcasterBuilder<TReceiver>.BroadcasterType_ToOne;
+            return (TReceiver) Activator.CreateInstance(type, new object[] { group, toConnectionId });
+        }
+
+        /// <summary>
+        /// Create a receiver proxy from the group. Can be use to broadcast messages to some clients.
+        /// </summary>
+        /// <typeparam name="TReceiver"></typeparam>
+        /// <param name="group"></param>
+        /// <param name="toConnectionIds"></param>
+        /// <returns></returns>
+        public static TReceiver CreateBroadcasterTo<TReceiver>(this IGroup group, Guid[] toConnectionIds)
+        {
+            var type = DynamicBroadcasterBuilder<TReceiver>.BroadcasterType_ToMany;
+            return (TReceiver) Activator.CreateInstance(type, new object[] { group, toConnectionIds });
+        }
+    }
 }