almanac

#!/usr/bin/env bash

while getopts "d:" OPTION; do
  case $OPTION in
      d)
          DIFFTOOL=${OPTARG}
          ;;
      *)
          echo "Unknow comandline switch: $OPTION"
          exit 1
  esac
done

if ! command -v yq &> /dev/null; then
  echo "Could not find 'yq'."
  echo "Please visit https://github.com/mikefarah/yq/#install"
  exit 1
fi

CH_CFG_PATH=~/.chia/mainnet/config/config.yaml
if [[ ! -f $CH_CFG_PATH ]]; then
  echo "Could not find $CH_CFG_PATH"
  exit 1
fi

if [[ "$(git rev-parse --show-toplevel 2>/dev/null)" != $(pwd -P) ]]; then
  echo "Not in chia-blockchain project root."
  exit 1
fi
python -m pip freeze | grep 'pyaml' > /dev/null 2>&1
RESULT=$?
if [[ $RESULT -eq 1 ]]; then
	echo "This requires the python module 'pyaml'."
  echo "Install this with 'pip install pyaml' and try again."
  exit 1
fi

GIT_REMOTE=$(git remote -v 2>/dev/null | head -n1 | awk '{print $2}')
REPO=$(echo "${GIT_REMOTE}" | sed -e 's,.*:\(.*/\)\?,,' -e 's/\.git$//')
if [[ $REPO != chia-blockchain ]]; then
  echo "You must run this command from the chia-blockchain project root."
  exit 1
fi


TMPCUR=$(mktemp /tmp/curconfXXXXXXXXXX)
TMPNEW=$(mktemp /tmp/newconfXXXXXXXXXX)
TMPNOPOOL=$(mktemp /tmp/nopoolXXXXXXXXXX)
CH_CFG_INIT=$PWD/chia/util/initial-config.yaml
CUR_TS=$(date "+%d%b%Y_%H-%M-%S")
CH_CFG_BAK=$CH_CFG_PATH-$CUR_TS
TOOL=${DIFFTOOL:-vimdiff}

yq --front-matter=process '.pool.pool_list=[]' "$CH_CFG_PATH" > "$TMPNOPOOL"
yq '.pool.pool_list' "$CH_CFG_PATH" -s '"/tmp/tmppool"'

python -m pyaml "$TMPNOPOOL" > "$TMPCUR"
python -m pyaml "$CH_CFG_INIT" > "$TMPNEW"

VIMDIFF_CONTROLS=$(bat -Ppl cmd-help <<- EOF

^[[38;5;186mVIMDIFF CONTROLS ~^[[0m
^[[38;5;203m|^[[0m^[[38;5;231m   ]c           | advance to the next block with differences |^[[0m
^[[38;5;203m|^[[0m^[[38;5;231m   [c           | reverse search for the previous block with differences |^[[0m
^[[38;5;203m|^[[0m^[[38;5;231m   do           | bring changes from the other file to the current file |^[[0m
^[[38;5;203m|^[[0m^[[38;5;231m   dp           | send changes from the current file to the other file |^[[0m
^[[38;5;203m|^[[0m^[[38;5;231m   zo           | unfold/unhide text |^[[0m
^[[38;5;203m|^[[0m^[[38;5;231m   zc           | refold/rehide text |^[[0m
^[[38;5;203m|^[[0m^[[38;5;231m   zr           | unfold both files completely |^[[0m
^[[38;5;203m|^[[0m^[[38;5;231m   zm           | fold both files completely |^[[0m

^[[38;5;203mNOTE ABOUT DIFF OBTAIN/PUT CONTROL^[[0m^[[38;5;231m:^[[0m
^[[38;5;231m-^[[0m^[[38;5;231m ^[[0m^[[38;5;203mNormal Mode^[[0m^[[38;5;231m:^[[0m^[[38;5;231m ^[[0m^[[38;5;186mUse "do" and "dp" controls when your cursor is on, or just one line under a block.^[[0m
^[[38;5;231m-^[[0m^[[38;5;231m ^[[0m^[[38;5;203mVisual Mode^[[0m^[[38;5;231m:^[[0m^[[38;5;231m ^[[0m^[[38;5;186mUse ":'<,'>diffget" and ":'<,'>diffput" when selecting lines of text.^[[0m
^[[38;5;231m  ^[[0m^[[38;5;186mSee also ":h copy-diffs".^[[0m

^[[38;5;186m"^[[0m^[[38;5;186m:diffupdate^[[0m^[[38;5;186m"^[[0m^[[38;5;231m ^[[0m^[[38;5;186mwill re-scan the files for changes.^[[0
^[[38;5;186mWhen you are finished editing, press "escape", and enter ":wqa" and press "enter".^[[0m

EOF
)

echo "You will now see a diff view of:"
echo "$CH_CFG_PATH"
echo "and"
echo "$CH_CFG_INIT"
echo "This will open in $TOOL."
if [[ $TOOL == "vimdiff" ]]; then
  echo -e "$VIMDIFF_CONTROLS"
  echo ""
fi
read -rp "Press enter to continue."

$TOOL "$TMPCUR" "$TMPNEW"

read -r -p "Do you want to replace your current Chia config file with your saved changes? [y/N] " CFG_CONFIRM
case "$CFG_CONFIRM" in
  [yY][eE][sS]|[yY])
    mv "$CH_CFG_PATH" "$CH_CFG_BAK"
    #cp "$TMPCUR" "$CH_CFG_PATH"
    yq '.pool.pool_list |=load("/tmp/tmppool.yml")' "$TMPCUR" > "$CH_CFG_PATH"
    ;;
  *)
    exit 0
    ;;
esac

rm "$TMPCUR" "$TMPNEW" /tmp/tmppool.yml