#--
# 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
  bindtextdomain("cyborghood", {:path => File.join(APP_ROOT, "locale"), :charset => "UTF-8"})

  PRODUCT = "CyborgHood"
  VERSION = "0.2.0~dev"

  # 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