diff --git a/laptop_full_setup.sh b/laptop_full_setup.sh index b06c8c5..212fc64 100755 --- a/laptop_full_setup.sh +++ b/laptop_full_setup.sh @@ -225,25 +225,35 @@ if $INSTALL_AI; then echo -e " ${BOLD}[2/6] Where should Ollama models be stored?${NC}" echo " (Models are large — 5-50GB each. A fast SSD or large HDD is ideal.)" echo "" - echo " Available mount points with >20GB free:" - df -h --output=target,avail,fstype 2>/dev/null \ - | awk 'NR>1 && $2~/[0-9]/ { - val=$2; unit=substr(val,length(val)); - num=substr(val,1,length(val)-1)+0; - if ((unit=="G" && num>=20) || unit=="T") print " " $0 - }' | head -10 + mapfile -t _MPTS < <( + df -h --output=target,avail,fstype 2>/dev/null \ + | awk 'NR>1 && $2~/[0-9]/ { + val=$2; unit=substr(val,length(val)); + num=substr(val,1,length(val)-1)+0; + if ((unit=="G" && num>=20) || unit=="T") print $0 + }' | head -10 + ) + echo " 0) Docker volume (default — /var/lib/docker/volumes/)" + for _i in "${!_MPTS[@]}"; do + printf " %d) %s\n" "$((_i+1))" "${_MPTS[$_i]}" + done echo "" - echo " 1) Docker volume (default — stored in /var/lib/docker/volumes/)" - echo " 2) Custom path (e.g. /mnt/ssd/ollama or /data/ollama)" - echo "" - read -rp " Choice [1]: " STORAGE_CHOICE - STORAGE_CHOICE="${STORAGE_CHOICE:-1}" - if [[ "$STORAGE_CHOICE" == "2" ]]; then - read -rp " Enter full path for Ollama models: " OLLAMA_HOST_PATH - OLLAMA_HOST_PATH="${OLLAMA_HOST_PATH%/}" # strip trailing slash + read -rp " Choice [0]: " STORAGE_CHOICE + STORAGE_CHOICE="${STORAGE_CHOICE:-0}" + if [[ "$STORAGE_CHOICE" == "0" ]]; then + OLLAMA_STORAGE="volume" + elif [[ "$STORAGE_CHOICE" =~ ^[0-9]+$ ]] && (( STORAGE_CHOICE >= 1 && STORAGE_CHOICE <= ${#_MPTS[@]} )); then + _MP=$(awk '{print $1}' <<< "${_MPTS[$(( STORAGE_CHOICE-1 ))]}") + OLLAMA_HOST_PATH="$_MP/ollama-models" + OLLAMA_STORAGE="bind" + ok "Ollama models → $OLLAMA_HOST_PATH" + else + # Fallback: treat input as a literal path + OLLAMA_HOST_PATH="${STORAGE_CHOICE%/}" [[ -z "$OLLAMA_HOST_PATH" ]] && die "No path entered." OLLAMA_STORAGE="bind" fi + unset _MPTS _MP _i fi # ── Q3: Storage for Kiwix ZIMs ──────────────────────────────────────────────── @@ -254,19 +264,30 @@ if $SVC_KIWIX; then echo -e " ${BOLD}[3/6] Where should Kiwix ZIM files be stored?${NC}" echo " (ZIMs are large — Wikipedia alone is ~46GB. Total collection ~130GB.)" echo "" - echo " Available mount points with >50GB free:" - df -h --output=target,avail,fstype 2>/dev/null \ - | awk 'NR>1 && $2~/[0-9]/ { - val=$2; unit=substr(val,length(val)); - num=substr(val,1,length(val)-1)+0; - if ((unit=="G" && num>=50) || unit=="T") print " " $0 - }' | head -10 + mapfile -t _MPTS < <( + df -h --output=target,avail,fstype 2>/dev/null \ + | awk 'NR>1 && $2~/[0-9]/ { + val=$2; unit=substr(val,length(val)); + num=substr(val,1,length(val)-1)+0; + if ((unit=="G" && num>=50) || unit=="T") print $0 + }' | head -10 + ) + echo " 0) Default: $KIWIX_DIR" + for _i in "${!_MPTS[@]}"; do + printf " %d) %s\n" "$((_i+1))" "${_MPTS[$_i]}" + done echo "" - echo " Default: $KIWIX_DIR" - read -rp " Press Enter to use default, or type a different path: " KIWIX_INPUT - if [[ -n "$KIWIX_INPUT" ]]; then - KIWIX_DIR="${KIWIX_INPUT%/}" + read -rp " Choice [0]: " KIWIX_CHOICE + KIWIX_CHOICE="${KIWIX_CHOICE:-0}" + if [[ "$KIWIX_CHOICE" != "0" ]] && [[ "$KIWIX_CHOICE" =~ ^[0-9]+$ ]] && (( KIWIX_CHOICE >= 1 && KIWIX_CHOICE <= ${#_MPTS[@]} )); then + _MP=$(awk '{print $1}' <<< "${_MPTS[$(( KIWIX_CHOICE-1 ))]}") + KIWIX_DIR="$_MP/kiwix" + ok "Kiwix ZIMs → $KIWIX_DIR" + elif [[ "$KIWIX_CHOICE" != "0" ]] && [[ -n "$KIWIX_CHOICE" ]]; then + # Fallback: treat as a literal path + KIWIX_DIR="${KIWIX_CHOICE%/}" fi + unset _MPTS _MP _i _KIWIX_CHOICE # ── Q4: Download ZIMs now? ───────────────────────────────────────────────── echo ""