#!/bin/bash

# SaMpeg version
SAMPEG_VER="2.2024d"

# Function to check if NVENC is supported
check_nvenc_support() {
    if ffmpeg -hide_banner -encoders 2>/dev/null | grep -qE "(nvenc|cuda)"; then
        if [ -e "/dev/nvidia0" ]; then
            echo "NVENC encoding supported"
            return 0
        else
            echo "No NVIDIA GPU detected. NVENC encoding not supported."
            return 1
        fi
    else
        echo "NVENC encoding not supported"
        return 1
    fi
}


# Function to generate concat.txt from a folder
generate_concat_file() {
    folder="$1"
    output="$2"
    cd "$folder" || exit
    files=$(ls *.mp4 | sort -n)
    for file in $files; do
        echo "file '$folder/$file'" >>"$output"
    done
    echo "Generated concat.txt"
}

# Function to combine clips using concat file and scale
combine_using_concat_file() {
    concat_file="$1"
    output="$2"
    resolution="${3:-3840:2160}"  # Default resolution is 4k (16:9)
    if check_nvenc_support; then
        ffmpeg -f concat -safe 0 -i "$concat_file" -vf "scale=$resolution:force_original_aspect_ratio=decrease,pad=$resolution:(ow-iw)/2:(oh-ih)/2" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    else
        ffmpeg -f concat -safe 0 -i "$concat_file" -vf "scale=$resolution:force_original_aspect_ratio=decrease,pad=$resolution:(ow-iw)/2:(oh-ih)/2" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    fi
    echo "Clips combined and scaled successfully"
}

# Function to crop a clip
crop_clip() {
    input="$1"
    output="$2"
    width="$3"
    height="$4"
    x_position="$5"
    y_position="$6"
    if check_nvenc_support; then
        ffmpeg -i "$input" -vf "crop=$width:$height:$x_position:$y_position" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    else
        ffmpeg -i "$input" -vf "crop=$width:$height:$x_position:$y_position" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    fi
    echo "Clip cropped successfully"
}

# Function to trim a clip from the beginning
trim_beginning() {
    input="$1"
    output="$2"
    start_time="$3"
    if check_nvenc_support; then
        ffmpeg -i "$input" -ss "$start_time" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    else
        ffmpeg -i "$input" -ss "$start_time" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    fi
    echo "Clip trimmed from beginning successfully"
}

# Function to trim a clip from the end
trim_end() {
    input="$1"
    output="$2"
    end_time="$3"
    if check_nvenc_support; then
        ffmpeg -i "$input" -to "$end_time" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    else
        ffmpeg -i "$input" -to "$end_time" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    fi
    echo "Clip trimmed from end successfully"
}

# Function to scale a clip
scale_clip() {
    input="$1"
    output="$2"
    resolution="${3:-3840x2160}"  # Default resolution is 4k (16:9)
    if check_nvenc_support; then
        ffmpeg -i "$input" -vf "scale=$resolution:force_original_aspect_ratio=decrease,pad=$resolution:(ow-iw)/2:(oh-ih)/2" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    else
        ffmpeg -i "$input" -vf "scale=$resolution:force_original_aspect_ratio=decrease,pad=$resolution:(ow-iw)/2:(oh-ih)/2" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    fi
    echo "Clip scaled successfully"
}

# Function to remove parts without audio from a clip
remove_silence() {
    input="$1"
    output="$2"
    ffmpeg -i "$input" -af silenceremove=1:0:-50dB "$output"
    echo "Silence removed successfully"
}

# Function for picture-in-picture
picture_in_picture() {
    input_main="$1"
    input_pip="$2"
    output="$3"
    x_scale="$4"
    y_scale="$5"
    x_position="$6"
    y_position="$7"
    if check_nvenc_support; then
        ffmpeg -i "$input_main" -i "$input_pip" -filter_complex "[1:v]scale=$x_scale:$y_scale [pip]; [0:v][pip]overlay=$x_position:$y_position" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    else
        ffmpeg -i "$input_main" -i "$input_pip" -filter_complex "[1:v]scale=$x_scale:$y_scale [pip]; [0:v][pip]overlay=$x_position:$y_position" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    fi
    echo "Picture-in-picture applied successfully"
}

# Function for image stabilization
image_stabilization() {
    input="$1"
    output="$2"
    if check_nvenc_support; then
        ffmpeg -i "$input" -vf vidstabdetect=shakiness=10:accuracy=15:result=transform.trf -f null -
        ffmpeg -i "$input" -vf vidstabtransform=input="transform.trf" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    else
        ffmpeg -i "$input" -vf vidstabdetect=shakiness=10:accuracy=15:result=transform.trf -f null -
        ffmpeg -i "$input" -vf vidstabtransform=input="transform.trf" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    fi
    echo "Image stabilization applied successfully"
}

# Function to set brightness of a clip
set_brightness() {
    input="$1"
    output="$2"
    brightness="$3"
    if check_nvenc_support; then
        ffmpeg -i "$input" -vf "eq=brightness=$brightness" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    else
        ffmpeg -i "$input" -vf "eq=brightness=$brightness" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    fi
    echo "Brightness set successfully"
}

# Function to set hue of a clip
set_hue() {
    input="$1"
    output="$2"
    hue=$(echo "$3 * 359" | bc)
    if check_nvenc_support; then
        ffmpeg -i "$input" -vf "hue=h=$hue" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    else
        ffmpeg -i "$input" -vf "hue=h=$hue" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    fi
    echo "Hue set successfully"
}

# Function to set saturation of a clip
set_saturation() {
    input="$1"
    output="$2"
    saturation="$3"
    if check_nvenc_support; then
        ffmpeg -i "$input" -vf "eq=saturation=$saturation" -c:v h264_nvenc -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    else
        ffmpeg -i "$input" -vf "eq=saturation=$saturation" -c:v libx264 -preset medium -crf 23 -c:a aac -b:a 128k "$output"
    fi
    echo "saturation set successfully"
}

# Function to record screen with optional webcam and microphone as picture-in-picture
record_screen() {
    screen_resolution=$(xrandr | awk '/ primary/{getline; print $1}')
    output="$1"
    include_webcam="$2"
    include_microphone="$3"
    include_audio="$4"

    # Set default webcam and microphone devices
    webcam_device="/dev/video0"  # Default webcam device
    microphone_device="default"        # Default microphone device

    # Set up inputs for webcam and microphone if included
    webcam_input=""
    microphone_input=""
    if [[ "$include_webcam" == "true" ]]; then
        webcam_input="-f v4l2 -video_size 320x180 -framerate 30 -thread_queue_size 225 -i /dev/video0"
    fi
    if [[ "$include_microphone" == "true" ]]; then
        if [[ ! "$mic" == "" ]]; then
            microphone_input="-f alsa -i default -ac 1 -f mp3 -acodec libmp3lame"
        fi
    fi

    # Set up audio output options
    audio_output=""
    if [[ "$include_audio" == "true" ]]; then
        if [[ ! "$audio_speakers" == "" ]]; then
            audio_output="-f alsa -i -i hw:Loopback,1,0 -ac 2 -f mp3 -acodec libmp3lame"
        fi
    fi

    # Determine screen capture method based on display server
    if [[ "$XDG_SESSION_TYPE" == "wayland" ]]; then
        # Wayland screen capture
        ffmpeg -f x11grab -thread_queue_size 450 -video_size "$screen_resolution" -framerate 45 -i "$DISPLAY" $webcam_input $microphone_input \
               -filter_complex "[0:v][1:v]overlay=10:10" \
               $audio_output \
               -r 30 -preset ultrafast -qp 0 \
               "$output"
    else
        # X11 screen capture
        ffmpeg -f x11grab -thread_queue_size 450 -video_size "$screen_resolution" -framerate 45 -i :0.0 $webcam_input $microphone_input \
               -filter_complex "[0:v][1:v]overlay=10:10" \
               $audio_output \
               -r 30 -preset ultrafast -qp 0 \
               "$output"
    fi


    echo "Screen recorded successfully"
}


# Function to add a layer of audio from a video and audio file
add_audio_layer() {
    input_video="$1"
    input_audio="$2"
    output="$3"
    ffmpeg -i "$input_video" -i "$input_audio" -c:v copy -c:a aac -strict experimental -shortest "$output"
    echo "Audio layer added successfully"
}

# Function to display the SaMpeg header
display_header() {
    toilet -f pagga -F gay "SaMpeg"
    echo "The video fusing interface."
    echo "Powered by FFmpeg."
    echo "------------"
}

# Function to display the command list
display_commands() {
    echo "Usage:"
    echo "  $0 generate-concat-file <folder> <output>"
    echo "    Generates a concat.txt file from a folder containing video files."

    echo "  $0 combine-using-concat-file <concat_file> <output> [resolution]"
    echo "    Combines video clips using the concat file and scales them to the specified resolution."

    echo "  $0 crop <input> <output> <width> <height> <x_position> <y_position>"
    echo "    Crops a video clip to the specified dimensions and position."

    echo "  $0 trim-beginning <input> <output> <start_time>"
    echo "    Trims the beginning of a video clip starting from the specified time."

    echo "  $0 trim-end <input> <output> <end_time>"
    echo "    Trims the end of a video clip ending at the specified time."

    echo "  $0 scale <input> <output> [resolution]"
    echo "    Scales a video clip to the specified resolution."

    echo "  $0 remove-silence <input> <output>"
    echo "    Removes parts of the video clip without audio."

    echo "  $0 picture-in-picture <input_main> <input_pip> <output> <x_scale> <y_scale> <x_position> <y_position>"
    echo "    Places a smaller video (picture-in-picture) onto a larger video."

    echo "  $0 image-stabilization <input> <output>"
    echo "    Stabilizes the video to reduce shaking."

    echo "  $0 record-screen <output> [include_webcam] [include_microphone] [include_audio]"
    echo "    Records the screen with optional webcam, microphone, and desktop audio."
    echo "    [include_webcam]: true/false, [include_microphone]: true/false, [include_audio]: true/false."
    
    echo "  $0 set-brightness <input> <output> <brightness_value>"
    echo "    Adjusts the brightness of the video. The value ranges from -1 to 1."

    echo "  $0 set-hue <input> <output> <hue_value>"
    echo "    Adjusts the hue of the video. The value ranges from 0 to 1."

    echo "  $0 set-saturation <input> <output> <saturation_value>"
    echo "    Adjusts the saturation of the video. The value ranges from -1 to infinity."

    echo "  $0 add-audio-layer <input_video> <input_audio> <output>"
    echo "     Adds a layer of audio and outputs to a video."

    echo "  $0 help"
    echo "    Displays this help message."
}


# Function to display help message
display_help() {
    display_header
    display_commands
}

# Main script

if [[ $# -lt 1 ]]; then
    display_help
    exit 1
fi

while [[ $# -gt 0 ]]; do
    option="$1"
    case $option in
    generate-concat-file)
        if [[ $# -lt 3 ]]; then
            echo "Usage: $0 generate-concat-file <folder> <output>"
            exit 1
        fi
        generate_concat_file "$2" "$3"
        ;;
    combine-using-concat-file)
        if [[ $# -lt 3 ]]; then
            echo "Usage: $0 combine-using-concat-file <concat_file> <output> [resolution]"
            exit 1
        fi
        combine_using_concat_file "$2" "$3" "$4"
        ;;
    crop)
        if [[ $# -ne 7 ]]; then
            echo "Usage: $0 crop <input> <output> <width> <height> <x_position> <y_position>"
            exit 1
        fi
        crop_clip "$2" "$3" "$4" "$5" "$6" "$7"
        ;;
    trim-beginning)
        if [[ $# -ne 4 ]]; then
            echo "Usage: $0 trim-beginning <input> <output> <start_time>"
            exit 1
        fi
        trim_beginning "$2" "$3" "$4"
        ;;
    trim-end)
        if [[ $# -ne 4 ]]; then
            echo "Usage: $0 trim-end <input> <output> <end_time>"
            exit 1
        fi
        trim_end "$2" "$3" "$4"
        ;;
    scale)
        if [[ $# -lt 3 ]]; then
            echo "Usage: $0 scale <input> <output> [resolution]"
            exit 1
        fi
        scale_clip "$2" "$3" "$4"
        ;;
    remove-silence)
        if [[ $# -ne 3 ]]; then
            echo "Usage: $0 remove-silence <input> <output>"
            exit 1
        fi
        remove_silence "$2" "$3"
        ;;
    picture-in-picture)
        if [[ $# -ne 8 ]]; then
            echo "Usage: $0 picture-in-picture <input_main> <input_pip> <output> <x_scale> <y_scale> <x_position> <y_position>"
            exit 1
        fi
        picture_in_picture "$2" "$3" "$4" "$5" "$6" "$7" "$8"
        ;;
    image-stabilization)
        if [[ $# -ne 3 ]]; then
            echo "Usage: $0 image-stabilization <input> <output>"
            exit 1
        fi
        image_stabilization "$2" "$3"
        ;;
    record-screen)
        if [[ $# -lt 2 ]]; then
            echo "Usage: $0 record-screen <output> [include_webcam] [include_microphone] [include_audio]"
            exit 1
        fi
        record_screen "$2" "$3" "$4"
        ;;
    help)
        display_help
        exit 0
        ;;
    add-audio-layer)
        if [[ $# -lt 3 ]]; then
            echo "Usage: $0 add-audio-layer <input_video> <input_audio> <output>"
            exit 1
        fi
        add_audio_layer "$2" "$3" "$4"
        ;;
    set-brightness)
        if [[ $# -lt 3 ]]; then
            echo "Usage: $0 set-brightness <input_video> <input_audio> <output>"
            exit 1
        fi
        set_brightness "$2" "$3" "$4"
        ;;
    set-hue)
        if [[ $# -lt 3 ]]; then
            echo "Usage: $0 set-hue <input_video> <input_audio> <output>"
            exit 1
        fi
        set_hue "$2" "$3" "$4"
        ;;
    set-saturation)
        if [[ $# -lt 3 ]]; then
            echo "Usage: $0 set-saturation <input_video> <input_audio> <output>"
            exit 1
        fi
        set_saturation "$2" "$3" "$4"
        ;;
    *)
        exit 1
        ;;
    esac
    shift
done