#!/usr/bin/env python3
"""
Path mapping utilities for NFOGuard
Handles conversion between external service paths and container paths
"""
import os
import re
from pathlib import Path

class PathMapper:
    """Handles path mapping between different environments"""
    
    def __init__(self, config):
        """Initialize path mapper with configuration"""
        self.radarr_roots = [path.strip() for path in config.radarr_root_folders.split(',') if path.strip()]
        self.sonarr_roots = [path.strip() for path in config.sonarr_root_folders.split(',') if path.strip()]
        self.movie_paths = [path.strip() for path in config.movie_paths.split(',') if path.strip()]
        self.tv_paths = [path.strip() for path in config.tv_paths.split(',') if path.strip()]

    def sonarr_path_to_container_path(self, sonarr_path: str) -> str:
        """Convert Sonarr path to container path using environment mappings"""
        # Sort roots by length (longest first) to avoid substring matching issues
        indexed_roots = [(i, root) for i, root in enumerate(self.sonarr_roots)]
        indexed_roots.sort(key=lambda x: len(x[1]), reverse=True)
        
        # Try to match against configured Sonarr root folders (longest first)
        for original_index, sonarr_root in indexed_roots:
            if sonarr_path.startswith(sonarr_root + '/') or sonarr_path == sonarr_root:
                # Map to corresponding TV path
                if original_index < len(self.tv_paths):
                    container_root = self.tv_paths[original_index]
                    relative_path = sonarr_path[len(sonarr_root):].lstrip('/')
                    result = str(Path(container_root) / relative_path) if relative_path else container_root
                    return result
        
        # No fallback - if path mapping fails, return original and let validation catch it
        return sonarr_path
    
    def radarr_path_to_container_path(self, radarr_path: str) -> str:
        """Convert Radarr path to container path using environment mappings"""
        # Sort roots by length (longest first) to avoid substring matching issues
        indexed_roots = [(i, root) for i, root in enumerate(self.radarr_roots)]
        indexed_roots.sort(key=lambda x: len(x[1]), reverse=True)
        
        # Try to match against configured Radarr root folders (longest first)
        for original_index, radarr_root in indexed_roots:
            if radarr_path.startswith(radarr_root + '/') or radarr_path == radarr_root:
                # Map to corresponding movie path
                if original_index < len(self.movie_paths):
                    container_root = self.movie_paths[original_index]
                    relative_path = radarr_path[len(radarr_root):].lstrip('/')
                    result = str(Path(container_root) / relative_path) if relative_path else container_root
                    return result
        
        # No fallback - if path mapping fails, return original and let validation catch it
        return radarr_path
    
    def container_path_to_host_path(self, container_path: str) -> str:
        """Convert container path back to host path if needed"""
        # This might be needed for file operations
        return container_path