Transform JSON payload to a graph file

json_to_graph.rb

#!/usr/bin/ruby

# To run in terminal:
#
#       `ruby json_to_graph.rb <json_input_file_path> <output_path>`
#
#
# Make sure you install the following gems:
#
#      `gem install ruby-graphviz`
#
# You'll also need to install graphviz to use the gem - see instructions here: https://graphviz.org/download/
# e.g. for installation in Mac
#
#      `brew install graphviz`
#

require 'ruby-graphviz'
require 'json'

json_input_file_path = ARGV[0]
output_path = ARGV[1]

file = File.read(json_input_file_path)
data = JSON.parse(file)

###############################################################################
# CHANGE ITEMS START - Only change values within this area                    #
###############################################################################
# Change the following to the array you want to convert to a graph
data_array = data['tags']

def get_id(single_entry)
  single_entry['tagId']
end
def get_parent_ids(single_entry)
  single_entry['parentTagIds']
end
def get_name(single_entry)
  single_entry['allNamesByLocale']['en'] # If name is the id and is used above, can return as empty string
end
###############################################################################
# CHANGE ITEMS END                                                            #
###############################################################################

data_graph = GraphViz.new( :G, :type => :digraph )
node_map = {}

# Create nodes
data_array.each do |single_entry|
  id = get_id(single_entry)
  name = get_name(single_entry)
  parent_ids = get_parent_ids(single_entry)

  puts "Creating node with id: #{id} and name: #{name}"
  node_map[id] = if name == '' || name.nil?
                  data_graph.add_nodes("id:#{id}")
                else
                  data_graph.add_nodes("id: #{id}, name: #{name}")
                end
end

# Create edges
data_array.each do |single_entry|
  id = get_id(single_entry)
  parent_ids = get_parent_ids(single_entry)

  unless parent_ids.nil?
    puts "Creating edge with id: #{id}"
    parent_ids.each do |parent_id|
      if node_map.key?(parent_id)
        data_graph.add_edges(node_map[parent_id], node_map[id])
      end
    end
  end
end


# If your graph ends up cropped or too pixelated, you can try to output it as svg
data_graph.output(path: output_path, png: 'graph.png' )
puts "Graph created at #{output_path}/graph.png"

# data_graph.output(path: output_path, svg: 'graph.svg' )
# puts "Graph created at #{output_path}/graph.svg"