Add Problem

Add getinput on aoc

sbin/aoc

@@ -1,14 +1,11 @@
 #!/bin/bash
 
 today() {
-  echo "Going to today"
   TODAYDIR="$AOCROOT/$CURRYEAR/day$CURRDAY" 
   if [ ! -d $TODAYDIR ]; then
-    echo "> Doesn't exist, making"
     mkdir $TODAYDIR
   fi
-  echo "> cd $TODAYDIR"
-  cd $TODAYDIR
+  echo $TODAYDIR
 }
 
 getsharelink() {
@@ -29,6 +26,7 @@ getsharelink() {
 getproblem() {
   YR=$CURRYEAR
   DY=$CURRDAY
+  DIRDY=$CURRDAY
 
   PWD=$(pwd)
   PWD=${PWD#$AOCROOT/}
@@ -38,7 +36,10 @@ getproblem() {
     DY=${DY/day0/}
     DY=${DY/day/}
   fi
-  cd "$AOCROOT/$YR/day$DY"
+  if [[ ! -d "$AOCROOT/$YR/day$DIRDY" ]]; then
+    mkdir "$AOCROOT/$YR/day$DIRDY"
+  fi
+  cd "$AOCROOT/$YR/day$DIRDY"
   if [[ -f "problem" ]]; then
     echo "Found problem file."
     if [[ $1 != "-f" ]]; then
@@ -51,10 +52,46 @@ getproblem() {
   # Remove any zero padding from day
   DY=$(echo "$CURRDAY"|awk '$0*=1')
   echo "Getting problem at $CURRYEAR/day/$DY"
-  # elinks -dump https://adventofcode.com/$CURRYEAR/day/$DY > problem
+  elinks -dump https://adventofcode.com/$CURRYEAR/day/$DY > problem
   vim problem
 }
 
+getinput() {
+  YR=$CURRYEAR
+  DY=$CURRDAY
+  DIRDY=$CURRDAY
+
+  PWD=$(pwd)
+  PWD=${PWD#$AOCROOT/}
+  if [[ "$PWD" =~ ^[1-9][0-9]{3}/day[0-2][0-9]$ ]]; then
+    YR=${PWD:0:4}
+    DY=${PWD:5:5}
+    DY=${DY/day0/}
+    DY=${DY/day/}
+  fi
+  if [[ ! -d "$AOCROOT/$YR/day$DIRDY" ]]; then
+    mkdir "$AOCROOT/$YR/day$DIRDY"
+  fi
+  cd "$AOCROOT/$YR/day$DIRDY"
+  if [[ -f "input" ]]; then
+    echo "Found input file."
+    if [[ $1 != "-f" ]]; then
+      # Check if the file already exists
+      echo "Not forcing overwrite."
+      exit 1
+    fi
+  fi
+
+  # Remove any zero padding from day
+  DY=$(echo "$CURRDAY"|awk '$0*=1')
+  echo "Getting input for $CURRYEAR/day/$DY"
+  echo "https://adventofcode.com/$CURRYEAR/day/$DY/input"
+  echo "session=$(getsessiontoken)"
+  curl --cookie "session=$(getsessiontoken)" https://adventofcode.com/$CURRYEAR/day/$DY/input > input
+  # elinks -dump https://adventofcode.com/$CURRYEAR/day/$DY/input > input
+  # elinks -dump https://adventofcode.com/$CURRYEAR/day/$DY/input
+}
+
 getsessiontoken() {
   FULLCOOKIE=$(grep "adventofcode" $HOME/.elinks/cookies)
   if [[ ! -n "$FULLCOOKIE" ]]; then
@@ -68,10 +105,16 @@ help() {
   echo "Usage: aoc [option]"
   echo "Available Options:"
   echo "" 
+  echo "  today           - Echo the current day's directory, creating it if it doesn't exist"
+  echo "" 
   echo "  getproblem      - Retrieve the problem for the day associated with the current directory"
   echo "                    If the current directory isn't a specific problem (\d\d\d\d/day\d\d/)"
   echo "                    then it changes to today's directory, making it if needed"
   echo ""
+  echo "  getinput        - Retrieve the input for the day associated with the current directory"
+  echo "                    If the current directory isn't a specific problem (\d\d\d\d/day\d\d/)"
+  echo "                    then it changes to today's directory, making it if needed"
+  echo ""
   echo "  getsessiontoken - Echo the session token from the elinks cookie file"
   echo "" 
   echo "  getsharelink    - Get the repo remote path for the main.go file in the current directory"
@@ -83,8 +126,14 @@ help() {
 }
 
 case $1 in
+  "today") 
+    today
+    ;;
   "getproblem") 
-    getproblem $2
+    getproblem
+    ;;
+  "getinput") 
+    getinput
     ;;
   "getsessiontoken")
     getsessiontoken