trimodel 0.0.1 → 0.0.2

data/lib/generators/trimodel/delete_generator.rb

@@ -0,0 +1,54 @@
+require 'rails/generators'
+
+module Trimodel
+  class DeleteGenerator < Rails::Generators::Base
+    desc "Delete an existing triple model relationship"
+    source_root File.expand_path(File.join(File.dirname(__FILE__), 'templates'))
+
+    class_option :models, :type => :array, :desc => "The model triplet from where the association will be removed"
+
+    def check_parameters
+      if options[:models].class == NilClass
+        puts "error: You need to give the models you want to associate"
+        puts "e.g. rails g trimodel:new --models a b c"
+      elsif options[:models].size != 3
+        puts "error: Wrong number of models, they should be 3"
+      end
+    end
+
+    def rollback_migrations
+      list_and_perform_on_files options[:models][0], options[:models][1] { |f,a,b| rollback_migration(f,a,b) }
+      list_and_perform_on_files options[:models][1], options[:models][2] { |f,a,b| rollback_migration(f,a,b) }
+    end
+
+    def delete_migration_files
+      list_and_perform_on_files options[:models][0], options[:models][1] { |f,a,b| delete_migration_file(f,a,b) }
+      list_and_perform_on_files options[:models][1], options[:models][2] { |f,a,b| delete_migration_file(f,a,b) }
+    end
+
+    def delete_trimodel_file
+      File.delete(Rails.root + "config/initializers/trimodel.rb")
+    end
+
+    private
+      def list_and_perform_on_files model_a, model_b
+        migration_files = Dir.entries(Rails.root + "db/migrate")
+        for file in migration_files
+          yield(file, model_a, model_b)
+        end
+      end
+
+      def delete_migration_file file, model_a, model_b
+        if /_create_#{model_a.pluralize.downcase}_#{model_b.pluralize.downcase}_trimodel/ =~ file
+          full_path = Rails.root.to_s + "/db/migrate/" + file
+          File.delete(full_path)
+        end
+      end
+
+      def rollback_migration file, model_a, model_b
+        if /_create_#{model_a.pluralize.downcase}_#{model_b.pluralize.downcase}_trimodel/ =~ file
+          %x[rake db:migrate:down VERSION=#{file[0,14]}]
+        end
+      end
+  end
+end

data/lib/generators/trimodel/new_generator.rb

@@ -0,0 +1,120 @@
+require 'rails/generators'
+require 'active_support/inflector'
+require 'rake'
+
+module Trimodel
+  class NewGenerator < Rails::Generators::Base
+    desc "Creates a new triple model relationship"
+    source_root File.expand_path(File.join(File.dirname(__FILE__), 'templates'))
+
+    class_option :models, :type => :array, :desc => "The triplet model that will be associated"
+
+    def check_parameters
+      if options[:models].class == NilClass
+        puts "error: You need to give the models you want to associate"
+        puts "e.g. rails g trimodel:new --models a b c"
+      elsif options[:models].size != 3
+        puts "error: Wrong number of models, they should be 3"
+      end
+    end
+
+    def create_migration_files
+      create_migration_file options[:models][0], options[:models][1]
+      #need to wait so the timestamp has a different value
+      #and forms a correct migration file name
+      sleep(1)
+      create_migration_file options[:models][1], options[:models][2]
+    end
+
+    def perform_migrations
+      %x[rake db:migrate]
+    end
+
+    def create_trimodel_file
+      File.open(Rails.root + "config/initializers/trimodel.rb",
+        File::CREAT|File::RDWR) do |f|
+          add_one_asoc_to_model f, options[:models][0], options[:models][1], options[:models][2]
+          add_two_asocs_to_model f, options[:models][1], options[:models][0], options[:models][2]
+          add_one_asoc_to_model f, options[:models][2], options[:models][1], options[:models][0]
+      end
+    end
+
+    private
+      def add_one_asoc_to_model file, model_a, model_b, model_c
+        file.write(create_class_definition(model_a))
+        file.write(add_n_to_n_association(model_a, model_b))
+        file.write(add_iterator_method(model_b, model_c))
+        file.write(add_end)
+      end
+
+      def add_two_asocs_to_model file, model_a, model_b, model_c
+        file.write(create_class_definition(model_a))
+        file.write(add_n_to_n_association(model_a, model_b))
+        file.write(add_n_to_n_association(model_a, model_c))
+        file.write(add_end)
+      end
+
+      def create_class_definition model
+        "class #{model} < ActiveRecord::Base\n"
+      end
+
+      def add_n_to_n_association model_a, model_b
+        "  has_and_belongs_to_many :#{model_b.pluralize.downcase}\n"
+      end
+
+      def add_iterator_method bridge_model, target_model
+        method_body=<<-eos
+
+  def #{target_model.pluralize.downcase}
+    records = []
+    self.#{bridge_model.pluralize.downcase}.each do |bm|
+      records << bm.#{target_model.pluralize.downcase}.map { |obj| obj }
+    end
+    records.flatten
+  end
+eos
+        method_body
+      end
+
+      def add_end
+        "end\n\n"
+      end
+
+      def create_migration_file model_a, model_b
+        path = Rails.root.to_s
+        path << "/db/migrate/#{create_timestamp}_"
+        path << "create_#{model_a.pluralize.downcase}_"
+        path << "#{model_b.pluralize.downcase}_trimodel_join_table.rb" 
+        File.open(path,File::CREAT|File::RDWR) do |fi|
+            fi.write(write_migration_code(model_a, model_b))
+        end
+      end
+
+      def create_timestamp
+        Time.now.utc.to_s.gsub('-','').gsub(':','').gsub(' ','')[0..-4]
+      end
+
+      def write_migration_code model_a, model_b
+        if (model_a.pluralize.downcase[0] < model_b.pluralize.downcase)
+          first, second = model_a.pluralize.downcase, model_b.pluralize.downcase
+        else
+          first, second = model_b.pluralize.downcase, model_a.pluralize.downcase
+        end
+        code=<<-eos
+class Create#{model_a.pluralize}#{model_b.pluralize}TrimodelJoinTable < ActiveRecord::Migration
+  def self.up
+    create_table :#{first}_#{second}, :id => false do |t|
+      t.integer :#{model_a.downcase}_id
+      t.integer :#{model_b.downcase}_id
+    end
+  end
+
+  def self.down
+    drop_table :#{first}_#{second}
+  end
+end
+eos
+        code
+      end
+  end
+end

data/lib/trimodel/engine.rb

@@ -0,0 +1,6 @@
+require 'rails'
+
+module Trimodel
+  class Engine < Rails::Engine
+  end
+end

data/trimodel.rb

@@ -0,0 +1,3 @@
+module Trimodel
+  require 'trimodel/engine' if defined?(Rails)
+end

metadata

@@ -1,15 +1,15 @@
 --- !ruby/object:Gem::Specification
 name: trimodel
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
-- Angelos
+- Angelos Kapsimanis
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-03-12 00:00:00.000000000 Z
+date: 2013-03-18 00:00:00.000000000 Z
 dependencies: []
 description: ! "      A gem that enables transparent relationships \n      between
   2 models associated through a third one.\n"
@@ -18,7 +18,10 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- lib/trimodel.rb
+- trimodel.rb
+- lib/trimodel/engine.rb
+- lib/generators/trimodel/new_generator.rb
+- lib/generators/trimodel/delete_generator.rb
 homepage: https://github.com/akxs14/trimodel
 licenses: []
 post_install_message: 
@@ -39,7 +42,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.24
+rubygems_version: 1.8.25
 signing_key: 
 specification_version: 3
 summary: A negotiator between 3 models

data/lib/trimodel.rb

@@ -1,3 +0,0 @@
-class Trimodel
-
-end