build.py

import os
import sys

OS = sys.platform
ARCH = "unknown"

if OS == "linux":
    ARCH = os.uname().machine
elif OS == "win32":
    ARCH = os.environ['PROCESSOR_ARCHITECTURE']

    if ARCH == "AMD64":
        ARCH = "x86_64"
    elif ARCH == "ARM64":
        ARCH = "aarch64"

def _print_stage(message):
    print(f"\033[44m=> {message}\033[0m")

def _run_command(command):
    os.system(command)

def _createDir(directory):
    if not os.path.exists(directory):
        os.makedirs(directory)

def _get_7z():
    _run_command("winget install -e --id 7zip.7zip --accept-source-agreements --accept-package-agreements")
    _print_stage("Installed 7z")

def _get_ffmpeg():
    FFMPEG_ARCH = ""

    if ARCH == "x86_64":
        FFMPEG_ARCH = "amd64"
    elif ARCH == "aarch64":
        FFMPEG_ARCH = "arm64"

    if FFMPEG_ARCH == "":
        _print_stage("Unsupported architecture")
        exit(1)
    
    if OS == "linux":
        _run_command(f"curl -L https://johnvansickle.com/ffmpeg/builds/ffmpeg-git-{FFMPEG_ARCH}-static.tar.xz -o ffmpeg.tar.xz")
        _run_command("tar -xf ffmpeg.tar.xz")
        _run_command(f"mv ./ffmpeg-git-*-{FFMPEG_ARCH}-static/ffmpeg ./out/ffmpeg")
        _run_command(f"rm -rf ffmpeg.tar.xz ffmpeg-git-*-{FFMPEG_ARCH}-static")
        _run_command(f"chmod +x out/ffmpeg")
    elif OS == "win32":
        PROGFILES = os.environ['ProgramFiles']
        ZIP_PATH = f"{PROGFILES}\\7-Zip\\7z.exe"

        _get_7z()
        _run_command(f'curl -L https://www.gyan.dev/ffmpeg/builds/ffmpeg-git-essentials.7z -o ffmpeg.7z')
        _run_command(f'"{ZIP_PATH}" e ffmpeg.7z -oout ffmpeg.exe -r')
        _run_command(f'del ffmpeg.7z')

    _print_stage("Downloaded ffmpeg")
    
def install_dependencies():
    print("=> Installing dependencies...")
    _run_command("pip install -r requirements.txt --break-system-packages")

def activate_venv():
    _run_command("python -m venv tvdownloader-venv")

    if OS == "linux":
        _run_command("source tvdownloader-venv/bin/activate")
    elif OS == "win32":
        _run_command("tvdownloader-venv\\Scripts\\activate")

    _print_stage("Activated virtual environment")

def deactivate_venv():
    _run_command("source deactivate")
    _print_stage("Deactivated virtual environment")

def clean():
    _print_stage("Cleaning up...")

    if OS == "linux":
        _run_command("rm -rf dist/ build/ out/ tvdownloader.spec")
    elif OS == "win32":
        _run_command("rmdir /s /q dist build out tvdownloader-venv")
        _run_command("del tvdownloader.spec")

def build_appimage():
    _print_stage(f"Building AppImage for {OS}-{ARCH}...")
    _run_command(f'curl -L https://github.com/AppImage/appimagetool/releases/download/continuous/appimagetool-{ARCH}.AppImage -o appimagetool.AppImage')
    _run_command('chmod +x appimagetool.AppImage')
    _createDir('out/appimage.AppDir/usr/bin')
    _run_command(f'cp out/tvdownloader-{OS}-{ARCH} out/appimage.AppDir/usr/bin/tvdownloader')
    _run_command('cp data/appimage/* out/appimage.AppDir/')
    _run_command('cp data/logo/logo-256x256.png out/appimage.AppDir/')
    _run_command('chmod +x out/appimage.AppDir/AppRun')
    _run_command('chmod +x out/appimage.AppDir/tvdownloader.desktop')
    _run_command(f'ARCH={ARCH} ./appimagetool.AppImage out/appimage.AppDir out/tvdownloader-{OS}-{ARCH}.AppImage')
    _run_command('rm -rf out/appimage.AppDir appimagetool.AppImage')
    _print_stage(f"Built out/tvdownloader-{OS}-{ARCH}.AppImage")


def build():
    clean()
    activate_venv()
    install_dependencies()
    _print_stage(f"Building TV Downloader for {OS}-{ARCH}...")
    _createDir("out")
    _get_ffmpeg()

    if OS == "linux":
        _run_command('pyinstaller -F --add-data "./out/ffmpeg:./ffmpeg/" --add-data "./data/logo/logo-64x64.png:./data/logo/" --windowed --onefile --noconsole --clean --name tvdownloader --icon="./data/logo/logo-256x256.ico"  gui/main.py')
        _run_command(f'mv dist/tvdownloader out/tvdownloader-{OS}-{ARCH}')
        _run_command('rm -rf dist/ build/ tvdownloader.spec ./out/ffmpeg')
        _run_command(f'chmod +x out/tvdownloader-{OS}-{ARCH}')
        _print_stage(f"Built out/tvdownloader-{OS}-{ARCH}")
        build_appimage()
    elif OS == "win32":
        _run_command('pyinstaller.exe -F --add-data "./out/ffmpeg.exe:./ffmpeg/" --add-data "./data/logo/*:./data/logo/" --windowed --onefile --noconsole --clean --name tvdownloader --icon="./data/logo/logo-256x256.ico"  gui/main.py')
        _run_command(f'powershell -command "Move-Item -Path dist/tvdownloader.exe -Destination out/tvdownloader-windows-{ARCH}.exe"')
        _run_command('rmdir /s /q dist build tvdownloader-venv')  
        _run_command('powershell -command "Remove-Item -Path out/ffmpeg.exe"')
        _run_command('powershell -command "Remove-Item -Path tvdownloader.spec"')
        _print_stage(f"Built out/tvdownloader-windows-{ARCH}.exe")

    deactivate_venv()

def main():
    args = sys.argv[1:]
    if len(args) == 0:
        _print_stage("Usage: python build.py [build|clean]")
        exit(1)
    elif args[0] == "build":
        build()
    elif args[0] == "clean":
        clean()
    

if __name__ == "__main__":
    main()