Contents

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

module RailsBestPractices
  module Prepares
    # Remember the model attributes.
    class SchemaPrepare < Core::Check
      interesting_nodes :command, :command_call
      interesting_files SCHEMA_FILE

      # all attribute types
      ATTRIBUTE_TYPES = %w(integer float boolean string text date time datetime binary)

      def initialize
        @model_attributes = Prepares.model_attributes
      end

      def start_command(node)
        if "create_table" == node.message.to_s
          @last_klazz = node.arguments.all[0].to_s.classify
        end
      end

      # check command_call node to remember the model attributes.
      def start_command_call(node)
        if ATTRIBUTE_TYPES.include? node.message.to_s
          attribute_name = node.arguments.all[0].to_s
          @model_attributes.add_attribute(@last_klazz, attribute_name, node.message.to_s)
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
rails_best_practices-1.4.0	lib/rails_best_practices/prepares/schema_prepare.rb