Contents

# encoding: utf-8
require 'rails_best_practices/core/check'

module RailsBestPractices
  module Prepares
    # Remember the model associations.
    class ModelPrepare < Core::Check

      def interesting_nodes
        [:class, :call]
      end

      def interesting_files
        MODEL_FILES
      end

      def initialize
        @associations = {}
      end

      # check class node to remember the last class name.
      def start_class(class_node)
        @last_klazz = class_node.class_name.to_s
      end

      # assign @associations to Prepare.model_associations.
      def end_class(class_node)
        Prepares.model_associations = @associations
      end

      # check call node to remember all assoications.
      #
      # the remembered association names (@associations) are like
      #     {
      #       :Project=>{
      #         "categories"=>:has_and_belongs_to_many,
      #         "project_manager"=>:has_one,
      #         "portfolio"=>:belongs_to,
      #         "milestones=>:has_many"
      #       }
      #     }
      def start_call(node)
        remember_association(node) if association_methods.include? node.message
      end

      # remember associations, with class to association names.
      def remember_association(association_node)
        association_meta = association_node.message
        association_name = association_node.arguments[1].to_s
        @associations[@last_klazz] ||= {}
        @associations[@last_klazz][association_name] = association_meta
      end

      # default rails association methods.
      def association_methods
        [:belongs_to, :has_one, :has_many, :has_and_belongs_to_many]
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
rails_best_practices-0.7.1	lib/rails_best_practices/prepares/model_prepare.rb
rails_best_practices-0.7.0	lib/rails_best_practices/prepares/model_prepare.rb
rails_best_practices-0.6.7	lib/rails_best_practices/prepares/model_prepare.rb