From 58f85b30ac7320dbcfe8be4a40b23c5ee68e4e77 Mon Sep 17 00:00:00 2001
From: Mike Lang <mikelang3000@gmail.com>
Date: Tue, 24 Jun 2025 00:36:40 +1000
Subject: [PATCH] youtubebot: Show the video title

This is an additional 1 quota per video, but with a cache (it's ok if it's out of date)
this should be no issue.
---
 zulip_bots/zulip_bots/youtubebot.py | 32 ++++++++++++++++++++++++++---
 1 file changed, 29 insertions(+), 3 deletions(-)

diff --git a/zulip_bots/zulip_bots/youtubebot.py b/zulip_bots/zulip_bots/youtubebot.py
index 25e42e3..7cd9f6d 100644
--- a/zulip_bots/zulip_bots/youtubebot.py
+++ b/zulip_bots/zulip_bots/youtubebot.py
@@ -11,6 +11,30 @@ from common.googleapis import GoogleAPIClient
 from .config import get_config
 from .zulip import Client
 
+
+_video_title_cache = {}
+def get_video_title(google, video_id):
+	if video_id not in _video_title_cache:
+		_video_title_cache[video_id] = _get_video_title(google, video_id)
+	return _video_title_cache[video_id]
+
+
+def _get_video_title(google, video_id):
+	resp = google.request("GET",
+		"https://www.googleapis.com/youtube/v3/videos",
+		params={
+			"part": "snippet",
+			"id": video_id,
+		}
+	)
+	resp.raise_for_status()
+	items = resp.json()["items"]
+	if not items:
+		# This should be rare, and it's ok to just crash
+		raise Exception(f"Request for video id {video_id} returned no results, was it deleted?")
+	return items[0]["snippet"]["title"]
+
+
 def get_comments(google, channel_id):
 	resp = google.request("GET",
 		"https://www.googleapis.com/youtube/v3/commentThreads",
@@ -39,10 +63,12 @@ def get_comments(google, channel_id):
 	return comments
 
 
-def show_comment(zulip, stream, topic, comment):
+def show_comment(zulip, google, stream, topic, comment):
+	title = get_video_title(google, comment["videoId"])
 	c = comment["snippet"]
 	author = f"[{c['authorDisplayName']}]({c['authorChannelUrl']})"
-	video = f"https://youtu.be/{comment['videoId']}"
+	url = f"https://youtu.be/{comment['videoId']}"
+	video = f"[{title}]({url})"
 	message = f"{author} commented on {video}:\n```quote\n{c['textDisplay']}\n```"
 	logging.info(f"Sending message to {stream}/{topic}: {message!r}")
 	# Empty stream acts as a dry-run mode
@@ -92,7 +118,7 @@ def main(conf_file, interval=60, one_off=0, stream="bot-spam", topic="Youtube Co
 				if comment["id"] in seen:
 					logging.debug(f"Comment {comment['id']} already seen, skipping")
 					continue
-				show_comment(zulip, stream, topic, comment)
+				show_comment(zulip, google, stream, topic, comment)
 				seen.append(comment["id"])
 		seen = seen[-keep:]