Class: Jpstation::Generators::ModelsGenerator

Inherits:
Rails::Generators::Base
  • Object
show all
Defined in:
lib/generators/jpstation/models_generator.rb

Instance Method Summary collapse

Instance Method Details

#generate_model ⇒ Object 



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
 
# File 'lib/generators/jpstation/models_generator.rb', line 8

def generate_model
  if File.exists?(File.join(destination_root, 'app/models/prefecture.rb'))
    inject_into_class "app/models/prefecture.rb", Prefecture, <<CODE
  has_many :railway_stations
  def railway_lines
    RailwayLine.joins(:railway_stations).merge(railway_stations.reorder(nil)).distinct.order(:sort, :id)
  end
CODE
  else
    Rails::Generators.invoke 'jpstation:model', ['prefecture']
  end

  %w{ railway_company railway_line railway_station }.each do |model_name|
    Rails::Generators.invoke 'jpstation:model', [model_name]
  end
end