Module: MigrationComments::ActiveRecord::ConnectionAdapters::AbstractAdapter::SchemaCreation

Defined in:
lib/migration_comments/active_record/connection_adapters/abstract_adapter/schema_creation.rb

Instance Method Summary collapse

Instance Method Details

#add_column_options!(sql, options) ⇒ Object 



9
10
11
12
13
14
15
16
 
# File 'lib/migration_comments/active_record/connection_adapters/abstract_adapter/schema_creation.rb', line 9

def add_column_options!(sql, options)
  sql = super(sql, options)
  if options.keys.include?(:comment) && !@conn.independent_comments?
    comment_definition = MigrationComments::ActiveRecord::ConnectionAdapters::CommentDefinition.new(nil, nil, options[:comment])
    sql << @conn.comment_sql(comment_definition)
  end
  sql
end

#column_options(o) ⇒ Object 



3
4
5
6
7
 
# File 'lib/migration_comments/active_record/connection_adapters/abstract_adapter/schema_creation.rb', line 3

def column_options(o)
  column_options = super(o)
  column_options[:comment] = o.comment.comment_text if o.comment.respond_to?(:comment_text)
  column_options
end

#visit_ColumnDefinition(o) ⇒ Object 



30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
 
# File 'lib/migration_comments/active_record/connection_adapters/abstract_adapter/schema_creation.rb', line 30

def visit_ColumnDefinition(o)
  if @conn.inline_comments? && o.type.to_sym == :primary_key && o.comment
    if ::ActiveRecord::VERSION::MAJOR >= 5
      comment_sql = super(o)
      comment_sql << @conn.comment_sql(o.comment)
    else
      sql_type = type_to_sql(o.type.to_sym, o.limit, o.precision, o.scale)
      column_sql = "#{quote_column_name(o.name)} #{sql_type}"
      add_column_options!(column_sql, column_options(o))
      column_sql
    end
  else
    super(o)
  end
end

#visit_TableDefinition(o) ⇒ Object 



18
19
20
21
22
23
24
25
26
27
28
 
# File 'lib/migration_comments/active_record/connection_adapters/abstract_adapter/schema_creation.rb', line 18

def visit_TableDefinition(o)
  if @conn.inline_comments?
    create_sql = "CREATE#{' TEMPORARY' if o.temporary} TABLE "
    create_sql << "#{quote_table_name(o.name)}#{@conn.comment_sql(o.comment)} ("
    create_sql << o.columns.map { |c| accept c }.join(', ')
    create_sql << ") #{o.options}"
    create_sql
  else
    super(o)
  end
end