dune-weaver/modules/core/playlist_manager.py

import json
import os
import logging
import asyncio
from typing import Optional
from modules.core import pattern_manager
from modules.core.state import state

# Configure logging
logger = logging.getLogger(__name__)

# Track the current playlist task so we can cancel it properly
_current_playlist_task: Optional[asyncio.Task] = None

# Global state
PLAYLISTS_FILE = os.path.join(os.getcwd(), "playlists.json")

# Ensure the file exists and contains at least an empty JSON object
if not os.path.isfile(PLAYLISTS_FILE):
    logger.info(f"Creating new playlists file at {PLAYLISTS_FILE}")
    with open(PLAYLISTS_FILE, "w") as f:
        json.dump({}, f, indent=2)

def load_playlists():
    """Load the entire playlists dictionary from the JSON file."""
    try:
        with open(PLAYLISTS_FILE, "r") as f:
            content = f.read().strip()
            if not content:
                logger.warning("Playlists file is empty, returning empty dict")
                return {}
            playlists = json.loads(content)
            logger.debug(f"Loaded {len(playlists)} playlists")
            return playlists
    except (json.JSONDecodeError, ValueError) as e:
        logger.error(f"Playlists file is corrupted, resetting to empty: {e}")
        save_playlists({})
        return {}

def save_playlists(playlists_dict):
    """Save the entire playlists dictionary back to the JSON file."""
    logger.debug(f"Saving {len(playlists_dict)} playlists to file")
    with open(PLAYLISTS_FILE, "w") as f:
        json.dump(playlists_dict, f, indent=2)

def list_all_playlists():
    """Returns a list of all playlist names."""
    playlists_dict = load_playlists()
    playlist_names = list(playlists_dict.keys())
    logger.debug(f"Found {len(playlist_names)} playlists")
    return playlist_names

def get_playlist(playlist_name):
    """Get a specific playlist by name."""
    playlists_dict = load_playlists()
    if playlist_name not in playlists_dict:
        logger.warning(f"Playlist not found: {playlist_name}")
        return None
    logger.debug(f"Retrieved playlist: {playlist_name}")
    return {
        "name": playlist_name,
        "files": playlists_dict[playlist_name]
    }

def create_playlist(playlist_name, files):
    """Create or update a playlist."""
    playlists_dict = load_playlists()
    playlists_dict[playlist_name] = files
    save_playlists(playlists_dict)
    logger.info(f"Created/updated playlist '{playlist_name}' with {len(files)} files")
    return True

def modify_playlist(playlist_name, files):
    """Modify an existing playlist."""
    logger.info(f"Modifying playlist '{playlist_name}' with {len(files)} files")
    return create_playlist(playlist_name, files)

def delete_playlist(playlist_name):
    """Delete a playlist."""
    playlists_dict = load_playlists()
    if playlist_name not in playlists_dict:
        logger.warning(f"Cannot delete non-existent playlist: {playlist_name}")
        return False
    del playlists_dict[playlist_name]
    save_playlists(playlists_dict)
    logger.info(f"Deleted playlist: {playlist_name}")
    return True

def add_to_playlist(playlist_name, pattern):
    """Add a pattern to an existing playlist."""
    playlists_dict = load_playlists()
    if playlist_name not in playlists_dict:
        logger.warning(f"Cannot add to non-existent playlist: {playlist_name}")
        return False
    playlists_dict[playlist_name].append(pattern)
    save_playlists(playlists_dict)
    logger.info(f"Added pattern '{pattern}' to playlist '{playlist_name}'")
    return True

def rename_playlist(old_name, new_name):
    """Rename an existing playlist."""
    if not new_name or not new_name.strip():
        logger.warning("Cannot rename playlist: new name is empty")
        return False, "New name cannot be empty"

    new_name = new_name.strip()

    playlists_dict = load_playlists()
    if old_name not in playlists_dict:
        logger.warning(f"Cannot rename non-existent playlist: {old_name}")
        return False, "Playlist not found"

    if old_name == new_name:
        return True, "Name unchanged"

    if new_name in playlists_dict:
        logger.warning(f"Cannot rename playlist: '{new_name}' already exists")
        return False, "A playlist with that name already exists"

    # Copy files to new key and delete old key
    playlists_dict[new_name] = playlists_dict[old_name]
    del playlists_dict[old_name]
    save_playlists(playlists_dict)
    logger.info(f"Renamed playlist '{old_name}' to '{new_name}'")
    return True, f"Playlist renamed to '{new_name}'"

async def cancel_current_playlist():
    """Cancel the current playlist task if one is running."""
    global _current_playlist_task
    if _current_playlist_task and not _current_playlist_task.done():
        logger.info("Cancelling existing playlist task...")
        _current_playlist_task.cancel()
        try:
            await _current_playlist_task
        except asyncio.CancelledError:
            logger.info("Playlist task cancelled successfully")
        except Exception as e:
            logger.warning(f"Error while cancelling playlist task: {e}")
        _current_playlist_task = None

async def run_playlist(playlist_name, pause_time=0, clear_pattern=None, run_mode="single", shuffle=False, runs_per_day=1):
    """Run a playlist with the given options.

    runs_per_day applies only when run_mode == "scheduled": the playlist will be played
    that many full cycles per 24 hours, evenly spread across the day's active hours
    (Still Sands periods are excluded from the daily quota).
    """
    global _current_playlist_task

    # Cancel any existing playlist task first
    await cancel_current_playlist()

    # Also stop any running pattern
    if pattern_manager.get_pattern_lock().locked():
        logger.info("Another pattern is running, stopping it first...")
        await pattern_manager.stop_actions()

    playlists = load_playlists()
    if playlist_name not in playlists:
        logger.error(f"Cannot run non-existent playlist: {playlist_name}")
        return False, "Playlist not found"

    file_paths = playlists[playlist_name]
    file_paths = [os.path.join(pattern_manager.THETA_RHO_DIR, file) for file in file_paths]

    if not file_paths:
        logger.warning(f"Cannot run empty playlist: {playlist_name}")
        return False, "Playlist is empty"

    try:
        extra = f", runs_per_day={runs_per_day}" if run_mode == "scheduled" else ""
        logger.info(f"Starting playlist '{playlist_name}' with mode={run_mode}, shuffle={shuffle}{extra}")
        # Set ALL playlist state variables BEFORE creating the async task.
        # This ensures state is correct even if the task doesn't start immediately
        # (important for TestClient which may cancel background tasks).
        state.current_playlist = file_paths
        state.current_playlist_name = playlist_name
        state.playlist_mode = run_mode
        state.current_playlist_index = 0
        _current_playlist_task = asyncio.create_task(
            pattern_manager.run_theta_rho_files(
                file_paths,
                pause_time=pause_time,
                clear_pattern=clear_pattern,
                run_mode=run_mode,
                shuffle=shuffle,
                runs_per_day=runs_per_day,
            )
        )
        return True, f"Playlist '{playlist_name}' is now running."
    except Exception as e:
        logger.error(f"Failed to run playlist '{playlist_name}': {str(e)}")
        return False, str(e)