Project

General

Profile

Download (6.81 KB) Statistics
| Branch: | Tag: | Revision:

root/lib/ldap_shadows/controller.rb @ 74f1db3a

#--
# LdapShadows, a Medium-level LDAP Access Library and Tool.
# 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 'active_ldap'
require 'ldap_shadows/activeldap_fixes'
require 'ldap_shadows/object'


module LdapShadows
class Controller
def initialize(mod_container = LdapShadows::Objects)
@mod_container = mod_container

clear_shadow
LdapShadows::LdapObject.mapper = self
end

def clear_shadow
@object_definitions = {}
@aspects = {}
# TODO: should replace @aspects properly one day
@aspects2 = {}
@shadow_dir = nil
end

def set_global_config(global_config)
@global_config = global_config
end

def get_global_config
@global_config
end

def set_aspect(aspect_name, aspect_def)
@aspects[aspect_name] = aspect_def

filename = File.join(Config::CFG_DIR, "hooks", "aspects", aspect_name.to_s.downcase + ".rb")
if File.exists?(filename)
klass_name = "LdapAspect" + aspect_name.to_s.capitalize
klass_content = IO.read(filename)
begin
Aspects.module_eval(<<-EOS)
class #{klass_name} < Aspect
#{klass_content}
end
EOS
rescue
STDERR.puts "Could not load Aspect plugin '#{aspect_name}'"
exit 1
end

@aspects2[aspect_name] = Aspects.const_get(klass_name)
end
end

def get_aspect(aspect_name)
@aspects[aspect_name.to_sym]
end

def get_aspect_klass(aspect_name)
@aspects2[aspect_name.to_sym]
end

def self.object_name_to_klass_name(obj_name)
"LdapObject" + obj_name.to_s.capitalize
end

def load_object(obj_name, obj_def)
obj_mapping = obj_def[:mapping]
klass_name = self.class.object_name_to_klass_name(obj_name)

# create class
@mod_container.module_eval(<<-EOS)
class #{klass_name} < LdapShadows::LdapObject; end
EOS

# configure class
klass = find_klass(obj_name)
klass.handle = obj_name
klass.presentation = obj_def[:presentation]
klass.mapper = self
klass.ldap_mapping obj_mapping.reject {|key, val| not ActiveLdap::Base::VALID_LDAP_MAPPING_OPTIONS.include?(key) }

# store definition for later relations processing
@object_definitions[obj_name] = obj_def
end

def find_klass(obj_name)
klass_name = self.class.object_name_to_klass_name(obj_name)
return nil unless @mod_container.const_defined?(klass_name)
@mod_container.const_get(klass_name)
end

# run it _once_ when all objects are loaded
def load_relations
@object_definitions.each_pair do |obj_name, obj_def|
obj_rel = {}
obj_rel.merge!(obj_def[:relations]) if obj_def.include?(:relations)
if obj_def[:presentation].has_key?(:allowed_aspects)
obj_def[:presentation][:allowed_aspects].each do |aspect|
aspect_data = get_aspect(aspect)
if aspect_data.nil?
STDERR.puts "Aspect '%s' is missing" % aspect
exit 1
end
obj_rel.merge!(aspect_data[:relations]) if aspect_data.has_key?(:relations) and aspect_data[:relations]
end
end
next if obj_rel.empty?

klass = find_klass(obj_name)

obj_relations_info = {}
obj_rel.each_pair do |field_name, rel|
foreign_klass = find_klass(rel[:object])
if foreign_klass.nil?
STDERR.puts "Relation '%s' for object '%s' is impossible: foreign object '%s' is missing" % [field_name, obj_name, rel[:object]]
exit 1
end
rel[:class_name] = foreign_klass.to_s

case rel[:type]
when :belongs_to
klass.belongs_to field_name, rel.reject {|key, val| not ActiveLdap::Associations::ClassMethods::VALID_BELONGS_TO_OPTIONS.include?(key) }
when :has_many
klass.has_many field_name, rel.reject {|key, val| not ActiveLdap::Associations::ClassMethods::VALID_HAS_MANY_OPTIONS.include?(key) }
else
raise "bug in '#{obj_name}' object relations (wrong type)"
end

obj_relations_info[field_name] = {
:foreign_klass => foreign_klass,
:single_value => ActiveLdap::Base.schema.attribute(rel[:foreign_key]).single_value?,
:read_only => rel[:read_only] || false
}
end
klass.relations_info = obj_relations_info
end
end

def objects
@object_definitions.keys.collect{|key| key.to_s }.sort
end

def load_shadow(shadow_name = nil)
clear_shadow

# TODO: local global config and use shadow name to load corresponding Shadow (or defautl Shadow if nil)
@shadow_dir = LdapShadows::Config::CFG_DIR

config_str = IO.read(File.join(@shadow_dir, "shadow.conf"))
config = YAML.load(config_str)
config_str_prv_filelist = [
File.join(ENV['HOME'], ".shadowwalker"),
File.join(@shadow_dir, "shadow_private.conf")
]
config_str_prv_filelist.each do |file|
if File.exists?(file)
config_str_prv = IO.read(file)
config.merge!(YAML.load(config_str_prv))
break
end
end
config.recursive_symbolize_keys!

$ldapctl.set_global_config(config[:presentation])

load_config_components("aspects") do |c_name, c_config|
$ldapctl.set_aspect(c_name, c_config)
end

ActiveLdap::Base.setup_connection(config[:ldap])

load_config_components("objects") do |c_name, c_config|
$ldapctl.load_object(c_name, c_config)
end

$ldapctl.load_relations
end

protected

def load_config_components(type)
c_config_dir = File.join(@shadow_dir, type)
c_config_pattern = File.join(c_config_dir, "**", "*.conf")

Dir.glob(c_config_pattern).each do |f|
next if f[0..0] == "."

c_name = File.basename(f).sub(".conf", "").to_sym
c_config = YAML.load(IO.read(f))
c_config.recursive_symbolize_keys!

yield(c_name, c_config)
end
end
end

# default location for mapped objects
module Objects
end

class Aspect
def hook_create
end

def hook_mod
end
end

module Aspects
end
end

(2-2/7)