Merge pull request #80 from iayushanand/master

added tickets, lyrics cmds and made afk lvl 30+

Author: iayushanand

.env.example

@@ -1,3 +1,4 @@
 TOKEN = 
 MODMAIL_WEBHOOK_URL =
-GEMINI_API_KEY = 
+GEMINI_API_KEY = 
+GITHUB_TOKEN = 

cogs/misc.py

@@ -6,15 +6,15 @@
 import random
 import re
 from typing import TYPE_CHECKING, Optional
-
+from discord import app_commands
 import button_paginator as pg
 import contextlib
 import discord
 from discord.ext import commands
-from ext.helpers import Spotify, grouper, ordinal_suffix_of, gemini_split_string
+from ext.helpers import Spotify, grouper, ordinal_suffix_of, gemini_split_string, get_lyrics, find_surrounding_lyrics, filter_banned_words
 from ext.http import Http
 from ext.ui.view import Piston
-
+import time
 import google.generativeai as genai
 import button_paginator as pg
 from googletrans import Translator
@@ -23,6 +23,44 @@
     from ext.models import CodingBot
 
 
+
+# I don't like the way its implemented but can't find a better way to do it so if someone can make this command better, that'll be great!
+
+@app_commands.context_menu(name = "translate")
+async def translate(interaction: discord.Interaction, message: discord.Message):
+    await interaction.response.defer(ephemeral = True)
+    trans = message.content
+    try:
+        translated = Translator(service_urls=[
+            'translate.google.com',
+            'translate.google.co.kr'
+        ]).translate(trans)
+    except Exception as e:
+        raise e
+    
+    embed = discord.Embed()
+
+    _from = translated.src.upper()
+    _to = translated.dest.upper()
+    
+    embed.add_field(
+        name = f"Original ({_from})",
+        value = trans,
+        inline = False
+    )
+    embed.add_field(
+        name = f"Translated ({_to})",
+        value = translated.text,
+        inline = False
+    )
+
+    await interaction.followup.send(
+        embed = embed,
+        ephemeral = True
+    )
+
+
+
 class Miscellaneous(commands.Cog, command_attrs=dict(hidden=False)):
     hidden = False
 
@@ -40,6 +78,9 @@ async def cog_check(self, ctx: commands.Context[CodingBot]) -> bool:
             return True
         await ctx.send("Please use commands in the server instead of dms")
         return False
+    
+    async def cog_load(self):
+        self.bot.tree.add_command(translate)
 
     @commands.hybrid_command(
         name="retry",
@@ -66,6 +107,7 @@ async def retry(self, ctx: commands.Context[CodingBot]):
         name="afk", aliases=["afk-set", "set-afk"], help="Sets your afk"
     )
     @commands.cooldown(1, 10, commands.BucketType.member)
+    @commands.has_role(734283436637814844) # lvl 30+
     async def afk(
         self, ctx: commands.Context[CodingBot], *, reason: Optional[str] = None
     ):
@@ -496,46 +538,42 @@ async def on_timeout():
 
         await paginator.start()
 
-    @commands.command(name = "translate")
-    async def translate(self, ctx: commands.Context, *, text: str = None):
-        if not ctx.message.reference and not text:
-            return await ctx.reply("Please reply a message or provide a text to translate!")
-
-        if text:
-            trans = text
-            message = ctx.message
-        else:
-            message = await ctx.channel.fetch_message(ctx.message.reference.message_id)
-            trans = message.content
 
-        try:
-            translated = Translator(service_urls=[
-                'translate.google.com',
-                'translate.google.co.kr'
-            ]).translate(trans)
-        except Exception as e:
-            raise e
+    @commands.hybrid_command(name = "lyric", aliases = ["lyrics"])
+    async def lyric(self, ctx: commands.Context, member: discord.Member = None):
+        member = member or ctx.author
 
-        embed = discord.Embed()
+        spotify_activity = None
+        for activity in member.activities:
+            if isinstance(activity, discord.Spotify):
+                spotify_activity = activity
+                break
+
+        if not spotify_activity:
+            await ctx.send(f"{member.display_name} is not listening to Spotify.")
+            return
+    
+        else:
+            duration = time.time()-spotify_activity.start.timestamp()
+            song_title = spotify_activity.title
+            song_artist = spotify_activity.artist
+            lyr = get_lyrics(song_title, song_artist)
+            if not lyr:
+                return await ctx.send(f"Lyrics not found for song - {song_title} - {song_artist}")
+            if lyr[1] == 1:
+                lyr = "\n".join(find_surrounding_lyrics(lyr[0], int(duration)))
+            else:
+                lyr = "\n".join(lyr.splitlines()[0:5])
+
+            lyr = filter_banned_words(lyr)
+            embed = discord.Embed(description = lyr, title = f"{song_title} - {song_artist}", color = spotify_activity.color)
+            embed.set_author(name = ctx.author, icon_url = ctx.author.avatar.url)
+            embed.set_thumbnail(url = spotify_activity.album_cover_url)
+
+            await ctx.send(embed = embed)
+
 
-        _from = translated.src.upper()
-        _to = translated.dest.upper()
-        
-        embed.add_field(
-            name = f"Original ({_from})",
-            value = trans,
-            inline = False
-        )
-        embed.add_field(
-            name = f"Translated ({_to})",
-            value = translated.text,
-            inline = False
-        )
 
-        await message.reply(
-            embed = embed,
-            allowed_mentions=discord.AllowedMentions.none()
-        )
 
 
 async def setup(bot: CodingBot):

cogs/ticket.py

@@ -0,0 +1,30 @@
+import discord
+from discord.ext import commands
+
+from ext.ui.view import CreateButton, CloseButton, TrashButton
+
+class TicketCog(commands.Cog):
+    def __init__(self, bot: commands.Bot):
+        self.bot = bot
+
+    @commands.Cog.listener(name = "on_ready")
+    async def before_ready(self):
+        self.bot.add_view(CreateButton())
+        self.bot.add_view(CloseButton())
+        self.bot.add_view(TrashButton())
+        await self.bot.change_presence(activity=discord.Activity(type = discord.ActivityType.listening, name = "Doghouse Game's Heart 😳"))
+        print(f"Logged in as: {self.bot.user.name}")
+    
+    @commands.command(name="ticket")
+    @commands.has_permissions(administrator=True)
+    async def ticket(self, ctx):
+        await ctx.send(
+            embed = discord.Embed(
+                description="🎫 **Click  on the button below to create a ticket**\nIf you need any help regarding punishments, roles, or you just have a general question, feel free to create a ticket and a staff member will get to you shortly!\nOpening a ticket without a valid reason will get you warned/blacklisted.\n\n__**Do not open support tickets for Coding Help. Doing so will get you warned.**__",
+                color = 0x8b6ffc
+            ),
+            view = CreateButton()
+        )
+
+async def setup(bot):
+    await bot.add_cog(TicketCog(bot))

ext/consts.py

@@ -132,6 +132,20 @@ def release_format(self):
                             UNIQUE(user_id, guild_id)
                         );"""
 
+
+TICKETS_CONFIG_SCHEMA = """
+            CREATE TABLE IF NOT EXISTS tickets (
+                message_id BIGINT PRIMARY KEY,
+                ticket_id BIGINT,
+                opened_by BIGINT,
+                closed_by BIGINT,
+                opened_at BIGINT,
+                closed_at BIGINT,
+                reason TEXT
+            )
+        """
+
+
 HELP_COMMAND = """
             Help command for Coding Bot
 
@@ -157,5 +171,12 @@ def release_format(self):
 MODMAIL_CLOSED = 1144842686579359837 # conch: 1144839107353256017
 STAFF_UPDATE_CHANNEL_ID = 1124612885365133412
 
+
+TICKET_REPO = "WhoIsConch/tcrtickets"
+TICKET_HANDLER_ROLE_ID = 788799215417032705
+OPEN_TICKET_CATEGORY = 788797663377883147
+CLOSED_TICKET_CATEGORY = 1287954797286391860
+TICKET_LOG_CHANNEL = 829936676021075999
+
 # mods pls fill this up
 TICKET_CATEGORY_ID = 0 # <--- this still hasn't been resolved but I don't think we need it ~ ayu

ext/helpers.py

@@ -11,16 +11,19 @@
 import traceback
 from io import BytesIO
 from typing import TYPE_CHECKING, Any, Callable, List, Optional, Tuple
-
+from lrclib import LrcLibAPI
 import aiohttp
 import discord
 import humanize
 from bs4 import BeautifulSoup
 from discord.ext import tasks
 from PIL import Image, ImageDraw, ImageFilter, ImageFont
 from cbvx import iml
+from ext.consts import TCR_STAFF_ROLE_ID, TICKET_REPO
 
-from ext.consts import TCR_STAFF_ROLE_ID
+import chat_exporter
+from github import Github
+import os
 
 if TYPE_CHECKING:
     from ext.models import CodingBot
@@ -781,3 +784,93 @@ def invert_string(text):
 
 def gemini_split_string(string, chunk_size=1000):
     return [string[i:i+chunk_size] for i in range(0, len(string), chunk_size)]
+
+
+
+
+# GET TRANSCRIPT
+async def get_transcript(member: discord.Member, channel: discord.TextChannel):
+    export = await chat_exporter.export(channel=channel)
+    file_name=f"{member.id}.html"
+    with open(f"storage/tickets/{file_name}", "w", encoding="utf-8") as f:
+        f.write(export)
+
+# UPLOAD TO GITHUB
+def upload(file_path: str, member_name: str, file_name: str):
+    github = Github(os.getenv("GITHUB_TOKEN"))
+    repo = github.get_repo(TICKET_REPO)
+    repo.create_file(
+        path=f"templates/tickets/{file_name}.html",
+        message="Ticket Log for {0}".format(member_name),
+        branch="main",
+        content=open(f"{file_path}","r",encoding="utf-8").read()
+    )
+    os.remove(file_path)
+
+    return file_name
+
+
+
+def get_lyrics(name: str, artist: str) -> str:
+    user_agent = (
+        "Mozilla/5.0 (X11; Linux x86_64; rv:125.0) Gecko/20100101 Firefox/125.0"
+    )
+
+    # Prepare the API
+    api = LrcLibAPI(user_agent=user_agent)
+    id = api.search_lyrics(track_name=name, artist_name=artist)
+
+    # Check if lyrics are available
+    if len(id) != 0:
+        lyrics = api.get_lyrics_by_id(id[0].id)
+        s_lyrics = lyrics.synced_lyrics, 1
+        if not s_lyrics:
+            s_lyrics = lyrics.plain_lyrics, 0
+        return s_lyrics
+    else:
+        return None
+
+
+def parse_timestamp_to_seconds(timestamp):
+    minutes, seconds = map(float, timestamp.split(':'))
+    return minutes * 60 + seconds
+
+def find_surrounding_lyrics(lyrics, target_seconds):
+    pattern = r'\[(\d{2}:\d{2}\.\d{2})\] (.+)'
+    matches = re.findall(pattern, lyrics)
+
+    parsed_lyrics = []
+
+    for match in matches:
+        timestamp, lyric = match
+        time_in_seconds = parse_timestamp_to_seconds(timestamp)
+        parsed_lyrics.append((time_in_seconds, lyric))
+
+    closest_index = min(range(len(parsed_lyrics)), key=lambda i: abs(parsed_lyrics[i][0] - target_seconds))
+
+    start_index = max(closest_index - 2, 0)
+    end_index = min(closest_index + 3, len(parsed_lyrics))
+
+    surrounding_lyrics = [parsed_lyrics[i][1] for i in range(start_index, end_index)]
+
+    return surrounding_lyrics
+
+
+
+def filter_banned_words(text: str):
+    with open("storage/banned_word.txt") as f:
+        words = f.read().split(", ")
+    
+    new_text = text
+    for word in words:
+        pattern = re.compile(re.escape(word), re.IGNORECASE)
+        
+        def censor(match):
+            matched_word = match.group()
+            word_length = len(matched_word) // 2
+            stars = "\*" * word_length
+            return f"{stars}{matched_word[word_length:]}"
+        
+        new_text = pattern.sub(censor, new_text)
+
+    return new_text

ext/models.py

@@ -36,6 +36,7 @@
     THANK_DATA_CONFIG_SCHEMA,
     THANK_INFO_CONFIG_SCHEMA,
     WARNINGS_CONFIG_SCHEMA,
+    TICKETS_CONFIG_SCHEMA,
     VERSION,
     Version,
 )
@@ -133,6 +134,7 @@ async def __aenter__(self) -> "Database":
         self.conn["afk"] = await aiosqlite.connect("./database/afk.db")
         self.conn["thanks"] = await aiosqlite.connect("./database/thanks.db")
         self.conn["metrics"] = await aiosqlite.connect("./database/metrics.db")
+        self.conn["tickets"] = await aiosqlite.connect("./database/tickets.db")
         await self.init_dbs()
         self.bot.logger.info("Finished creating all connections")
         return self
@@ -164,6 +166,9 @@ async def init_dbs(self):
 
         async with self.cursor("metrics") as cursor:
             await cursor.execute(MESSAGE_METRIC_SCHEMA)
+        
+        async with self.cursor("tickets") as cursor:
+            await cursor.execute(TICKETS_CONFIG_SCHEMA)
 
         await self.commit()