#!/usr/bin/env bash

# /etc/vz/conf/vps.premount

. /etc/vz/vz.conf
. "${VE_CONFFILE}"

# Add NAT rules
VZ_PORT_CONFIG="/etc/vz/ports/${HOSTNAME}"
if [[ -r "$VZ_PORT_CONFIG" ]]; then
        PORTS=$(cat "$VZ_PORT_CONFIG")
        if [[ -n "$PORTS" ]]; then
		. /etc/vz/nat.conf
                for VAR in VZ_DNAT_CHAIN VZ_DNAT_INTERFACE; do
                        if [[ -z "${!VAR}" ]]; then
                                echo "Error adding NAT rules: ${VAR} is not set."
                                exit 1
                        fi
                done
                echo "Adding NAT rules for ports ${PORTS}"
                /sbin/iptables -t nat -I "$VZ_DNAT_CHAIN" -i "$VZ_DNAT_INTERFACE" -p tcp -m multiport --dports "$PORTS" -j DNAT --to-destination "${IP_ADDRESS}" -m comment --comment "VZ ${VEID}"
        fi
fi