#!/bin/bash
###############################################################################
#                                                                             #
# IPFire.org - A linux based firewall                                         #
# Copyright (C) 2010  Michael Tremer & Christian Schmidt                      #
#                                                                             #
# This program is free software: you can redistribute it and/or modify        #
# it under the terms of the GNU General Public License as published by        #
# the Free Software Foundation, either version 3 of the License, or           #
# (at your option) any later version.                                         #
#                                                                             #
# This program is distributed in the hope that it will be useful,             #
# but WITHOUT ANY WARRANTY; without even the implied warranty of              #
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               #
# GNU General Public License for more details.                                #
#                                                                             #
# You should have received a copy of the GNU General Public License           #
# along with this program.  If not, see <http://www.gnu.org/licenses/>.       #
#                                                                             #
###############################################################################

COLOUR_GREEN="\\033[1;32m"
COLOUR_RED="\\033[1;31m"
COLOUR_NORMAL="\\033[0;39m"
COLOUR_YELLOW="\\033[1;35m"

COLOUR_BOLD="\\033[1;39m"
COLOUR_DOWN=${COLOUR_RED}
COLOUR_ERROR=${COLOUR_RED}
COLOUR_OK=${COLOUR_GREEN}
COLOUR_UP=${COLOUR_GREEN}
COLOUR_WARN=${COLOUR_YELLOW}

COLOUR_ENABLED=${COLOUR_GREEN}
COLOUR_DISABLED=${COLOUR_RED}

declare -A COLOUR_LOG
COLOUR_LOG["DEBUG"]=${COLOUR_NORMAL}
COLOUR_LOG["INFO"]=${COLOUR_NORMAL}
COLOUR_LOG["ERROR"]=${COLOUR_RED}
COLOUR_LOG["WARNING"]=${COLOUR_YELLOW}

COLOUR_STP_FORWARDING=${COLOUR_GREEN}
COLOUR_STP_DISCARDING=${COLOUR_RED}
COLOUR_STP_LEARNING=${COLOUR_YELLOW}
COLOUR_STP_BLOCKING=${COLOUR_YELLOW}

COLOUR_VARIABLES="COLOUR_GREEN COLOUR_RED COLOUR_NORMAL COLOUR_YELLOW"
COLOUR_VARIABLES="${COLOUR_VARIABLES} COLOUR_BOLD COLOUR_DOWN COLOUR_ERROR"
COLOUR_VARIABLES="${COLOUR_VARIABLES} COLOUR_ERROR COLOUR_OK COLOUR_UP"
COLOUR_VARIABLES="${COLOUR_VARIABLES} COLOUR_WARM COLOUR_ENABLED COLOUR_DISABLED"
COLOUR_VARIABLES="${COLOUR_VARIABLES} COLOUR_LOG COLOUR_STP_FORWARDING"
COLOUR_VARIABLES="${COLOUR_VARIABLES} COLOUR_STP_DISCARDING COLOUR_STP_LEARNING"
COLOUR_VARIABLES="${COLOUR_VARIABLES} COLOUR_STP_BLOCKING"

function colours_disable() {
	local line
	for line in ${COLOUR_VARIABLES}; do
		unset ${line}
	done
}

function colours_auto_disable() {
	if [ "${TERM}" = "dumb" ]; then
		colours_disable
	fi
}