# Copyright (C) 2026 Rootiest
# SPDX-License-Identifier: AGPL-3.0-or-later

# CATEGORY
#   13-media-and-utilities
#
# SYNOPSIS
#   dng2avif [-h] [-i <file>] [-o <file>] [-q <n>] [-s <n>] [input.dng]
#
# DESCRIPTION
#   Converts a DNG raw image to a 10-bit HDR AVIF using a three-step pipeline:
#   develop with ImageMagick, encode with ffmpeg+avifenc, sync metadata with
#   exiftool. Requires magick, ffmpeg, avifenc, and exiftool.
#
# ARGUMENTS
#   -i, --input FILE    Input DNG file
#   -o, --output FILE   Output AVIF file (defaults to input basename)
#   -q, --quality N     Encoding quality 0-100 (default: 92)
#   -s, --speed N       Encoder speed 0-10 (default: 3, 0 = slowest)
#   -h, --help          Show help message
#
# EXIT STATUS
#   0  Conversion complete
#   1  File not found, missing dependency, or encode step failed
#
# EXAMPLE
#   dng2avif photo.dng
#   dng2avif -q 85 -s 5 -i shot.dng -o out.avif
function dng2avif --description 'Convert DNG raw to 10-bit HDR AVIF'
    set -l options (fish_opt -s h -l help)
    set -a options (fish_opt -s i -l input -r)
    set -a options (fish_opt -s o -l output -r)
    set -a options (fish_opt -s q -l quality -r)
    set -a options (fish_opt -s s -l speed -r)

    argparse $options -- $argv
    or return

    # Help Screen
    if set -q _flag_help; or test (count $argv) -eq 0 -a -z "$_flag_input"
        set -l c_head (set_color --bold cyan)
        set -l c_cmd (set_color --bold)
        set -l c_flag (set_color yellow)
        set -l c_dim (set_color brblack)
        set -l c_reset (set_color normal)
        echo "$c_head""Usage:$c_reset $c_cmd""dng2avif$c_reset $c_flag""[options]$c_reset $c_dim""[input.dng]$c_reset"
        echo ""
        echo "$c_head""Options:$c_reset"
        echo "  $c_flag-i$c_reset, $c_flag--input$c_reset $c_dim""FILE$c_reset    Input DNG file"
        echo "  $c_flag-o$c_reset, $c_flag--output$c_reset $c_dim""FILE$c_reset   Output AVIF file (defaults to input name)"
        echo "  $c_flag-q$c_reset, $c_flag--quality$c_reset $c_dim""N$c_reset     Encoding quality 0-100 (default: 92)"
        echo "  $c_flag-s$c_reset, $c_flag--speed$c_reset $c_dim""N$c_reset       Encoder speed 0-10 (default: 3, 0=slowest)"
        echo "  $c_flag-h$c_reset, $c_flag--help$c_reset          Show this help message"
        return 0
    end

    # Handle Input Source
    set -l input ""
    if set -q _flag_input
        set input $_flag_input
    else
        set input $argv[1]
    end

    if not test -f "$input"
        echo (set_color red)"Error: File '$input' not found."(set_color normal)
        return 1
    end

    # Configuration
    set -l quality 92
    if set -q _flag_quality
        set quality $_flag_quality
    end

    set -l speed 3
    if set -q _flag_speed
        set speed $_flag_speed
    end

    set -l basename (string replace -r '\.dng$' '' $input)
    set -l output "$basename.avif"
    if set -q _flag_output
        set output $_flag_output
    end

    set -l temp_pnm "$basename"_temp.pnm

    # Dependency check
    for cmd in magick ffmpeg avifenc exiftool
        if not type -q $cmd
            echo (set_color red)"Error: $cmd is not installed."(set_color normal)
            return 1
        end
    end

    # Step 1: Develop
    echo -n (set_color blue)"Step 1/3: Developing... "(set_color normal)
    magick "$input" -depth 16 pnm:"$temp_pnm" &>/dev/null
    if test $status -ne 0
        echo (set_color red)"FAILED"
        return 1
    end
    echo (set_color green)"DONE"(set_color normal)

    # Step 2: Encode
    echo -n (set_color blue)"Step 2/3: Encoding (Q:$quality, S:$speed)... "(set_color normal)
    ffmpeg -i "$temp_pnm" -vf "format=yuv444p10le" -f yuv4mpegpipe -strict -1 - 2>/dev/null | avifenc -q $quality -s $speed --stdin -d 10 -y 444 --cicp 12/1/1 -r f -o "$output" &>/dev/null
    if test $status -ne 0
        echo (set_color red)"FAILED"
        rm -f "$temp_pnm"
        return 1
    end
    echo (set_color green)"DONE"(set_color normal)

    # Step 3: Metadata
    echo -n (set_color blue)"Step 3/3: Syncing Metadata... "(set_color normal)
    exiftool -tagsFromFile "$input" -all:all "$output" -overwrite_original &>/dev/null
    if test $status -ne 0
        echo (set_color red)"FAILED"
    else
        echo (set_color green)"DONE"(set_color normal)
    end

    # Final Cleanup
    test -f "$temp_pnm"; and rm "$temp_pnm"

    set -l size (stat -c '%s' "$output" | numfmt --to=iec)
    echo (set_color yellow)"Complete: $output ($size)"(set_color normal)
end