#!/bin/bash

DISCOVERD_URL="http://127.0.0.1:5050/v1/continue"
#DISCOVERD_URL="http://192.168.0.11:5050/v1/continue"
function update() {
    jq "$1" data.json > temp.json || echo "Error: update $1 to json failed"
    mv temp.json data.json
}

function get_system_info(){
    PRODUCT=$(dmidecode -s system-product-name)
    FAMILY=$(dmidecode -t system|grep "Family"|cut -d ":" -f2)
    VERSION=$(dmidecode -s system-version)
    SERIAL=$(dmidecode -s system-serial-number)
    MANUFACTURER=$(dmidecode -s system-manufacturer)
    UUID=$(dmidecode -s system-uuid)
    #FQDN=$(hostname -f)
    FQDN='Hostname'
    echo '{"system":{}}' > data.json
    update ".system[\"product\"] = \"$PRODUCT\""
    update ".system[\"family\"] = \"$FAMILY\""
    update ".system[\"fqdn\"] = \"$FQDN\""
    update ".system[\"version\"] = \"$VERSION\""
    update ".system[\"serial\"] = \"$SERIAL\""
    update ".system[\"manufacturer\"] = \"$MANUFACTURER\""
    update ".system[\"uuid\"] = \"$UUID\""
}

function get_cpu_info(){
    REAL=$(cat /proc/cpuinfo |grep "physical id"|sort |uniq|wc -l)
    TOTAL=$(cat /proc/cpuinfo |grep "processor"|wc -l)
    update ".cpu[\"real\"] = $REAL"
    update ".cpu[\"total\"] = $TOTAL"

    for i in $(seq $TOTAL)
    do
        if [ ! -z "$i" ]; then
        SPEC_MODEL=$(cat /proc/cpuinfo | grep name | cut -f2 -d:|sed -n $i"p")
        SPEC_FRE=$(cat /proc/cpuinfo | grep MHz | cut -f2 -d:|sed -n $i"p")
        update ".cpu[\"spec_$i\"] = {model:\"$SPEC_MODEL\", frequency:$SPEC_FRE}"
        fi
    done
}

function get_memory_info(){
    PHY_NUM=$(dmidecode -t memory|grep "Physical Memory Array"|wc -l)
    TOTAL_MEM=$(cat /proc/meminfo |grep MemTotal |cut -d ":" -f2)
    update ".memory[\"total\"] = \"$TOTAL_MEM\""
    for num in $(seq $PHY_NUM)
    do
        SLOTS=$(dmidecode -t memory |grep "Number Of Devices" |cut -d ":" -f2|sed -n $num"p")
        MAX_CAP=$(dmidecode -t memory |grep "Maximum Capacity" |cut -d ":" -f2|sed -n $num"p")
        update ".memory[\"phy_memory_$num\"] = {slots:\"$SLOTS\", maximum_capacity:\"$MAX_CAP\"}"

        for i in $(seq $SLOTS)
        do
            if [ ! -z "$i" ]; then
                DEVICE_FRE=$(dmidecode -t memory |grep "Speed" |cut -d ":" -f2|sed -n $i"p")
                DEVICE_TYPE=$(dmidecode -t memory |grep 'Type:' |grep -v "Error Correction Type"|cut -d ":" -f2|sed -n $i"p")
                DEVICE_SIZE=$(dmidecode -t memory |grep Size |cut -d ":" -f2|sed -n $i"p")
                update ".memory[\"phy_memory_$num\"][\"devices_$i\"] = {frequency:\"$DEVICE_FRE\", type:\"$DEVICE_TYPE\", size:\"$DEVICE_SIZE\"}"
            fi
        done
    done
}

function get_net_info(){
    for iface in $(ls /sys/class/net/ | grep -v lo)
    do
        NAME=$iface
        MAC=$(ip link show $iface | awk '/ether/ {print $2}')
        IP=$(ip addr show $iface | awk '/inet / { sub(/\/.*/, "", $2); print $2 }')
        NETMASK=$(ifconfig $iface | grep netmask | awk '{print $4}')
        STATE=$(ip link show $iface | awk '/mtu/ {print $3}')
        PCI=$(ethtool -i $iface|grep "bus-info"|cut -d " " -f2)
        CURRENT_SPEED=$(ethtool $iface |grep Speed |awk -F " " '{print $2}')
        LINE=$(ethtool $iface|grep -n "Supported pause frame use"|awk -F ":" '{print $1}')
        #LINE=$[ LINE - 1 ]
        LINE_SPEED=$(ethtool $iface|grep -n "Supported link modes"|awk -F ":" '{print $1}')
        if [ $LINE -eq $LINE_SPEED ]; then
            MAX_SPEED=$(ethtool $iface|grep "Supported link modes"|cut -d ":" -f2)  
        else     
            MAX_SPEED=$(ethtool $iface |sed -n $LINE"p"|awk -F " " '{print $1}')
        fi

        UP="UP"
        if [[ "$STATE" =~ "$UP" ]]; then
            STATE="up"
        else
            STATE="down"
        fi
        if [ ! -z "$MAC" ]; then
            update ".interfaces[\"$iface\"] = {mac: \"$MAC\", ip: \"$IP\", netmask: \"$NETMASK\", name: \"$iface\", max_speed: \"$MAX_SPEED\", state: \"$STATE\", pci: \"$PCI\", current_speed: \"$CURRENT_SPEED\"}"
        fi
    done
}

function get_disk_info(){
    for disk in $(fdisk -l|grep Disk|grep "/dev" |cut -d ":" -f1|awk -F "/" '{print $NF}')
    do
        DISK_NAME=$disk
        DISK_SIZE=$(fdisk -l|grep Disk|grep "/dev" |grep $disk|cut -d "," -f2)
        #DISK_DISK=$(ls -l /dev/disk/by-path/|grep $disk"$"|awk '{print $9}')
        DISK_EXTRA_1=$(ls -l /dev/disk/by-id/|grep $disk"$"|awk '{print $9}'|sed -n 1p)
        DISK_EXTRA_2=$(ls -l /dev/disk/by-id/|grep $disk"$"|awk '{print $9}'|sed -n 2p)
        MODEL=$(hdparm  -I /dev/sda |grep Model | cut -d ":" -f2)
        REMOVABLE=$(hdparm  -I /dev/sda |grep removable|awk '{print $4}')
        update ".disk[\"$disk\"] = {name: \"$DISK_NAME\", size: \"$DISK_SIZE\", disk: \"$DISK_DISK\", model: \"$MODEL\", removable: \"$REMOVABLE\",extra: [\"$DISK_EXTRA_1\", \"$DISK_EXTRA_2\"]}"
    done
}

function main(){
    get_system_info
    get_cpu_info
    get_memory_info
    get_net_info
    get_disk_info
}
main

update ".ipmi_address = \"127.0.0.1\""
update ".data_name = \"baremetal_source\""

echo Collected:
cat data.json

RESULT=$(eval curl -i -X POST \
       "-H 'Accept: application/json'" \
       "-H 'Content-Type: application/json'" \
       "-d @data.json" \
       "$DISCOVERD_URL")

if echo $RESULT | grep "HTTP/1.0 4"; then
    echo "Ironic API returned error: $RESULT"
fi

echo "Node is now discovered! Halting..."
sleep 5