#!/bin/bash
set -euo pipefail
script_dirpath="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
root_dirpath="$(cd "${script_dirpath}/.." && pwd)"

REVERT="NO"
COPY="NO"
MOVE_NEW_FILES="NO"
EE_CODE_DIR="../windmill-ee-private/"

while [[ $# -gt 0 ]]; do
  case $1 in
    -r|--revert)
      # If EE files have been substituted, this will revert them to their initial content. 
      # This relies on `git restore` so the EE files must not be committed to the repo for 
      # this to work (commit hooks should prevent this from happening, as well as the fact
      # that we're using symlinks by default).
      REVERT="YES"
      MOVE_NEW_FILES="YES"
      shift
      ;;
    -c|--copy)
      # By default, EE files are symlinked. Pass this option to do a real copy instead.
      # This might be necessary if you want to build the Docker Image as Docker COPY seems
      # to not follow symlinks. For local development, symlinks should be preferred as they
      # adds a safeguards EE files can't be commited to the OSS repo.
      COPY="YES"
      shift # past argument
      ;;
    -m|--move-new-files)
      # This moves all new EE files from the public repository to the private repository.
      MOVE_NEW_FILES="YES"
      shift # past argument
      ;;
    -d|--dir)
      # Path to the local directory of the windmill-ee-private repository. By defaults, it
      # assumes it is cloned next to the Windmill OSS repo.
      EE_CODE_DIR="$2"
      shift # past argument
      shift # past value
      ;;
    -*|--*)
      echo "Unknown option $1"
      exit 1
      ;;
    *)
      POSITIONAL_ARGS+=("$1") # save positional arg
      shift # past argument
      ;;
  esac
done

if [[ $EE_CODE_DIR == /* ]]; then
  EE_CODE_DIR="${EE_CODE_DIR}"
else
  EE_CODE_DIR="${root_dirpath}/${EE_CODE_DIR}"
fi
echo "EE code directory = ${EE_CODE_DIR} | Revert = ${REVERT}"


if [ ! -d "${EE_CODE_DIR}" ]; then
  echo "Windmill EE repo not found, please clone it next to this repository (or use the --dir option) and try again"
  echo ">   git clone git@github.com:windmill-labs/windmill-ee-private.git"
  echo ""
  exit 0
fi

if [ "$REVERT" == "YES" ]; then
  for ee_file in $(find ${EE_CODE_DIR} -name "*ee.rs"); do
    ce_file="${ee_file/${EE_CODE_DIR}/}"
    ce_file="${root_dirpath}/backend/${ce_file}"
    rm ${ce_file} || true
  done
elif [ "$MOVE_NEW_FILES" == "NO" ]; then
  # This replaces all files in current repo with alternative EE files in windmill-ee-private
  for ee_file in $(find "${EE_CODE_DIR}" -name "*ee.rs"); do
  ce_file="${ee_file/${EE_CODE_DIR}/}"
  ce_file="${root_dirpath}/backend/${ce_file}"
    if [ "$COPY" == "YES" ]; then
      cp "${ee_file}" "${ce_file}"
      echo "File copied '${ee_file}' -->> '${ce_file}'"
    else
      ln -s "${ee_file}" "${ce_file}" || true
      echo "Symlink created '${ee_file}' -->> '${ce_file}'"
    fi
  done
fi

if [ "$MOVE_NEW_FILES" == "YES" ]; then
  for ce_file in $(find "${root_dirpath}"/backend/windmill-*/src/ -name "*ee.rs"); do
    backend_dirpath="${root_dirpath}/backend/"
    ee_file="${ce_file/${backend_dirpath}/}"
    ee_file="${EE_CODE_DIR}${ee_file}"
    if [ ! -f "${ee_file}" ]; then
      mv "${ce_file}" "${ee_file}"
      if [ ! "$REVERT" == "YES" ]; then
        ln -s "${ee_file}" "${ce_file}"
      fi
      echo "File moved '${ce_file}' -->> '${ee_file}'"
    fi
  done
fi