mra/OnBoard-Live
1
0
Fork 0
mirror of https://github.com/MichaByte/OnBoard-Live.git synced 2025-12-06 07:13:41 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

mostly working active stream system. it doesn't remove inactive streams immediately

Browse source
This commit is contained in:
Micha Albert 2024-07-30 14:54:02 -04:00
parent 05aace1ccf
commit 0a523a06d2
No known key found for this signature in database
GPG key ID: 33149159A417BBCE
1 changed files with 153 additions and 91 deletions
Download patch file Download diff file Expand all files Collapse all files

244
stream/backend/main.py
View file

@ -1,39 +1,51 @@
import json import json
import logging
import os import os
from contextlib import asynccontextmanager from contextlib import asynccontextmanager
from pickle import FALSE
from random import choice from random import choice
from secrets import token_hex from secrets import token_hex
import httpx import httpx
import uvicorn import uvicorn
from apscheduler.schedulers.asyncio import AsyncIOScheduler
from apscheduler.triggers.interval import IntervalTrigger
from dotenv import load_dotenv from dotenv import load_dotenv
from fastapi import FastAPI, Request, Response, Depends from fastapi import Depends, FastAPI, Request, Response
from fastapi.middleware.cors import CORSMiddleware from fastapi.middleware.cors import CORSMiddleware
from prisma import Prisma from prisma import Prisma
from slack_bolt.adapter.fastapi.async_handler import AsyncSlackRequestHandler from slack_bolt.adapter.fastapi.async_handler import AsyncSlackRequestHandler
from slack_bolt.async_app import AsyncAck, AsyncApp from slack_bolt.async_app import AsyncAck, AsyncApp
import asyncio
load_dotenv() load_dotenv()
active_stream = "" active_stream = {}
active_streams = []
scheduler = AsyncIOScheduler()
async def update_active(): async def update_active():
global active_stream global active_stream
global active_streams
async with httpx.AsyncClient() as client: async with httpx.AsyncClient() as client:
streams = (await client.get("http://localhost:9997/v3/paths/list")).json()[ streams_raw = (await client.get("http://localhost:9997/v3/paths/list")).json()[
"items" "items"
] ]
active_streams = [] streams = []
for stream in streams_raw:
streams.append({"name": stream["name"], "ready": stream["ready"]})
for stream in streams: for stream in streams:
if stream["ready"]: if stream["ready"] and stream not in active_streams:
active_streams.append(stream) active_streams.append(stream)
print(active_streams)
try: try:
if active_stream == "": if len(active_streams) == 0:
print("No active streams")
return
if active_stream == {}:
active_stream = choice(active_streams)["name"] active_stream = choice(active_streams)["name"]
new_stream = choice(active_streams)["name"] new_stream = choice(active_streams)["name"]
if len(active_streams) == 1:
return
while new_stream == active_stream: while new_stream == active_stream:
new_stream = choice(active_streams)["name"] new_stream = choice(active_streams)["name"]
await db.connect() await db.connect()
@ -46,16 +58,46 @@ async def update_active():
except Exception: except Exception:
return return
async def check_for_new():
async def init_active_update_task(): global active_stream
while True: global active_streams
asyncio.create_task(update_active()) async with httpx.AsyncClient() as client:
await asyncio.sleep(5 * 60) streams_raw = (await client.get("http://localhost:9997/v3/paths/list")).json()[
"items"
]
streams_have_changed = False
streams = []
for stream in streams_raw:
streams.append({"name": stream["name"], "ready": stream["ready"]})
for stream in streams:
for active_stream in active_streams:
if stream["name"] == active_stream["name"] and not stream["ready"]:
active_streams.remove(active_stream)
streams_have_changed = True
for active_stream in active_streams:
if stream["name"] == active_stream["name"] and not stream["ready"]:
active_streams.remove(active_stream)
streams_have_changed = True
if stream["ready"] and [s["name"] for s in streams] not in [s["name"] for s in active_streams]:
streams_have_changed = True
active_streams.append(stream)
new_active_streams = []
[new_active_streams.append(x) for x in active_streams if x not in new_active_streams]
active_streams = new_active_streams
if streams_have_changed:
print("Streams have changed!")
streams_have_changed = False
await update_active()
return
if len(active_streams) == 0:
print("No active streams")
active_stream = {}
@asynccontextmanager @asynccontextmanager
async def lifespan(app: FastAPI): async def lifespan(app: FastAPI):
asyncio.create_task(init_active_update_task()) await update_active()
scheduler.start()
scheduler.add_job(update_active, IntervalTrigger(minutes=5))
scheduler.add_job(check_for_new, IntervalTrigger(seconds=3))
await db.connect() await db.connect()
async with httpx.AsyncClient() as client: async with httpx.AsyncClient() as client:
for stream in await db.stream.find_many(): for stream in await db.stream.find_many():
@ -65,6 +107,7 @@ async def lifespan(app: FastAPI):
) )
await db.disconnect() await db.disconnect()
yield yield
scheduler.shutdown()
api = FastAPI(lifespan=lifespan) # type: ignore api = FastAPI(lifespan=lifespan) # type: ignore
@ -105,8 +148,7 @@ async def get_user_by_id(user_id: str):
@api.get("/api/v1/active_stream") @api.get("/api/v1/active_stream")
async def get_active_stream(): async def get_active_stream():
global active_stream return active_stream["name"] if "name" in active_stream else ""
return active_stream
@bolt.event("app_home_opened") @bolt.event("app_home_opened")
@ -196,6 +238,16 @@ async def approve(ack, body):
) )
await bolt.client.chat_postMessage( await bolt.client.chat_postMessage(
channel=sumbitter_convo["channel"]["id"], channel=sumbitter_convo["channel"]["id"],
blocks=[
{
"type": "section",
"text": {
"type": "plain_text",
"text": f"Welcome to OnBoard Live! Your stream key is {new_stream.key}. To use your stream key the easy way, go to <https://live.onboard.hackclub.com/{new_stream.key}/publish|this link>. You can also use it in OBS with the server URL of rtmp://live.onboard.hackclub.com:1935",
"emoji": True,
},
}
],
) )
await db.disconnect() await db.disconnect()
@ -204,6 +256,9 @@ async def approve(ack, body):
async def handle_application_submission(ack, body): async def handle_application_submission(ack, body):
await ack() await ack()
user = body["user"]["id"] user = body["user"]["id"]
print(user + "has applied")
with open("applicants.txt", "a") as f:
f.write(user + "\n")
sumbitter_convo = await bolt.client.conversations_open(users=user, return_im=True) sumbitter_convo = await bolt.client.conversations_open(users=user, return_im=True)
user_real_name = (await bolt.client.users_info(user=user))["user"]["real_name"] user_real_name = (await bolt.client.users_info(user=user))["user"]["real_name"]
user_verified = "" user_verified = ""
@ -220,90 +275,97 @@ async def handle_application_submission(ack, body):
) )
await bolt.client.chat_postMessage( await bolt.client.chat_postMessage(
channel=sumbitter_convo["channel"]["id"], channel=sumbitter_convo["channel"]["id"],
text=f"Your application has been submitted! We will review it shortly. Please do not send another application - If you haven't heard back in over 48 hours, or you forgot something in your application, please message <@{os.environ['ADMIN_SLACK_ID']}>! Here's a copy of your responses for your reference:\nSome info on your project(s): {body['view']['state']['values']['project-info']['project-info-body']['value']}{f'\nPlease fill out <https://forms.hackclub.com/eligibility?program=Onboard%20Live&slack_id={user}|the verification form>! We can only approve your application once this is done.' if not user_verified else ''}", text=f"Your application has been submitted! We will review it shortly. Please do not send another application - If you haven't heard back in over 48 hours, or you forgot something in your application, please message <@{os.environ['ADMIN_SLACK_ID']}>! Here's a copy of your responses for your reference:\nSome info on your project(s): {body['view']['state']['values']['project-info']['project-info-body']['value']}\n{f'Please fill out <https://forms.hackclub.com/eligibility?program=Onboard%20Live&slack_id={user}|the verification form>! We can only approve your application once this is done.' if not user_verified else ''}",
) )
admin_convo = await bolt.client.conversations_open( admin_convo = await bolt.client.conversations_open(
users=os.environ["ADMIN_SLACK_ID"], return_im=True users=os.environ["ADMIN_SLACK_ID"], return_im=True
) )
will_behave = False will_behave = True
boxes = body["view"]["state"]["values"]["c+wGI"]["checkboxes"]["selected_options"] print(body["view"]["state"]["values"])
if len(boxes) == 1 and boxes[0]["value"] == "value-1": # boxes = body["view"]["state"]["values"]["kAgeY"]["checkboxes"]["selected_options"]
will_behave = True # if len(boxes) == 1 and boxes[0]["value"] == "value-1":
await bolt.client.chat_postMessage( # will_behave = True
channel=admin_convo["channel"]["id"], print(
text="New OnBoard Live application!", await bolt.client.chat_postMessage(
blocks=[ channel=os.environ["ADMIN_SLACK_ID"],
{ text="New OnBoard Live application!",
"type": "section", blocks=[
"text": { {
"type": "mrkdwn", "type": "section",
"text": ":siren-real: New OnBoard Live application! :siren-real:", "text": {
"type": "mrkdwn",
"text": ":siren-real: New OnBoard Live application! :siren-real:",
},
}, },
}, {
{ "type": "section",
"type": "section", "text": {
"text": { "type": "plain_text",
"type": "plain_text", "text": f":technologist: Name: {user_real_name}",
"text": f":technologist: Name: {user_real_name}", "emoji": True,
"emoji": True, },
}, },
}, {
{ "type": "section",
"type": "section", "text": {
"text": { "type": "plain_text",
"type": "plain_text", "text": f":white_check_mark: Is verified: {user_verified}",
"text": f":white_check_mark: Is verified: {user_verified}", "emoji": True,
"emoji": True, },
}, },
}, {
{ "type": "section",
"type": "section", "text": {
"text": { "type": "plain_text",
"type": "plain_text", "text": f":hammer_and_wrench: Will make: {body['view']['state']['values']['project-info']['project-info-body']['value']}",
"text": f":hammer_and_wrench: Will make: {body['view']['state']['values']['project-info']['project-info-body']['value']}", "emoji": True,
"emoji": True, },
}, },
}, {
{ "type": "section",
"type": "section", "text": {
"text": { "type": "plain_text",
"type": "plain_text", "text": f":pray: Will behave on stream: {will_behave}",
"text": f":pray: Will behave on stream: {will_behave}", "emoji": True,
"emoji": True, },
}, },
}, {
{ "type": "section",
"type": "section", "text": {
"text": { "type": "plain_text",
"type": "plain_text", "text": f"Slack ID: {user}",
"text": f"Slack ID: {user}", "emoji": True,
"emoji": True, },
}, },
}, {
{ "type": "actions",
"type": "actions", "elements": [
"elements": [ {
{ "type": "button",
"type": "button", "text": {
"text": { "type": "plain_text",
"type": "plain_text", "emoji": True,
"emoji": True, "text": "Approve",
"text": "Approve", },
"style": "primary",
"value": "approve",
"action_id": "approve",
}, },
"style": "primary", {
"value": "approve", "type": "button",
"action_id": "approve", "text": {
}, "type": "plain_text",
{ "emoji": True,
"type": "button", "text": "Deny",
"text": {"type": "plain_text", "emoji": True, "text": "Deny"}, },
"style": "danger", "style": "danger",
"value": "deny", "value": "deny",
"action_id": "deny", "action_id": "deny",
}, },
], ],
}, },
], ],
)
) )