jamesmoriarty/call-graph

require 'erb'

require 'call_graph/printers/dot/line'

module CallGraph
  module Printers
    class Dot
      attr_reader :config

      def initialize(config)
        @config = config
      end

      def to_s
        ERB.new(TEMPLATE, trim_mode: '>').result(binding)
      end

      private

      def lines
        IO.read(config.path(:tmp))
          .split("\n")
          .uniq
          .map { |line| line.split(',') }
          .map { |c, r, id| Line.new(c, r, label: id) }
      end

      TEMPLATE = <<~EOF.chomp
        digraph call_graph {
        <% lines.each do |line| %>
          "<%= line.caller %>" -> "<%= line.receiver %>" [<% line.attributes.each do |(name, value)| %><%= name %>="<%= value %>"<% end %>];
        <% end %>}
      EOF
    end
  end
end