File: model_generator.rb

package info (click to toggle)
gitlab 17.6.5-19
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 629,368 kB
  • sloc: ruby: 1,915,304; javascript: 557,307; sql: 60,639; xml: 6,509; sh: 4,567; makefile: 1,239; python: 406
file content (29 lines) | stat: -rw-r--r-- 928 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
# frozen_string_literal: true

require 'rails/generators'
require 'rails/generators/active_record/model/model_generator'

module Model
  class ModelGenerator < ActiveRecord::Generators::ModelGenerator
    source_root File.expand_path('../../../generator_templates/active_record/migration/', __dir__)

    def create_migration_file
      return if skip_migration_creation?

      if options[:indexes] == false
        attributes.each { |a| a.attr_options.delete(:index) if a.reference? && !a.has_index? }
      end

      migration_template "create_table_migration.rb", File.join(db_migrate_path, "create_#{table_name}.rb")
    end

    def create_database_dictionary
      template "database_dictionary.yml", Rails.root.join("db", "docs", "#{table_name}.yml").to_s
    end

    # Override to find templates from superclass as well
    def source_paths
      super + [self.class.superclass.default_source_root]
    end
  end
end