Class: ActiveRecord::Generators::FastTreeGenerator

Inherits:
Base
  • Object
show all
Includes:
FastTree::Generators::OrmHelpers
Defined in:
lib/generators/active_record/fast_tree_generator.rb

Instance Method Summary collapse

Instance Method Details

#copy_fast_tree_migrationObject



12
13
14
15
16
17
18
# File 'lib/generators/active_record/fast_tree_generator.rb', line 12

def copy_fast_tree_migration
  if (behavior == :invoke && model_exists?) || (behavior == :revoke && migration_exists?(table_name))
    migration_template "migration_existing.rb", "db/migrate/add_fast_tree_to_#{table_name}.rb", migration_version: migration_version
  else
    migration_template "migration.rb", "db/migrate/fast_tree_create_#{table_name}.rb", migration_version: migration_version
  end
end

#generate_modelObject



20
21
22
# File 'lib/generators/active_record/fast_tree_generator.rb', line 20

def generate_model
  invoke "active_record:model", [name], migration: false unless model_exists? && behavior == :invoke
end

#migration_dataObject



24
25
26
27
28
29
30
# File 'lib/generators/active_record/fast_tree_generator.rb', line 24

def migration_data
<<RUBY
## Pointers
t.integer :l_ptr
t.integer :r_ptr
RUBY
end

#migration_versionObject



41
42
43
44
45
# File 'lib/generators/active_record/fast_tree_generator.rb', line 41

def migration_version
  if rails5?
    "[#{Rails::VERSION::MAJOR}.#{Rails::VERSION::MINOR}]"
  end
end

#postgresql?Boolean



36
37
38
39
# File 'lib/generators/active_record/fast_tree_generator.rb', line 36

def postgresql?
  config = ActiveRecord::Base.configurations[Rails.env]
  config && config['adapter'] == 'postgresql'
end

#rails5?Boolean



32
33
34
# File 'lib/generators/active_record/fast_tree_generator.rb', line 32

def rails5?
  Rails.version.start_with? '5'
end