Puppet catalog to TGF graph format converter

catalog2tgf.rb

#!/usr/bin/ruby

require 'json'
require 'tgf'

catalog_file_name = 'catalog.json'
graph_file_name = 'foo.tgf'

catalog = JSON.parse(File.read(catalog_file_name))

nodes = catalog['data']['resources']
edges = catalog['data']['edges']

edges.map! do |entry|
	from = entry['source'].gsub(' ', '_')
	to = entry['target'].gsub(' ', '_')
	TGF::Edge.new("#{from} #{to}")
end

nodes.map! do |entry|
	label = "#{entry['type']}[#{entry['title']}]"
	id = label.gsub(' ', '_')
	TGF::Node.new("#{id} #{label}")
end

graph_file = File.open(graph_file_name, 'w')
TGF.write(graph_file, nodes, edges)

Gemfile

source 'https://rubygems.org'

gem 'tgf', '~> 1.1.0'