#!/bin/bash

MANIFESTS_DIR="${KUBEVIRTCI_PATH}/cluster/${KUBEVIRT_PROVIDER}/sriov-components/manifests"

KUSTOMIZE_MULTUS_DIR="${MANIFESTS_DIR}/multus"
MULTUS_MANIFEST="${CUSTOM_MANIFESTS}/multus.yaml"

CUSTOM_MANIFESTS="${KUBEVIRTCI_CONFIG_PATH}/${KUBEVIRT_PROVIDER}/manifests"
SRIOV_COMPONENTS_MANIFEST="${CUSTOM_MANIFESTS}/sriov-components.yaml"

SRIOV_DEVICE_PLUGIN_CONFIG_TEMPLATE="${MANIFESTS_DIR}/sriovdp-config.yaml.in"
SRIOV_DEVICE_PLUGIN_CONFIG="${CUSTOM_MANIFESTS}/sriovdp-config.yaml"

PATCH_SRIOVDP_RESOURCE_PREFIX_TEMPLATE="${MANIFESTS_DIR}/patch-sriovdp-resource-prefix.yaml.in"
PATCH_SRIOVDP_RESOURCE_PREFIX="${CUSTOM_MANIFESTS}/patch-sriovdp-resource-prefix.yaml"

PATCH_NODE_SELECTOR_TEMPLATE="${MANIFESTS_DIR}/patch-node-selector.yaml.in"
PATCH_NODE_SELECTOR="${CUSTOM_MANIFESTS}/patch-node-selector.yaml"

KUBECONFIG="${KUBEVIRTCI_CONFIG_PATH}/$KUBEVIRT_PROVIDER/.kubeconfig"
KUBECTL="${KUBEVIRTCI_CONFIG_PATH}/$KUBEVIRT_PROVIDER/.kubectl --kubeconfig=${KUBECONFIG}"

SRIOV_DRA_MANIFEST="${MANIFESTS_DIR}/dra/sriov_dra.yaml"
SRIOV_DRA_NAMESPACE="dra-driver-sriov"

NETWORK_RESOURCES_INJECTOR_AUTH_MANIFEST="${MANIFESTS_DIR}/network_resources_injector/auth.yaml"
NETWORK_RESOURCES_INJECTOR_SERVICE_MANIFEST="${MANIFESTS_DIR}/network_resources_injector/service.yaml"
NETWORK_RESOURCES_INJECTOR_SERVER_MANIFEST="${MANIFESTS_DIR}/network_resources_injector/server.yaml"

function _kubectl() {
    ${KUBECTL} "$@"
}

function _retry() {
  local -r tries=$1
  local -r wait_time=$2
  local -r action=$3
  local -r wait_message=$4
  local -r waiting_action=$5

  eval $action
  local return_code=$?
  for i in $(seq $tries); do
    if [[ $return_code -ne 0 ]]; then
      echo "[$i/$tries] $wait_message"
      eval $waiting_action
      sleep $wait_time
      eval $action
      return_code=$?
    else
      return 0
    fi
  done

  return 1
}

function _check_all_pods_ready() {
  all_pods_ready_condition=$(_kubectl get pods -A --no-headers -o custom-columns=':.status.conditions[?(@.type == "Ready")].status')
  if [ "$?" -eq 0 ]; then
    pods_not_ready_count=$(grep -cw False <<<"$all_pods_ready_condition")
    if [ "$pods_not_ready_count" -eq 0 ]; then
      return 0
    fi
  fi

  return 1
}

# not using kubectl wait since with the sriov operator the pods get restarted a couple of times and this is
# more reliable
function sriov_components::wait_pods_ready() {
  local -r tries=30
  local -r wait_time=10

  local -r wait_message="Waiting for all pods to become ready.."
  local -r error_message="Not all pods were ready after $(($tries * $wait_time)) seconds"

  local -r get_pods='_kubectl get pods --all-namespaces'
  local -r action="_check_all_pods_ready"

  set +x
  trap "set -x" RETURN

  if ! _retry "$tries" "$wait_time" "$action" "$wait_message" "$get_pods"; then
    echo $error_message
    return 1
  fi

  echo "all pods are ready"
  return 0
}

function sriov_components::wait_allocatable_resource() {
  local -r node=$1
  local resource_name=$2
  local -r expected_value=$3

  local -r tries=48
  local -r wait_time=10

  local -r wait_message="wait for $node node to have allocatable resource: $resource_name: $expected_value"
  local -r error_message="node $node doesnt have allocatable resource $resource_name:$expected_value"

  # it is necessary to add '\' before '.' in the resource name.
  resource_name=$(echo $resource_name | sed s/\\./\\\\\./g)
  local -r action='_kubectl get node $node -ocustom-columns=:.status.allocatable.$resource_name --no-headers | grep -w $expected_value'

  if ! _retry $tries $wait_time "$action" "$wait_message"; then
    echo $error_message
    return 1
  fi

  return 0
}

function sriov_components::deploy_multus() {
  _kubectl kustomize "$KUSTOMIZE_MULTUS_DIR" > "$MULTUS_MANIFEST"

  echo "Deploying Multus:"
  cat "$MULTUS_MANIFEST"

  _kubectl apply -f "$MULTUS_MANIFEST"

  return 0
}

function sriov_components::deploy() {
  local -r pf_names=$1
  local -r drivers=$2
  local -r resource_prefix=$3
  local -r resource_name=$4
  local -r label_key=$5
  local -r label_value=$6

  _create_custom_manifests_dir 
  _prepare_node_selector_patch "$label_key" "$label_value"
  _prepare_sriovdp_resource_prefix_patch "$resource_prefix"
  _prepare_device_plugin_config \
    "$pf_names" \
    "$resource_name" \
    "$drivers"
  _deploy_sriov_components

  return 0
}

function sriov_components::deploy_dra() {
  _kubectl create namespace "$SRIOV_DRA_NAMESPACE"
  _kubectl apply -f "$SRIOV_DRA_MANIFEST"

  return 0
}

function sriov_components::deploy_network_resources_injector() {
  _kubectl apply -f "$NETWORK_RESOURCES_INJECTOR_AUTH_MANIFEST"
  _kubectl apply -f "$NETWORK_RESOURCES_INJECTOR_SERVICE_MANIFEST"
  _kubectl apply -f "$NETWORK_RESOURCES_INJECTOR_SERVER_MANIFEST"
  return 0
}

function _create_custom_manifests_dir() {
  mkdir -p "$CUSTOM_MANIFESTS"

  cp -f $(find "$MANIFESTS_DIR"/*.yaml) "$CUSTOM_MANIFESTS"

  return 0
}

function _prepare_node_selector_patch() {
  local -r label_key=$1
  local -r label_value=$2

  (
    export LABEL_KEY=$label_key
    export LABEL_VALUE=$label_value
    envsubst < "$PATCH_NODE_SELECTOR_TEMPLATE" > "$PATCH_NODE_SELECTOR"
  )
}

function _prepare_sriovdp_resource_prefix_patch() {
  local -r resource_prefix=$1

  (
    export RESOURCE_PREFIX=$resource_prefix
    envsubst < "$PATCH_SRIOVDP_RESOURCE_PREFIX_TEMPLATE" > "$PATCH_SRIOVDP_RESOURCE_PREFIX"
  )
}

function _prepare_device_plugin_config() {
  local -r pf_names=$1
  local -r resource_name=$2
  local -r drivers=$3

  (
    export RESOURCE_NAME=$resource_name
    export DRIVERS=$(_format_json_array "$drivers")
    export PF_NAMES=$(_format_json_array "$pf_names")
    envsubst < "$SRIOV_DEVICE_PLUGIN_CONFIG_TEMPLATE" > "$SRIOV_DEVICE_PLUGIN_CONFIG"
  )

  return 0
}

function _format_json_array() {
  local -r string=$1

  local json_array="$string"
  # Replace all spaces with ",": aa bb -> aa","bb
  local -r replace='","'
  json_array="${json_array// /$replace}"

  # Add opening quotes for first element, and closing quotes for last element
  # aa","bb -> "aa","bb"
  json_array="\"${json_array}\""

  # Add brackets: "aa","bb" -> ["aa","bb"]
  json_array="[${json_array}]"

  echo "$json_array"
}

function _deploy_sriov_components() {
  _kubectl kustomize "$CUSTOM_MANIFESTS" >"$SRIOV_COMPONENTS_MANIFEST"

  echo "Deploying SRIOV components:"
  cat "$SRIOV_COMPONENTS_MANIFEST"

  _kubectl apply -f "$SRIOV_COMPONENTS_MANIFEST"

  return 0
}