Class: SequenceDiagramGenerator

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

Class Method Summary collapse

Class Method Details

.generate(runtime) ⇒ Object 



7
8
9
10
11
12
13
14
15
 
# File 'lib/sequence_diagram_generator.rb', line 7

def self.generate runtime
	index = 0
	total = runtime.sequences.length
	runtime.sequences.values.each do |sequence|
		index += 1
		puts "Generating sequence diagram [#{index}/#{total}]: #{sequence.name}.png"
		generate_sequence sequence
	end
end

.generate_sequence(sequence) ⇒ Object 



17
18
19
20
21
22
23
24
25
26
27
28
29
30
 
# File 'lib/sequence_diagram_generator.rb', line 17

def self.generate_sequence sequence
	file = "docs/diagrams/#{sequence.name}.png"
	text = make_code sequence
	response = Net::HTTP.post_form(URI.parse('http://www.websequencediagrams.com/index.php'), 'style' => 'omegapple', 'message' => text)
	if response.body =~ /img: "(.+)"/
		url = "http://www.websequencediagrams.com/#{ $1 }"
		url.match(/(png)=(.+)$/)
		open url do |content|
			File.open file, 'wb' do |file|
				file.puts content.read
			end
		end
	end
end

.make_code(sequence) ⇒ Object 



32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
 
# File 'lib/sequence_diagram_generator.rb', line 32

def self.make_code sequence
	str = "title #{sequence.name} #{sequence.doc}\n"
	sequence.steps.each do |step|
		direction = step.direction
		head = ''
		case direction.direction
		when :left
			head = direction.server.name + ' -> ' + direction.client.name
		when :right
			head = direction.client.name + ' -> ' + direction.server.name
		end
		strParams = step.message.members.map { |m| m.name } .join ', '
		str += head + ': ' + step.message.name + '(' + strParams + ') ' + step.doc + "\n"
	end
	str
end