|
#--
|
|
# CyborgHood, a distributed system management software.
|
|
# Copyright (c) 2009 Marc Dequènes (Duck) <Duck@DuckCorp.org>
|
|
#
|
|
# 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/>.
|
|
#++
|
|
|
|
$KCODE = 'UTF8'
|
|
require 'jcode'
|
|
require 'singleton'
|
|
require 'yaml'
|
|
require 'log4r'
|
|
require 'gettext'
|
|
require "cyborghood/config"
|
|
require "cyborghood/info"
|
|
require "cyborghood/lang_additions"
|
|
|
|
module CyborgHood
|
|
include GetText
|
|
bindtextdomain("cyborghood", {:path => Config::L10N_DIR, :charset => "UTF-8"})
|
|
|
|
# severities: :dangerous :unrecoverable :processable :ignorable
|
|
# categories:
|
|
# - db (not a protocol?)
|
|
# + ldap
|
|
# + sql
|
|
# - protocol
|
|
# + imap
|
|
# + mail
|
|
# - service
|
|
# + dns
|
|
class CyberError < StandardError
|
|
attr_accessor :severity, :category
|
|
|
|
#include GetTextSupport
|
|
|
|
def initialize(severity, category, message)
|
|
@severity = severity
|
|
@category = category
|
|
super(message)
|
|
end
|
|
end
|
|
|
|
class Logger < Log4r::Logger
|
|
include Singleton
|
|
|
|
LOG_FORMAT = "[%5l - %d] #PREFIX#%m"
|
|
|
|
def output_level(level)
|
|
l = case level
|
|
when :verbose
|
|
Log4r::DEBUG
|
|
when :quiet
|
|
Log4r::FATAL
|
|
else # normal
|
|
Log4r::WARN
|
|
end
|
|
@main_outputter.level = l
|
|
end
|
|
|
|
def log_to_file(filename)
|
|
file_outputter = Log4r::FileOutputter.new(filename, :filename => filename)
|
|
file_outputter.formatter = @default_formatter
|
|
self.outputters << file_outputter
|
|
end
|
|
|
|
def set_prefix(prefix = "")
|
|
prefix ||= ""
|
|
log_format = LOG_FORMAT.gsub("#PREFIX#", prefix)
|
|
@default_formatter = Log4r::PatternFormatter.new(:pattern => log_format)
|
|
self.outputters.each{|outputter| outputter.formatter = @default_formatter }
|
|
end
|
|
|
|
private
|
|
|
|
def initialize
|
|
super(PRODUCT)
|
|
|
|
self.level = Log4r::DEBUG
|
|
@main_outputter = Log4r::Outputter.stdout
|
|
@main_outputter.level = Log4r::WARN
|
|
self.outputters = [@main_outputter]
|
|
|
|
set_prefix()
|
|
end
|
|
end
|
|
|
|
class Config
|
|
include Singleton
|
|
|
|
CONFFILE_GLOBAL = "cyborghood.conf"
|
|
CONFFILE_BOT = "cyborghood_%s.conf"
|
|
|
|
def self.load(name = nil)
|
|
# load all config parts
|
|
config = fetch_config
|
|
unless name.nil?
|
|
config_spc = fetch_config(name)
|
|
config = merge_configs(config, config_spc) if config_spc
|
|
end
|
|
|
|
# create config object for easier access
|
|
@@config = config.to_ostruct.freeze
|
|
end
|
|
|
|
def method_missing(sym, *args, &block)
|
|
begin
|
|
@@config.send sym, *args, &block
|
|
rescue
|
|
nil
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def self.fetch_config(name = nil)
|
|
begin
|
|
if name.nil?
|
|
filename = CONFFILE_GLOBAL
|
|
else
|
|
filename = sprintf(CONFFILE_BOT, name)
|
|
end
|
|
filepath = File.join(File.expand_path(CFG_DIR), filename)
|
|
str = File.read(filepath)
|
|
YAML.load(str)
|
|
rescue
|
|
logger.fatal "Problems fetching configuration file '" + filepath + "': " + $!
|
|
exit 1
|
|
end
|
|
end
|
|
|
|
def self.merge_configs(conf1, conf2)
|
|
new_conf = conf1.dup
|
|
conf2.each_pair do |k, v|
|
|
if conf1.has_key?(k) and conf1[k].is_a?(Hash) and v.is_a?(Hash)
|
|
new_conf[k] = merge_configs(conf1[k], v)
|
|
else
|
|
new_conf[k] = v
|
|
end
|
|
end
|
|
return new_conf
|
|
end
|
|
end
|
|
|
|
class I18n
|
|
include Singleton
|
|
|
|
def initialize
|
|
@config = Config.instance
|
|
end
|
|
|
|
def available_languages
|
|
list = ['en'] + Dir.new(L10N_DIR).select{|d| File.directory?(d) and d[0..0] != "." }
|
|
# local admin can restrict available languages
|
|
# (may be useful if l10n is partial due to third party plugins)
|
|
list = list & @config.i18n.restricted_language_set if @config.i18n.restricted_language_set
|
|
list = ['en'] if list.empty?
|
|
list
|
|
end
|
|
|
|
def set_language(lang)
|
|
set_locale(lang)
|
|
end
|
|
|
|
def set_default_language
|
|
set_language('en')
|
|
end
|
|
|
|
def set_language_for_user(user)
|
|
if user.nil?
|
|
set_default_language
|
|
else
|
|
logger.debug "User preference for language: " + user.preferredLanguage
|
|
|
|
lang = user.prefered_language(self.available_languages)
|
|
if lang.nil?
|
|
logger.debug "No available langage fits the user preference, using english"
|
|
lang = 'en'
|
|
else
|
|
logger.debug "Language better fitting user preference: " + lang
|
|
end
|
|
|
|
set_language(lang)
|
|
end
|
|
end
|
|
end
|
|
end
|