ixtlan-guard 0.1.0 → 0.4.0

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2010 Kristian Meier
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/lib/generators/active_record/templates/flavor_migration.rb

@@ -0,0 +1,13 @@
+class Create<%= association_name.camelize %> < ActiveRecord::Migration
+  def self.up
+    create_table :<%= association_name %>, :id => false, :force => true do |t|
+      <% [file_name, group_name, user_name].sort.each do |name| -%>
+      t.integer :<%= name %>_id
+      <% end -%>
+    end
+  end
+
+  def self.down
+    drop_table :<%= association_name %>
+  end
+end

data/lib/generators/active_record/templates/flavor_model.rb

@@ -0,0 +1,8 @@
+class <%= association_class_name(plural_name) %> < <%= parent_class_name.classify %>
+<% attributes.select {|attr| attr.reference? }.each do |attribute| -%>
+  belongs_to :<%= attribute.name %>
+<% end -%>
+<% [name, group_name, user_name].sort.each do |ref_name| -%>
+  belongs_to :<%= ref_name %>
+<% end -%>
+end

data/lib/generators/active_record/templates/group_model.rb

@@ -0,0 +1,43 @@
+class <%= group_class_name %> < <%= parent_class_name.classify %>
+<% attributes.select {|attr| attr.reference? }.each do |attribute| -%>
+  belongs_to :<%= attribute.name %>
+<% end -%>
+
+  has_and_belongs_to_many :<%= plural_user_name %>
+
+  ROOT = 'root'
+  ADMIN = 'admin'
+
+  def self.admin_group
+    find_by_<%= attributes.first.name %>(ADMIN)
+  end
+
+  def self.root_group
+    find_by_<%= attributes.first.name %>(ROOT)
+  end
+
+  def admin?
+    <%= attributes.first.name %> == ADMIN
+  end
+
+  def root?
+    <%= attributes.first.name %> == ROOT
+  end
+
+  def self.get(id_or_<%= attributes.first.name %>_or_<%= file_name %>)
+    case id_or_<%= attributes.first.name %>_or_<%= file_name %>
+    when Fixnum
+      find(id_or_<%= attributes.first.name %>_or_<%= file_name %>)
+    when String
+      find_by_<%= attributes.first.name %>(id_or_<%= attributes.first.name %>_or_<%= file_name %>)
+    when Symbol
+      find_by_<%= attributes.first.name %>(id_or_<%= attributes.first.name %>_or_<%= file_name %>.to_s)
+    else
+      id_or_<%= attributes.first.name %>_or_<%= file_name %>
+    end
+  end
+
+  def to_name
+    <%= group_field_name %>
+  end
+end

data/lib/generators/active_record/templates/group_user_migration.rb

@@ -0,0 +1,13 @@
+class Create<%= group_user_name.camelize %> < ActiveRecord::Migration
+  def self.up
+    create_table :<%= group_user_name %>, :id => false, :force => true do |t|
+      <% [group_name, user_name].sort.each do |name| -%>
+      t.integer :<%= name %>_id
+      <% end -%>
+    end
+  end
+
+  def self.down
+    drop_table :<%= group_user_name %>
+  end
+end

data/lib/generators/active_record/templates/user_model.rb

@@ -0,0 +1,124 @@
+require 'ixtlan/guard/models/user_update_manager'
+class <%= user_class_name %> < <%= parent_class_name.classify %>
+<% attributes.select {|attr| attr.reference? }.each do |attribute| -%>
+  belongs_to :<%= attribute.name %>
+<% end -%>
+
+  has_and_belongs_to_many :<%= plural_group_name %>
+
+  attr_accessor :id
+
+  def self.update_manager
+    @manager ||= Ixtlan::Guard::Models::UserUpdateManager.new( :group_model => <%= group_class_name %>, :user_id => :<%= user_name %>_id, :group_id => :<%= group_name %>_id, :plural_group_name => :<%= plural_group_name %> )
+  end
+
+  alias :create! :create
+  def self.create(params = {})
+    u = self.new
+    u.current_user = params.delete("current_user") || params.delete(:current_user)
+    u.update_attributes(params)
+    u
+  end
+
+  def update_attributes(params)
+    result = []
+ <% flavors.each do |flavor| %>
+    # update <%= flavor.pluralize %>
+    if result.all?{|a| a} 
+      result << self.class.update_manager.update(self, params, :flavor_id => :<%= flavor %>_id, :flavors_for_group => :<%= flavor.pluralize %>_for_<%= group_name %>, :association_model => <%= association_class_name(flavor.pluralize) %>) 
+    end
+<% end -%> 
+
+    # update <%= plural_group_name %>
+    self.class.update_manager.update_groups(self, params) if result.all?{|a| a}
+  end
+
+  unless respond_to? :current_user
+    # do not use attr_accessor to allow them to be used 
+    # for mass_assignmnet_protection
+    def current_user=(u)
+      @current_user = u
+    end
+    
+    def current_user
+      @current_user
+    end
+
+    after_save do
+      @current_user = nil
+    end
+  end
+
+  def root?
+    <%= plural_group_name %>.detect {|g| g.root? } != nil
+  end
+
+<% flavors.each do |flavor| 
+   plural_flavor = flavor.pluralize  %>
+  def managed_<%= plural_flavor %>_for_<%= group_name %>(gid)
+    <%= group_name %> = <%= group_class_name %>.get(gid)
+    if <%= group_name %>.root?
+      []
+    else
+      existing = <%= plural_flavor %>_for_<%= group_name %>(<%= group_name %>)
+      managed = <%= plural_flavor %>_for_<%= group_name %>(<%= group_class_name %>.admin_group)
+      (managed - (managed - existing))
+    end
+  end
+
+  def <%= plural_flavor %>_for_<%= group_name %>(group_or_id)
+    if root?
+      <%= flavor.camelize %>.all
+    else
+      <%= group_name %> = <%= group_class_name %>.get(group_or_id)
+      <%= association_class_name(plural_flavor) %>.all(:conditions => ["<%= user_name %>_id=? and <%= group_name %>_id=?", id, <%= group_name %>.id]).collect { |dgu| dgu.<%= flavor %> }
+    end
+  end
+
+  def <%= flavor %>_<%= group_name %>?(<%= flavor %>, <%= group_name %>)
+    <%= association_class_name(plural_flavor) %>.count(:conditions => ["<%= user_name %>_id=? and <%= group_name %>_id=? and <%= flavor %>_id=?", id, <%= group_name %>.id, <%= flavor %>.id]) == 1 
+  end
+
+<% end -%>
+
+  def all_<%= plural_group_name %>
+    if root?
+      <%= group_class_name %>.all
+    else
+      <%= plural_group_name %>
+    end
+  end
+
+  def to_xml
+    to_hash.to_xml(:root => "<%= file_name %>")
+  end
+
+  def to_json
+    { :<%= file_name %> => to_hash }.to_json
+  end
+
+  def to_hash
+    map = attributes.dup
+    g = (map[:<%= plural_group_name %>] = [])
+    <%= plural_group_name %>.each do |<%= group_name %>|
+      gg = <%= group_name %>.attributes.dup
+      gg.delete("<%= group_name %>_id")
+      gg.delete("<%= file_name %>_id")
+<% flavors.each do |flavor| 
+   plural_flavor = flavor.pluralize  %>
+      <%= plural_flavor %> = <%= plural_flavor %>_for_<%= group_name %>(<%= group_name %>)
+      if <%= plural_flavor %>.size > 0
+        d = (gg[:<%= plural_flavor %>] = [])
+        <%= plural_flavor %>.each do |<%= flavor %>|
+          dd = <%= flavor %>.attributes.dup
+          dd.delete("created_at")
+          dd.delete("updated_at")
+          d << dd
+        end
+      end
+<% end -%>
+      g << gg
+    end
+    map
+  end
+end

data/lib/generators/active_record/user_management_models_generator.rb

@@ -0,0 +1,202 @@
+require 'rails/generators/active_record/model/model_generator'
+module ActiveRecord
+  class UserManagementModelsGenerator < ::ActiveRecord::Generators::ModelGenerator
+
+    source_paths << ActiveRecord::Generators::ModelGenerator.source_root
+
+    source_root File.expand_path('../templates', __FILE__)
+    
+    arguments.clear # clear name argument from NamedBase
+    
+    argument :params, :type => :array, :required => true
+
+    remove_hook_for :test_framework
+
+    def name
+      @name ||= user_params[0]
+    end
+
+    def attributes
+      @attributes ||= user_params[1,10000]
+    end
+
+    def attributes=(a = nil)
+      @attributes = a
+    end
+
+    protected
+    def user_params
+      retrieve_params(@params)
+    end
+
+    def group_params
+      result = retrieve_params(@params[user_params.size, 10000])
+      case result.size
+      when 0
+        ["group", "name:string"]
+      when 1
+        result << "name:string"
+      else
+        result
+      end
+    end
+
+    def flavor_params(index, count = 0)
+      count = user_params.size + retrieve_params(@params[user_params.size, 10000]).size if count == 0
+      if(index > 0)
+        count += retrieve_params(@params[count, 10000]).size
+        if count != @params.size
+          flavor_params(index - 1, count)
+        end
+      else
+        retrieve_params(@params[count, 10000])
+      end
+    end
+
+    def retrieve_params(params)
+      done = nil
+      params.select do |para|
+        if done.nil?
+          done = false
+          true
+        elsif done
+          false
+        else
+          done = (para =~ /[^:]+\:[^:]+/).nil?
+          !done
+        end
+      end
+    end
+
+    public
+    
+    check_class_collision
+    
+    class_option :migration,  :type => :boolean
+    class_option :timestamps, :type => :boolean
+    class_option :parent,     :type => :string, :desc => "The parent class for the generated model"
+
+    def user_name
+      @user_name ||= user_params[0].singularize
+    end
+
+    def group_name
+      @group_name ||= group_params[0].singularize
+    end
+
+    def group_field_name
+      @group_field_name ||= (group_params[1] || "name:").sub(/\:.*/, '')
+    end
+
+    def plural_user_name
+      @plural_user_name ||= user_name.pluralize
+    end
+
+    def plural_group_name
+      @plural_group_name ||= group_name.pluralize
+    end
+
+    def user_class_name
+      @user_class_name ||= user_name.camelize
+    end
+
+    def group_class_name
+      @group_class_name ||= group_name.camelize
+    end
+
+    def association_name
+      @association_name ||= [plural_name, plural_group_name, plural_user_name].sort.join("_")
+    end
+
+    def group_user_name
+      @group_user_name ||= [plural_group_name, plural_user_name].sort.join("_")
+    end
+
+    def flavors
+      @flavors ||= params_array[2,10000].collect {|c| c[0].to_s }
+    end
+
+    def create_migration_file
+      return unless options[:migration] && options[:parent].nil?
+      
+      index = 0
+      params_array.each do |params|
+        setup(params)
+        migration_template "migration.rb", "db/migrate/create_#{table_name}.rb"
+        index += 1
+        if(index > 2)
+          migration_template "flavor_migration.rb", "db/migrate/create_#{association_name}.rb"
+        
+        end
+      end
+      migration_template "group_user_migration.rb", "db/migrate/create_#{group_user_name}.rb"  
+    end
+
+    def create_model_file
+      setup(user_params)
+      template 'user_model.rb', File.join('app/models', class_path, "#{file_name}.rb")
+      
+      setup(group_params)
+      template 'group_model.rb', File.join('app/models', class_path, "#{file_name}.rb")
+      
+      if params_array.size > 2
+        params_array[2,10000].each do |params|
+          setup(params)
+          template 'model.rb', File.join('app/models', class_path, "#{file_name}.rb")
+          template 'flavor_model.rb', File.join('app/models', class_path, "#{association_name.singularize}.rb")
+        end
+      end
+    end
+    
+    def create_module_file
+      params_array.each do |params|
+        assign_names!(params[0])
+        unless class_path.empty?
+          template 'module.rb', File.join('app/models', "#{class_path.join('/')}.rb") if behavior == :invoke
+        end
+      end
+    end
+    
+    protected
+
+    def association_class_name(plural_flavor)
+      @association_class_name ||= {}
+      @association_class_name[plural_flavor] ||= [plural_flavor, plural_group_name, plural_user_name].sort.join("_").singularize.camelize
+    end
+
+    def setup(params)      
+      @name = params[0]
+      assign_names!(params[0])
+      self.attributes = params[1, 10000]
+      parse_attributes!
+      @file_path = nil
+      @class_name = nil
+      @human_name = nil
+      @plural_name = nil
+      @i18n_scope = nil
+      @table_name = nil
+      @singular_table_name = nil
+      @plural_table_name = nil
+      @plural_file_name = nil
+      @route_url = nil
+      @association_name = nil
+    end
+
+    def params_array
+      @array ||= begin
+                   result = [user_params, group_params]
+                   index = 0
+                   while( flavor = flavor_params(index) ) do
+                     result << flavor if flavor.size > 0
+                     index += 1
+                   end
+                   result
+                 end
+    end
+
+    def parent_class_name
+      options[:parent] || "ActiveRecord::Base"
+    end
+    
+  end
+end

data/lib/generators/erb/user_management_controller_generator.rb

@@ -0,0 +1,10 @@
+require 'rails/generators/erb'
+require 'rails/generators/resource_helpers'
+
+module Erb
+  class UserManagementControllerGenerator < ::Erb::Generators::ScaffoldGenerator
+  
+    source_root File.expand_path('../templates', __FILE__)
+    
+  end
+end