#!/bin/sh

USDA_DB_FILE="sr28asc.zip"
USDA_DB_URL="https://www.ars.usda.gov/ARSUserFiles/80400525/Data/SR/SR28/dnload/$USDA_DB_FILE"
NUTR_DEF_FILE="NUTR_DEF.txt"
SCRIPT_DIR="$(dirname "$0")"
USDA_DIR="$SCRIPT_DIR/../usda"
DB="ieat"
COLLECTION="usda-nutrdef"

if [ ! -d "$USDA_DIR" ]; then
    if ! mkdir "$USDA_DIR"; then
        echo 1>&2 "Failed to create directory: '$USDA_DIR'."
        exit 1
    fi
fi
if [ ! -f "$USDA_DIR/$USDA_DB_FILE" ]; then
    if ! wget -P "$USDA_DIR" "$USDA_DB_URL"; then
        echo 1>&2 "Error downloading USDA ASCII database."
        exit 1
    fi
fi
if ! unzip -o "$USDA_DIR/$USDA_DB_FILE" $NUTR_DEF_FILE -d "$USDA_DIR"; then
    echo 1>&2 "Error extracting USDA ASCII database."
    echo 1>&2 "File, '$USDA_DIR/$USDA_DB_FILE', may be corrupted."
    exit 1
fi
"$SCRIPT_DIR"/nutr_def.jq "$USDA_DIR"/$NUTR_DEF_FILE |
    mongoimport --drop -d $DB -c $COLLECTION