1 files changed, 9 insertions, 1 deletions

diff --git a/contrib/angoumenu b/contrib/angoumenu
index 04bbf95..85ca2e4 100755
--- a/contrib/angoumenu
+++ b/contrib/angoumenu
@@ -3,4 +3,12 @@ ANGOU_DIR="${ANGOU_DIR:-~/.angou}"
 DMENU="dmenu"
 
 cd "$ANGOU_DIR"
-find . -name '*.gpg' | sort | sed 's,^\./,,;s,\.gpg$,,' | $DMENU | xargs angou copy
+file=$(find . -type f -printf '%P\n' | sort | sed 's,^\./,,' | $DMENU)
+
+if [ -z "$1" ]; then
+        angou copy "$file"
+elif [ "$1" = "totp" ]
+        angou totp "$file"
+else
+        exit 1
+fi