From 08247c0ee4eef44866a964a3575c2166a48018cd Mon Sep 17 00:00:00 2001
From: Philipp Wolfer <ph.wolfer@gmail.com>
Date: Thu, 11 May 2023 08:33:20 +0200
Subject: [PATCH] replaygain2: replace extension based type detection with type
 check

Fixes OGG Vorbis files not being detected as supported
---
 plugins/replaygain2/__init__.py | 47 ++++++++++++++++++++++-----------
 1 file changed, 31 insertions(+), 16 deletions(-)

diff --git a/plugins/replaygain2/__init__.py b/plugins/replaygain2/__init__.py
index 2c9140b9..393647c1 100755
--- a/plugins/replaygain2/__init__.py
+++ b/plugins/replaygain2/__init__.py
@@ -42,7 +42,21 @@
 from PyQt5.QtWidgets import QFileDialog
 
 from picard import log
-from picard.formats.vorbis import OggOpusFile
+from picard.formats import (
+    AiffFile,
+    ASFFile,
+    FLACFile,
+    MonkeysAudioFile,
+    MP3File,
+    MP4File,
+    OggFLACFile,
+    OggOpusFile,
+    OggSpeexFile,
+    OggTheoraFile,
+    OggVorbisFile,
+    WAVFile,
+    WavPackFile,
+)
 from picard.album import Album
 from picard.track import Track, NonAlbumTrack
 from picard.util import thread
@@ -52,6 +66,7 @@
                                  register_album_action)
 from picard.plugins.replaygain2.ui_options_replaygain2 import Ui_ReplayGain2OptionsPage
 
+
 CLIP_MODE_DISABLED = 0
 CLIP_MODE_POSITIVE = 1
 CLIP_MODE_ALWAYS = 2
@@ -65,20 +80,20 @@
 OPUS_MODE_STANDARD = 0
 OPUS_MODE_R128 = 1
 
-SUPPORTED_EXTS = (
-    ".mp2",
-    ".mp3",
-    ".flac",
-    ".ogg",
-    ".oga",
-    ".spx",
-    ".opus",
-    ".m4a",
-    ".wv",
-    ".ape",
-    ".wma",
-    ".wav",
-    ".aiff"
+SUPPORTED_FORMATS = (
+    AiffFile,
+    ASFFile,
+    FLACFile,
+    MonkeysAudioFile,
+    MP3File,
+    MP4File,
+    OggFLACFile,
+    OggOpusFile,
+    OggSpeexFile,
+    OggTheoraFile,
+    OggVorbisFile,
+    WAVFile,
+    WavPackFile,
 )
 
 TABLE_HEADER = (
@@ -174,7 +189,7 @@ def calculate_replaygain(tracks, options):
         if not track.files:
             continue
         file = track.files[0]
-        if not file.EXTENSIONS or file.EXTENSIONS[0] not in SUPPORTED_EXTS:
+        if not type(file) in SUPPORTED_FORMATS:
             raise Exception(f"ReplayGain 2.0: File '{file.filename}' is of unsupported format")
         files.append(file.filename)
         valid_tracks.append(track)