1 files changed, 10 insertions, 1 deletions

diff --git a/angou b/angou
index 845670b..c1d06fc 100755
--- a/angou
+++ b/angou
@@ -6,6 +6,9 @@ ANGOU_DIR=${ANGOU_DIR:-"$HOME/.angou/"}
 ANGOU_CLIPBOARD=${ANGOU_CLIPBOARD:-"xsel -i"}
 # sets command to use when listing directory structure
 ANGOU_LIST=${ANGOU_LIST:-"tree"}
+# custom command for generating passwords/users
+ANGOU_GEN_PASS=${ANGOU_GEN_PASS:-"random pass"}
+ANGOU_GEN_USER=${ANGOU_GEN_USER:-"random user"}
 
 mkdir -p "$ANGOU_DIR"
 cd "$ANGOU_DIR" || exit
@@ -35,6 +38,11 @@ edit() {
 
         tmpfile="$(mktemp)"
 
+        if [ "$2" = "generate" ]; then
+                $ANGOU_GEN_PASS >"$tmpfile"
+                echo "Username: $($ANGOU_GEN_USER)" >>"$tmpfile"
+        fi
+
         # edit a copy of a password if it already exists
         [ -f "$1" ] && $ANGOU_DECRYPT "$1" > "$tmpfile"
 
@@ -46,7 +54,7 @@ edit() {
 
 # print usage
 usage() {
-        printf "Usage: %s [help|copy|edit|view [file or directory]]\n" "$0"
+        printf "Usage: %s [help|copy|edit|generate|view [file or directory]]\n" "$0"
         printf "\tThe edit and copy command expects a file as an argument,\n"
         printf "\twhile the view command expects a directory as an argument.\n"
         printf "\tThe arguments for view and edit are relative to\n"
@@ -61,6 +69,7 @@ case "$option" in
         help) usage ;;
         copy) copy "$2" ;;
         edit) edit "$2" ;;
+        generate) edit "$2" generate ;;
         view) view "$2" ;;
         *) view "$1" ;;
 esac