Add more flexibility to usage of lolcat-gen

diff --git a/lolcat-gen b/lolcat-gen
index 00b60adfc8fa026ef5057f8d0b8072b669ff482a..333bf7379cd1302848933601dd64d2b100e7a8bf 100755 (executable)
--- a/lolcat-gen
+++ b/lolcat-gen
@@ -1,10 +1,23 @@
 #!/usr/bin/env bash
 # Generates lolcated logo for my shell rc
 
+if [[ ${2} ]]; then
+  FILE=${2}
+else
+  FILE="~/Code/personal-logo/square/unicode/small"
+fi
+
+if [[ ${1} ]]; then
+  NUM=${1}
+else
+  NUM="100"
+fi
+
 [[ -d ~/.local/share/zsh/lolcat/ ]] || mkdir -p ~/.local/share/zsh/lolcat
-rm -rf ~/.local/share/zsh/lolcat/*
 
-for (( i = 0; i < $1 ; i++ )); do
-  cat ~/Code/personal-logo/ascii/ascii-small-white | lolcat --force > ~/.local/share/zsh/lolcat/$i
+rm ~/.local/share/zsh/lolcat/*
+
+for (( i = 0; i < ${NUM} ; i++ )); do
+  cat $FILE | lolcat --force > ~/.local/share/zsh/lolcat/$i
   echo -ne "$i/$1 done" \\r
 done