Notes to Dot converter

note_to_dot.rb

note = File.read(ARGV[0])

gid = 1
prev_indent = 0
stack = [0]

labels = ["Root"]
edges = []

note.lines.map { _1.delete!("\n") }.map do |line|
	indent = /^ */.match(line)[0].size

	current_gid = gid
	gid += 1

	labels[current_gid] = line.gsub(/^ *- ?/, '')

	if indent > prev_indent # New subtree
		stack.push(current_gid - 1)
	elsif indent < prev_indent # Back to parent
		((prev_indent - indent) / 2).times { stack.pop }
	end	

	edges.push([stack.last, current_gid])

	prev_indent = indent
end

# p labels
# p edges

puts("digraph {")

labels.each_with_index do |label, i|
	puts("\t#{i}[label=\"#{label}\"]")
end

puts

edges.each do |(from, to)|
	puts("\t#{from} -> #{to}")
end

puts("}")