neoarz/Syntrel
1
0
Fork 0
mirror of https://github.com/neoarz/Syntrel.git synced 2025-12-25 03:40:11 +01:00
Code Issues Packages Projects Releases Wiki Activity

chore: ruff formatting

Browse Source
This commit is contained in:
neoarz
2025-11-02 23:32:52 -05:00
parent 2ce2c69a87
commit 1eff6c9f53
91 changed files with 3824 additions and 2518 deletions
Download Patch File Download Diff File Expand all files Collapse all files

405
cogs/utilities/codepreview.py
View File

@@ -9,40 +9,44 @@ import json
def codepreview_command():
LANGUAGE_MAP = {
'.py': 'python',
'.js': 'js',
'.ts': 'ts',
'.java': 'java',
'.cpp': 'cpp',
'.c': 'c',
'.cs': 'cs',
'.go': 'go',
'.rs': 'rust',
'.rb': 'ruby',
'.php': 'php',
'.html': 'html',
'.css': 'css',
'.json': 'json',
'.xml': 'xml',
'.yaml': 'yaml',
'.yml': 'yaml',
'.ini': 'ini',
'.toml': 'toml',
'.lua': 'lua',
'.sh': 'bash',
'.md': 'markdown',
'.sql': 'sql',
'.diff': 'diff',
'.txt': '',
".py": "python",
".js": "js",
".ts": "ts",
".java": "java",
".cpp": "cpp",
".c": "c",
".cs": "cs",
".go": "go",
".rs": "rust",
".rb": "ruby",
".php": "php",
".html": "html",
".css": "css",
".json": "json",
".xml": "xml",
".yaml": "yaml",
".yml": "yaml",
".ini": "ini",
".toml": "toml",
".lua": "lua",
".sh": "bash",
".md": "markdown",
".sql": "sql",
".diff": "diff",
".txt": "",
}
async def send_embed(context, embed: discord.Embed, *, ephemeral: bool = False) -> None:
async def send_embed(
context, embed: discord.Embed, *, ephemeral: bool = False
) -> None:
interaction = getattr(context, "interaction", None)
if interaction is not None:
if interaction.response.is_done():
await interaction.followup.send(embed=embed, ephemeral=ephemeral)
else:
await interaction.response.send_message(embed=embed, ephemeral=ephemeral)
await interaction.response.send_message(
embed=embed, ephemeral=ephemeral
)
else:
await context.send(embed=embed)
@@ -50,7 +54,7 @@ def codepreview_command():
for ext, lang in LANGUAGE_MAP.items():
if filename.endswith(ext):
return lang
return ''
return ""
async def fetch_github_content(url):
try:
@@ -64,12 +68,12 @@ def codepreview_command():
async def fetch_pr_diff(owner, repo, pr_number):
try:
api_url = f'https://api.github.com/repos/{owner}/{repo}/pulls/{pr_number}'
api_url = f"https://api.github.com/repos/{owner}/{repo}/pulls/{pr_number}"
headers = {
'Accept': 'application/vnd.github.v3.diff',
'User-Agent': 'Discord-Bot'
"Accept": "application/vnd.github.v3.diff",
"User-Agent": "Discord-Bot",
}
async with aiohttp.ClientSession() as session:
async with session.get(api_url, headers=headers) as response:
if response.status == 200:
@@ -81,98 +85,95 @@ def codepreview_command():
async def fetch_pr_info(owner, repo, pr_number):
try:
api_url = f'https://api.github.com/repos/{owner}/{repo}/pulls/{pr_number}'
api_url = f"https://api.github.com/repos/{owner}/{repo}/pulls/{pr_number}"
headers = {
'Accept': 'application/vnd.github.v3+json',
'User-Agent': 'Discord-Bot'
"Accept": "application/vnd.github.v3+json",
"User-Agent": "Discord-Bot",
}
async with aiohttp.ClientSession() as session:
async with session.get(api_url, headers=headers) as response:
if response.status == 200:
pr_data = await response.json()
return {
'title': pr_data.get('title', ''),
'number': pr_data.get('number', pr_number),
'state': pr_data.get('state', ''),
'merged': pr_data.get('merged', False),
'additions': pr_data.get('additions', 0),
'deletions': pr_data.get('deletions', 0),
'changed_files': pr_data.get('changed_files', 0),
'user': pr_data.get('user', {}).get('login', ''),
'base_branch': pr_data.get('base', {}).get('ref', ''),
'head_branch': pr_data.get('head', {}).get('ref', '')
"title": pr_data.get("title", ""),
"number": pr_data.get("number", pr_number),
"state": pr_data.get("state", ""),
"merged": pr_data.get("merged", False),
"additions": pr_data.get("additions", 0),
"deletions": pr_data.get("deletions", 0),
"changed_files": pr_data.get("changed_files", 0),
"user": pr_data.get("user", {}).get("login", ""),
"base_branch": pr_data.get("base", {}).get("ref", ""),
"head_branch": pr_data.get("head", {}).get("ref", ""),
}
except Exception:
pass
return None
def parse_github_url(url):
pr_pattern = r'https://github\.com/([^/]+)/([^/]+)/pull/(\d+)(?:/files)?'
pr_pattern = r"https://github\.com/([^/]+)/([^/]+)/pull/(\d+)(?:/files)?"
pr_match = re.match(pr_pattern, url)
if pr_match:
owner, repo, pr_number = pr_match.groups()
return {
'type': 'pr',
'owner': owner,
'repo': repo,
'pr_number': pr_number
}
raw_pattern = r'https://raw\.githubusercontent\.com/([^/]+)/([^/]+)/([^/]+)/(.+?)$'
return {"type": "pr", "owner": owner, "repo": repo, "pr_number": pr_number}
raw_pattern = (
r"https://raw\.githubusercontent\.com/([^/]+)/([^/]+)/([^/]+)/(.+?)$"
)
raw_match = re.match(raw_pattern, url)
if raw_match:
owner, repo, branch, filepath = raw_match.groups()
return {
'type': 'file',
'owner': owner,
'repo': repo,
'branch': branch,
'filepath': filepath,
'raw_url': url,
'start_line': None,
'end_line': None
"type": "file",
"owner": owner,
"repo": repo,
"branch": branch,
"filepath": filepath,
"raw_url": url,
"start_line": None,
"end_line": None,
}
pattern = r'https://github\.com/([^/]+)/([^/]+)/blob/([^/]+)/(.+?)(?:#L(\d+)(?:-L(\d+))?)?$'
pattern = r"https://github\.com/([^/]+)/([^/]+)/blob/([^/]+)/(.+?)(?:#L(\d+)(?:-L(\d+))?)?$"
match = re.match(pattern, url)
if match:
owner, repo, branch, filepath, start_line, end_line = match.groups()
raw_url = f'https://raw.githubusercontent.com/{owner}/{repo}/{branch}/{filepath}'
raw_url = (
f"https://raw.githubusercontent.com/{owner}/{repo}/{branch}/{filepath}"
)
return {
'type': 'file',
'owner': owner,
'repo': repo,
'branch': branch,
'filepath': filepath,
'raw_url': raw_url,
'start_line': int(start_line) if start_line else None,
'end_line': int(end_line) if end_line else None
"type": "file",
"owner": owner,
"repo": repo,
"branch": branch,
"filepath": filepath,
"raw_url": raw_url,
"start_line": int(start_line) if start_line else None,
"end_line": int(end_line) if end_line else None,
}
return None
def extract_lines(content, start_line, end_line):
lines = content.split('\n')
lines = content.split("\n")
if start_line and end_line:
return '\n'.join(lines[start_line-1:end_line])
return "\n".join(lines[start_line - 1 : end_line])
elif start_line:
return lines[start_line-1] if start_line <= len(lines) else content
return lines[start_line - 1] if start_line <= len(lines) else content
return content
@commands.hybrid_command(
name="codepreview",
description="Preview code from GitHub URLs",
)
@app_commands.describe(
url="GitHub URL to preview code from"
)
@app_commands.describe(url="GitHub URL to preview code from")
async def codepreview(self, context, url: str = None):
if isinstance(context.channel, discord.DMChannel):
embed = discord.Embed(
@@ -180,8 +181,10 @@ def codepreview_command():
description="This command can only be used in servers.",
color=0xE02B2B,
)
embed.set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
embed.set_author(
name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp"
)
interaction = getattr(context, "interaction", None)
if interaction is not None:
if not interaction.response.is_done():
@@ -198,8 +201,10 @@ def codepreview_command():
description="The bot needs the `send messages` permission in this channel.",
color=0xE02B2B,
)
embed.set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
embed.set_author(
name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp"
)
interaction = getattr(context, "interaction", None)
if interaction is not None:
if not interaction.response.is_done():
@@ -209,12 +214,18 @@ def codepreview_command():
else:
await context.send(embed=embed, ephemeral=True)
return
if not url or not url.strip():
if context.message and context.message.reference and context.message.reference.resolved:
if (
context.message
and context.message.reference
and context.message.reference.resolved
):
replied_message = context.message.reference.resolved
if hasattr(replied_message, 'content') and replied_message.content:
github_pattern = r'https://(?:github\.com|raw\.githubusercontent\.com)/[^\s]+'
if hasattr(replied_message, "content") and replied_message.content:
github_pattern = (
r"https://(?:github\.com|raw\.githubusercontent\.com)/[^\s]+"
)
urls = re.findall(github_pattern, replied_message.content)
if urls:
url = urls[0]
@@ -223,7 +234,10 @@ def codepreview_command():
title="Error",
description="No GitHub URL found in the replied message.",
color=0xE02B2B,
).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
).set_author(
name="Utility",
icon_url="https://yes.nighty.works/raw/8VLDcg.webp",
)
await send_embed(context, embed, ephemeral=True)
return
else:
@@ -231,7 +245,10 @@ def codepreview_command():
title="Error",
description="The replied message has no content to extract GitHub URL from.",
color=0xE02B2B,
).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
).set_author(
name="Utility",
icon_url="https://yes.nighty.works/raw/8VLDcg.webp",
)
await send_embed(context, embed, ephemeral=True)
return
else:
@@ -239,23 +256,31 @@ def codepreview_command():
title="Error",
description="Please provide a GitHub URL or reply to a message containing a GitHub URL.",
color=0xE02B2B,
).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
).set_author(
name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp"
)
await send_embed(context, embed, ephemeral=True)
return
if not url.startswith('https://github.com/') and not url.startswith('https://raw.githubusercontent.com/'):
if not url.startswith("https://github.com/") and not url.startswith(
"https://raw.githubusercontent.com/"
):
embed = discord.Embed(
title="Error",
description="Please provide a valid GitHub URL.",
color=0xE02B2B,
).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
).set_author(
name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp"
)
await send_embed(context, embed, ephemeral=True)
return
# Check if bot has send messages permission before starting processing
try:
test_embed = discord.Embed(title="Testing permissions...", color=0x7289DA)
test_embed.set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
test_embed.set_author(
name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp"
)
await context.channel.send(embed=test_embed, delete_after=0.1)
except discord.Forbidden:
embed = discord.Embed(
@@ -263,8 +288,10 @@ def codepreview_command():
description="The bot needs the `send messages` permission to execute this command.",
color=0xE02B2B,
)
embed.set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
embed.set_author(
name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp"
)
interaction = getattr(context, "interaction", None)
if interaction is not None:
if not interaction.response.is_done():
@@ -274,135 +301,177 @@ def codepreview_command():
else:
await context.send(embed=embed, ephemeral=True)
return
parsed = parse_github_url(url)
if not parsed:
embed = discord.Embed(
title="Error",
description="Invalid GitHub URL format. Please provide a valid GitHub blob URL or PR URL.",
color=0xE02B2B,
).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
).set_author(
name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp"
)
await send_embed(context, embed, ephemeral=True)
return
if parsed.get('type') == 'pr':
pr_info = await fetch_pr_info(parsed['owner'], parsed['repo'], parsed['pr_number'])
diff_content = await fetch_pr_diff(parsed['owner'], parsed['repo'], parsed['pr_number'])
if parsed.get("type") == "pr":
pr_info = await fetch_pr_info(
parsed["owner"], parsed["repo"], parsed["pr_number"]
)
diff_content = await fetch_pr_diff(
parsed["owner"], parsed["repo"], parsed["pr_number"]
)
if not pr_info or not diff_content:
embed = discord.Embed(
title="Error",
description="Failed to fetch pull request information. The PR might not exist or be accessible.",
color=0xE02B2B,
).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
).set_author(
name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp"
)
await send_embed(context, embed, ephemeral=True)
return
pr_url = f"https://github.com/{parsed['owner']}/{parsed['repo']}/pull/{parsed['pr_number']}"
if pr_info['merged']:
pr_color = 0x6f42c1
if pr_info["merged"]:
pr_color = 0x6F42C1
pr_status = "Merged"
elif pr_info['state'] == 'open':
elif pr_info["state"] == "open":
pr_color = 0x57F287
pr_status = "Open"
else:
pr_color = 0xE02B2B
pr_status = "Closed"
embed = discord.Embed(
title=f"Pull Request #{pr_info['number']}: {pr_info['title'][:100]}",
description=f"**Repository:** [{parsed['owner']}/{parsed['repo']}]({pr_url})\n**Author:** {pr_info['user']}\n**Status:** {pr_status}",
color=pr_color,
)
embed.set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
embed.add_field(name="Changes", value=f"**+{pr_info['additions']}** / **-{pr_info['deletions']}**", inline=True)
embed.add_field(name="Files Changed", value=f"{pr_info['changed_files']}", inline=True)
embed.add_field(name="Branches", value=f"`{pr_info['base_branch']}` ← `{pr_info['head_branch']}`", inline=False)
embed.set_footer(text=f"Requested by {context.author.name}", icon_url=context.author.display_avatar.url)
embed.set_author(
name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp"
)
embed.add_field(
name="Changes",
value=f"**+{pr_info['additions']}** / **-{pr_info['deletions']}**",
inline=True,
)
embed.add_field(
name="Files Changed", value=f"{pr_info['changed_files']}", inline=True
)
embed.add_field(
name="Branches",
value=f"`{pr_info['base_branch']}` ← `{pr_info['head_branch']}`",
inline=False,
)
embed.set_footer(
text=f"Requested by {context.author.name}",
icon_url=context.author.display_avatar.url,
)
interaction = getattr(context, "interaction", None)
if interaction is not None and not interaction.response.is_done():
await interaction.response.defer(ephemeral=True)
await context.channel.send(embed=embed)
max_diff_length = 1900
max_lines = 100
diff_lines = diff_content.split('\n')
diff_lines = diff_content.split("\n")
if len(diff_lines) > max_lines:
diff_lines = diff_lines[:max_lines]
diff_lines.append(f"\n... ({len(diff_content.split(chr(10))) - max_lines} more lines omitted)")
diff_lines.append(
f"\n... ({len(diff_content.split(chr(10))) - max_lines} more lines omitted)"
)
current_chunk = ""
for line in diff_lines:
test_chunk = current_chunk + line + '\n'
test_chunk = current_chunk + line + "\n"
if len(test_chunk) + 10 > max_diff_length:
if current_chunk.strip():
remaining_lines = len(diff_lines) - len(current_chunk.split('\n'))
remaining_lines = len(diff_lines) - len(
current_chunk.split("\n")
)
if remaining_lines > 0:
current_chunk += f"\n... ({remaining_lines} more lines omitted)"
await context.channel.send(f"```diff\n{current_chunk.rstrip()}\n```")
current_chunk += (
f"\n... ({remaining_lines} more lines omitted)"
)
await context.channel.send(
f"```diff\n{current_chunk.rstrip()}\n```"
)
break
else:
current_chunk = test_chunk
else:
if current_chunk.strip():
await context.channel.send(f"```diff\n{current_chunk.rstrip()}\n```")
await context.channel.send(
f"```diff\n{current_chunk.rstrip()}\n```"
)
if interaction is not None:
try:
await interaction.delete_original_response()
except:
pass
return
content = await fetch_github_content(parsed['raw_url'])
content = await fetch_github_content(parsed["raw_url"])
if not content:
embed = discord.Embed(
title="Error",
description="Failed to fetch content from GitHub. The file might not exist or be accessible.",
color=0xE02B2B,
).set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
).set_author(
name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp"
)
await send_embed(context, embed, ephemeral=True)
return
if parsed['start_line']:
code = extract_lines(content, parsed['start_line'], parsed['end_line'])
if parsed["start_line"]:
code = extract_lines(content, parsed["start_line"], parsed["end_line"])
line_info = f" (Lines {parsed['start_line']}"
if parsed['end_line']:
if parsed["end_line"]:
line_info += f"-{parsed['end_line']}"
line_info += ")"
else:
code = content
line_info = ""
code_lines = code.split('\n')
code_lines = code.split("\n")
if len(code_lines) > 100:
code = '\n'.join(code_lines[:100])
code = "\n".join(code_lines[:100])
code += f"\n\n... ({len(code_lines) - 100} more lines omitted)"
filename = parsed['filepath'].split('/')[-1]
filename = parsed["filepath"].split("/")[-1]
language = get_language_from_filename(filename)
embed = discord.Embed(
title="Code Preview",
description=f"**Repository URL:** [{parsed['owner']}/{parsed['repo']}]({url})",
color=0x7289DA,
)
embed.set_author(name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp")
embed.add_field(name="File", value=f"`{parsed['filepath']}`{line_info}", inline=True)
embed.set_author(
name="Utility", icon_url="https://yes.nighty.works/raw/8VLDcg.webp"
)
embed.add_field(
name="File", value=f"`{parsed['filepath']}`{line_info}", inline=True
)
embed.add_field(name="Branch", value=f"`{parsed['branch']}`", inline=True)
embed.set_footer(text=f"Requested by {context.author.name}", icon_url=context.author.display_avatar.url)
embed.set_footer(
text=f"Requested by {context.author.name}",
icon_url=context.author.display_avatar.url,
)
code_block = f"```{language}\n{code}\n```"
if len(code_block) > 1990:
interaction = getattr(context, "interaction", None)
if interaction is not None:
@@ -411,19 +480,21 @@ def codepreview_command():
await context.channel.send(embed=embed)
else:
await context.channel.send(embed=embed)
max_code_length = 1980 - len(language) - 8
code_lines = code.split('\n')
code_lines = code.split("\n")
current_chunk = []
current_length = 0
for line in code_lines:
line_length = len(line) + 1
if current_length + line_length > max_code_length:
remaining_lines = len(code_lines) - len(current_chunk)
if remaining_lines > 0:
current_chunk.append(f"\n... ({remaining_lines} more lines omitted)")
chunk_text = '\n'.join(current_chunk)
current_chunk.append(
f"\n... ({remaining_lines} more lines omitted)"
)
chunk_text = "\n".join(current_chunk)
await context.channel.send(f"```{language}\n{chunk_text}\n```")
break
else:
@@ -431,9 +502,9 @@ def codepreview_command():
current_length += line_length
else:
if current_chunk:
chunk_text = '\n'.join(current_chunk)
chunk_text = "\n".join(current_chunk)
await context.channel.send(f"```{language}\n{chunk_text}\n```")
if interaction is not None:
try:
await interaction.delete_original_response()
@@ -454,4 +525,4 @@ def codepreview_command():
await context.channel.send(embed=embed)
await context.channel.send(code_block)
return codepreview
return codepreview