0
-$:.unshift File.dirname(__FILE__)
0
- MODEL_DIR = File.join(RAILS_ROOT, "app/models")
0
- FIXTURE_DIR = File.join(RAILS_ROOT, "test/fixtures")
0
- PREFIX = "== Schema Information"
0
- # Simple quoting for the default column value
0
- when NilClass then "NULL"
0
- when TrueClass then "TRUE"
0
- when FalseClass then "FALSE"
0
- when Float, Fixnum, Bignum then value.to_s
0
- # BigDecimals need to be output in a non-normalized form and quoted.
0
- when BigDecimal then value.to_s('F')
0
- # Use the column information in an ActiveRecord class
0
- # to create a comment block containing a line for
0
- # each column. The line contains the column name,
0
- # the type (and length), and any optional attributes
0
- def self.get_schema_info(klass, header)
0
- info = "# #{header}\n#\n"
0
- info << "# Table name: #{klass.table_name}\n#\n"
0
- max_size = klass.column_names.collect{|name| name.size}.max + 1
0
- klass.columns.each do |col|
0
- attrs << "default(#{quote(col.default)})" if col.default
0
- attrs << "not null" unless col.null
0
- attrs << "primary key" if col.name == klass.primary_key
0
- col_type = col.type.to_s
0
- if col_type == "decimal"
0
- col_type << "(#{col.precision}, #{col.scale})"
0
- col_type << "(#{col.limit})" if col.limit
0
- info << sprintf("# %-#{max_size}.#{max_size}s:%-15.15s %s\n", col.name, col_type, attrs.join(", "))
0
- # Add a schema block to a file. If the file already contains
0
- # a schema info block (a comment starting
0
- # with "Schema as of ..."), remove it first.
0
- def self.annotate_one_file(file_name, info_block)
0
- if File.exist?(file_name)
0
- content = File.read(file_name)
0
- # Remove old schema info
0
- content.sub!(/^# #{PREFIX}.*?\n(#.*\n)*\n/, '')
0
- File.open(file_name, "w") { |f| f.puts info_block + content }
0
- # Given the name of an ActiveRecord class, create a schema
0
- # info block (basically a comment containing information
0
- # on the columns and their types) and put it at the front
0
- # of the model and fixture source files.
0
- def self.annotate(klass, header)
0
- info = get_schema_info(klass, header)
0
- model_file_name = File.join(MODEL_DIR, klass.name.underscore + ".rb")
0
- annotate_one_file(model_file_name, info)
0
- fixture_file_name = File.join(FIXTURE_DIR, klass.table_name + ".yml")
0
- annotate_one_file(fixture_file_name, info)
0
- # Return a list of the model files to annotate. If we have
0
- # command line arguments, they're assumed to be either
0
- # the underscore or CamelCase versions of model names.
0
- # Otherwise we take all the model files in the
0
- # app/models directory.
0
- def self.get_model_names
0
- Dir.chdir(MODEL_DIR) do
0
- models = Dir["**/*.rb"]
0
- # We're passed a name of things that might be
0
- # ActiveRecord models. If we can find the class, and
0
- # if its a subclass of ActiveRecord::Base,
0
- # then pas it to the associated block
0
- def self.do_annotations
0
- version = ActiveRecord::Migrator.current_version rescue 0
0
- header << "\n# Schema version: #{version}"
0
- self.get_model_names.each do |m|
0
- class_name = m.sub(/\.rb$/,'').camelize
0
- klass = class_name.split('::').inject(Object){ |klass,part| klass.const_get(part) }
0
- if klass < ActiveRecord::Base && !klass.abstract_class?
0
- annotated << class_name
0
- self.annotate(klass, header)
0
- puts "Unable to annotate #{class_name}: #{e.message}"
0
- puts "Annotated #{annotated.join(', ')}"
0
\ No newline at end of file
) is private.
