#!/bin/sh

col_pname=""
col_error=""
col_info=""
col_info2=""
col_rst=""

echo "${col_info}Re-run this script to update (from Github) and rebuild the module.$col_rst"

#
# Clone or pull
#

ZINIT_HOME="${ZDOTDIR:-$HOME}/.zinit"

if ! test -d "$ZINIT_HOME"; then
    mkdir "$ZINIT_HOME"
    chmod g-rwX "$ZINIT_HOME"
fi

echo ">>> Downloading zdharma/zplugin module to $ZINIT_HOME/mod-bin"
if test -d "$ZINIT_HOME/mod-bin/.git"; then
    cd "$ZINIT_HOME/mod-bin"
    git pull origin master
else
    cd "$ZINIT_HOME"
    git clone --depth 10 https://github.com/zdharma/zplugin.git mod-bin
fi
echo ">>> Done"

#
# Build the module
#

cd "$ZINIT_HOME/mod-bin/zmodules"
echo "$col_pname== Building module zdharma/zplugin, running: a make clean, then ./configure and then make ==$col_rst"
echo "$col_pname== The module sources are located at: $ZINIT_HOME/mod-bin/zmodules ==$col_rst"
test -f Makefile && { [ "$1" = "--clean" ] && {
      echo "$col_info2-- make distclean --$col_rst"
      make distclean
      true
  } || {
      echo "$col_info2-- make clean (pass --clean to invoke \`make distclean') --$col_rst"
      make clean
  }
}
echo "$col_info2-- ./configure --$col_rst"
CPPFLAGS=-I/usr/local/include CFLAGS="-g -Wall -O3" LDFLAGS=-L/usr/local/lib ./configure --disable-gdbm --without-tcsetpgrp && {
  echo "$col_info2-- make --$col_rst"
  make && {
    echo "${col_info}Module has been built correctly.$col_rst"
    echo "To load the module, add following 2 lines to .zshrc, at top:"
    echo "    module_path+=( \"$ZINIT_HOME/mod-bin/zmodules/Src\" )"
    echo "    zmodload zdharma/zplugin"
    echo ""
    echo "After loading, use command \`zpmod' to communicate with the module."
    echo "See \`zpmod -h' for more information. There are two main features,"
    echo "invocation of \`zpmod source-study' which shows \`source' profile"
    echo "data, and guaranteed, automatic compilation of any sourced script"
    echo "while the module is loaded (check with Zsh command \`zmodload')."
  } || {
      echo "${col_error}Module didn't build.$col_rst. You can copy the error messages and submit"
      echo "error-report at: https://github.com/zdharma/zinit/issues"
  }
}