Merge pull request #36 from bobloy/infochannel-develop

Infochannel initial commit and leaver readme update

README.md

@@ -16,7 +16,8 @@ Cog Function
 | forcemention | **Alpha** | <details><summary>Mentions unmentionable roles</summary>Very simple cog, mention doesn't persist</details> |
 | hangman | **Alpha** | <details><summary>Play a game of hangman</summary>Some visual glitches and needs more customization</details> |
 | howdoi | **Incomplete** | <details><summary>Ask coding questions and get results from StackExchange</summary>Not yet functional</details> |
-| leaver | **Alpha** | <details><summary>Send a message in a channel when a user leaves the server</summary>Just released, please report bugs</details> |
+| leaver | **Beta** | <details><summary>Send a message in a channel when a user leaves the server</summary>Seems to be functional, please report any bugs or suggestions</details> |
+| infochannel | **Alpha** | <details><summary>Create a channel to display server info</summary>Just released, please report bugs</details> |
 | lovecalculator | **Alpha** | <details><summary>Calculate the love between two users</summary>[Snap-Ons] Just updated to V3</details> |
 | lseen | **Alpha** | <details><summary>Track when a member was last online</summary>Alpha release, please report bugs</details> |
 | nudity | **Incomplete** | <details><summary>Checks for NSFW images posted in non-NSFW channels</summary>Library this is based on has a bug, waiting for author to merge my PR</details> |

infochannel/__init__.py

@@ -0,0 +1,5 @@
+from .infochannel import InfoChannel
+
+
+def setup(bot):
+    bot.add_cog(InfoChannel(bot))

infochannel/info.json

@@ -0,0 +1,19 @@
+{
+  "author": [
+    "Bobloy"
+  ],
+  "bot_version": [
+    3,
+    0,
+    0
+  ],
+  "description": "Create a channel with updating server info",
+  "hidden": false,
+  "install_msg": "Thank you for installing InfoChannel. Get started with `[p]load infochannel`, then `[p]help InfoChannel`",
+  "requirements": [],
+  "short": "Updating server info channel",
+  "tags": [
+    "bobloy",
+    "utils"
+  ]
+}

infochannel/infochannel.py

@@ -0,0 +1,121 @@
+from typing import Any
+
+import discord
+from redbot.core import Config, commands, checks
+from redbot.core.bot import Red
+
+Cog: Any = getattr(commands, "Cog", object)
+
+
+class InfoChannel(Cog):
+    """
+    Create a channel with updating server info
+
+    Less important information about the cog
+    """
+
+    def __init__(self, bot: Red):
+        self.bot = bot
+        self.config = Config.get_conf(
+            self, identifier=731101021116710497110110101108, force_registration=True
+        )
+
+        default_guild = {
+            "channel_id": None,
+            "category_id": None,
+            "member_count": True,
+            "channel_count": False,
+        }
+
+        self.config.register_guild(**default_guild)
+
+    @commands.command()
+    @checks.admin()
+    async def infochannel(self, ctx: commands.Context):
+        """
+        Toggle info channel for this server
+        """
+
+        def check(m):
+            return (
+                m.content.upper() in ["Y", "YES", "N", "NO"]
+                and m.channel == ctx.channel
+                and m.author == ctx.author
+            )
+
+        guild: discord.Guild = ctx.guild
+        channel_id = await self.config.guild(guild).channel_id()
+        if channel_id is not None:
+            channel: discord.VoiceChannel = guild.get_channel(channel_id)
+        else:
+            channel: discord.VoiceChannel = None
+
+        if channel_id is not None and channel is None:
+            await ctx.send("Info channel has been deleted, recreate it?")
+        elif channel_id is None:
+            await ctx.send("Enable info channel on this server?")
+        else:
+            await ctx.send("Info channel is {}. Delete it?".format(channel.mention))
+
+        msg = await self.bot.wait_for("message", check=check)
+
+        if msg.content.upper() in ["N", "NO"]:
+            await ctx.send("Cancelled")
+            return
+
+        if channel is None:
+            await self.make_infochannel(guild)
+        else:
+            await self.delete_infochannel(guild, channel)
+
+        if not await ctx.tick():
+            await ctx.send("Done!")
+
+    async def make_infochannel(self, guild: discord.Guild):
+        category: discord.CategoryChannel = await guild.create_category("────Server Stats────")
+
+        overwrites = {guild.default_role: discord.PermissionOverwrite(connect=False)}
+
+        channel = await guild.create_voice_channel(
+            "Placeholder", category=category, reason="InfoChannel make", overwrites=overwrites
+        )
+
+        await self.config.guild(guild).channel_id.set(channel.id)
+        await self.config.guild(guild).category_id.set(category.id)
+
+        await self.update_infochannel(guild)
+
+    async def delete_infochannel(self, guild: discord.Guild, channel: discord.VoiceChannel):
+        await channel.category.delete(reason="InfoChannel delete")
+        await channel.delete(reason="InfoChannel delete")
+        await self.config.guild(guild).clear()
+
+    async def update_infochannel(self, guild: discord.Guild):
+        guild_data = await self.config.guild(guild).all()
+
+        channel_id = guild_data["channel_id"]
+        if channel_id is None:
+            return
+
+        channel: discord.VoiceChannel = guild.get_channel(channel_id)
+
+        if channel is None:
+            return
+
+        name = ""
+        if guild_data["member_count"]:
+            name += "Members: {} ".format(guild.member_count)
+
+        if guild_data["channel_count"]:
+            name += "─ Channels: {}".format(len(guild.channels))
+
+        if name == "":
+            name = "Stats not enabled"
+
+        await channel.edit(reason="InfoChannel update", name=name)
+
+    async def on_member_join(self, member: discord.Member):
+        await self.update_infochannel(member.guild)
+
+    async def on_member_remove(self, member: discord.Member):
+        await self.update_infochannel(member.guild)