#!/bin/bash -e

export BASE_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
export SCRIPT_DIR="$BASE_DIR/scripts"
export FUNCTIONS_DIR="$BASE_DIR/functions"
export DEPLOY_DIR="$BASE_DIR/deploy"

export IMG_DATE
export WORK_DIR

export IMG_NAME
export HOSTNAME
export USERNAME
export PASSWORD

export STAGE
export STAGE_DIR
export STAGE_WORK_DIR
export ROOTFS_DIR
export BOOTFS_DIR

export IMG_FILE
export ZIP_FILE

export QUILT_NO_DIFF_INDEX=1
export QUILT_NO_DIFF_TIMESTAMPS=1
export QUILT_REFRESH_ARGS='-p ab'

source "$FUNCTIONS_DIR/logging.sh"
source "$SCRIPT_DIR/common.sh"
source "$FUNCTIONS_DIR/dependencies_check.sh"

main() {
  dependencies_check "$BASE_DIR/depends"

  source "$BASE_DIR/config"

  if [ "$(id -u)" != '0' ]; then
    echo 'Please run as root' 1>&2
    exit 1
  fi

  if [ -z "$IMG_NAME" ]; then
    echo 'IMG_NAME not set' 1>&2
    exit 1
  fi

  if [ -z "$HOSTNAME" ]; then
    echo 'HOSTNAME not set' 1>&2
    exit 1
  fi

  if [ -z "$USERNAME" ]; then
    echo 'USERNAME not set' 1>&2
    exit 1
  fi

  if [ -z "$PASSWORD" ]; then
    echo 'PASSWORD not set' 1>&2
    exit 1
  fi

  IMG_DATE="$(date +%Y-%m-%d)"
  WORK_DIR="$BASE_DIR/work/$IMG_DATE-$IMG_NAME"

  STAGE='stage0'
  STAGE_DIR="$BASE_DIR/$STAGE"
  STAGE_WORK_DIR="$WORK_DIR/$STAGE"
  ROOTFS_DIR="$STAGE_WORK_DIR/rootfs"
  BOOTFS_DIR="$ROOTFS_DIR/boot"

  IMG_FILE="$STAGE_WORK_DIR/$IMG_DATE-${IMG_NAME}.img"
  ZIP_FILE="$DEPLOY_DIR/image_$IMG_DATE-${IMG_NAME}.zip"

  tput setaf 2 # Green color
  echo "$IMG_DATE $(date +"%T")"
  echo
  echo "Work dir:       $WORK_DIR"
  echo "Stage dir:      $STAGE_DIR"
  echo "Stage work dir: $STAGE_WORK_DIR"
  echo
  echo "Root FS dir: $ROOTFS_DIR"
  echo "Boot FS dir: $BOOTFS_DIR"
  echo
  echo "Image file: $IMG_FILE"
  echo "ZIP file:   $ZIP_FILE"
  echo
  tput sgr0 # No color

  mkdir -p "$WORK_DIR"
  mkdir -p "$DEPLOY_DIR"

  unmount "$STAGE_WORK_DIR"

  run_sub_stage "$STAGE_DIR/00-substage"

  unmount "$STAGE_WORK_DIR"
}

run_sub_stage() {
  log_begin "$1"
  pushd "$1" > /dev/null

  for i in {00..99}; do
    task_patches "$1/$i-patches"
    task_run     "$1/$i-run.sh"
  done

  popd > /dev/null
  log_end "$1"
}

task_patches() {
  if [ -d "$1" ]; then
    local SUB_STAGE_DIR=$(dirname "$1")
    local SUB_STAGE_NAME=$(basename "$SUB_STAGE_DIR")

    log_begin "$1"
    pushd "$STAGE_WORK_DIR" > /dev/null

    rm -rf .pc
    rm -rf *-pc

    export QUILT_PATCHES="$1"

    SUB_STAGE_QUILT_PATCH_DIR="$SUB_STAGE_NAME-pc"
    mkdir -p "$SUB_STAGE_QUILT_PATCH_DIR"
    ln -snf "$SUB_STAGE_QUILT_PATCH_DIR" .pc

    if [ -e "$1/EDIT" ]; then
      tput setaf 3 # Yellow color
      echo 'Dropping into bash to edit patches...'
      echo 'Tutorial: https://raphaelhertzog.com/2012/08/08/how-to-use-quilt-to-manage-patches-in-debian-packages/'
      echo 'Example:'
      echo '  quilt new XX-name-of-the-patch.diff'
      echo '  quilt edit rootfs/path/to/file'
      echo '  quilt diff'
      echo '  quilt refresh'
      tput sgr0 # No color

      bash
    fi

    quilt upgrade
    RC=0
    quilt push -a || RC=$?

    case "$RC" in
    0|2)
      ;;
    *)
      false
      ;;
    esac

    popd > /dev/null
    log_end "$1"
  fi
}

task_run() {
  if [ -x "$1" ]; then
    log_begin "$1"

    "$1"

    log_end "$1"
  fi
}

main