1 files changed, 10 insertions, 10 deletions

diff --git a/tools/wiki/diary b/tools/wiki/diary
index 924c99c..d7257de 100755
--- a/tools/wiki/diary
+++ b/tools/wiki/diary
@@ -4,17 +4,17 @@
 #echo
 YEAR=22
 
-read -p 'Month? ' -r MONTH
-MONTH=$(printf "%02d" $MONTH)
+while read -p 'Month? ' -r MONTH; do
+  [ -z $MONTH ] && MONTH=$(date +%m)
+  MONTH=$(printf "%02d" $MONTH)
+  date -d $YEAR-$MONTH-01 &>/dev/null && break
+done
 
 while read -p 'Day? ' -r DAY; do
+  [ -z $DAY ] && DAY=$(date +%d)
   DAY=$(printf "%02d" $DAY)
-  FILE=~/vimwiki/diary/20$YEAR-$MONTH-$DAY.md
-
-  if [ -e $FILE ]; then
-    vim -c 'Goyo' $FILE
-    break
-  else
-    echo No diary at 20$YEAR-$MONTH-$DAY
-  fi
+  date -d $YEAR-$MONTH-$DAY &>/dev/null && break
 done
+
+FILE=~/vimwiki/diary/20$YEAR-$MONTH-$DAY.md
+vim -c 'Goyo' $FILE