From 5a0d879b5b9046a909d20228f70b4d00f8fb42cc Mon Sep 17 00:00:00 2001
From: DarkCat09 <gh@dc09.ru>
Date: Sat, 27 Apr 2024 19:58:20 +0400
Subject: [PATCH] Untested pre/post-processors (ID3 + correct info dict),
 v0.0.1

---
 id3pp.py | 62 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)
 create mode 100644 id3pp.py

diff --git a/id3pp.py b/id3pp.py
new file mode 100644
index 0000000..e8a5960
--- /dev/null
+++ b/id3pp.py
@@ -0,0 +1,62 @@
+from mutagen import mp3, id3
+from yt_dlp.postprocessor import PostProcessor
+
+
+ENC_UTF8 = 3
+
+
+def add_tags(file: mp3.MP3, info: dict[str, str | int]) -> None:
+
+    file['TIT2'] = id3.TIT2(encoding=ENC_UTF8, text=info['track'])
+    file['TPE1'] = id3.TPE1(encoding=ENC_UTF8, text=info['artist'])
+    if 'album' in info:
+        file['TALB'] = id3.TALB(encoding=ENC_UTF8, text=info['album'])
+    if 'release_year' in info:
+        file['TDRC'] = id3.TDRC(encoding=ENC_UTF8, text=str(info['release_year']))
+    if 'track_number' in info:
+        file['TRCK'] = id3.TRCK(encoding=ENC_UTF8, text=str(info['track_number']))
+
+
+class InfoYouTubePP(PostProcessor):
+    '''Generates YT Music fields in info if necessary.
+    Must be run before downloading and post-processing with
+    FFmpegExtractAudioPP and ID3YouTubePP, so use only with
+    when<="before_dl" ("pre_process" also suits, see yt_dlp.utils.POSTPROCESS_WHEN)
+    for correct file path and ID3 tags'''
+
+    def run(self, information):
+
+        if not 'track' in information:
+            information['track'] = information['title']
+        if not 'artist' in information:
+            information['artist'] = information['channel'].removesuffix(' - Topic')
+
+        if not 'album' in information:
+            try:
+                information['album'] = information['playlist'].removeprefix('Album \u2013 ')
+            except KeyError:
+                pass
+
+        if not 'track_number' in information:
+            try:
+                information['track_number'] = information['playlist_index']
+            except KeyError:
+                pass
+        # here was some code for disc_number,
+        # but it is even not in mutagen ID3 fields list
+
+        return [], information
+
+
+class ID3YouTubePP(PostProcessor):
+    '''Inserts ID3 tags after InfoYouTubePP and FFmpegExtractAudioPP,
+    triggers searching and parsing lyrics from Genius'''
+
+    def run(self, information):
+
+        add_tags(
+            mp3.MP3(information['filepath']),
+            information,
+        )
+
+        return [], information