# This file is part of Fail2Ban.
#
# Fail2Ban 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 2 of the License, or
# (at your option) any later version.
#
# Fail2Ban 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 Fail2Ban; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

# Author: Cyril Jaquier
# 
# $Revision: 288 $

__author__ = "Cyril Jaquier"
__version__ = "$Revision: 288 $"
__date__ = "$Date: 2006-08-22 23:59:51 +0200 (Tue, 22 Aug 2006) $"
__copyright__ = "Copyright (c) 2004 Cyril Jaquier"
__license__ = "GPL"

from server.server import ServerUnknownJail
from server.server import ServerDuplicateJail
import logging

# Gets the instance of the logger.
logSys = logging.getLogger("fail2ban.client.config")

##
# Beautify the output of the client.
#
# Fail2ban server only return unformatted return codes which need to be
# converted into user readable messages.

class Beautifier:
	
	def __init__(self, cmd = None):
		self.inputCmd = cmd

	def setInputCmd(self, cmd):
		self.inputCmd = cmd
		
	def getInputCmd(self):
		return self.inputCmd
		
	def beautify(self, response):
		logSys.debug("Beautify " + `response` + " with " + `self.inputCmd`)
		msg = response
		if self.inputCmd[0:1] == ['status']:
			if len(self.inputCmd) > 1:
				msg = "Status for the jail: " + self.inputCmd[1] + "\n"
				msg = msg + "|- " + response[0][0] + "\n"
				msg = msg + "|  |- " + response[0][1][0][0] + ":\t\t" + `response[0][1][0][1]` + "\n"
				msg = msg + "|  `- " + response[0][1][1][0] + ":\t\t" + `response[0][1][1][1]` + "\n"
				msg = msg + "`- " + response[1][0] + "\n"
				msg = msg + "   |- " + response[1][1][0][0] + ":\t\t" + `response[1][1][0][1]` + "\n"
				msg = msg + "   `- " + response[1][1][1][0] + ":\t\t" + `response[1][1][1][1]`
			else:
				msg = "Status\n"
				msg = msg + "|- " + response[0][0] + ":\t" + `response[0][1]` + "\n"
				msg = msg + "`- " + response[1][0] + ":\t\t" + response[1][1]
		elif self.inputCmd[1:2] == ['loglevel']:
			msg = "Current logging level is "
			if response == 1:
				msg = msg + "ERROR"
			elif response == 2:
				msg = msg + "WARN"
			elif response == 3:
				msg = msg + "INFO"
			elif response == 4:
				msg = msg + "DEBUG"
			else:
				msg = msg + `response`
		elif self.inputCmd == ["stop"]:
			if response == None:
				msg = "Shutdown successful"
		return msg

	def beautifyError(self, response):
		logSys.debug("Beautify (error) " + `response` + " with " + `self.inputCmd`)
		msg = response
		if isinstance(response, ServerUnknownJail):
			msg = "Sorry but the jail '" + response[0] + "' does not exist"
		elif isinstance(response, IndexError):
			msg = "Sorry but the command is invalid"
		elif isinstance(response, ServerDuplicateJail):
			msg = "The jail '" + response[0] + "' already exists"
		return msg