import asyncio
from typing import List, Dict
from dataclasses import dataclass
from operator import itemgetter

@dataclass
class LoraCache:
    """Cache structure for LoRA data"""
    raw_data: List[Dict]
    sorted_by_name: List[Dict]
    sorted_by_date: List[Dict]
    folders: List[str]
    
    def __post_init__(self):
        self._lock = asyncio.Lock()

    async def resort(self, name_only: bool = False):
        """Resort all cached data views"""
        async with self._lock:
            self.sorted_by_name = sorted(
                self.raw_data, 
                key=lambda x: x['model_name'].lower()  # Case-insensitive sort
            )
            if not name_only:
                self.sorted_by_date = sorted(
                    self.raw_data, 
                    key=itemgetter('modified'), 
                    reverse=True
                )
            # Update folder list
            self.folders = sorted(list(set(
                l['folder'] for l in self.raw_data
            )))

    async def update_preview_url(self, file_path: str, preview_url: str) -> bool:
        """Update preview_url for a specific lora in all cached data
        
        Args:
            file_path: The file path of the lora to update
            preview_url: The new preview URL
            
        Returns:
            bool: True if the update was successful, False if the lora wasn't found
        """
        async with self._lock:
            # Update in raw_data
            for item in self.raw_data:
                if item['file_path'] == file_path:
                    item['preview_url'] = preview_url
                    break
            else:
                return False  # Lora not found
                
            # Update in sorted lists (references to the same dict objects)
            for item in self.sorted_by_name:
                if item['file_path'] == file_path:
                    item['preview_url'] = preview_url
                    break
                    
            for item in self.sorted_by_date:
                if item['file_path'] == file_path:
                    item['preview_url'] = preview_url
                    break
                    
            return True