From 96287130e6228c11dc539bc096fdf4bbdafc8296 Mon Sep 17 00:00:00 2001
From: Armaan Bhojwani <me@armaanb.net>
Date: Sun, 10 Jan 2021 11:35:29 -0500
Subject: [PATCH] Add more flexibility to usage of lolcat-gen

---
 lolcat-gen | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/lolcat-gen b/lolcat-gen
index 00b60ad..333bf73 100755
--- 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
-- 
2.39.2