#!/usr/bin/env bash

if ! type -f bpkg-realpath &>/dev/null; then
  echo "error: bpkg-realpath not found, aborting"
  exit 1
else
  # shellcheck disable=SC2230
  # shellcheck source=lib/realpath/realpath.sh
  source "$(which bpkg-realpath)"
fi

if ! type -f bpkg-utils &>/dev/null; then
  echo "error: bpkg-utils not found, aborting"
  exit 1
else
  # shellcheck source=lib/utils/utils.sh
  source "$(which bpkg-utils)"
fi

if ! type -f bpkg-env &>/dev/null; then
  echo "error: bpkg-env not found, aborting"
  exit 1
else
  # shellcheck disable=SC2230
  # shellcheck source=lib/env/env.sh
  source "$(which bpkg-env)"
fi

if ! type -f bpkg-install &>/dev/null; then
  echo "error: bpkg-install not found, aborting"
  exit 1
else
  # shellcheck source=lib/install/install.sh
  source "$(which bpkg-install)"
fi

if ! type -f bpkg-package &>/dev/null; then
  echo "error: bpkg-package not found, aborting"
  exit 1
else
  # shellcheck source=lib/package/package.sh
  source "$(which bpkg-package)"
fi

bpkg_initrc

## output usage
usage () {
  echo 'usage: bpkg-run [-h|--help]'
  echo '   or: bpkg-run [-h|--help] [command]'
  echo '   or: bpkg-run [-s|--source] <package> [command]'
  echo '   or: bpkg-run [-s|--source] <user>/<package> [command]'
}

runner () {
  local cmd="$1"
  shift
  # shellcheck disable=SC2068
  eval "$cmd"
  return 0
}

bpkg_run () {
  local should_emit_source=0
  local should_source=0
  local should_clean=0
  local needs_name=0
  local dest=''
  local name=''

  for opt in "$@"; do
    case "$opt" in
      -h|--help)
        usage
        return 0
        ;;

      -s|--source)
        should_source=1
        shift
        ;;

      --emit-source)
        should_emit_source=1
        shift
        ;;

      -c|--clean)
        should_clean=1
        shift
        ;;

      -t|--target|--name)
        needs_name=1
        shift
        ;;

      *)
        if (( 1 == needs_name )); then
          name="$opt"
          shift
        fi

        break
        ;;
    esac
  done

  if [ -n "$1" ]; then
    local cmd="$(bpkg_package commands "$1" 2>/dev/null)"

    if [ -n "$cmd" ]; then
      local root="$(dirname "$(bpkg_package --path)")"
      cd "$root" || return 1
      # shellcheck disable=SC2230
      # shellcheck source=lib/env/env.sh
      source "$(which bpkg-env)"

      local parts=()
      # shellcheck disable=SC2086
      read -r -a parts <<< $cmd
      local args=()
      local prefix="${parts[0]}"

      for (( i = 1; i < ${#parts[@]}; i++ )); do
        if [[ "${parts[$i]}" =~ \*.\* ]]; then
          local found=($(find . -path "${parts[$i]}"))
          if (( ${#found[@]} > 0 )); then
            args+=("${found[@]}")
          else
            # shellcheck disable=SC2086
            args+=($($(which ls) ${parts[$i]} 2>/dev/null))
          fi
        else
          args+=("${parts[$i]}")
        fi
      done

      shift
      # shellcheck disable=SC2068
      runner "$prefix ${args[*]}" $@
      return $?
    fi
  fi

  if which "$name" 2>/dev/null; then
    return 0
  fi

  if ! pushd . >/dev/null; then
    bpkg_error "Failed to 'pushd' to current working directory."
    return 1
  fi

  if (( 0 == should_clean )); then
    dest=$(bpkg_install --no-prune -g "${name:-$1}" | grep 'Cloning' | sed 's/.* to //g' | xargs echo)
  else
    dest=$(bpkg_install -g "${name:-$1}" | grep 'Cloning' | sed 's/.* to //g' | xargs echo)
  fi

  if [ -z "$dest" ]; then
    if (( 1 == should_emit_source )); then
      bpkg_error "The source '${name:-1}' was not found locally in a 'bpkg.json' or published as a package."
    else
      bpkg_error "The command '$1' was not found locally in a 'bpkg.json' or published as a package."
    fi
    return 1
  fi

  if ! cd "$dest" >/dev/null; then
    bpkg_error "Failed to change directory to package: '$dest'."
    return 1
  fi

  local pkgname="$(bpkg_package name 2>/dev/null)"
  if [ -n "$pkgname" ]; then
    name="$pkgname"
  fi

  if (( 1 == should_emit_source )); then
    if which "$name" 2>/dev/null; then
      :
    elif test -f "$1"; then
      bpkg_realpath "$1"
    elif [ -n "$1" ] && which "$1" 2>/dev/null; then
      :
    fi
  else
    shift

    if (( 1 == should_source )); then
      # shellcheck disable=SC1090
      source "$(which "$name")"
    else
      # shellcheck disable=SC2230
      # shellcheck source=lib/env/env.sh
      source "$(which bpkg-env)"

      cmd="$(bpkg_package commands "$1" 2>/dev/null)"

      if [ -n "$cmd" ]; then
        local parts=()
        # shellcheck disable=SC2086
        read -r -a parts <<< $cmd
        local args=()
        local prefix="${parts[0]}"

        for (( i = 1; i < ${#parts[@]}; i++ )); do
          if [[ "${parts[$i]}" =~ \*.\* ]]; then
            local found=($(find . -path "${parts[$i]}"))
            if (( ${#found[@]} > 0 )); then
              args+=("${found[@]}")
            else
              # shellcheck disable=SC2086
              args+=($($(which ls) ${parts[$i]} 2>/dev/null))
            fi
          fi
        done

        shift
        # shellcheck disable=SC2068
        runner "$prefix ${args[*]}" $@
      fi

      # shellcheck disable=SC2068
      "$(which "$name")" $@
    fi
  fi

  if ! popd >/dev/null; then
    bpkg_error "Failed to 'popd' to previous working directory."
    return 1
  fi

  return $?
}

## Use as lib or perform install
if [[ ${BASH_SOURCE[0]} != "$0" ]]; then
  export -f bpkg_run
else
  bpkg_run "$@"
  exit $?
fi