path: root/stikman

#! /bin/sh

# https://gitlab.com/sinanmohd/stikman

####################
## User variables ##
####################

# dir where stickers are stored
stick_path="${HOME}"/.local/share/sticker
# resolution of stickers during creation
stick_resolution=512 #p
# path where stickers are temporarily stored during ouput
output_path="${HOME}"/.cache/sticker
# output format of stickers
output_format=webp
# output resolution of stickers
output_resolution=480 #p
# sleep time during output, increade this if you are stuck at the upload screen
output_wait_time=.2
# location where shared sticker packs are uploaded
upload_locations="https://0.vern.cc
                  https://0x0.st
                  https://bin.kernal.eu
                  https://bin.sinanmohd.com
                  https://envs.sh
                  https://null.slipfox.xyz
                  https://ttm.sh
                  https://x0.at"

image_picker=sxiv
command -v nsxiv 1>/dev/null && image_picker=nsxiv

print_help() {
	while IFS= read -r Line; do
		printf "%s\n" "$Line";
	done <<EOF
Usage: stikman [OPTION]

stikman install <pack.zip>      install sticker pack
stikman rm                      remove sticker pack
stikman view                    view sticker pack
stikman add     <image file>    add a new item to sticker pack 
stikman share                   get a shareable link for sticker pack
stikman rename                  rename a sticker pack
stikman fetch   <link>          install pack from a link 
stikman create  <directory>     make pack from images inside the directory
stikman help                    print this menu

EOF
}

get_stick_grp() {
    find "$stick_path" -mindepth 1 -maxdepth 1 -type d -printf '%f\n' | dmenu -p "select sticker pack" -l 25
}

get_stick() {
    stick_grp="$(get_stick_grp)"
    if [ -z "$stick_grp" ]
    then
        notify-send "stikman" "stickers are not installed :( \nrun stikman help"
        printf "stickers are not installed :( \nrun stikman help\n"
        exit
    fi

    "$image_picker" -bfqrto -k 1 "$stick_path"/"$stick_grp" 2>/dev/null | tail -n 1
}

coom() {
    convert "$1"    \
    -strip \
    -quality 70% \
    -resize "$output_resolution" \
    "$output_path"/sticker."$output_format"
}

to_webp() {
    dep_ck "mogrify"
    mogrify \
    -strip \
    -resize "$stick_resolution" \
    -path "$1" \
    -format webp ./"$2"/*.jpeg ./"$2"/*.jpg ./"$2"/*.png ./"$2"/*.heif ./"$2"/*.heic ./"$2"/*.avif > /dev/null 2>&1 
}

input_stick() {
    printf "file://%s/sticker.%s" "$output_path" "$output_format" | xclip -selection clipboard -t text/uri-list
    xdotool key a
    xdotool key BackSpace
    xdotool key ctrl+v
    sleep "$output_wait_time" && xdotool key enter
}

dep_ck () {
    for Dep; do
        if ! command -v "$Dep" 1>/dev/null; then
            printf "%s not found. Please install it.\n" "$Dep" >&2
            command -v pacman 1>/dev/null && printf "run pacman -F %s to see which program they belong to\n" "$Dep"
            exit
        fi
    done
    unset Dep
}

get_pack_name() {
    basename "$(find "$1" -maxdepth 1 -type d | sed 1d | tail -n 1)"
}

pack_install() {
    [ -z "$1" ] && { print_help; exit; } 
    dep_ck "unzip"
    [ ! -d "$stick_path" ] && mkdir -p "$stick_path"

    mkdir -p "$output_path"/installing
    unzip -d "$output_path"/installing "$1" > /dev/null 2>&1

    pack_name="$(get_pack_name "$output_path"/installing)"
    if [ -z "$pack_name" ] || [ -n "${pack_name##*[!0-9]*}" ]
    then
        pack_name=${1%.zip}
        mkdir "$stick_path"/"$pack_name"
        mv "$output_path"/installing/* "$stick_path"/"$pack_name"
    else
        cp -r "$output_path"/installing/"$pack_name" "$stick_path" 
    fi

    to_webp "$stick_path"/"$pack_name" "$stick_path"/"$pack_name"
    cd "$stick_path"/"$pack_name" || exit
    rm ./*.jpeg ./*.jpg ./*.png ./*.heif ./*.heic ./*.avif > /dev/null 2>&1 
    rm -r "$output_path"/installing
}

pack_remove() {
    stick_grp="$(get_stick_grp)"
    [ -n "$stick_grp" ] && rm -r "${stick_path:?}"/"$stick_grp" &&
         printf "sticker pack %s was successfully removed\n" "$stick_grp"
}

get_upload_location() {
    echo "$upload_locations" | shuf -n1 | tr -d ' '
}

pack_share() {
    dep_ck "curl"
    stick_grp="$(get_stick_grp)"
    cd "$stick_path" || exit
    zip -r "$output_path"/share.zip "$stick_grp" > /dev/null 2>&1
    cd "$output_path" || exit
    curl -F'file=@share.zip' "$(get_upload_location)" | tr -d '\n' | xclip -sel clip
    rm share.zip 
    notify-send "stikman" "output coppied to clipboard"
}

pack_fetch() {
    [ -z "$1" ] && { print_help; exit; } 
    dep_ck "curl"
    curl -o "$output_path"/shared.zip "$1"
    pack_install "$output_path"/shared.zip
    rm "$output_path"/shared.zip
}

pack_view() {
    stick_grp="$(get_stick_grp)"
    "$image_picker" -bfrt "$stick_path"/"$stick_grp"
}

pack_rename() {
    stick_grp="$(get_stick_grp)"
    new_name="$(echo "" | dmenu -p "Enter new pack name")"
    mv "$stick_path"/"$stick_grp" "$stick_path"/"$new_name"
}

pack_create() {
    [ -z "$1" ] && { print_help; exit; } 
    printf "converting files...\n"
    mkdir -p "$stick_path"/"$1"
    to_webp "$stick_path"/"$1" "$1"
    cp "$1"/*.webp "$stick_path"/"$1" > /dev/null 2>&1
}

stick_add() {
    [ -z "$1" ] && { print_help; exit; } 
    stick_grp="$(get_stick_grp)"
    convert -strip \
            -resize "$stick_resolution" \
            "$1" "$stick_path"/"$stick_grp"/"${1%.*}".webp
}

main() {
    dep_ck "convert" "xdotool" "xclip" "$image_picker" "dmenu" 
    [ ! -d "$output_path" ] && mkdir -p "$output_path"
    # rm "$output_path"/sticker."$output_format"
    stick_sel="$(get_stick)"
    coom "$stick_sel"
    input_stick
}

case "$1" in
    "") main ;;
    install) pack_install "$2" ;;
    rm) pack_remove ;;
    view) pack_view ;;
    rename) pack_rename ;;
    add) stick_add "$2";;
    share) pack_share ;;
    fetch) pack_fetch "$2" ;;
    create) pack_create "$2" ;;
    help) print_help ;;
    *) print_help;;
esac