init repo

This commit is contained in:
0ry5 2025-03-30 18:21:42 +02:00
parent ba8b46c8a7
commit a8a5fed56f
4 changed files with 275 additions and 0 deletions

7
.example.env Normal file
View file

@ -0,0 +1,7 @@
DISCORD_TOKEN="DISCORT_BOT_TOKEN"
CHANNEL_ID="DISCORD_CHANNEL_ID"
GITHUB_TOKEN="GITHUB_TOKEN_TO_PUSH_TO_MAIN"
REPO_OWNER="netz39"
REPO_NAME="wwww.netz39.de"
BASE_BRANCH="main"
MAINTAINERS='[DISCORD_IDS_OF_BRANCH_MAINTAINERS]'

173
bot.py Normal file
View file

@ -0,0 +1,173 @@
import json
import re
import discord
from discord.ext import commands
from dotenv import load_dotenv
from os import getenv
from github_connector import create_or_update_file, delete_file
load_dotenv("./.env")
CHANNEL_ID = int(getenv("CHANNEL_ID"))
MAINTAINERS = json.loads(getenv("MAINTAINERS"))
DISCORD_TOKEN = getenv("DISCORD_TOKEN")
print(MAINTAINERS)
# Intents are required to listen to events like messages
intents = discord.Intents.default()
intents.message_content = True
intents.members = True
intents.reactions = True
intents.dm_messages = True
# Create an instance of the Bot
bot = commands.Bot(command_prefix="!", intents=intents)
def getFileName(event):
return f"{event.start_time.strftime("%Y-%m-%d")}_{event.name}"
def getFilePath(event):
return f"_events/{event.start_time.strftime("%Y")}/discord-event-{event.id}.md"
def getEventYML(event):
return f'# event imported from discord manual changes may be overwritten\n---\n \nlayout: event\ntitle: "{event.name}"\nauthor: "Netz39 e.V." \nevent:\n start: {event.start_time.strftime("%Y-%m-%d %H:%M:%S")} \n end: {event.end_time.strftime("%Y-%m-%d %H:%M:%S")} \n organizer: "Netz39 Team <kontakt@netz39.de>" \n location: "Leibnizstr. 18, 39104 Magdeburg"\n---'
def addMaintainer(maintainer):
MAINTAINERS.append(maintainer)
with open(".env", "r") as fileR:
# read a list of lines into data
data = fileR.readlines()
data[6] = f"MAINTAINERS='{json.dumps(MAINTAINERS)}'"
with open(".env", "w") as fileW:
fileW.writelines(data)
async def handleEvent(event):
eventYML = getEventYML(event)
if "netz39" in event.location.lower() or (
"leibniz" in event.location.lower() and "18" in event.location
):
if event.creator_id in MAINTAINERS:
create_or_update_file(
getFilePath(event),
eventYML,
f"new event: {getFileName(event)}",
)
channel = bot.get_channel(CHANNEL_ID)
await channel.send(
f"📅 New event '{event.name}' on {event.start_time.strftime("%Y-%m-%d")}."
)
else:
eventJSON = json.dumps(
{
"yml": eventYML,
"path": getFilePath(event),
"fileName": getFileName(event),
"name": event.name,
"date": event.start_time.strftime("%Y-%m-%d"),
},
indent=2,
)
for maintainer in MAINTAINERS:
user = await bot.fetch_user(event.creator_id)
await sendDm(
maintainer,
f"{user.name} added a new Event on {event.start_time.strftime("%Y")}: {event.name}. Like this message to approve.\n\n\n```json\n{eventJSON}```",
)
# Event: When the bot is ready
@bot.event
async def on_ready():
print(f"Logged in as {discord.user}")
# Command: !hello
@bot.command()
async def hello(ctx):
await ctx.send(
"Hello, I am a bot to add events from discord to the Netz39 calendar on the website!"
)
@bot.event
async def on_ready():
print(f"Logged in as {bot.user}")
channel = bot.get_channel(CHANNEL_ID)
await channel.send("Hello everyone! I'm online and ready to go! 📅")
@bot.event
async def on_scheduled_event_update(event):
await handleEvent(event)
@bot.event
async def on_scheduled_event_create(event):
await handleEvent(event)
async def sendDm(userID, message):
user = await bot.fetch_user(userID)
await user.send(message)
@bot.event
async def on_reaction_add(reaction, user):
if not user.id in MAINTAINERS:
return # Ignore non-maintainers
if user.bot:
return # Ignore bot reactions
if (
isinstance(reaction.message.channel, discord.DMChannel)
and reaction.message.author.bot
): # Check if it's a DM and from the bot
if reaction.emoji == "👍":
messageJSON = json.loads(
re.search(
"(?s)(?<=```json).*?(?=```)", reaction.message.content
).group()
)
create_or_update_file(
messageJSON["path"],
messageJSON["yml"],
f"new event: {messageJSON["fileName"]}",
)
channel = bot.get_channel(CHANNEL_ID)
await channel.send(
f"📅 New event '{messageJSON["name"]}' on {messageJSON["date"]}."
)
@bot.event
async def on_message(message):
if isinstance(message.channel, discord.DMChannel):
global MAINTAINERS
author = message.author
if "add maintainer" in message.content.lower() and author.id in MAINTAINERS:
newMaintainer = message.content.lower().replace("add maintainer ", "")
if not newMaintainer in MAINTAINERS:
addMaintainer(int(newMaintainer))
await sendDm(
newMaintainer,
"You are now a Maintainer for posthorn! Create an event located at 'Netz39 e.V.' and it will be added to the calendar.",
)
await message.channel.send("Maintainer added!")
await bot.process_commands(message)
@bot.event
async def on_scheduled_event_delete(event):
delete_file(getFilePath(event))
channel = bot.get_channel(CHANNEL_ID)
await channel.send(f"❌ Event has been canceled: {event.name}")
bot.run(DISCORD_TOKEN)

71
github_connector.py Normal file
View file

@ -0,0 +1,71 @@
import requests
import json
from dotenv import load_dotenv
from os import getenv
import base64
load_dotenv()
# load env vars
REPO_OWNER = getenv("REPO_OWNER")
REPO_NAME = getenv("REPO_NAME")
BASE_BRANCH = getenv("BASE_BRANCH")
GITHUB_TOKEN = getenv("GITHUB_TOKEN")
# GitHub API URL
API_URL = f"https://api.github.com/repos/{REPO_OWNER}/{REPO_NAME}"
# Headers for authentication
HEADERS = {
"Authorization": f"token {GITHUB_TOKEN}",
"Accept": "application/vnd.github.v3+json",
}
def get_file_sha(file_path):
url = f"{API_URL}/contents/{file_path}?ref={BASE_BRANCH}"
response = requests.get(url, headers=HEADERS)
if response.status_code == 200:
return response.json()["sha"]
else:
raise Exception(f"Error getting file SHA: {response.json()}")
def delete_file(file_path):
url = f"{API_URL}/contents/{file_path}"
data = {
"message": f"Deleting file: {file_path}",
"sha": get_file_sha(file_path),
"branch": BASE_BRANCH,
}
response = requests.delete(url, headers=HEADERS, json=data)
if not response.status_code == 200 and not response.status_code == 404:
raise Exception(f"Error deleting file: {response.json()}")
def create_or_update_file(file_path, content, commit_message):
url = f"{API_URL}/contents/{file_path}"
b = base64.b64encode(bytes(content, "utf-8"))
base64_str = b.decode("utf-8")
# Check if file exists
response = requests.get(url, headers=HEADERS)
if response.status_code == 200:
sha = response.json()["sha"]
else:
sha = None
data = {
"message": commit_message,
"content": base64_str,
"branch": BASE_BRANCH,
}
if sha:
data["sha"] = sha # If file exists, update it
response = requests.put(url, headers=HEADERS, json=data)
if not response.status_code in [200, 201]:
raise Exception(f"Error committing file: {response.json()}")

24
requirements.txt Normal file
View file

@ -0,0 +1,24 @@
aiohappyeyeballs==2.6.1
aiohttp==3.11.14
aiosignal==1.3.2
attrs==25.3.0
certifi==2025.1.31
cffi==1.17.1
charset-normalizer==3.4.1
cryptography==44.0.2
Deprecated==1.2.18
discord.py==2.5.2
frozenlist==1.5.0
idna==3.10
multidict==6.2.0
propcache==0.3.1
pycparser==2.22
PyGithub==2.6.1
PyJWT==2.10.1
PyNaCl==1.5.0
python-dotenv==1.1.0
requests==2.32.3
typing_extensions==4.13.0
urllib3==2.3.0
wrapt==1.17.2
yarl==1.18.3