#!/bin/bash
# Copyright 2015 Mellanox Technologies, Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
# implied.
# See the License for the specific language governing permissions and
# limitations under the License.

readonly SCRIPT_NAME=`basename $0`

function logger_print () {
  priority=$1
  msg=$2
  logger -t $SCRIPT_NAME "${priority}: ${msg}"
  echo $(date +"%Y-%m-%d %H:%M:%S") $SCRIPT_NAME ${priority}: ${msg} >> /var/log/mellanox-plugin.log
}

function get_mlnx_param () {
  param="$1"
  val=$(ruby -r hiera -r yaml -e "hiera = Hiera.new(:config => '/etc/puppet/hiera.yaml'); mlnx = hiera.lookup 'mellanox-plugin', [], {};  puts mlnx['"$param"']")
  echo $val
}

function get_distro () {
  if [[ -f /etc/redhat-release ]]; then
    dist='redhat';
  elif [[ -f /etc/debian_version ]]; then
    dist='ubuntu';
  else
    logger_print error "Could not detect linux distribution"
  fi
  echo $dist
}

readonly DISTRO=`get_distro`
readonly DRIVER=`get_mlnx_param driver`
readonly SRIOV=`get_mlnx_param sriov`
readonly USER_NUM_OF_VFS=`get_mlnx_param num_of_vfs`
readonly ISER=`get_mlnx_param iser`
readonly MAX_VFS=64
readonly MIN_VFS=1