ppm

#!/bin/bash

# Persistent Playlist Manager - M3U8 Playlist Manager
# This script manages persistent playlists using m3u8 files and hard links
#
# Kaya-Sem
# Version 1.0

# Configuration
DATA_DIR="$HOME/Music/PPM"
ALBUMS_DIR="$DATA_DIR/playlists"
MYPLAYCOMMAND="mpv --shuffle"

# Initialize the data directory structure
init_directories() {
    if [ ! -d "$ALBUMS_DIR" ]; then
        mkdir -p "$ALBUMS_DIR"
    fi
}

# Display usage information
show_usage() {
    local script_name="$(basename "$0")"
    cat >&2 << EOF
Usage: $script_name <command> [options]

Commands:
  list                                         List all playlists
  tracks <playlist>                            List songs in a playlist
  add <playlist> [song1] [song2] ...           Add songs to a playlist
  remove <playlist> <song1> [song2] ...        Remove songs from a playlist
  create [playlist1] [playlist2] ...           Create new playlists
  delete <playlist>                            Delete a playlist
  play <playlist>                              Play a playlist using the configured player

Examples:
  $script_name list
  $script_name tracks 'My Favorites'
  $script_name add 'Rock Hits' /path/to/song1.flac /path/to/song2.mp3
  $script_name remove 'Rock Hits' song1.mp3 song2.mp3
  $script_name create 'Rock Hits' 'My Favorites'
  $script_name create < playlist_names.txt
  $script_name add 'Rock Hits' <<< 'song1.mp3'
  $script_name play 'Rock Hits'
EOF
}

# Check if playlist exists
playlist_exists() {
    local playlist_name="$1"
    [ -d "$ALBUMS_DIR/$playlist_name" ] && [ -f "$ALBUMS_DIR/$playlist_name/$playlist_name.m3u8" ]
}

# Check if song exists in playlist
song_in_playlist() {
    local playlist_name="$1"
    local song_name="$2"
    local playlist_file="$ALBUMS_DIR/$playlist_name/$playlist_name.m3u8"
    
    if [ -f "$playlist_file" ]; then
        grep -q "^#EXTINF:.*$song_name$" "$playlist_file" 2>/dev/null
    else
        return 1
    fi
}

# Get absolute path of a file
get_absolute_path() {
    local file_path="$1"
    if [[ "$file_path" == /* ]]; then
        echo "$file_path"
    else
        echo "$(pwd)/$file_path"
    fi
}

# Validate song file
validate_song_file() {
    local song_path="$1"
    local abs_path=$(get_absolute_path "$song_path")
    
    if [ ! -f "$abs_path" ]; then
        echo -e "track file not found: $song_path" >&2
        return 1
    fi
    
    # Check if it's an audio file (basic check)
    local extension="${abs_path##*.}"
    case "${extension,,}" in
        mp3|m4a|flac|wav|ogg|aac|wma)
            return 0
            ;;
        *)
            echo -e "$song_path may not be an audio file" >&2
            return 0
            ;;
    esac
}

# List all playlists
list_playlists() {
    if [ ! -d "$ALBUMS_DIR" ] || [ -z "$(ls -A "$ALBUMS_DIR" 2>/dev/null)" ]; then
        echo "No playlists found." >&2
        return
    fi
    
    local count=0
    for playlist_dir in "$ALBUMS_DIR"/*/; do
        if [ -d "$playlist_dir" ]; then
            local playlist_name=$(basename "$playlist_dir")
            local playlist_file="$playlist_dir$playlist_name.m3u8"
            if [ -f "$playlist_file" ]; then
                local song_count=$(grep -c "^#EXTINF:" "$playlist_file" 2>/dev/null)
                echo -e "$playlist_name ($song_count tracks)"
                ((count++))
            fi
        fi
    done
    
    if [ $count -eq 0 ]; then
        echo "No playlists found."
    fi
}

# List songs in a playlist
list_songs() {
    local playlist_name="$1"
    
    if [ -z "$playlist_name" ]; then
        echo -e "${RED}Error:${NC} Playlist name is required"
        echo "Usage: $(basename "$0") list-songs <playlist_name>"
        exit 1
    fi
    
    local playlist_file="$ALBUMS_DIR/$playlist_name/$playlist_name.m3u8"
    
    if [ ! -f "$playlist_file" ]; then
        echo -e "playlist '$playlist_name' not found" >&2
        exit 1
    fi

    
    local count=0
    while IFS= read -r line; do
        if [[ "$line" =~ ^#EXTINF: ]]; then
            # Extract song name from EXTINF line
            local song_name=$(echo "$line" | sed 's/^#EXTINF:.*,//')
            echo -e "$song_name"
            ((count++))
        fi
    done < "$playlist_file"
    
    if [ $count -eq 0 ]; then
        echo "No songs in this playlist."
    fi
}

# Create new playlists
create_playlists() {
    local playlists=()
    
    # If arguments provided, use them
    if [ $# -gt 0 ]; then
        playlists=("$@")
    else
        # Read from STDIN
        while IFS= read -r line; do
            # Skip empty lines and comments
            if [[ -n "$line" && ! "$line" =~ ^[[:space:]]*# ]]; then
                playlists+=("$line")
            fi
        done
    fi
    
    if [ ${#playlists[@]} -eq 0 ]; then
        echo -e "no playlist names provided" >&2
        echo "Usage: $(basename "$0") create [playlist1] [playlist2] ..." >&2
        echo "       echo 'playlist1' | $(basename "$0") create" >&2
        exit 1
    fi
    
    local created_count=0
    local skipped_count=0
    
    for playlist_name in "${playlists[@]}"; do
        # Trim whitespace
        playlist_name=$(echo "$playlist_name" | xargs)
        
        if [ -z "$playlist_name" ]; then
            continue
        fi
        
        if playlist_exists "$playlist_name"; then
            echo -e "'$playlist_name' already exists" >&2
            ((skipped_count++))
            continue
        fi
        
        # Create playlist directory
        local playlist_dir="$ALBUMS_DIR/$playlist_name"
        mkdir -p "$playlist_dir"
        
        # Create m3u8 file with header
        cat > "$playlist_dir/$playlist_name.m3u8" << EOF
#EXTM3U
# Created by Persistent Playlist Manager
# Playlist: $playlist_name
# Created: $(date)

EOF
        
    done
    
}

# Add songs to a playlist
add_songs() {
    local playlist_name="$1"
    shift
    
    if [ -z "$playlist_name" ]; then
        echo -e "playlist name is required" >&2
        echo "Usage: $(basename "$0") add <playlist_name> [song1] [song2] ..." >&2
        echo "       echo 'song1.mp3' | $(basename "$0") add <playlist_name>" >&2
        exit 1
    fi
    
    local songs=()
    
    # If arguments provided, use them
    if [ $# -gt 0 ]; then
        songs=("$@")
    else
        # Read from STDIN
        while IFS= read -r line; do
            # Skip empty lines and comments
            if [[ -n "$line" && ! "$line" =~ ^[[:space:]]*# ]]; then
                songs+=("$line")
            fi
        done
    fi
    
    if [ ${#songs[@]} -eq 0 ]; then
        echo -e "No tracks provided" >&2
        echo "Usage: $(basename "$0") add <playlist_name> [song1] [song2] ..." >&2
        echo "       echo 'song1.mp3' | $(basename "$0") add <playlist_name>" >&2
        exit 1
    fi
    
    # Create playlist if it doesn't exist
    if ! playlist_exists "$playlist_name"; then
        create_playlists "$playlist_name"
    fi
    
    local playlist_dir="$ALBUMS_DIR/$playlist_name"
    local playlist_file="$playlist_dir/$playlist_name.m3u8"
    local added_count=0
    local skipped_count=0
    
    for song_path in "${songs[@]}"; do
        # Trim whitespace
        song_path=$(echo "$song_path" | xargs)
        
        if [ -z "$song_path" ]; then
            continue
        fi
        
        if ! validate_song_file "$song_path"; then
            continue
        fi
        
        local abs_path=$(get_absolute_path "$song_path")
        local song_name=$(basename "$song_path")
        
        # Check if song is already in playlist
        if song_in_playlist "$playlist_name" "$song_name"; then
            echo -e "'$song_name' already in playlist '$playlist_name'" >&2
            ((skipped_count++))
            continue
        fi
        
        # Create hard link in playlist directory
        local link_path="$playlist_dir/$song_name"
        
        # Handle duplicate filenames
        local counter=1
        local original_name="$song_name"
        while [ -f "$link_path" ]; do
            local name_without_ext="${original_name%.*}"
            local extension="${original_name##*.}"
            song_name="${name_without_ext}_$counter.$extension"
            link_path="$playlist_dir/$song_name"
            ((counter++))
        done
        
        # Create hard link
        if ! ln "$abs_path" "$link_path" 2>/dev/null; then
            echo -e "Failed to add '$song_name' to playlist '$playlist_name'" >&2
        else
            # Add to m3u8 file
            echo "#EXTINF:-1,$song_name" >> "$playlist_file"
            echo "$link_path" >> "$playlist_file"
        fi
    done
    
}

# Remove songs from a playlist
remove_songs() {
    local playlist_name="$1"
    shift
    
    if [ -z "$playlist_name" ]; then
        echo -e "playlist name is required" >&2
        echo "Usage: $(basename "$0") remove <playlist> <song1> [song2] ..." >&2
        exit 1
    fi
    
    if [ $# -eq 0 ]; then
        echo -e "at least one track is required" >&2
        echo "Usage: $(basename "$0") remove <playlist> <song1> [song2] ..." >&2
        exit 1
    fi
    
    local playlist_dir="$ALBUMS_DIR/$playlist_name"
    local playlist_file="$playlist_dir/$playlist_name.m3u8"
    
    if [ ! -f "$playlist_file" ]; then
        echo -e "playlist '$playlist_name' not found" >&2
        exit 1
    fi
    
    local removed_count=0
    local not_found_count=0
    
    for song_name in "$@"; do
        if song_in_playlist "$playlist_name" "$song_name"; then
            # Create temporary file without the song
            local temp_file=$(mktemp)
            local in_song_block=false
            local skip_next=false
            
            while IFS= read -r line; do
                if [[ "$line" =~ ^#EXTINF: ]]; then
                    # Check if this is the song we want to remove
                    local current_song=$(echo "$line" | sed 's/^#EXTINF:.*,//')
                    if [ "$current_song" = "$song_name" ]; then
                        skip_next=true
                        # Remove hard link if it exists
                        local next_line=$(head -n 1)
                        if [ -f "$next_line" ]; then
                            rm -f "$next_line"
                        fi
                        continue
                    fi
                    in_song_block=true
                    echo "$line" >> "$temp_file"
                elif [ "$skip_next" = true ]; then
                    skip_next=false
                    continue
                else
                    echo "$line" >> "$temp_file"
                fi
            done < "$playlist_file"
            
            # Replace original file with temp file
            mv "$temp_file" "$playlist_file"
        else
            echo -e "track '$song_name' not found in playlist '$playlist_name'" >&2
        fi
    done
    
}

# Delete a playlist
delete_playlist() {
    local playlist_name="$1"
    
    if [ -z "$playlist_name" ]; then
        echo -e "playlist name is required" >&2
        echo "Usage: $(basename "$0") delete <playlist>" >&2
        exit 1
    fi
    
    if ! playlist_exists "$playlist_name"; then
        echo -e "playlist '$playlist_name' not found" >&2
        exit 1
    fi
    
    local playlist_dir="$ALBUMS_DIR/$playlist_name"
    local playlist_file="$playlist_dir/$playlist_name.m3u8"
    
    # Remove hard links for songs in this playlist
    while IFS= read -r line; do
        if [[ "$line" =~ ^[^#] ]] && [ -f "$line" ]; then
            rm -f "$line"
        fi
    done < "$playlist_file"
    
    # Remove the entire playlist directory
    rm -rf "$playlist_dir"
    
}

# Play a playlist
play_playlist() {
    local playlist_name="$1"
    if [ -z "$playlist_name" ]; then
        echo -e "${RED}Error:${NC} Playlist name is required"
        echo "Usage: $(basename "$0") play <playlist_name>"
        exit 1
    fi
    local playlist_dir="$ALBUMS_DIR/$playlist_name"
    local playlist_file="$playlist_dir/$playlist_name.m3u8"
    if [ ! -f "$playlist_file" ]; then
        echo -e "${RED}Error:${NC} Playlist '$playlist_name' not found"
        exit 1
    fi
    (cd "$playlist_dir" && eval $MYPLAYCOMMAND "$playlist_name.m3u8")
}

# Main script logic
main() {
    # Initialize directories
    init_directories
    
    case "${1:-}" in
        "list")
            list_playlists
            ;;
        "tracks")
            list_songs "$2"
            ;;
        "add")
            shift
            add_songs "$@"
            ;;
        "remove")
            shift
            remove_songs "$@"
            ;;
        "create")
            shift
            create_playlists "$@"
            ;;
        "delete")
            delete_playlist "$2"
            ;;
        "play")
            play_playlist "$2"
            ;;
        "help"|"-h"|"--help"|"")
            show_usage
            ;;
        *)
            echo -e "unknown command '$1'" >&2
            echo "Use '$(basename "$0") help' for usage information" >&2
            exit 1
            ;;
    esac
}

# Run main function with all arguments
main "$@"