More code re-organizing

src/engines/ffmpeg/ffmpeg_downloader.py

@@ -0,0 +1,112 @@
+import logging
+import os
+import platform
+import re
+
+import requests
+
+from src.engines.core.downloader_core import download_and_extract_app
+
+logger = logging.getLogger()
+supported_formats = ['.zip', '.tar.xz', '.dmg']
+
+
+class FFMPEGDownloader:
+
+    # macOS FFMPEG mirror maintained by Evermeet - https://evermeet.cx/ffmpeg/
+    macos_url = "https://evermeet.cx/pub/ffmpeg/"
+
+    # Linux FFMPEG mirror maintained by John van Sickle - https://johnvansickle.com/ffmpeg/
+    linux_url = "https://johnvansickle.com/ffmpeg/"
+
+    # macOS FFMPEG mirror maintained by GyanD - https://www.gyan.dev/ffmpeg/builds/
+    windows_download_url = "https://github.com/GyanD/codexffmpeg/releases/download/"
+    windows_api_url = "https://api.github.com/repos/GyanD/codexffmpeg/releases"
+
+    @classmethod
+    def get_macos_versions(cls):
+        response = requests.get(cls.macos_url)
+        response.raise_for_status()
+
+        link_pattern = r'>(.*\.zip)[^\.]'
+        link_matches = re.findall(link_pattern, response.text)
+
+        return [link.split('-')[-1].split('.zip')[0] for link in link_matches]
+
+    @classmethod
+    def get_linux_versions(cls):
+
+        # Link 1 / 2 - Current Version
+        response = requests.get(cls.linux_url)
+        response.raise_for_status()
+        current_release = re.findall(r'release: ([\w\.]+)', response.text)[0]
+
+        # Link 2 / 2 - Previous Versions
+        response = requests.get(os.path.join(cls.linux_url, 'old-releases'))
+        response.raise_for_status()
+        releases = list(set(re.findall(r'href="ffmpeg-([\w\.]+)-.*">ffmpeg', response.text)))
+        releases.sort(reverse=True)
+        releases.insert(0, current_release)
+        return releases
+
+    @classmethod
+    def get_windows_versions(cls):
+        response = requests.get(cls.windows_api_url)
+        response.raise_for_status()
+
+        versions = []
+        all_git_releases = response.json()
+        for item in all_git_releases:
+            if re.match(r'^[0-9.]+$', item['tag_name']):
+                versions.append(item['tag_name'])
+        return versions
+
+    @classmethod
+    def find_most_recent_version(cls, system_os, cpu, lts_only=False):
+        pass
+
+    @classmethod
+    def download_engine(cls, version, download_location, system_os=None, cpu=None):
+        system_os = system_os or platform.system().lower().replace('darwin', 'macos')
+        cpu = cpu or platform.machine().lower().replace('amd64', 'x64')
+
+        # Verify requested version is available
+        remote_url = None
+        versions_per_os = {'linux': cls.get_linux_versions, 'macos': cls.get_macos_versions, 'windows': cls.get_windows_versions}
+        if not versions_per_os.get(system_os):
+            logger.error(f"Cannot find version list for {system_os}")
+            return
+        if version not in versions_per_os[system_os]():
+            logger.error(f"Cannot find FFMPEG version {version} for {system_os}")
+
+        # Platform specific naming cleanup
+        if system_os == 'macos':
+            remote_url = os.path.join(cls.macos_url, f"ffmpeg-{version}.zip")
+            download_location = os.path.join(download_location, f'ffmpeg-{version}-{system_os}-{cpu}')  # override location to match linux
+        elif system_os == 'linux':
+            release_dir = 'releases' if version == cls.get_linux_versions()[0] else 'old-releases'
+            remote_url = os.path.join(cls.linux_url, release_dir, f'ffmpeg-{version}-{cpu}-static.tar.xz')
+        elif system_os == 'windows':
+            remote_url = f"{cls.windows_download_url.strip('/')}/{version}/ffmpeg-{version}-full_build.zip"
+
+        # Download and extract
+        try:
+            logger.info(f"Requesting download of ffmpeg-{version}-{system_os}-{cpu}")
+            download_and_extract_app(remote_url=remote_url, download_location=download_location)
+
+            # naming cleanup to match existing naming convention
+            if system_os == 'linux':
+                os.rename(os.path.join(download_location, f'ffmpeg-{version}-{cpu}-static'),
+                          os.path.join(download_location, f'ffmpeg-{version}-{system_os}-{cpu}'))
+            elif system_os == 'windows':
+                os.rename(os.path.join(download_location, f'ffmpeg-{version}-full_build'),
+                          os.path.join(download_location, f'ffmpeg-{version}-{system_os}-{cpu}'))
+
+        except IndexError:
+            logger.error("Cannot download requested engine")
+
+
+if __name__ == "__main__":
+    logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+    # print(FFMPEGDownloader.download_engine('6.0', '/Users/brett/zordon-uploads/engines/'))
+    print(FFMPEGDownloader.download_engine(version='6.0', download_location='/Users/brett/zordon-uploads/engines/'))