feat: improve bash scripts

Author: coko7

README.md

@@ -46,9 +46,11 @@ Options:
           Print help
 ```
 
-3. You can also use the small bash script `pull_all.sh` that uses the `vegapull` CLI to download data for all existing packs:
+3. You can also use the example bash scripts that directly use the `vegapull` CLI to download data for all existing packs:
 ```console
-coko7@example:~$ bash pull_all.sh
+coko7@example:~$ bash scripts/pull-all.sh
+# the `gum` one is more complete but requires some additional tooling to install in your shell:
+coko7@example:~$ bash scripts/pull-all-gum.sh
 ```
 
 ## 🃏 Supported card fields

pull_all.sh

@@ -0,0 +1,95 @@
+#!/usr/bin/env bash
+
+LANGUAGES="🇬🇧 english\n🇯🇵 japanese"
+language=$(echo -e "$LANGUAGES" | gum choose --header="Language:" | cut -d' ' -f2)
+if [ -z "$language" ]; then
+    exit 1
+fi
+
+default_data_dir="data/$language"
+data_dir=$(gum input \
+    --value="$default_data_dir" \
+    --placeholder="Enter a path..." \
+    --prompt="📦 TCG Download location> ")
+if [ -z "$data_dir" ]; then
+    exit 1
+fi
+
+echo "📦 TCG Download location: $data_dir"
+
+if [ -d "$data_dir" ]; then
+    if ! gum confirm "📁 Directory '$data_dir' is about to be wiped to hold new data, do you want to proceed?"; then
+        exit 1
+    fi
+
+    rm -rf "$data_dir"
+fi
+
+mkdir "$data_dir"
+echo -e "📁 Created directory: $data_dir\n"
+
+if ! gum spin --title="VegaPull packs..." -- vegapull --language "$language" packs > "$data_dir/packs.json"; then
+    echo "Failed to pull list of packs using vegapull. Aborted" >&2
+    exit 1
+fi
+
+pack_count=$(jq length "$data_dir"/packs.json)
+
+echo -e "✅ Successfully found $pack_count packs!\n"
+
+function pull_cards() {
+    local index=1
+    local packs
+    packs=$(cat "$data_dir/packs.json")
+
+    while read -r pack_id; do
+        pack_title=$(echo "$packs" | jq --arg pack_id "$pack_id" -r '.[] | select(.id == $pack_id) | .title_parts.title')
+        message="[$index/$pack_count] VagaPulling cards for: $pack_title ($pack_id)..."
+
+        if ! gum spin --title="$message" -- \
+            vegapull --language "$language" cards "$pack_id" > "$data_dir/cards_$pack_id.json"; then
+            echo "Failed to pull cards using vegapull. Aborted" >&2
+            return 1
+        else
+            echo "[$index/$pack_count] Succesfully VegaPulled cards for: $pack_title ($pack_id) ✅"
+        fi
+
+        ((index++))
+    done < <( echo "$packs" | jq -r '.[].id')
+
+    echo "✅ Successfully downloaded data for $index packs!"
+}
+
+if ! pull_cards; then
+    exit 1
+fi
+
+function download_images() {
+    local index=1
+    local packs
+    packs=$(cat "$data_dir/packs.json")
+
+    while read -r pack_id; do
+        local output_dir="$data_dir/images/$pack_id"
+        pack_title=$(echo "$packs" | jq --arg pack_id "$pack_id" -r '.[] | select(.id == $pack_id) | .title_parts.title')
+        echo "[$index/$pack_count] VagaPulling images for: $pack_title ($pack_id)..."
+
+        if ! vegapull --language "$language" images --output-dir="$output_dir" "$pack_id" -vv; then
+            echo "Failed to pull images using vegapull. Aborted" >&2
+            return 1
+        else
+            echo "[$index/$pack_count] Succesfully VegaPulled images for: $pack_title ($pack_id) ✅"
+        fi
+
+        ((index++))
+    done < <( echo "$packs" | jq -r '.[].id')
+
+    echo "✅ Successfully downloaded data for $index packs!"
+
+}
+
+if gum confirm "🖼️ Download card images as well?"; then
+    download_images
+fi
+
+echo "✅ Successfully filled the punk records with latest data"

scripts/pull-all-gum.sh

@@ -0,0 +1,64 @@
+#!/usr/bin/env bash
+
+LANGUAGE="japanese"
+VEGA_DATA=data/$LANGUAGE
+
+if [ -d "$VEGA_DATA" ]; then
+    read -rp "The $VEGA_DATA is about to be wiped to hold the new data, do you want to proceed? (y/N) " confirm
+    case $confirm in
+        [Yy]* ) ;;
+        * ) echo "Aborted" >&2; exit 1 ;;
+    esac
+
+    rm -rf $VEGA_DATA
+fi
+
+mkdir $VEGA_DATA
+echo -e "Created dir: $VEGA_DATA\n"
+
+echo "VegaPulling the list of packs ($LANGUAGE)..."
+
+if ! ./target/release/vegapull --language $LANGUAGE packs > $VEGA_DATA/packs.json; then
+    echo "Failed to pull list of packs using vegapull. Aborted" >&2
+    exit 1
+fi
+
+count=$(jq length $VEGA_DATA/packs.json)
+
+echo -e "Successfully pulled $count packs!\n"
+
+function pull_cards() {
+    local index=1
+    local packs
+    packs=$(cat $VEGA_DATA/packs.json)
+
+    while read -r id; do
+        echo -n "[$index/$count] VagaPulling cards for pack '$id'..."
+        if ! ./target/release/vegapull --language $LANGUAGE cards "$id" > "$VEGA_DATA/cards_$id.json"; then
+            echo "Failure"
+            echo "Failed to pull cards using vegapull. Aborted" >&2
+            return 1
+        fi
+
+        echo " OK"
+        ((index++))
+    done < <( echo "$packs" | jq -r '.[].id')
+
+    echo "Successfully download data for $index packs!"
+}
+
+if ! pull_cards; then
+    exit 1
+fi
+
+function download_images() {
+    echo "NOT IMPLEMENTED YET"
+}
+
+read -rp "Download card images as well? (y/N) " confirm
+case $confirm in
+    [Yy]* ) download_images ;;
+    * ) ;;
+esac
+
+echo "Successfully filled the punk records with latest data"