root/lib/cyborghood/base.rb @ d32ee48a
d32ee48a | Marc Dequenes | #--
|
|
# 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/>.
|
|||
#++
|
|||
d31869c4 | Marc Dequenes | $KCODE = 'UTF8'
|
|
require 'jcode'
|
|||
require 'singleton'
|
|||
require 'yaml'
|
|||
require 'log4r'
|
|||
require "cyborghood/lang_additions"
|
|||
module CyborgHood
|
|||
PRODUCT = "CyborgHood"
|
|||
VERSION = "0.0.1"
|
|||
4b58d4fd | Marc Dequenes | # severities: # :dangerous :unrecoverable :ignorable
|
|
# categories:
|
|||
# - db
|
|||
# + ldap
|
|||
# + sql
|
|||
# - protocol
|
|||
# + imap
|
|||
# - service
|
|||
# + dns
|
|||
323a6bb6 | Marc Dequenes | class CyberError < StandardError
|
|
4b58d4fd | Marc Dequenes | attr_accessor :severity, :category
|
|
#include GetTextSupport
|
|||
def initialize(severity, category, message)
|
|||
@severity = severity
|
|||
@category = category
|
|||
super(message)
|
|||
end
|
|||
end
|
|||
class Logger < Log4r::Logger
|
|||
d31869c4 | Marc Dequenes | include Singleton
|
|
790f61f9 | Marc Dequenes | LOG_FORMAT = "[%5l - %d] #PREFIX#%m"
|
|
d31869c4 | Marc Dequenes | ||
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
|
|||
790f61f9 | Marc Dequenes | 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
|
|||
d31869c4 | Marc Dequenes | private
|
|
def initialize
|
|||
super(PRODUCT)
|
|||
self.level = Log4r::DEBUG
|
|||
@main_outputter = Log4r::Outputter.stdout
|
|||
@main_outputter.level = Log4r::WARN
|
|||
self.outputters = [@main_outputter]
|
|||
790f61f9 | Marc Dequenes | ||
set_prefix()
|
|||
d31869c4 | Marc Dequenes | end
|
|
end
|
|||
class Config
|
|||
include Singleton
|
|||
CFG_DIR = './config/'
|
|||
c7904e2c | Marc Dequenes | WORK_DIR = './var/lib/cyborghood/'
|
|
d31869c4 | Marc Dequenes | ||
CONFFILE_GLOBAL = "cyborghood.conf"
|
|||
CONFFILE_BOT = "cyborghood_%s.conf"
|
|||
def self.load(name = nil)
|
|||
# load all config parts
|
|||
config = fetch_config
|
|||
b28fbd1e | Marc Dequenes | unless name.nil?
|
|
config_spc = fetch_config(name)
|
|||
config = merge_configs(config, config_spc) if config_spc
|
|||
end
|
|||
d31869c4 | Marc Dequenes | ||
# 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
|
|||
c7904e2c | Marc Dequenes | # temporary
|
|
def work_dir
|
|||
WORK_DIR
|
|||
end
|
|||
d31869c4 | Marc Dequenes | 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
|