initial code upload

main.py

@@ -0,0 +1,146 @@
+import discord
+from discord.ext import commands
+import redditwarp.SYNC
+import json
+import os
+import aiohttp
+import asyncio
+
+# Load configuration
+config_path = os.path.join(os.path.dirname(__file__), "config.json")
+with open(config_path, "r") as f:
+    config = json.load(f)
+
+bot_token = config.get("bot_token")
+prefix = config.get("prefix", ";")
+command_name = config.get("command_name", "rediddy")
+comment_limit = config.get("comment_limit", 2)
+delete_user_msg = config.get("delete_user_message_on_dismiss", True)
+uptime_kuma_url = config.get("uptime_kuma_url", "")
+uptime_kuma_interval = config.get("uptime_kuma_interval", 60)
+
+# Color mapping
+COLORS = {
+    "blue": discord.Color.blue(),
+    "green": discord.Color.green(),
+    "red": discord.Color.red(),
+    "gold": discord.Color.gold(),
+    "purple": discord.Color.purple(),
+}
+sub_color = COLORS.get(
+    config.get("embed_color_submission", "blue"), discord.Color.blue()
+)
+comment_color = COLORS.get(
+    config.get("embed_color_comments", "green"), discord.Color.green()
+)
+
+# Setup Discord bot intents
+intents = discord.Intents.default()
+intents.message_content = True
+
+bot = commands.Bot(command_prefix=prefix, intents=intents)
+reddit_client = redditwarp.SYNC.Client()
+
+
+@bot.event
+async def on_ready():
+    print(f"Logged in as {bot.user} (ID: {bot.user.id})")
+    print("------")
+    if uptime_kuma_url:
+        bot.loop.create_task(uptime_kuma_heartbeat())
+
+
+async def uptime_kuma_heartbeat():
+    async with aiohttp.ClientSession() as session:
+        while not bot.is_closed():
+            try:
+                async with session.get(uptime_kuma_url) as resp:
+                    if resp.status == 200:
+                        print("Uptime Kuma heartbeat sent.")
+                    else:
+                        print(f"Uptime Kuma error: {resp.status}")
+            except Exception as e:
+                print(f"Uptime Kuma connection failed: {e}")
+            await asyncio.sleep(uptime_kuma_interval)
+
+
+class RediddyView(discord.ui.View):
+    def __init__(self, author_msg: discord.Message, submission_url: str):
+        super().__init__(timeout=60)
+        self.author_msg = author_msg
+        self.add_item(discord.ui.Button(label="Open Submission", url=submission_url))
+
+    @discord.ui.button(label="Dismiss", style=discord.ButtonStyle.danger)
+    async def dismiss(
+        self, interaction: discord.Interaction, button: discord.ui.Button
+    ):
+        if delete_user_msg:
+            try:
+                await self.author_msg.delete()
+            except discord.Forbidden:
+                pass  # May not have permission to delete user messages
+        await interaction.message.delete()
+
+
+@bot.command(name=command_name)
+async def fetch_reddit_submission(ctx, subreddit_name: str, *, query: str):
+    """Searches Reddit and returns the top post and comments in embeds."""
+    try:
+        # Search for the subreddit name to get a valid subreddit object/ID
+        sr_list = reddit_client.p.subreddit.search_names(subreddit_name)
+        if not sr_list:
+            await ctx.send(f"Subreddit '{subreddit_name}' not found.")
+            return
+
+        target_sr = sr_list[0]
+
+        # Search for submissions within that subreddit
+        submissions = reddit_client.p.submission.search(target_sr, query, 1, sort="top")
+
+        found = False
+        for submission_brief in submissions:
+            found = True
+            # Get full submission details
+            submission = reddit_client.p.submission.get(submission_brief.id)
+            title = submission.d.get("title", "Reddit Post")
+            selftext = submission.d.get("selftext", "No content.")
+            permalink = f"https://www.reddit.com{submission.d.get('permalink')}"
+
+            # Submission Embed
+            main_embed = discord.Embed(
+                title=title,
+                url=permalink,
+                description=selftext[:2000] if selftext else "No content.",
+                color=sub_color,
+            )
+            main_embed.set_author(name=f"r/{target_sr}")
+
+            embeds = [main_embed]
+
+            # Fetch comments based on limit
+            try:
+                comment_tree = reddit_client.p.comment_tree.fetch(
+                    submission.id, limit=comment_limit, sort="top"
+                )
+                for i, child in enumerate(comment_tree.children[:comment_limit]):
+                    comment_body = child.value.body
+                    comment_embed = discord.Embed(
+                        title=f"Top Comment #{i+1}",
+                        description=comment_body[:1000],
+                        color=comment_color,
+                    )
+                    embeds.append(comment_embed)
+            except Exception as e:
+                print(f"Error fetching comments: {e}")
+
+            view = RediddyView(ctx.message, permalink)
+            await ctx.send(embeds=embeds, view=view)
+
+        if not found:
+            await ctx.send(f"No results found for '{query}' in {target_sr}.")
+
+    except Exception as e:
+        await ctx.send(f"An error occurred: {e}")
+
+
+bot.run(bot_token)