#!/bin/bash

echo -e "${_GREEN}LOADING FUNCTIONS${_RESET}"

function prompt_confirm() {
  while true; do
    read -r -n 1 -p "${1:-Continue?} [y/n]: " REPLY
    case $REPLY in
      [yY]) echo ; return 0 ;;
      [nN]) echo ; return 1 ;;
      *) printf " \033[31m %s \n\033[0m" "invalid input [y/n]"
    esac 
  done  
}

function show_settings () {
	echo "${_CYAN}Settings from loaded environment:${_RESET}"
        echo "${_GREEN}Pool URL:${_RESET}: $_POOLS_URL"
        echo "${_GREEN}Recieve Wallet:${_RESET}: $_POOLS_USER"
        echo "${_GREEN}Email Address:${_RESET}: $_POOLS_PASS"
}

function show_mysettings () {
        if [ -f $_WORK_DIR/$_MYSETTINGS_FILE ]; then
		load_mysettings
		echo "${_CYAN}Settings from mysettings.sh:${_RESET}"
		echo "${_GREEN}Pool URL:${_RESET}: $_POOLS_URL"
	        echo "${_GREEN}Recieve Wallet:${_RESET}: $_POOLS_USER"
	        echo "${_GREEN}Email Address:${_RESET}: $_POOLS_PASS"
        fi
}

function load_mysettings () {
	if [ -f $_WORK_DIR/$_MYSETTINGS_FILE ]; then
                . $_MYSETTINGS_FILE
        fi
}

function my_config_json () {

	_CONFIG_JSON=$(jo -p -B api=$(jo id=$_API_ID worker-id=$_API_WORKER_ID) autosave=$_AUTOSAVE background=$_BACKGROUND colors=$_COLORS cpu=$(jo enabled=$_CPU_ENABLED huge-pages=$_CPU_HUGE_PAGES hw-aes=$_CPU_HW_AES priority=$_CPU_PRIORITY asm=$_CPU_ASM cn=$(jo -a $_CPU_CN) cn/0=$_CPU_CN_0 cn-lite/0=$_CPU_CN_LITE_0) opencl=$(jo enabled=$_OPENCL_ENABLED cache=$_OPENCL_CACHE loader=$_OPENCL_LOADER platform=$_OPENCL_PLATFORM) donate-level=$_DONATE_LEVEL donate-over-proxy=$_DONATE_OVER_PROXY http=$(jo enabled=$_HTTP_ENABLED host=$_HTTP_HOST port=$_HTTP_PORT access-token=$_HTTP_ACCESS_TOKEN restricted=$_HTTP_RESTRICTED) log-file=$_LOG_FILE pools=$(jo -a $(jo algo=$_POOLS_ALGO url=$_POOLS_URL user=$_POOLS_USER pass=$_POOLS_PASS rig-id=$_POOLS_RIG_ID nicehash=$_POOLS_NICEHASH keepalive=$_POOLS_KEEPALIVE enabled=$_POOLS_ENABLED tls=$_POOLS_TLS tls-fingerprint=$_POOLS_TLS_FINDERPRINT daemon=$_POOLS_DAEMON)) print-time=$_PRINT_TIME retries=$_RETRIES retry-pause=$_RETRIES_PAUSE syslog=$_SYSLOG user-agent=$_USER_AGENT watch=$_WATCH)

}

function build_xmrig () {
        if [ -d $_XMRIG_BUILD_LOCATION ]; then
                # Remove existing build directory
                rm -rf $_XMRIG_BUILD_LOCATION

                # Make new build directory
                mkdir $_XMRIG_BUILD_LOCATION
        else
                # Make new build directory
                mkdir $_XMRIG_BUILD_LOCATION
        fi

        # Change to build directory
        cd $_XMRIG_BUILD_LOCATION

        # Configure cmake scafolding
        cmake ..

        # Compile the software
        make
}


function start_xmrig () {
        screen -dmS $_XMRIG_SCREEN xmrig --config=$_XMRIG_CONFIG_LOCATION
}


function stop_xmrig () {
        screen -S $_XMRIG_SCREEN -X kill
}

function check_cpu () {
	# Check if CPU supports AES-NI
	cpuid | grep -i aes > hw-aes.txt
	if grep -q true "hw-aes.txt"; then
        	_AES_NI=true
	else
        	_AES_NI=false
	fi
	rm hw-aes.txt

	# Get number of available CPU Cores
	_ENV_CORE=$(nproc --all)

	# Get CPU L3 cache value
	_ENV_CORE_L3=$(getconf LEVEL3_CACHE_SIZE)
}

function calc_threads () {
	# Each thread needs 2048 byte (2 Megabyte)
	# Calculate number of threads supported with L3 cache size

	_ENV_THREAD_QTY=$(( ($_ENV_CORE_L3/2048)/1000 ))

	# Affine number of threads for _CPU_CN
	_COUNTER="$(( $_ENV_THREAD_QTY-1 ))"

	_ENV_CPU_THREAD_AFFINITY="$(seq -s " " 0 $_COUNTER)"

}

function calc_hugepages () {
         # Setting according to cpu cores but xmrig will not use all
         # change to make value use _ENV_CORE_THREADS
         # Check that hugepages set in /etc/sysctl.conf
       
         _ENV_CHECK="nr_hugepages"
     
         if sudo grep -q $_ENV_CHECK /etc/sysctl.conf; then
                 sudo sysctl -p
         else    
                 # Set value in current env
                 sudo sysctl -w vm.nr_hugepages="$_ENV_THREAD_QTY"
                 # Add value to sysctl
                 sudo sysctl -p
         fi

}

echo -e "${_CYAN}FUNCTIONS LOADED${_RESET}"