Adding Hosts to Zabbix Using Bash Script

				
					#!/bin/bash

# inventory_update.sh

# Purpose: 	This script adds hosts and respective IP addresses to the Zabbix database. It takes
# 	   	    the path to a file with a list of hostnames and IP addresses as arguments.

# Zabbix PostgreSQL database credentials
DB_USER="zabbix"
DB_PASSWORD="zabbix"
DB_NAME="zabbix"
DB_HOST="192.168.1.4"
DB_PORT="5432"

# Function to check if a file exists
check_file_exists() {
    if [ ! -f "$1" ]; then
        echo "File $1 does not exist."
        exit 1
    fi
}

# Function to check if a host already exists in the database
host_exists() {
    local hostname="$1"
    local existing_host

    existing_host=$(PGPASSWORD="$DB_PASSWORD" psql -U "$DB_USER" -h "$DB_HOST" -p "$DB_PORT" -d "$DB_NAME" -t -c \
    "SELECT hostid FROM hosts WHERE host = '$hostname';" | sed 's/^ *//;s/ *$//')

    if [ -n "$existing_host" ]; then
        
	echo -e "Host \033[1;31m$hostname\033[0m already exists with host ID: \033[1;31m$existing_host\033[0m."; sleep 0.5
        return 0
    else
        return 1
    fi
}

# Function to get the next host ID
get_next_host_id() {
    local next_host_id

    next_host_id=$(PGPASSWORD="$DB_PASSWORD" psql -U "$DB_USER" -h "$DB_HOST" -p "$DB_PORT" -d "$DB_NAME" -t -c \
    "SELECT COALESCE(MAX(hostid), 0) + 1 FROM hosts;" | sed 's/^ *//;s/ *$//')

    if [ -z "$next_host_id" ]; then
        echo "Failed to retrieve the next host ID."
        exit 1
    fi

    echo "$next_host_id"
}

# Function to get the next interface ID
get_next_interface_id() {
    local next_int_id

    next_int_id=$(PGPASSWORD="$DB_PASSWORD" psql -U "$DB_USER" -h "$DB_HOST" -p "$DB_PORT" -d "$DB_NAME" -t -c \
    "SELECT COALESCE(MAX(interfaceid), 0) + 1 FROM interface;" | sed 's/^ *//;s/ *$//')

    if [ -z "$next_int_id" ]; then
        echo "Failed to retrieve the next interface ID."
        exit 1
    fi

    echo "$next_int_id"
}

# Function to add host to Zabbix
add_host() {
    local hostname="$1"
    local ip_address="$2"
    local host_id
    local interface_id

    echo "Adding host $hostname with IP $ip_address..."

    # Check if the host already exists
    if host_exists "$hostname"; then
        echo "Skipping addition of $hostname as it already exists."
        return 0
    fi

    # Get the next host ID
    host_id=$(get_next_host_id)
    if [ -z "$host_id" ]; then
        echo "Failed to get the next host ID."
        return 1
    fi

    echo "Next host ID: $host_id"

    local sql_host="INSERT INTO hosts (hostid, host, name, status) VALUES ('$host_id', '$hostname', '$hostname', 0) RETURNING hostid;"
    local result
    result=$(PGPASSWORD="$DB_PASSWORD" psql -U "$DB_USER" -h "$DB_HOST" -p "$DB_PORT" -d "$DB_NAME" -t -c "$sql_host" 2>&1 | sed 's/^ *//;s/ *$//')

    if [[ "$result" == *"ERROR"* || -z "$result" ]]; then
        echo "Failed to add host $hostname. SQL Result: $result"
        return 1
    fi

    # Get the next interface ID
    interface_id=$(get_next_interface_id)
    if [ -z "$interface_id" ]; then
        echo "Failed to get the next interface ID."
        return 1
    fi

    echo "Next interface ID: $interface_id"

    # Insert the IP address into the interface table
    local sql_interface="INSERT INTO interface (interfaceid, hostid, main, ip, dns, port) VALUES ('$interface_id', '$host_id','1', '$ip_address', '$hostname', 10050);"
    result=$(PGPASSWORD="$DB_PASSWORD" psql -U "$DB_USER" -h "$DB_HOST" -p "$DB_PORT" -d "$DB_NAME" -c "$sql_interface" 2>&1)

    if [[ "$result" == *"ERROR"* ]]; then
        echo "Failed to add IP address $ip_address for host ID $host_id. SQL Result: $result"
        return 1
    fi

    echo "Successfully added IP address $ip_address for host $hostname (host ID: $host_id, interface ID: $interface_id)."
    return 0
}

# Main function to process the hosts file
process_hosts_file() {
    local hosts_file="$1"

    while IFS= read -r line; do
        # Skip empty lines or lines starting with #
        [[ -z "$line" || "$line" =~ ^# ]] && continue
        
        # Extract the hostname and IP address
        hostname=$(echo "$line" | awk '{print $1}')
        ip_address=$(echo "$line" | awk '{print $2}')
        
        # Check if both hostname and IP are provided
        if [ -z "$hostname" ] || [ -z "$ip_address" ]; then
            echo "Invalid entry: $line. Skipping..."
            continue
        fi
        
        # Add host and its IP address
        add_host "$hostname" "$ip_address"
        if [ $? -eq 0 ]; then
            echo -e "Successfully processed host \033[1;32m$hostname\033[0m with IP \033[1;32m$ip_address\033[0m."; sleep 0.5
            echo -e "";
        fi

    done < "$hosts_file"
}

# Main script execution
main() {
    # Check if the user provided the correct number of arguments
    if [ $# -ne 1 ]; then
        echo "Usage: $0 <hosts_file>"
        exit 1
    fi

    # File containing hosts and IPs
    local hosts_file="$1"

    # Ensure the file exists
    check_file_exists "$hosts_file"

    # Process the hosts file
    process_hosts_file "$hosts_file"
}

# Call the main function with all script arguments
main "$@"