Skip to content

Commit

Permalink
src/webhooks.py: Add webhooks for uf-mil-electrical github
Browse files Browse the repository at this point in the history
  • Loading branch information
@cbrxyz
cbrxyz committed Jul 20, 2024
1 parent 73c60ba commit e5cbd09
Show file tree
Hide file tree
Showing 2 changed files with 102 additions and 35 deletions.
47 changes: 36 additions & 11 deletions src/bot.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -92,11 +92,13 @@ class MILBot(commands.Bot):
general_channel: discord.TextChannel
member_services_channel: discord.TextChannel
errors_channel: discord.TextChannel
github_updates_channel: discord.TextChannel
software_github_channel: discord.TextChannel
software_projects_channel: discord.TextChannel
electrical_category_channel: discord.CategoryChannel
software_category_channel: discord.CategoryChannel
operations_leaders_channel: discord.TextChannel
software_leaders_channel: discord.TextChannel
electrical_leaders_channel: discord.TextChannel

# Emojis
loading_emoji: str
Expand Down Expand Up @@ -294,12 +296,42 @@ async def fetch_vars(self) -> None:
assert isinstance(software_leaders_channel, discord.TextChannel)
self.software_leaders_channel = software_leaders_channel

github_updates_channel = discord.utils.get(
electrical_leaders_channel = discord.utils.get(
self.active_guild.text_channels,
name="electrical-leadership",
)
assert isinstance(electrical_leaders_channel, discord.TextChannel)
self.electrical_leaders_channel = electrical_leaders_channel

software_category_channel = discord.utils.get(
self.active_guild.categories,
name="Software",
)
assert isinstance(software_category_channel, discord.CategoryChannel)
self.software_category_channel = software_category_channel

electrical_category_channel = discord.utils.get(
self.active_guild.categories,
name="Electrical",
)
assert isinstance(electrical_category_channel, discord.CategoryChannel)
self.electrical_category_channel = electrical_category_channel

el_github_updates_channel = discord.utils.get(
self.active_guild.text_channels,
name="github-updates",
category=self.software_category_channel,
)
assert isinstance(el_github_updates_channel, discord.TextChannel)
self.software_github_channel = el_github_updates_channel

el_github_updates_channel = discord.utils.get(
self.active_guild.text_channels,
name="github-updates",
category=self.electrical_category_channel,
)
assert isinstance(github_updates_channel, discord.TextChannel)
self.github_updates_channel = github_updates_channel
assert isinstance(el_github_updates_channel, discord.TextChannel)
self.electrical_github_channel = el_github_updates_channel

errors_channel = discord.utils.get(
self.active_guild.text_channels,
Expand All @@ -315,13 +347,6 @@ async def fetch_vars(self) -> None:
assert isinstance(software_projects_channel, discord.TextChannel)
self.software_projects_channel = software_projects_channel

software_category_channel = discord.utils.get(
self.active_guild.categories,
name="Software",
)
assert isinstance(software_category_channel, discord.CategoryChannel)
self.software_category_channel = software_category_channel

# Roles
egn4912_role = discord.utils.get(
self.active_guild.roles,
Expand Down
90 changes: 66 additions & 24 deletions src/webhooks.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
import datetime
from typing import TYPE_CHECKING

import discord
from discord.ext import commands
from discord.ext.ipc.objects import ClientPayload
from discord.ext.ipc.server import Server
Expand Down Expand Up @@ -57,6 +58,26 @@ def natural_wrap(self, text: str) -> str:
return text[: text.index("\n")][:1000] + "..."
return text[:1000]

def updates_channel(self, repository_or_login: dict | str) -> discord.TextChannel:
login = (
repository_or_login
if isinstance(repository_or_login, str)
else repository_or_login["owner"]["login"]
)
if login.startswith("uf-mil-electrical"):
return self.bot.electrical_github_channel
return self.bot.software_github_channel

def leaders_channel(self, repository_or_login: dict | str) -> discord.TextChannel:
login = (
repository_or_login
if isinstance(repository_or_login, str)
else repository_or_login["owner"]["login"]
)
if login.startswith("uf-mil-electrical"):
return self.bot.electrical_leaders_channel
return self.bot.software_leaders_channel

@Server.route()
async def push(self, payload: ClientPayload):
gh = payload.github_data
Expand All @@ -72,6 +93,7 @@ async def push(self, payload: ClientPayload):
repo = f"[{gh['repository']['full_name']}]({self.url(gh['repository'], html=True)})"
compare = f"[diff]({gh['compare']})"
commit_count = len(gh["commits"])
updates_channel = self.updates_channel(gh["repository"])
if gh["ref"] == "refs/heads/master" or gh["ref"] == "refs/heads/main":
if commit_count == 1:
by_statement = (
Expand All @@ -80,7 +102,7 @@ async def push(self, payload: ClientPayload):
else ""
)
message = f"\"{self.natural_wrap(gh['head_commit']['message'])}\""
await self.bot.github_updates_channel.send(
await updates_channel.send(
f"{name} {pushed} a commit{by_statement} to {branch} in {repo} ({compare}): {message}",
)
else:
Expand All @@ -89,7 +111,7 @@ async def push(self, payload: ClientPayload):
for commit in gh["commits"]
]
formatted_commits_str = "\n".join(formatted_commits)
await self.bot.github_updates_channel.send(
await updates_channel.send(
f"{name} {pushed} {commit_count} commits to {branch} in {repo} ({compare}):\n{formatted_commits_str}",
)

Expand All @@ -100,7 +122,8 @@ async def star_created(self, payload: ClientPayload):
# [User A](link) starred [repo_name](link)
name = f"[{await self.real_name(gh['sender']['login'])}]({self.url(gh['sender'],html=True)})"
repo = f"[{gh['repository']['full_name']}]({self.url(gh['repository'], html=True)})"
await self.bot.github_updates_channel.send(f"{name} added a star to {repo}")
updates_channel = self.updates_channel(gh["repository"])
await updates_channel.send(f"{name} added a star to {repo}")

@Server.route()
async def issues_opened(self, payload: ClientPayload):
Expand All @@ -113,7 +136,8 @@ async def issues_opened(self, payload: ClientPayload):
issue = f"[#{gh['issue']['number']}]({self.url(gh['issue'], html=True)})"
repo = f"[{gh['repository']['full_name']}]({self.url(gh['repository'], html=True)})"
title = f"\"{gh['issue']['title']}\""
await self.bot.github_updates_channel.send(
updates_channel = self.updates_channel(gh["repository"])
await updates_channel.send(
f"{name} opened issue {issue} in {repo}: {title}",
)

Expand All @@ -133,7 +157,8 @@ async def issues_closed(self, payload: ClientPayload):
else "not planned"
)
title = f"\"{gh['issue']['title']}\""
await self.bot.github_updates_channel.send(
updates_channel = self.updates_channel(gh["repository"])
await updates_channel.send(
f"{name} closed issue {issue} as {state} in {repo}: {title}",
)

Expand All @@ -151,7 +176,8 @@ async def organization_member_invited(self, payload: ClientPayload):
teams = ", ".join(
[f"[{team['name']}]({self.url(team, html=True)})" for team in teams_resp],
)
await self.bot.software_leaders_channel.send(
updates_channel = self.updates_channel(gh["organization"]["login"])
await updates_channel.send(
f"{name} invited {invited} to {org} in the following teams: {{{teams}}}",
)

Expand All @@ -162,7 +188,8 @@ async def organization_member_added(self, payload: ClientPayload):
# [User A](link) accepted an invitation to join [organization_name](link)
name = f"[{await self.real_name(gh['sender']['login'])}]({self.url(gh['sender'], html=True)})"
org = f"[{gh['organization']['login']}]({self.url(gh['organization'])})"
await self.bot.software_leaders_channel.send(
updates_channel = self.updates_channel(gh["organization"]["login"])
await updates_channel.send(
f"{name} accepted an invitation to join {org}",
)

Expand All @@ -173,7 +200,8 @@ async def organization_member_removed(self, payload: ClientPayload):
# [User A](link) was removed from [organization_name](link)
name = f"[{await self.real_name(gh['sender']['login'])}]({self.url(gh['sender'], html=True)})"
org = f"[{gh['organization']['login']}]({self.url(gh['organization'])})"
await self.bot.software_leaders_channel.send(f"{name} was removed from {org}")
updates_channel = self.updates_channel(gh["organization"]["login"])
await updates_channel.send(f"{name} was removed from {org}")

@Server.route()
async def pull_request_opened(self, payload: ClientPayload):
Expand All @@ -184,7 +212,8 @@ async def pull_request_opened(self, payload: ClientPayload):
pr = f"[#{gh['pull_request']['number']}]({self.url(gh['pull_request'], html=True)})"
repo = f"[{gh['repository']['full_name']}]({self.url(gh['repository'], html=True)})"
title = f"\"{gh['pull_request']['title']}\""
await self.bot.github_updates_channel.send(
updates_channel = self.updates_channel(gh["repository"])
await updates_channel.send(
f"{name} opened pull request {pr} in {repo}: {title}",
)

Expand All @@ -203,7 +232,8 @@ async def pull_request_closed(self, payload: ClientPayload):
if gh["pull_request"]["user"]["login"] != gh["sender"]["login"]
else ""
)
await self.bot.github_updates_channel.send(
updates_channel = self.updates_channel(gh["repository"])
await updates_channel.send(
f"{name} {'merged' if gh['pull_request']['merged'] else 'closed'} pull request {pr}{by_statement} in {repo}: {title}",
)

Expand All @@ -216,7 +246,8 @@ async def pull_request_review_requested(self, payload: ClientPayload):
requested = f"[{await self.real_name(gh['requested_reviewer']['login'])}]({self.url(gh['requested_reviewer'], html=True)})"
pr = f"[#{gh['pull_request']['number']}]({self.url(gh['pull_request'], html=True)})"
repo = f"[{gh['repository']['full_name']}]({self.url(gh['repository'], html=True)})"
await self.bot.github_updates_channel.send(
updates_channel = self.updates_channel(gh["repository"])
await updates_channel.send(
f"{name} requested a review from {requested} on pull request {pr} in {repo}",
)

Expand All @@ -228,7 +259,8 @@ async def pull_request_review_submitted(self, payload: ClientPayload):
name = f"[{await self.real_name(gh['sender']['login'])}]({self.url(gh['sender'], html=True)})"
pr = f"[#{gh['pull_request']['number']}]({self.url(gh['pull_request'], html=True)})"
repo = f"[{gh['repository']['full_name']}]({self.url(gh['repository'], html=True)})"
await self.bot.github_updates_channel.send(
updates_channel = self.updates_channel(gh["repository"])
await updates_channel.send(
f"{name} submitted a review on pull request {pr} in {repo}",
)

Expand All @@ -241,7 +273,8 @@ async def commit_comment(self, payload: ClientPayload):
commit = f"[`{gh['comment']['commit_id'][:7]}`]({self.url(gh['comment'], html=True)})"
repo = f"[{gh['repository']['full_name']}]({self.url(gh['repository'], html=True)})"
comment = f"\"{self.natural_wrap(gh['comment']['body'])}\""
await self.bot.github_updates_channel.send(
updates_channel = self.updates_channel(gh["repository"])
await updates_channel.send(
f"{name} commented on commit {commit} in {repo}: {comment}",
)

Expand All @@ -254,7 +287,8 @@ async def issues_comment_created(self, payload: ClientPayload):
issue = f"[#{gh['issue']['number']}]({self.url(gh['issue'], html=True)})"
repo = f"[{gh['repository']['full_name']}]({self.url(gh['repository'], html=True)})"
comment = f"\"{self.natural_wrap(gh['comment']['body'])}\""
await self.bot.github_updates_channel.send(
updates_channel = self.updates_channel(gh["repository"])
await updates_channel.send(
f"{name} commented on issue {issue} in {repo}: {comment}",
)

Expand All @@ -267,12 +301,13 @@ async def issues_assigned(self, payload: ClientPayload):
assigned = f"[{await self.real_name(gh['assignee']['login'])}]({self.url(gh['assignee'], html=True)})"
issue = f"[#{gh['issue']['number']}]({self.url(gh['issue'], html=True)})"
repo = f"[{gh['repository']['full_name']}]({self.url(gh['repository'], html=True)})"
updates_channel = self.updates_channel(gh["repository"])
if gh["assignee"]["login"] == gh["sender"]["login"]:
await self.bot.github_updates_channel.send(
await updates_channel.send(
f"{name} self-assigned issue {issue} in {repo}",
)
else:
await self.bot.github_updates_channel.send(
await updates_channel.send(
f"{name} assigned {assigned} to issue {issue} in {repo}",
)

Expand All @@ -285,12 +320,13 @@ async def issues_unassigned(self, payload: ClientPayload):
unassigned = f"[{await self.real_name(gh['assignee']['login'])}]({self.url(gh['assignee'], html=True)})"
issue = f"[#{gh['issue']['number']}]({self.url(gh['issue'], html=True)})"
repo = f"[{gh['repository']['full_name']}]({self.url(gh['repository'], html=True)})"
updates_channel = self.updates_channel(gh["repository"])
if gh["assignee"]["login"] == gh["sender"]["login"]:
await self.bot.github_updates_channel.send(
await updates_channel.send(
f"{name} unassigned themself from issue {issue} in {repo}",
)
else:
await self.bot.github_updates_channel.send(
await updates_channel.send(
f"{name} unassigned {unassigned} from issue {issue} in {repo}",
)

Expand All @@ -303,8 +339,9 @@ async def pull_request_edited(self, payload: ClientPayload):
pr = f"[#{gh['pull_request']['number']}]({self.url(gh['pull_request'], html=True)})"
repo = f"[{gh['repository']['full_name']}]({self.url(gh['repository'], html=True)})"
title = f"\"{gh['pull_request']['title']}\""
updates_channel = self.updates_channel(gh["repository"])
if gh["changes"]["title"]:
await self.bot.github_updates_channel.send(
await updates_channel.send(
f"{name} edited the title of pull request {pr} in {repo} to {title}",
)

Expand All @@ -318,8 +355,9 @@ async def membership_added(self, payload: ClientPayload):
added = f"[{await self.real_name(gh['member']['login'])}]({self.url(gh['member'], html=True)})"
team = f"[{gh['team']['name']}]({self.url(gh['team'], html=True)})"
org = f"[{gh['organization']['login']}]({self.url(gh['organization'])})"
leaders_channel = self.leaders_channel(gh["organization"]["login"])
if "core" in team.lower():
await self.bot.software_leaders_channel.send(
await leaders_channel.send(
f"{name} added {added} to {team} in {org}",
)

Expand All @@ -330,7 +368,8 @@ async def public(self, payload: ClientPayload):
# [User A](link) made [repo_name](link) public
name = f"[{await self.real_name(gh['sender']['login'])}]({self.url(gh['sender'], html=True)})"
repo = f"[{gh['repository']['full_name']}]({self.url(gh['repository'], html=True)})"
await self.bot.github_updates_channel.send(f"{name} made {repo} public")
updates_channel = self.updates_channel(gh["repository"])
await updates_channel.send(f"{name} made {repo} public")

@Server.route()
async def repository_created(self, payload: ClientPayload):
Expand All @@ -339,7 +378,8 @@ async def repository_created(self, payload: ClientPayload):
# [User A](link) created [repo_name](link)
name = f"[{await self.real_name(gh['sender']['login'])}]({self.url(gh['sender'], html=True)})"
repo = f"[{gh['repository']['full_name']}]({self.url(gh['repository'], html=True)})"
await self.bot.github_updates_channel.send(f"{name} created {repo}")
updates_channel = self.updates_channel(gh["repository"])
await updates_channel.send(f"{name} created {repo}")

@Server.route()
async def repository_deleted(self, payload: ClientPayload):
Expand All @@ -348,7 +388,8 @@ async def repository_deleted(self, payload: ClientPayload):
# [User A](link) deleted [repo_name](link)
name = f"[{await self.real_name(gh['sender']['login'])}]({self.url(gh['sender'], html=True)})"
repo = f"[{gh['repository']['full_name']}]({self.url(gh['repository'], html=True)})"
await self.bot.github_updates_channel.send(f"{name} deleted {repo}")
updates_channel = self.updates_channel(gh["repository"])
await updates_channel.send(f"{name} deleted {repo}")

@Server.route()
async def check_suite_completed(self, payload: ClientPayload):
Expand Down Expand Up @@ -377,7 +418,8 @@ async def check_suite_completed(self, payload: ClientPayload):
commit = f"[`{gh['check_suite']['head_sha'][:7]}`](<https://github.com/{gh['repository']['full_name']}/commit/{gh['check_suite']['head_sha']}>)"
repo = f"[{gh['repository']['full_name']}]({self.url(gh['repository'], html=True)})"
branch = f"`{gh['check_suite']['head_branch']}`"
await self.bot.software_leaders_channel.send(
leaders_channel = self.leaders_channel(gh["repository"])
await leaders_channel.send(
f"{failed_count} failed ({failed_links_str}) on commit {commit} by {name} in {repo} on {branch}",
)

Expand Down

0 comments on commit e5cbd09

Please sign in to comment.