Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 98e2683b80 | |||
| cba7a701bf | |||
| 884f1c6dd2 |
@@ -4,6 +4,7 @@
|
|||||||
import argparse
|
import argparse
|
||||||
import asyncio
|
import asyncio
|
||||||
import datetime
|
import datetime
|
||||||
|
import html
|
||||||
import os
|
import os
|
||||||
import pickle
|
import pickle
|
||||||
import re
|
import re
|
||||||
@@ -177,29 +178,31 @@ def add_video_to_playlist(youtube, playlist_id, video_id, retry_count=6):
|
|||||||
continue
|
continue
|
||||||
raise error
|
raise error
|
||||||
|
|
||||||
|
|
||||||
def get_video_info(youtube, video_id):
|
def get_video_info(youtube, video_id):
|
||||||
"""Check whether a YouTube video is music and return its title."""
|
"""Check whether a YouTube video is music and return its title and channel."""
|
||||||
try:
|
try:
|
||||||
video_details = youtube.videos().list(id=video_id, part="snippet").execute()
|
video_details = youtube.videos().list(id=video_id, part="snippet").execute()
|
||||||
|
|
||||||
# Check if the video actually exists/is accessible
|
# Check if the video actually exists/is accessible
|
||||||
if not video_details.get("items"):
|
if not video_details.get("items"):
|
||||||
return False, "[Video unavailable or private]"
|
return False, "[Video unavailable or private]", ""
|
||||||
|
|
||||||
snippet = video_details["items"][0]["snippet"]
|
snippet = video_details["items"][0]["snippet"]
|
||||||
|
|
||||||
# Check if the video category is Music (10) or Entertainment (24)
|
# Check if the video category is Music (10) or Entertainment (24)
|
||||||
is_music = snippet.get("categoryId") in ("10", "24")
|
is_music = snippet.get("categoryId") in ("10", "24")
|
||||||
title = snippet.get("title", "[Unknown Title]")
|
title = snippet.get("title", "[Unknown Title]")
|
||||||
|
channel = snippet.get("channelTitle", "[Unknown Channel]")
|
||||||
|
|
||||||
return is_music, title
|
return is_music, title, channel
|
||||||
|
|
||||||
except errors.HttpError as error:
|
except errors.HttpError as error:
|
||||||
print(f"YouTube API error fetching info for {video_id}: {error}")
|
print(f"YouTube API error fetching info for {video_id}: {error}")
|
||||||
return False, "[Error fetching title]"
|
return False, "[Error fetching title]", ""
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"Unexpected error fetching info for {video_id}: {e}")
|
print(f"Unexpected error fetching info for {video_id}: {e}")
|
||||||
return False, "[Error fetching title]"
|
return False, "[Error fetching title]", ""
|
||||||
|
|
||||||
|
|
||||||
async def send_intro_message(client, sender, room_id):
|
async def send_intro_message(client, sender, room_id):
|
||||||
@@ -256,6 +259,7 @@ async def send_playlist_of_all(client, sender, room_id, playlist_id):
|
|||||||
content={"msgtype": "m.text", "body": reply_msg},
|
content={"msgtype": "m.text", "body": reply_msg},
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
async def message_callback(conn, cursor, youtube, client, room, event):
|
async def message_callback(conn, cursor, youtube, client, room, event):
|
||||||
"""Event handler for received messages."""
|
"""Event handler for received messages."""
|
||||||
sender = event.sender
|
sender = event.sender
|
||||||
@@ -273,7 +277,9 @@ async def message_callback(conn, cursor, youtube, client, room, event):
|
|||||||
event.server_timestamp / 1000, datetime.UTC # millisec to sec
|
event.server_timestamp / 1000, datetime.UTC # millisec to sec
|
||||||
)
|
)
|
||||||
current_time = datetime.datetime.now(datetime.UTC)
|
current_time = datetime.datetime.now(datetime.UTC)
|
||||||
recent = current_time - timestamp_sec < datetime.timedelta(seconds=30)
|
|
||||||
|
# Account for up to 5 minutes of clock drift
|
||||||
|
recent = abs(current_time - timestamp_sec) < datetime.timedelta(minutes=5)
|
||||||
|
|
||||||
if body == "!parkerbot" and recent:
|
if body == "!parkerbot" and recent:
|
||||||
await send_intro_message(client, sender, room.room_id)
|
await send_intro_message(client, sender, room.room_id)
|
||||||
@@ -296,16 +302,29 @@ async def message_callback(conn, cursor, youtube, client, room, event):
|
|||||||
for link in youtube_links:
|
for link in youtube_links:
|
||||||
video_id = link.split("v=")[-1].split("&")[0].split("/")[-1]
|
video_id = link.split("v=")[-1].split("&")[0].split("/")[-1]
|
||||||
|
|
||||||
# Safely fetch the category check and the title
|
# Safely fetch the category check, title, and channel
|
||||||
is_music_vid, title = get_video_info(youtube, video_id)
|
is_music_vid, title, channel = get_video_info(youtube, video_id)
|
||||||
|
|
||||||
# Send the title to the channel so people know what the link is
|
# Send the title to the channel so people know what the link is
|
||||||
# Only do this for recent messages to prevent spam during backwards-sync
|
# Only do this for recent messages to prevent spam during backwards-sync
|
||||||
if recent:
|
if recent:
|
||||||
|
plain_text = f"🎵 {title}"
|
||||||
|
if channel:
|
||||||
|
plain_text += f" - {channel}"
|
||||||
|
|
||||||
|
# Escape HTML characters to prevent broken rendering in Matrix
|
||||||
|
escaped_text = html.escape(plain_text)
|
||||||
|
html_text = f"<em><span data-mx-color='#808080'>{escaped_text}</span></em>"
|
||||||
|
|
||||||
await client.room_send(
|
await client.room_send(
|
||||||
room_id=room.room_id,
|
room_id=room.room_id,
|
||||||
message_type="m.room.message",
|
message_type="m.room.message",
|
||||||
content={"msgtype": "m.text", "body": f"▶️ {title}"},
|
content={
|
||||||
|
"msgtype": "m.text",
|
||||||
|
"body": plain_text,
|
||||||
|
"format": "org.matrix.custom.html",
|
||||||
|
"formatted_body": html_text
|
||||||
|
},
|
||||||
)
|
)
|
||||||
|
|
||||||
# Only add to the playlist if it's categorized as music/entertainment
|
# Only add to the playlist if it's categorized as music/entertainment
|
||||||
|
|||||||
Reference in New Issue
Block a user