#!/bin/bash

SCRIPT="envsetup.sh"
MACHINE="$1"
BUILDDIR="build"
DISTRO="poky-sota-systemd"
BASE_CONF="local.conf.base.append"

if [ -n "$ZSH_VERSION" ]; then
    # be more compatible with bash
    setopt shwordsplit
fi

# A definition of a dictionary with a list of configuration files that must be appended
# to resulting conf/local.conf file for each particular distribution.
declare -A supported_distros
supported_distros[poky-sota-systemd]="local.conf.systemd.append"
supported_distros[poky-sota]="local.conf.base.append"
supported_distros[poky]="local.conf.systemd.append local.conf.nonostree.append"

usage () {
  echo "Usage: ${SCRIPT} <machine> [builddir] [distro=< poky-sota-systemd | poky-sota | poky >]"
}

[[ "$#" -lt 1 ]] && { usage; return 1; }
[[ "$#" -ge 2 ]] && { BUILDDIR="$2"; }
[[ "$#" -eq 3 ]] && { DISTRO="$3"; }

# detect if this script is sourced: see http://stackoverflow.com/a/38128348/6255594
SOURCED=0
if [[ -n "$ZSH_EVAL_CONTEXT" ]]; then
  [[ "$ZSH_EVAL_CONTEXT" =~ :file$ ]] && { SOURCED=1; SOURCEDIR=$(cd "$(dirname -- "$0")" && pwd -P); }
elif [[ -n "$BASH_VERSION" ]]; then
  [[ "$0" != "${BASH_SOURCE[0]}" ]] && { SOURCED=1; SOURCEDIR=$(cd "$(dirname -- "${BASH_SOURCE[0]}")" && pwd -P); }
fi

if [[ $SOURCED -ne 1 ]]; then
  echo "Error: this script needs to be sourced in a supported shell" >&2
  echo "Please check that the current shell is bash or zsh and run this script as '. $0 <args>'" >&2
  exit 1
fi

METADIR=${METADIR:-${SOURCEDIR}/../..}

if [[ ! -f "${BUILDDIR}/conf/local.conf" ]]; then
  if [ -z "${supported_distros[$DISTRO]}" ]; then
    echo "The specified distro $DISTRO is not supported"
    usage
    return 1
  fi

  source "$METADIR/poky/oe-init-build-env" "$BUILDDIR"

  echo "METADIR  := \"\${@os.path.abspath('${METADIR}')}\"" >> conf/bblayers.conf
  cat "${METADIR}/meta-updater/conf/include/bblayers/sota.inc" >> conf/bblayers.conf
  cat "${METADIR}/meta-updater/conf/include/bblayers/sota_${MACHINE}.inc" >> conf/bblayers.conf

  sed -e "s/##MACHINE##/$MACHINE/g" \
      -e "s/##DISTRO##/$DISTRO/g" \
         "${METADIR}/meta-updater/conf/$BASE_CONF" >> conf/local.conf

  for config in ${supported_distros[$DISTRO]}; do
    if [[ "$BASE_CONF" != "$config" ]]; then
      cat "${METADIR}/meta-updater/conf/$config" >> conf/local.conf
    fi
  done
else
  source "$METADIR/poky/oe-init-build-env" "$BUILDDIR"
fi