Merge pull request #9 from AryToNeX/master

Proper arguments support and support w3m backend

nekofetch

@@ -1,24 +1,61 @@
 #!/bin/sh
 
 tmpfile="$(mktemp)"
+imgtype=""
+use_height=""
 
+while :; do
     case "$1" in
         "--nsfw"|"nsfw"|"-n"|"n")
+            if [ "$imgtype" = "" ]; then
                 [ "$DEBUG" = "true" ] && echo "Getting a nsfw image"
-        imgurl="https://nekos.life/api/v2/img/cum_jpg"
+                imgtype="nsfw"
+            else
+                echo "$imgtype images were already chosen! Please only specify either sfw or nsfw!"
+            fi
             ;;
         "--sfw"|"sfw"|"-s"|"s")
+            if [ "$imgtype" = "" ]; then
                 [ "$DEBUG" = "true" ] && echo "Getting a sfw image"
-        imgurl="https://nekos.life/api/v2/img/neko"
+                imgtype="sfw"
+            else
+                echo "$imgtype images were already chosen! Please only specify either sfw or nsfw!"
+            fi
+            ;;
+        "--w3m"|"w3m"|"--img"|"img"|"-i"|"i")
+            [ "$DEBUG" = "true" ] && echo "Using w3m image backend for neofetch"
+            use_w3m="true"
+            ;;
+        "--height"|"-h")
+            if [ "$use_height" = "" ]; then
+                use_height="$2"
+                [ "$DEBUG" = "true" ] && echo "Using height $height"
+            else
+                echo "You can only specify the height argument once!"
+            fi
+            shift
             ;;
         *)
-        [ "$DEBUG" = "true" ] && echo "Could not interpret as either sfw or nsfw. Defaulting to sfw."
+            if [ -z "$1" ]; then
-        imgurl="https://nekos.life/api/v2/img/neko"
+                break
+            fi
+            echo "Could not interpret parameter '$1'."
             ;;
     esac
+    shift
+done
 
-echo "$2" | grep -qE '^[0-9]+$' && height="$2" || height="$(($(stty size | awk '{print $1}') - 5))"
+if [ "$imgtype" = "nsfw" ]; then
-[ "$DEBUG" = "true" ] && echo "Using height $height"
+    imgurl="https://nekos.life/api/v2/img/cum_jpg"
+else
+    imgurl="https://nekos.life/api/v2/img/neko"
+fi
+
+if [ "$use_height" != "" ]; then
+    height="$use_height"
+else
+    height="$(($(stty size | awk '{print $1}') - 5))"
+fi
 
 url=$(curl -fsSL "$imgurl" | jq -r ".url")
 
@@ -31,6 +68,8 @@ if [ "$TERM" = "xterm-kitty" ]; then
     fi
 elif [ "$LC_TERMINAL" = "iTerm2" ]; then
     neofetch --iterm2 "$tmpfile.jpg"
+elif [ "$use_w3m" = "true" ]; then
+    neofetch --w3m "$tmpfile.jpg"
 else
     jp2a --height="$height" "$tmpfile.jpg" > "$tmpfile"
     neofetch --source "$tmpfile"