#!/bin/bash

select_mode=false
clip_mode=false
jpeg=false

# Parse arguments
while [[ $# -gt 0 ]]; do
    case $1 in
        -s|--select)
            select_mode=true
            shift
            ;;
        -c|--clip)
            clip_mode=true
            shift
            ;;
        -j|--jpg|--jpeg)
            jpeg=true
            shift
            ;;
        *)
            echo "Usage: $(basename "$0") [-s|--select] [-c|--clip] [-j|--jpg|--jpeg]"
            exit 1
            ;;
    esac
done

# Handle different combinations
if [ "$clip_mode" = true ]; then
    if [ "$select_mode" = true ]; then
        maim -us
    else
        maim
    fi | xclip -selection clipboard -t image/png
else
    filename=~/$(date +%s%3N)
    #filename=~/$(date +%Y%m%d_%H%M%S)
    if [ "$select_mode" = true ]; then
        maim -us "$filename.png"
    else
        maim "$filename.png"
    fi
    if [ "$jpeg" = true ]; then
        magick "$filename.png" "$filename.jpg"
        rm "$filename.png"
    fi
fi