Ajout de la gestion des publications Patreon dans la base de données. Création de la table patreon_post avec les colonnes nécessaires et mise à jour des migrations pour intégrer ces nouvelles colonnes. Intégration de la vérification des publications Patreon dans le bot Discord et ajout de liens vers Patreon dans l'interface utilisateur.

2026-03-07 11:01:58 +01:00
parent a2ade9deeb
commit 540d23a3cf
9 changed files with 654 additions and 11 deletions
@@ -26,6 +26,7 @@ from discordbot.moderation import (
 	transfer_message_context_menu
 )
 from discordbot.welcome import sendWelcomeMessage, sendLeaveMessage, updateInviteCache
+from discordbot.patreon import checkPatreonPosts
 from discordbot.youtube import checkYouTubeVideos
 from discordbot.auto_rooms import on_voice_state_update_auto_rooms, on_raw_reaction_add_auto_rooms, on_message_auto_rooms, cleanup_orphaned_auto_rooms
 from protondb import searhProtonDb
@@ -76,6 +77,7 @@ class DiscordBot(discord.Client):
 		self.loop.create_task(self.updateHumbleBundle())
 		self.loop.create_task(self.updateYouTube())
 		self.loop.create_task(self.updateFreeLoot())
+		self.loop.create_task(self.updatePatreon())

 	async def on_disconnect(self):
 		webapp.config["BOT_STATUS"]["discord_connected"] = False
@@ -105,6 +107,11 @@ class DiscordBot(discord.Client):
 			await checkFreeLootAndNotify(self)
 			await asyncio.sleep(30*60)

+	async def updatePatreon(self):
+		while not self.is_closed():
+			await checkPatreonPosts(self)
+			await asyncio.sleep(10*60)
+
 	def getAllTextChannel(self) -> list[TextChannel]:
 		channels = []
 		for channel in self.get_all_channels():
@@ -0,0 +1,282 @@
+import asyncio
+import logging
+import re
+import xml.etree.ElementTree as ET
+
+import requests
+from discord import Client
+
+from database import db
+from database.helpers import ConfigurationHelper
+from database.models import PatreonPost
+from webapp import webapp
+
+logger = logging.getLogger('patreon-notification')
+logger.setLevel(logging.INFO)
+
+_patreon_first_check = True
+
+
+def _get_mention_content() -> str:
+	raw = ConfigurationHelper().getValue("patreon_mention")
+	if not raw or not str(raw).strip():
+		return ""
+	parts = []
+	for s in str(raw).strip().split(","):
+		s = s.strip()
+		if s == "everyone":
+			parts.append("@everyone")
+		elif s == "here":
+			parts.append("@here")
+		elif s.isdigit():
+			parts.append(f"<@&{s}>")
+	return " ".join(parts) if parts else ""
+
+
+def _strip_html(html: str, max_len: int = 300) -> str:
+	"""Extrait le texte brut depuis du HTML et tronque."""
+	if not html:
+		return ""
+	text = re.sub(r'<br\s*/?>', '\n', html)
+	text = re.sub(r'<[^>]+>', '', text)
+	text = re.sub(r'&nbsp;', ' ', text)
+	text = re.sub(r'&amp;', '&', text)
+	text = re.sub(r'&lt;', '<', text)
+	text = re.sub(r'&gt;', '>', text)
+	text = re.sub(r'&#\d+;', '', text)
+	text = re.sub(r'\n{3,}', '\n\n', text).strip()
+	if len(text) > max_len:
+		text = text[:max_len].rsplit(' ', 1)[0] + '...'
+	return text
+
+
+def _extract_image(html: str) -> str | None:
+	"""Extrait la première URL d'image depuis le contenu HTML."""
+	if not html:
+		return None
+	match = re.search(r'<img[^>]+src=["\']([^"\']+)["\']', html)
+	if match:
+		url = match.group(1)
+		if url.startswith("http"):
+			return url
+	return None
+
+
+def _parse_item(item, creator_name: str) -> dict | None:
+	"""Parse un <item> RSS et retourne un dict avec les métadonnées."""
+	guid_elem = item.find('guid')
+	if guid_elem is None or not guid_elem.text:
+		return None
+	title_elem = item.find('title')
+	link_elem = item.find('link')
+	desc_elem = item.find('description')
+	pub_elem = item.find('pubDate')
+	return {
+		'guid': guid_elem.text.strip(),
+		'title': title_elem.text if title_elem is not None else 'Nouveau post',
+		'link': link_elem.text if link_elem is not None else '',
+		'description': desc_elem.text if desc_elem is not None else '',
+		'published_at': pub_elem.text if pub_elem is not None else '',
+		'creator': creator_name,
+	}
+
+
+def _fetch_rss() -> tuple[list[dict], str] | None:
+	"""Fetch le RSS Patreon et retourne (posts, creator_name) ou None."""
+	helper = ConfigurationHelper()
+	creator = helper.getValue("patreon_creator")
+	if not creator or not str(creator).strip():
+		return None
+
+	rss_url = f"https://www.patreon.com/rss/{str(creator).strip()}"
+
+	try:
+		response = requests.get(rss_url, timeout=15)
+	except Exception as e:
+		logger.error(f"Patreon: erreur réseau lors de la récupération du RSS: {e}")
+		return None
+
+	if response.status_code != 200:
+		logger.error(f"Patreon: HTTP {response.status_code} pour {rss_url}")
+		return None
+
+	try:
+		root = ET.fromstring(response.content)
+	except ET.ParseError as e:
+		logger.error(f"Patreon: erreur de parsing XML: {e}")
+		return None
+
+	creator_name = creator
+	channel_elem = root.find('.//channel/title')
+	if channel_elem is not None and channel_elem.text:
+		creator_name = channel_elem.text
+
+	items = root.findall('.//item')
+	posts = []
+	for item in items:
+		parsed = _parse_item(item, creator_name)
+		if parsed:
+			posts.append(parsed)
+
+	return (posts, creator_name)
+
+
+def _build_embed(post: dict):
+	import discord
+
+	title = post.get('title') or 'Nouveau post Patreon'
+	link = post.get('link') or ''
+	description = _strip_html(post.get('description') or '', max_len=350)
+	creator = post.get('creator') or 'Patreon'
+	image_url = _extract_image(post.get('description') or '')
+
+	helper = ConfigurationHelper()
+	try:
+		color = int(helper.getValue('patreon_embed_color') or 'F96854', 16)
+	except (ValueError, TypeError):
+		color = 0xF96854
+
+	embed = discord.Embed(
+		title=title,
+		url=link if link.startswith("http") else None,
+		color=color,
+	)
+
+	if description:
+		embed.description = description
+
+	embed.set_author(
+		name=creator,
+		icon_url="https://c5.patreon.com/external/favicon/favicon-32x32.png",
+	)
+
+	if image_url:
+		embed.set_image(url=image_url)
+
+	embed.set_footer(text="MamieHenriette \u2022 Patreon")
+
+	return embed
+
+
+async def checkPatreonPosts(bot: Client):
+	global _patreon_first_check
+	with webapp.app_context():
+		helper = ConfigurationHelper()
+		if not helper.getValue("patreon_enable"):
+			return
+
+		channel_id = helper.getIntValue("patreon_channel_id")
+		if not channel_id:
+			return
+
+		channel = bot.get_channel(channel_id)
+		if not channel:
+			logger.warning("Patreon: canal Discord introuvable")
+			return
+
+		result = await asyncio.to_thread(_fetch_rss)
+		if not result:
+			return
+
+		posts, creator_name = result
+
+		if not posts:
+			logger.info("Patreon: aucun post trouvé dans le flux RSS")
+			return
+
+		if _patreon_first_check:
+			logger.info("Patreon: première vérification, synchronisation sans notification")
+			for post_data in posts:
+				guid = post_data['guid']
+				if not PatreonPost.query.get(guid):
+					try:
+						db.session.add(PatreonPost(
+							guid=guid,
+							title=post_data['title'],
+							link=post_data['link'],
+							description=post_data['description'],
+							published_at=post_data['published_at'],
+							notified=False,
+						))
+						db.session.commit()
+					except Exception as e:
+						logger.error(f"Patreon: erreur de synchronisation pour {guid}: {e}")
+						db.session.rollback()
+			_patreon_first_check = False
+			return
+
+		for post_data in posts:
+			guid = post_data['guid']
+
+			if PatreonPost.query.get(guid):
+				continue
+
+			try:
+				embed = _build_embed(post_data)
+				content = _get_mention_content()
+				await channel.send(content=content or None, embed=embed)
+				db.session.add(PatreonPost(
+					guid=guid,
+					title=post_data['title'],
+					link=post_data['link'],
+					description=post_data['description'],
+					published_at=post_data['published_at'],
+					notified=True,
+				))
+				db.session.commit()
+				logger.info(f"Patreon: notification envoyée pour '{post_data['title']}'")
+			except Exception as e:
+				logger.error(f"Patreon: envoi Discord échoué pour {guid}: {e}")
+				db.session.rollback()
+
+
+async def _send_post_to_discord_async(bot: Client, guid: str) -> tuple[bool, str]:
+	"""Envoie un post Patreon sur Discord (appel manuel). Retourne (succès, message)."""
+	helper = ConfigurationHelper()
+	channel_id = helper.getIntValue("patreon_channel_id")
+	if not channel_id:
+		return (False, "Aucun canal Discord configuré pour Patreon.")
+	channel = bot.get_channel(channel_id)
+	if not channel:
+		return (False, "Canal Discord introuvable.")
+
+	post_db = PatreonPost.query.get(guid)
+	if not post_db:
+		return (False, "Post introuvable en base de données.")
+
+	creator = helper.getValue("patreon_creator") or "Patreon"
+	# Tenter de récupérer le nom du créateur depuis le RSS
+	result = _fetch_rss()
+	creator_name = result[1] if result else creator
+
+	post_data = {
+		'title': post_db.title or 'Nouveau post',
+		'link': post_db.link or '',
+		'description': post_db.description or '',
+		'creator': creator_name,
+	}
+
+	try:
+		embed = _build_embed(post_data)
+		content = _get_mention_content()
+		await channel.send(content=content or None, embed=embed)
+		post_db.notified = True
+		db.session.commit()
+		return (True, "Notification envoyée sur Discord.")
+	except Exception as e:
+		logger.error(f"Patreon: envoi manuel échoué pour {guid}: {e}")
+		db.session.rollback()
+		return (False, str(e))
+
+
+def send_post_to_discord_sync(bot: Client, guid: str) -> tuple[bool, str]:
+	"""Appel synchrone pour envoyer un post sur Discord (depuis la webapp)."""
+	try:
+		future = asyncio.run_coroutine_threadsafe(
+			_send_post_to_discord_async(bot, guid),
+			bot.loop,
+		)
+		return future.result(timeout=15)
+	except Exception as e:
+		logger.error(f"Patreon: send_post_to_discord_sync: {e}")
+		return (False, str(e))