|
#--
|
|
# 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 "cyborghood/lang_additions"
|
|
require 'gettext'
|
|
|
|
module CyborgHood
|
|
include GetText
|
|
|
|
PRODUCT = "CyborgHood"
|
|
VERSION = "0.2.0~dev"
|
|
|
|
bindtextdomain(PRODUCT, {:path => File.join(APP_ROOT, "locale"), :charset => "UTF-8"})
|
|
textdomain(PRODUCT)
|
|
#textdomain_to(self.class, PRODUCT + "_" + self.human_name)
|
|
|
|
# severities: # :dangerous :unrecoverable :ignorable
|
|
# categories:
|
|
# - db
|
|
# + ldap
|
|
# + sql
|
|
# - protocol
|
|
# + imap
|
|
# - 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
|
|
|
|
CFG_DIR = './config/'
|
|
WORK_DIR = './var/lib/cyborghood/'
|
|
|
|
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
|
|
|
|
# temporary
|
|
def work_dir
|
|
WORK_DIR
|
|
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
|
|
end
|