main.py

"""
Spotify Download Utils - Main Application
A modular tool for processing Spotify downloads: decompression, organization, and metadata enhancement.
"""

import sys
import os

# Add src directory to Python path
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))

from src.config import (
    COMPRESSED_FILES_FOLDER,
    OUTPUT_FOLDER,
    STRINGS_TO_REMOVE,
    UNRAR_TOOL_PATH,
    COMPRESSED_EXTENSIONS
)

from src.processors import (
    uncompress_files,
    process_loose_files,
    rename_folders,
    rename_files_with_track_numbers,
    process_album_folders
)


def main():
    """
    Main execution function.
    Orchestrates the entire processing pipeline.
    """
    try:
        print("=" * 60)
        print("🎵 Spotify Download Utils v2.0")
        print("=" * 60)
        print()
        
        # Step 1: Decompress files
        print("🚀 Step 1/5: Decompressing files...")
        uncompress_files(
            COMPRESSED_FILES_FOLDER,
            OUTPUT_FOLDER,
            UNRAR_TOOL_PATH,
            COMPRESSED_EXTENSIONS
        )
        print("✅ Decompression finished.\n")
        
        # Step 2: Process loose files
        print("🚀 Step 2/5: Processing loose files...")
        process_loose_files(
            COMPRESSED_FILES_FOLDER,
            OUTPUT_FOLDER,
            STRINGS_TO_REMOVE
        )
        print("✅ Loose files processing finished.\n")
        
        # Step 3: Rename folders
        print("🚀 Step 3/5: Renaming folders...")
        rename_folders(OUTPUT_FOLDER, STRINGS_TO_REMOVE)
        print("✅ Folder renaming finished.\n")
        
        # Step 4: Rename files with track numbers
        print("🚀 Step 4/5: Adding track numbers to files...")
        rename_files_with_track_numbers(OUTPUT_FOLDER)
        print("✅ File renaming finished.\n")
        
        # Step 5: Add album year
        print("🚀 Step 5/5: Adding album years...")
        process_album_folders(OUTPUT_FOLDER)
        print("✅ Album year processing finished.\n")
        
        print("=" * 60)
        print("✅ All processing completed successfully!")
        print("=" * 60)
        
    except Exception as e:
        print(f"\n❌ Error: {e}")
        sys.exit(1)


if __name__ == "__main__":
    main()