magic_model_generator 0.3.0

data/examples/singleview/variable_defn_dependency.rb

@@ -0,0 +1,11 @@
+class VariableDefnDependency < ActiveRecord::Base
+  belongs_to :variable_defn, :class_name => 'VariableDefn', :foreign_key => :variable_defn_id
+  validates_presence_of :variable_defn_id
+  validates_numericality_of :variable_defn_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :effective_start_date
+  validates_presence_of :effective_end_date
+  validates_presence_of :dependent_variable_defn_id
+  validates_numericality_of :dependent_variable_defn_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :dependency_count
+  validates_numericality_of :dependency_count, :allow_nil => false, :only_integer => true
+end

data/examples/singleview/variable_defn_direct.rb

@@ -0,0 +1,15 @@
+class VariableDefnDirect < ActiveRecord::Base
+  belongs_to :variable_defn, :class_name => 'VariableDefn', :foreign_key => :variable_defn_id
+  validates_presence_of :variable_defn_id
+  validates_numericality_of :variable_defn_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :effective_start_date
+  validates_presence_of :effective_end_date
+  validates_presence_of :last_modified
+  validates_presence_of :view_name
+  validates_length_of :view_name, :allow_nil => false, :maximum => 30
+  validates_presence_of :column_name
+  validates_length_of :column_name, :allow_nil => false, :maximum => 30
+  validates_presence_of :column_nr
+  validates_numericality_of :column_nr, :allow_nil => false, :only_integer => true
+  validates_length_of :description, :allow_nil => true, :maximum => 255
+end

data/examples/singleview/variable_defn_history.rb

@@ -0,0 +1,22 @@
+class VariableDefnHistory < ActiveRecord::Base
+  belongs_to :variable_defn, :class_name => 'VariableDefn', :foreign_key => :variable_defn_id
+  belongs_to :atlanta_operator, :class_name => 'AtlantaOperator', :foreign_key => :atlanta_operator_id
+  belongs_to :atlanta_group, :class_name => 'AtlantaGroup', :foreign_key => :atlanta_group_id
+  validates_presence_of :variable_defn_id
+  validates_numericality_of :variable_defn_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :variable_defn_name
+  validates_length_of :variable_defn_name, :allow_nil => false, :maximum => 73
+  validates_presence_of :last_modified
+  validates_numericality_of :atlanta_operator_id, :allow_nil => true, :only_integer => true
+  validates_numericality_of :atlanta_group_id, :allow_nil => true, :only_integer => true
+  validates_presence_of :effective_start_date
+  validates_presence_of :effective_end_date
+  validates_presence_of :data_type_code
+  validates_numericality_of :data_type_code, :allow_nil => false, :only_integer => true
+  validates_presence_of :variable_type_code
+  validates_numericality_of :variable_type_code, :allow_nil => false, :only_integer => true
+  validates_presence_of :context_code
+  validates_numericality_of :context_code, :allow_nil => false, :only_integer => true
+  validates_presence_of :application_env_code
+  validates_numericality_of :application_env_code, :allow_nil => false, :only_integer => true
+end

data/examples/singleview/work_force.rb

@@ -0,0 +1,25 @@
+class WorkForce < ActiveRecord::Base
+  belongs_to :work_force, :class_name => 'WorkForce', :foreign_key => :work_force_id
+  belongs_to :atlanta_operator, :class_name => 'AtlantaOperator', :foreign_key => :atlanta_operator_id
+  belongs_to :atlanta_group, :class_name => 'AtlantaGroup', :foreign_key => :atlanta_group_id
+  has_many :atlanta_profile_wf_maps, :class_name => 'AtlantaProfileWfMap', :foreign_key => :work_force_id
+  has_many :customer_query_progresses, :class_name => 'CustomerQueryProgress', :foreign_key => :work_force_id
+  has_many :work_forces, :class_name => 'WorkForce', :foreign_key => :work_force_id
+  has_many :atlanta_profiles, :through => atlanta_profile_wf_maps
+  has_many :customer_queries, :through => customer_query_progresses
+  has_many :atlanta_operators, :through => customer_query_progresses
+  has_many :atlanta_groups, :through => work_forces
+  has_many :atlanta_operators, :through => work_forces
+  validates_presence_of :work_force_id
+  validates_numericality_of :work_force_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :work_force_name
+  validates_length_of :work_force_name, :allow_nil => false, :maximum => 30
+  validates_presence_of :description
+  validates_length_of :description, :allow_nil => false, :maximum => 255
+  validates_presence_of :last_modified
+  validates_presence_of :atlanta_operator_id
+  validates_numericality_of :atlanta_operator_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :atlanta_group_id
+  validates_numericality_of :atlanta_group_id, :allow_nil => false, :only_integer => true
+  validates_length_of :version_str, :allow_nil => true, :maximum => 255
+end

data/examples/singleview/xpsi_addr_val_commtax21.rb

@@ -0,0 +1,24 @@
+class XpsiAddrValCommtax21 < ActiveRecord::Base
+  validates_presence_of :state_abbrev
+  validates_length_of :state_abbrev, :allow_nil => false, :maximum => 2
+  validates_presence_of :county_name_uppercase
+  validates_length_of :county_name_uppercase, :allow_nil => false, :maximum => 15
+  validates_presence_of :city_name_uppercase
+  validates_length_of :city_name_uppercase, :allow_nil => false, :maximum => 25
+  validates_presence_of :zip_begin
+  validates_length_of :zip_begin, :allow_nil => false, :maximum => 5
+  validates_presence_of :zip_end
+  validates_length_of :zip_end, :allow_nil => false, :maximum => 5
+  validates_presence_of :seqnr
+  validates_numericality_of :seqnr, :allow_nil => false, :only_integer => true
+  validates_presence_of :geocode
+  validates_length_of :geocode, :allow_nil => false, :maximum => 12
+  validates_presence_of :authority_code
+  validates_numericality_of :authority_code, :allow_nil => false, :only_integer => true
+  validates_presence_of :state_name
+  validates_length_of :state_name, :allow_nil => false, :maximum => 20
+  validates_presence_of :county_name
+  validates_length_of :county_name, :allow_nil => false, :maximum => 15
+  validates_presence_of :city_name
+  validates_length_of :city_name, :allow_nil => false, :maximum => 25
+end

data/examples/singleview/xpsi_addr_val_msag.rb

@@ -0,0 +1,24 @@
+class XpsiAddrValMsag < ActiveRecord::Base
+  validates_presence_of :street_name_uppercase
+  validates_length_of :street_name_uppercase, :allow_nil => false, :maximum => 48
+  validates_presence_of :community_name_uppercase
+  validates_length_of :community_name_uppercase, :allow_nil => false, :maximum => 32
+  validates_presence_of :state_abbrev
+  validates_length_of :state_abbrev, :allow_nil => false, :maximum => 2
+  validates_presence_of :low_number
+  validates_length_of :low_number, :allow_nil => false, :maximum => 8
+  validates_presence_of :high_number
+  validates_length_of :high_number, :allow_nil => false, :maximum => 8
+  validates_presence_of :seqnr
+  validates_numericality_of :seqnr, :allow_nil => false, :only_integer => true
+  validates_length_of :prefix_dir, :allow_nil => true, :maximum => 2
+  validates_length_of :postfix_dir, :allow_nil => true, :maximum => 2
+  validates_presence_of :street_name
+  validates_length_of :street_name, :allow_nil => false, :maximum => 48
+  validates_length_of :street_suffix, :allow_nil => true, :maximum => 8
+  validates_length_of :odd_even_ind, :allow_nil => true, :maximum => 1
+  validates_presence_of :community_name
+  validates_length_of :community_name, :allow_nil => false, :maximum => 32
+  validates_length_of :esn, :allow_nil => true, :maximum => 5
+  validates_length_of :exchange, :allow_nil => true, :maximum => 4
+end

data/examples/singleview/xpsi_addr_val_zip.rb

@@ -0,0 +1,14 @@
+class XpsiAddrValZip < ActiveRecord::Base
+  validates_presence_of :state_abbrev
+  validates_length_of :state_abbrev, :allow_nil => false, :maximum => 2
+  validates_presence_of :county_name_uppercase
+  validates_length_of :county_name_uppercase, :allow_nil => false, :maximum => 15
+  validates_presence_of :city_name_uppercase
+  validates_length_of :city_name_uppercase, :allow_nil => false, :maximum => 25
+  validates_presence_of :zip_begin
+  validates_length_of :zip_begin, :allow_nil => false, :maximum => 5
+  validates_presence_of :zip_end
+  validates_length_of :zip_end, :allow_nil => false, :maximum => 5
+  validates_presence_of :seqnr
+  validates_numericality_of :seqnr, :allow_nil => false, :only_integer => true
+end

data/examples/singleview/xpsi_ne_batch.rb

@@ -0,0 +1,20 @@
+class XpsiNeBatch < ActiveRecord::Base
+  belongs_to :xpsi_ne_batch, :class_name => 'XpsiNeBatch', :foreign_key => :xpsi_ne_batch_id
+  belongs_to :error_message, :class_name => 'ErrorMessage', :foreign_key => :error_message_id
+  has_many :xpsi_ne_batches, :class_name => 'XpsiNeBatch', :foreign_key => :xpsi_ne_batch_id
+  has_many :error_messages, :through => xpsi_ne_batches
+  validates_presence_of :xpsi_ne_batch_id
+  validates_numericality_of :xpsi_ne_batch_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :source
+  validates_length_of :source, :allow_nil => false, :maximum => 80
+  validates_presence_of :create_date
+  validates_numericality_of :sequence_number, :allow_nil => true, :only_integer => true
+  validates_presence_of :ne_filename
+  validates_length_of :ne_filename, :allow_nil => false, :maximum => 255
+  validates_numericality_of :error_message_id, :allow_nil => true, :only_integer => true
+  validates_length_of :error_message, :allow_nil => true, :maximum => 255
+  validates_presence_of :revoke_ind_code
+  validates_numericality_of :revoke_ind_code, :allow_nil => false, :only_integer => true
+  validates_presence_of :event_count
+  validates_numericality_of :event_count, :allow_nil => false, :only_integer => true
+end

data/examples/singleview/xpsi_product_batch.rb

@@ -0,0 +1,18 @@
+class XpsiProductBatch < ActiveRecord::Base
+  belongs_to :product_instance, :class_name => 'ProductInstance', :foreign_key => :product_instance_id
+  validates_presence_of :product_instance_id
+  validates_numericality_of :product_instance_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :effective_date
+  validates_presence_of :entity_type
+  validates_length_of :entity_type, :allow_nil => false, :maximum => 255
+  validates_presence_of :entity_id
+  validates_numericality_of :entity_id, :allow_nil => false, :only_integer => true
+  validates_numericality_of :svcda_id, :allow_nil => true, :only_integer => true
+  validates_numericality_of :seq_nr, :allow_nil => true, :only_integer => true
+  validates_presence_of :entity_status
+  validates_numericality_of :entity_status, :allow_nil => false, :only_integer => true
+  validates_numericality_of :old_entity_status, :allow_nil => true, :only_integer => true
+  validates_presence_of :entity_action
+  validates_numericality_of :entity_action, :allow_nil => false, :only_integer => true
+  validates_length_of :addn_info, :allow_nil => true, :maximum => 1999
+end

data/examples/singleview/xpsi_product_status.rb

@@ -0,0 +1,14 @@
+class XpsiProductStatus < ActiveRecord::Base
+  belongs_to :product_instance, :class_name => 'ProductInstance', :foreign_key => :product_instance_id
+  validates_presence_of :product_instance_id
+  validates_numericality_of :product_instance_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :effective_date
+  validates_presence_of :entity_type
+  validates_length_of :entity_type, :allow_nil => false, :maximum => 255
+  validates_presence_of :entity_id
+  validates_numericality_of :entity_id, :allow_nil => false, :only_integer => true
+  validates_numericality_of :svcda_id, :allow_nil => true, :only_integer => true
+  validates_numericality_of :seq_nr, :allow_nil => true, :only_integer => true
+  validates_presence_of :entity_status
+  validates_numericality_of :entity_status, :allow_nil => false, :only_integer => true
+end

data/examples/singleview/xpsi_tax_error.rb

@@ -0,0 +1,34 @@
+class XpsiTaxError < ActiveRecord::Base
+  belongs_to :xpsi_tax_error, :class_name => 'XpsiTaxError', :foreign_key => :xpsi_tax_error_id
+  belongs_to :normalised_event, :class_name => 'NormalisedEvent', :foreign_key => :normalised_event_id
+  belongs_to :service, :class_name => 'Service', :foreign_key => :service_id
+  belongs_to :customer_node, :class_name => 'CustomerNode', :foreign_key => :customer_node_id
+  belongs_to :error_message, :class_name => 'ErrorMessage', :foreign_key => :error_message_id
+  belongs_to :adjustment, :class_name => 'Adjustment', :foreign_key => :adjustment_id
+  has_many :xpsi_tax_errors, :class_name => 'XpsiTaxError', :foreign_key => :xpsi_tax_error_id
+  has_many :customer_nodes, :through => xpsi_tax_errors
+  has_many :adjustments, :through => xpsi_tax_errors
+  has_many :error_messages, :through => xpsi_tax_errors
+  has_many :services, :through => xpsi_tax_errors
+  has_many :normalised_events, :through => xpsi_tax_errors
+  validates_presence_of :xpsi_tax_error_id
+  validates_numericality_of :xpsi_tax_error_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :normalised_event_id
+  validates_numericality_of :normalised_event_id, :allow_nil => false, :only_integer => true
+  validates_numericality_of :service_id, :allow_nil => true, :only_integer => true
+  validates_numericality_of :customer_node_id, :allow_nil => true, :only_integer => true
+  validates_presence_of :tax_module_code
+  validates_numericality_of :tax_module_code, :allow_nil => false, :only_integer => true
+  validates_length_of :tax_module_return_value, :allow_nil => true, :maximum => 40
+  validates_numericality_of :tax_type_code, :allow_nil => true, :only_integer => true
+  validates_length_of :tax_key, :allow_nil => true, :maximum => 255
+  validates_numericality_of :amount, :allow_nil => true
+  validates_presence_of :adjust_ind
+  validates_numericality_of :adjust_ind, :allow_nil => false, :only_integer => true
+  validates_presence_of :error_message_id
+  validates_numericality_of :error_message_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :error_message
+  validates_length_of :error_message, :allow_nil => false, :maximum => 255
+  validates_numericality_of :adjustment_id, :allow_nil => true, :only_integer => true
+  validates_presence_of :error_date
+end

data/examples/singleview/xpsi_trt_revoke.rb

@@ -0,0 +1,68 @@
+class XpsiTrtRevoke < ActiveRecord::Base
+  belongs_to :adjustment, :class_name => 'Adjustment', :foreign_key => :adjustment_id
+  validates_presence_of :task_id
+  validates_numericality_of :task_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :effective_date
+  validates_presence_of :cq_customer_query_id
+  validates_numericality_of :cq_customer_query_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :qp_progress_seqnr
+  validates_numericality_of :qp_progress_seqnr, :allow_nil => false, :only_integer => true
+  validates_length_of :revoke_function, :allow_nil => true, :maximum => 255
+  validates_numericality_of :letter_type, :allow_nil => true, :only_integer => true
+  validates_length_of :products_suspended, :allow_nil => true, :maximum => 255
+  validates_numericality_of :adjustment_id, :allow_nil => true, :only_integer => true
+  validates_length_of :general_1, :allow_nil => true, :maximum => 255
+  validates_length_of :general_2, :allow_nil => true, :maximum => 255
+  validates_length_of :general_3, :allow_nil => true, :maximum => 255
+  validates_length_of :general_4, :allow_nil => true, :maximum => 255
+  validates_length_of :general_5, :allow_nil => true, :maximum => 255
+  validates_length_of :general_6, :allow_nil => true, :maximum => 255
+  validates_length_of :general_7, :allow_nil => true, :maximum => 255
+  validates_length_of :general_8, :allow_nil => true, :maximum => 255
+  validates_length_of :general_9, :allow_nil => true, :maximum => 255
+  validates_length_of :general_10, :allow_nil => true, :maximum => 255
+  validates_presence_of :cq_last_modified
+  validates_numericality_of :cq_customer_node_id, :allow_nil => true, :only_integer => true
+  validates_presence_of :cq_atlanta_group_id
+  validates_numericality_of :cq_atlanta_group_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :cq_atlanta_operator_id
+  validates_numericality_of :cq_atlanta_operator_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :cq_query_status_code
+  validates_numericality_of :cq_query_status_code, :allow_nil => false, :only_integer => true
+  validates_numericality_of :cq_query_resolution_code, :allow_nil => true, :only_integer => true
+  validates_numericality_of :cq_query_work_force_id, :allow_nil => true, :only_integer => true
+  validates_numericality_of :cq_progress_seqnr, :allow_nil => true, :only_integer => true
+  validates_presence_of :cq_open_date
+  validates_presence_of :cq_response_due_date
+  validates_presence_of :cq_next_callback_date
+  validates_presence_of :cq_resolution_due_date
+  validates_length_of :cq_query_text, :allow_nil => true, :maximum => 2000
+  validates_length_of :cq_resolution_text, :allow_nil => true, :maximum => 2000
+  validates_length_of :cq_general_1, :allow_nil => true, :maximum => 255
+  validates_length_of :cq_general_2, :allow_nil => true, :maximum => 255
+  validates_length_of :cq_general_3, :allow_nil => true, :maximum => 255
+  validates_length_of :cq_general_4, :allow_nil => true, :maximum => 255
+  validates_length_of :cq_general_5, :allow_nil => true, :maximum => 255
+  validates_length_of :cq_general_6, :allow_nil => true, :maximum => 255
+  validates_length_of :cq_general_7, :allow_nil => true, :maximum => 255
+  validates_length_of :cq_general_8, :allow_nil => true, :maximum => 255
+  validates_length_of :cq_general_9, :allow_nil => true, :maximum => 255
+  validates_length_of :cq_general_10, :allow_nil => true, :maximum => 255
+  validates_presence_of :qp_last_modified
+  validates_numericality_of :qp_open_query_ind_code, :allow_nil => true, :only_integer => true
+  validates_presence_of :qp_work_force_id
+  validates_numericality_of :qp_work_force_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :qp_open_date
+  validates_presence_of :qp_response_due_date
+  validates_length_of :qp_progress_text, :allow_nil => true, :maximum => 2000
+  validates_length_of :qp_action_text, :allow_nil => true, :maximum => 2000
+  validates_presence_of :qp_atlanta_operator_id
+  validates_numericality_of :qp_atlanta_operator_id, :allow_nil => false, :only_integer => true
+  validates_length_of :qp_resolution_text, :allow_nil => true, :maximum => 2000
+  validates_length_of :cda_index1_value, :allow_nil => true, :maximum => 255
+  validates_length_of :cda_result1_value, :allow_nil => true, :maximum => 255
+  validates_length_of :cda_result2_value, :allow_nil => true, :maximum => 255
+  validates_length_of :cda_result3_value, :allow_nil => true, :maximum => 255
+  validates_length_of :cda_result4_value, :allow_nil => true, :maximum => 255
+  validates_length_of :cda_result5_value, :allow_nil => true, :maximum => 255
+end

data/examples/singleview/xpsi_trt_revoke_prod_inst.rb

@@ -0,0 +1,12 @@
+class XpsiTrtRevokeProdInst < ActiveRecord::Base
+  belongs_to :product_instance, :class_name => 'ProductInstance', :foreign_key => :product_instance_id
+  validates_presence_of :task_id
+  validates_numericality_of :task_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :effective_date
+  validates_presence_of :cq_customer_query_id
+  validates_numericality_of :cq_customer_query_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :qp_progress_seqnr
+  validates_numericality_of :qp_progress_seqnr, :allow_nil => false, :only_integer => true
+  validates_presence_of :product_instance_id
+  validates_numericality_of :product_instance_id, :allow_nil => false, :only_integer => true
+end

data/examples/singleview/xpsi_update_capture.rb

@@ -0,0 +1,35 @@
+class XpsiUpdateCapture < ActiveRecord::Base
+  belongs_to :xpsi_update_capture, :class_name => 'XpsiUpdateCapture', :foreign_key => :xpsi_update_capture_id
+  belongs_to :task_queue, :class_name => 'TaskQueue', :foreign_key => :task_queue_id
+  has_many :xpsi_update_captures, :class_name => 'XpsiUpdateCapture', :foreign_key => :xpsi_update_capture_id
+  has_many :task_queues, :through => xpsi_update_captures
+  validates_presence_of :xpsi_update_capture_id
+  validates_numericality_of :xpsi_update_capture_id, :allow_nil => false, :only_integer => true
+  validates_presence_of :capture_type_code
+  validates_numericality_of :capture_type_code, :allow_nil => false, :only_integer => true
+  validates_presence_of :last_modified
+  validates_presence_of :effective_extract_date
+  validates_presence_of :seqnr
+  validates_numericality_of :seqnr, :allow_nil => false, :only_integer => true
+  validates_numericality_of :task_queue_id, :allow_nil => true, :only_integer => true
+  validates_length_of :general_1, :allow_nil => true, :maximum => 255
+  validates_length_of :general_2, :allow_nil => true, :maximum => 255
+  validates_length_of :general_3, :allow_nil => true, :maximum => 255
+  validates_length_of :general_4, :allow_nil => true, :maximum => 255
+  validates_length_of :general_5, :allow_nil => true, :maximum => 255
+  validates_length_of :general_6, :allow_nil => true, :maximum => 255
+  validates_length_of :general_7, :allow_nil => true, :maximum => 255
+  validates_length_of :general_8, :allow_nil => true, :maximum => 255
+  validates_length_of :general_9, :allow_nil => true, :maximum => 255
+  validates_length_of :general_10, :allow_nil => true, :maximum => 255
+  validates_length_of :general_11, :allow_nil => true, :maximum => 255
+  validates_length_of :general_12, :allow_nil => true, :maximum => 255
+  validates_length_of :general_13, :allow_nil => true, :maximum => 255
+  validates_length_of :general_14, :allow_nil => true, :maximum => 255
+  validates_length_of :general_15, :allow_nil => true, :maximum => 255
+  validates_length_of :general_16, :allow_nil => true, :maximum => 255
+  validates_length_of :general_17, :allow_nil => true, :maximum => 255
+  validates_length_of :general_18, :allow_nil => true, :maximum => 255
+  validates_length_of :general_19, :allow_nil => true, :maximum => 255
+  validates_length_of :general_20, :allow_nil => true, :maximum => 255
+end

data/lib/activerecord/base.rb

@@ -0,0 +1,12 @@
+#TODO: Use :dependent for FK cascade?
+
+module ActiveRecord
+  class Base
+    class << self
+      public
+        def get_unique_index_columns
+            self.connection.indexes(self.table_name, "#{self.name} Indexes").select { |index| index.unique && index.columns.size == 1 }.map{ |index| index.columns.first }
+        end
+    end
+  end
+end

data/lib/hash_tricks.rb

@@ -0,0 +1,21 @@
+class Hash
+  # lets through the keys in the argument
+  # >> {:one => 1, :two => 2, :three => 3}.pass(:one)
+  # => {:one=>1}
+  def pass(*keys)
+    tmp = self.clone
+    keys = keys[0] if keys[0].is_a? Array
+    tmp.delete_if {|k,v| ! keys.include?(k) }
+    tmp
+  end
+
+	def merge_with_prefix(prefix, hash)
+		merged = clone
+		hash.each {|key, value| merged.merge!("#{prefix}_#{key}" => value)}
+		merged
+	end
+
+	def merge_with_prefix!(prefix, hash)
+		hash.each {|key, value| merge!("#{prefix}_#{key}" => value)}
+	end
+end

data/lib/magic_model_generator.rb

@@ -0,0 +1,8 @@
+module MagicModelsGenerator
+  def self.logger; @@logger ||= RAILS_DEFAULT_LOGGER rescue Logger.new(STDERR); end
+end
+
+Dir[File.join(File.dirname(__FILE__), '**/*.rb')].sort.each { |lib| require lib }
+
+
+#require 'magic_model_generator/validations'

data/lib/magic_model_generator/schema.rb

@@ -0,0 +1,205 @@
+require 'pp'
+module MagicModelsGenerator
+
+  class Schema
+    class << self
+      cattr_accessor :superklass,
+        :belongs_to_associations,
+        :has_some_associations,
+        :has_many_through_associations
+
+      @@models = nil
+
+      def logger
+        @@logger ||= MagicModelsGenerator.logger
+      end
+
+      def models
+        load_schema if @@models.nil?
+        @@models
+      end
+
+      def tables
+        load_schema if @@tables.nil?
+        @@tables
+      end
+
+      def table_names
+        load_schema if @@table_names.nil?
+        @@table_names
+      end
+
+      def fks_on_table(table_name)
+        load_schema if @@models.nil?
+        @@fks_by_table[table_name.to_s] || []
+      end
+
+      # active record only support 2 column link tables, otherwise use a model table, has_many and through
+      def is_link_table?(table_name)
+        load_schema if @@models.nil?
+        return @@link_tables[table_name] if ! @@link_tables[table_name].nil?
+        column_names = @conn.columns(table_name).map{|x| x.name }
+        @@link_tables[table_name] = ! column_names.include?("id") && column_names.length == 2 && column_names.select { |x| x =~ /_id$/ } == column_names
+        return @@link_tables[table_name]
+      end
+
+      def link_tables_for_class(klass)
+        load_schema if @@models.nil?
+      end
+
+      def load_schema
+        return if ! @@models.nil?
+        @@superklass ||= ActiveRecord::Base
+        raise "No database connection" if !(@conn = @@superklass.connection)
+
+        @@models = ModelHash.new
+        @@tables = Hash.new
+        @@fks_by_table = Hash.new
+        @@link_tables = Hash.new
+        @@belongs_to_associations = Hash.new
+        @@has_some_associations = Hash.new
+        @@has_many_through_associations = Hash.new
+
+        @@table_names = @conn.tables.sort
+
+        # Work out which tables are in the model and which aren't
+        @@table_names.each do |table_name|
+
+          # a model table then...
+          model_class_name = ActiveRecord::Base.class_name(table_name)
+          logger.debug "Got a model table: #{table_name} => class #{model_class_name}"
+
+          @@models[model_class_name] = table_name
+          @@tables[table_name] = model_class_name
+
+          # Process FKs?
+          if @conn.supports_fetch_foreign_keys?
+            table_names.each do |table_name|
+              logger.debug "Getting FKs for #{table_name}"
+              @@fks_by_table[table_name] = Array.new
+              @conn.foreign_key_constraints(table_name).each do |fk|
+                logger.debug "Got one: #{fk}"
+                @@fks_by_table[table_name].push(fk)
+              end # do each fk
+            end # each table
+          end
+
+          # Try to work out our link tables now...
+          #@@models.keys.sort.each{|klass| process_table(@@models[klass.to_s])}
+          #@@link_tables.keys.sort.each{|table_name| process_link_table(table_name) if @@link_tables[table_name]}
+        end
+        
+        models.each do |model_name, table_name|
+          @@belongs_to_associations[model_name] = []
+          @@has_some_associations[model_name] = []
+          @@has_many_through_associations[model_name] = []
+        end
+        
+        logger.debug "Loaded all models, now generating associations..."
+        
+        models.keys.sort.each do |model_name|
+          puts "Generating for #{model_name}..."
+          table_name = models[model_name]
+          generate_associations(model_name, table_name)
+        end
+
+      end
+
+      # Used by the generator to access the association code lines for the generated class
+      def associations(klass)
+        @@belongs_to_associations[klass.to_s] +
+        @@has_some_associations[klass.to_s] +
+        @@has_many_through_associations[klass.to_s]
+      end
+
+    protected
+      def generate_associations(model_name, table_name)
+      	belongs_to_klass = model_name.constantize rescue return
+        logger.debug "Processing model table #{table_name}"
+
+        processed_columns = Hash.new
+
+        # ok, so let's look at the foreign keys on the table...
+        
+        fks_on_table(table_name).each do |fk|
+          logger.debug "Found FK column by suffix _id [#{fk.foreign_key}]"
+          has_some_klass = Inflector.classify(fk.reference_table).constantize rescue next
+          processed_columns[fk.foreign_key] = { :has_some_klass => has_some_klass }
+          processed_columns[fk.foreign_key].merge! add_has_some_belongs_to(belongs_to_klass, fk.foreign_key, has_some_klass) rescue next
+        end
+
+        column_names = @conn.columns(table_name).map{ |x| x.name}
+        column_names.each do |column_name|
+          next if not column_name =~ /_id$/
+          logger.debug "Found FK column by suffix _id [#{column_name}]"
+          if processed_columns.key?(column_name)
+            logger.debug "Skipping, already processed"
+            next
+          end
+          has_some_klass = Inflector.classify(column_name.sub(/_id$/,"")).constantize rescue next
+          processed_columns[column_name] = { :has_some_klass => has_some_klass }
+          begin
+	          processed_columns[column_name].merge! add_has_some_belongs_to(belongs_to_klass, column_name, has_some_klass)
+	        rescue
+	        	puts $!
+	        	next
+	       	end
+        end
+
+        # is this a link table with attributes? (has_many through?)
+        return if processed_columns.keys.length < 2
+
+        processed_columns.keys.each do |key1|
+          processed_columns.keys.each do |key2|
+            next if key1 == key2
+						has_some_class = processed_columns[key1][:has_some_class].to_s
+						@@has_many_through_associations[has_some_class] <<
+						  "has_many :#{processed_columns[key2][:belongs_to_name].to_s.pluralize.to_sym}, :through => :#{processed_columns[key2][:has_some_name]}"
+						logger.debug @@has_many_through_associations[has_some_class].last
+          end
+        end
+      end
+
+      def add_has_some_belongs_to(belongs_to_klass, belongs_to_fk, has_some_klass)
+          logger.debug "Trying to add a #{belongs_to_klass} belongs_to #{has_some_klass}..."
+
+          # so this is a belongs_to & has_some style relationship...
+          # is it a has_many, or a has_one? Well, let's assume a has_one has a unique index on the column please... good db design, haha!
+          unique = belongs_to_klass.get_unique_index_columns.include?(belongs_to_fk)
+          belongs_to_name = belongs_to_fk.sub(/_id$/, '').to_sym
+          @@belongs_to_associations[belongs_to_klass.to_s] << "belongs_to :#{belongs_to_name}, :class_name => '#{has_some_klass}', :foreign_key => :#{belongs_to_fk.to_sym}"
+          logger.debug @@belongs_to_associations[belongs_to_klass.to_s].last
+
+          # work out if we need a prefix
+          has_some_name = ((unique ? belongs_to_klass.table_name.singularize : belongs_to_klass.table_name.pluralize) + (belongs_to_name.to_s == has_some_klass.table_name.singularize ? "" : "_as_"+belongs_to_name.to_s)).downcase.to_sym
+          method = unique ? :has_one : :has_many
+          @@has_some_associations[has_some_klass.to_s] << "#{method} :#{has_some_name}, :class_name => '#{belongs_to_klass.to_s}', :foreign_key => :#{belongs_to_fk.to_sym}"
+          logger.debug @@has_some_associations[has_some_klass.to_s].last
+
+          return { :method => method,
+          				 :belongs_to_name => belongs_to_name,
+          				 :has_some_name => has_some_name,
+          				 :has_some_class => has_some_klass  }
+			end
+
+    end
+  end
+
+  class ModelHash < Hash
+    def unenquire(class_id)
+      @enquired ||= {}
+      @enquired[class_id = class_id.to_s] = false
+    end
+
+    def enquired?(class_id)
+      @enquired ||= {}
+      @enquired[class_id.to_s]
+    end
+
+    def [](class_id)
+      enquired?(class_id = class_id.to_s)
+      @enquired[class_id] = true
+      super(class_id)
+    end
+  end
+end