Project

General

Profile

Download (3.81 KB) Statistics
| Branch: | Tag: | Revision:
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