Class: Diagrammer::Generator

Inherits:
Object
  • Object
show all
Defined in:
lib/diagrammer/generator.rb

Instance Method Summary collapse

Constructor Details

#initialize(output:, models: nil, introspector: ModelIntrospector.new(models: models)) ⇒ Generator

Returns a new instance of Generator.



5
6
7
8
 
# File 'lib/diagrammer/generator.rb', line 5

def initialize(output:, models: nil, introspector: ModelIntrospector.new(models: models))
  @output = output
  @introspector = introspector
end

Instance Method Details

#call ⇒ Object 



10
11
12
13
14
15
16
17
18
19
20
 
# File 'lib/diagrammer/generator.rb', line 10

def call
  diagram = @introspector.call
  html = HtmlRenderer.new(
    diagram: diagram,
    title: 'Database Diagram',
    notice: notice_for(diagram)
  ).call

  File.write(@output, html)
  @output.to_s
end