Class: ExclusiveArcGenerator

Inherits:
ActiveRecord::Generators::Base
  • Object
show all
Defined in:
lib/generators/exclusive_arc_generator.rb

Constant Summary collapse

Error = 
Class.new(StandardError)

Instance Method Summary collapse

Constructor Details

#initialize(*args) ⇒ ExclusiveArcGenerator

Returns a new instance of ExclusiveArcGenerator.

Raises:



16
17
18
19
 
# File 'lib/generators/exclusive_arc_generator.rb', line 16

def initialize(*args)
  raise Error, "must supply a Model, arc, and at least two belong_tos" if args[0].size <= 3
  super
end

Instance Method Details

#create_exclusive_arc_migration ⇒ Object 



21
22
23
24
25
26
 
# File 'lib/generators/exclusive_arc_generator.rb', line 21

def create_exclusive_arc_migration
  migration_template(
    "migration.rb.erb",
    "db/migrate/#{migration_file_name}"
  )
end

#inject_exclusive_arc_into_model ⇒ Object 



28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
 
# File 'lib/generators/exclusive_arc_generator.rb', line 28

def inject_exclusive_arc_into_model
  indents = "  " * (class_name.scan("::").count + 1)
  inject_into_class(
    model_file_path,
    class_name.demodulize,
    "      \#{indents}include ExclusiveArc::Model\n    RB\n  )\n  gsub_file(\n    model_file_path,\n    /has_exclusive_arc :\#{arc}(.*)$/,\n    \"\"\n  )\n  inject_into_file(\n    model_file_path,\n    <<~RB,\n      \#{indents}has_exclusive_arc \#{model_exclusive_arcs}\n    RB\n    after: \"include ExclusiveArc::Model\\n\"\n  )\nend\n"