feat: discord rpc

1 files changed, 25 insertions, 0 deletions

diff --git a/yt_music/__yt_music__.py b/yt_music/__yt_music__.py
index 5e19c2d..9d8c0a3 100644
--- a/yt_music/__yt_music__.py
+++ b/yt_music/__yt_music__.py
@@ -3,7 +3,9 @@ import re
 import subprocess
 import platform
 import os
+import time
 
+from pypresence import Presence
 import httpx
 import fzf
 
@@ -12,6 +14,11 @@ headers = {
 }
 
 
+start = int(time.time())
+client_id = "1075791459409723472"
+RPC = Presence(client_id)
+RPC.connect()
+
 client = httpx.Client(headers=headers, timeout=None)
 
 base_url = "https://vid.puffyan.us"
@@ -72,6 +79,15 @@ def download(video_id, video_title):
 
 def play_loop(video_id, video_title):
    
+    RPC.update(
+        large_image = f"http://img.youtube.com/vi/{video_id}/0.jpg",
+        large_text = "haha checkmate spotify plebs",
+        small_image = "youtube_music_icon_svg",
+        small_text = "yt-music",
+        start = start,
+        details = f"{video_title} - loop",
+    )
+
     args = [
         MPV_EXECUTABLE,
         f"https://music.youtube.com/watch?v={video_id}",
@@ -87,6 +103,15 @@ def play_loop(video_id, video_title):
 
 def play(video_id, video_title):
    
+    RPC.update(
+        large_image = f"http://img.youtube.com/vi/{video_id}/0.jpg",
+        large_text = "haha checkmate spotify plebs",
+        small_image = "youtube_music_icon_svg",
+        small_text = "yt-music",
+        start = start,
+        details = video_title,
+    )
+
     args = [
         MPV_EXECUTABLE,
         f"https://music.youtube.com/watch?v={video_id}",