Project

General

Profile

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

root/lib/ldap_shadows/controller.rb @ 7da3bb75

89d8bebc Marc Dequènes (Duck)
#--
# 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'
b52f0f7d Marc Dequènes (Duck)
require 'ldap_shadows/activeldap_fixes'
89d8bebc Marc Dequènes (Duck)
require 'ldap_shadows/object'


module LdapShadows
class Controller
def initialize(mod_container = LdapShadows::Objects)
@mod_container = mod_container
74f1db3a Marc Dequènes (Duck)
clear_shadow
LdapShadows::LdapObject.mapper = self
end

def clear_shadow
89d8bebc Marc Dequènes (Duck)
@object_definitions = {}
@aspects = {}
74f1db3a Marc Dequènes (Duck)
# TODO: should replace @aspects properly one day
dba6a81b Marc Dequènes (Duck)
@aspects2 = {}
74f1db3a Marc Dequènes (Duck)
@shadow_dir = nil
89d8bebc Marc Dequènes (Duck)
end

1744c478 Marc Dequènes (Duck)
def set_global_config(global_config)
@global_config = global_config
end

def get_global_config
@global_config
end

89d8bebc Marc Dequènes (Duck)
def set_aspect(aspect_name, aspect_def)
@aspects[aspect_name] = aspect_def
dba6a81b Marc Dequènes (Duck)
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
89d8bebc Marc Dequènes (Duck)
end

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

dba6a81b Marc Dequènes (Duck)
def get_aspect_klass(aspect_name)
@aspects2[aspect_name.to_sym]
end

89d8bebc Marc Dequènes (Duck)
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)
bbb82941 Marc Dequènes (Duck)
klass.handle = obj_name
89d8bebc Marc Dequènes (Duck)
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)
b6fb956a Marc Dequènes (Duck)
obj_def[:presentation][:allowed_aspects].each do |aspect|
aspect_data = get_aspect(aspect)
if aspect_data.nil?
STDERR.puts "Aspect '%s' is missing" % aspect
cee99c3d Marc Dequènes (Duck)
exit 1
end
b6fb956a Marc Dequènes (Duck)
obj_rel.merge!(aspect_data[:relations]) if aspect_data.has_key?(:relations) and aspect_data[:relations]
89d8bebc Marc Dequènes (Duck)
end
end
next if obj_rel.empty?

klass = find_klass(obj_name)

43bd8057 Marc Dequènes (Duck)
obj_relations_info = {}
89d8bebc Marc Dequènes (Duck)
obj_rel.each_pair do |field_name, rel|
foreign_klass = find_klass(rel[:object])
1d23b8e8 Marc Dequènes (Duck)
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
89d8bebc Marc Dequènes (Duck)
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
43bd8057 Marc Dequènes (Duck)
obj_relations_info[field_name] = {
:foreign_klass => foreign_klass,
34cb4054 Marc Dequènes (Duck)
:single_value => ActiveLdap::Base.schema.attribute(rel[:foreign_key]).single_value?,
:read_only => rel[:read_only] || false
43bd8057 Marc Dequènes (Duck)
}
89d8bebc Marc Dequènes (Duck)
end
43bd8057 Marc Dequènes (Duck)
klass.relations_info = obj_relations_info
89d8bebc Marc Dequènes (Duck)
end
end
6089b33a Marc Dequènes (Duck)
def objects
@object_definitions.keys.collect{|key| key.to_s }.sort
end
74f1db3a Marc Dequènes (Duck)
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
89d8bebc Marc Dequènes (Duck)
end

# default location for mapped objects
module Objects
end
dba6a81b Marc Dequènes (Duck)
class Aspect
def hook_create
end

def hook_mod
end
end

module Aspects
end
89d8bebc Marc Dequènes (Duck)
end