require 'rubygems'

require 'slf4r/ruby_logger'
Slf4r::LoggerFacade4RubyLogger.level = ::Logger::DEBUG
require 'do_sqlite3'
require 'pathname'
$LOAD_PATH << Pathname(__FILE__).dirname.parent.expand_path + 'lib'

require 'ldap_resource'
#require 'ldap_facade_mock' # uncomment this to use the mock facade
require 'adapters/ldap_adapter'
require 'adapters/memory_adapter'

DataMapper.setup(:default, 'sqlite3::memory:')
DataMapper.setup(:ldap, {
                   :adapter  => 'ldap',
                   :host => 'localhost',
                   :port => '389',
                   :base => "dc=example,dc=com",
                   :bind_name => "cn=admin,dc=example,dc=com",
                   :password => "behappy"   
})
DataMapper.setup(:memory, {:adapter  => 'memory'})

class User
  include DataMapper::Resource
  property :id,        Serial, :field => "uidnumber"
  property :login,     String, :field => "uid", :unique_index => true
  property :hashed_password,  String, :field => "userpassword", :access => :private
  property :name,      String, :field => "cn"
  property :mail,      String
  property :age,       Integer, :field => "postalcode"
  property :alive,     Boolean, :field => "gecos"

  has n, :roles

  has n, :group_users

  def groups
    groups = GroupUser.all(:memberuid => id).collect{ |gu| gu.group }
    def groups.user=(user)
      @user = user
    end
    groups.user = self
    def groups.<<(group)
      unless member? group
        GroupUser.create(:memberuid => @user.id, :gidnumber => group.id)
        super
      end
      self
    end
    def groups.delete(group)
      gu = GroupUser.first(:memberuid => @user.id, :gidnumber => group.id)
      if gu
        gu.destroy
        super
      end
    end
    groups
  end

  dn_prefix { |user| "uid=#{user.login}"}

  treebase "ou=people"

  ldap_properties do |user|
    properties = { :objectclass => ["inetOrgPerson", "posixAccount", "shadowAccount"], :loginshell => "/bin/bash", :gidnumber => "10000" }
    properties[:sn] = "#{user.name.sub(/.*\ /, "")}"
    properties[:givenname] = "#{user.name.sub(/\ .*/, "")}"
    properties[:homedirectory] = "/home/#{user.login}"
    properties
  end

  def password=(password)
    attribute_set(:hashed_password, Ldap::Digest.ssha(password, "--#{Time.now}--#{login}--")) if password
  end
end

class Role
  include DataMapper::Resource
  property :id,       Serial, :field => "gidnumber"
  property :name,     String, :field => "cn"
  
  dn_prefix { |role| "cn=#{role.name}" }
  
  treebase "ou=groups"
  
  ldap_properties {:objectclass => "posixGroup"}

  belongs_to :user
end

class Group
  include DataMapper::Resource
  property :id,       Serial, :field => "gidnumber"
  property :name,     String, :field => "cn"
  
  dn_prefix { |group| "cn=#{group.name}" }
  
  treebase "ou=groups"
  
  ldap_properties {:objectclass => "posixGroup"}
end

class GroupUser
  include DataMapper::Resource
 
  dn_prefix { |group_user| "cn=#{group_user.group.name}" }
  
  treebase "ou=groups"
  
  multivalue_field :memberuid
  
  ldap_properties do |group_user|
    {:cn=>"#{group_user.group.name}",  :objectclass => "posixGroup"}
  end

  property :user_id, Integer, :key => true, :field => "memberuid"
  property :group_id, Integer, :key => true, :field => "gidnumber"

  def group
    Group.get!(group_id)
  end

  def group=(group)
    group_id = group.id
  end

  def user
    User.get!(user_id)
  end

  def user=(user)
    user_id = user.id
  end
end
DataMapper.auto_migrate!(:default)