Contents

# frozen_string_literal: true

module CobraCommander
  # Module for pretty printing dependency trees
  module Output
    def self.print(tree, format)
      output = format == "list" ? Output::FlatList.new(tree) : Output::Tree.new(tree)
      puts output.to_s
    end

    # Flattens a tree and prints unique items
    class FlatList
      def initialize(tree)
        @tree = tree
      end

      def to_s
        @tree.flatten.map(&:name)
      end
    end

    # Prints the tree in a nice tree form
    class Tree
      attr_accessor :tree

      SPACE  = "    "
      BAR    = "│   "
      TEE    = "├── "
      CORNER = "└── "

      def initialize(tree)
        @tree = tree
      end

      def to_s
        StringIO.new.tap do |io|
          io.puts @tree.name
          list_dependencies(io, @tree)
        end.string
      end

    private

      def list_dependencies(io, deps, outdents = [])
        deps.dependencies.each do |dep|
          decide_on_line(io, deps, dep, outdents)
        end
        nil
      end

      def decide_on_line(io, parent, dep, outdents)
        if parent.dependencies.last != dep
          add_tee(io, outdents, dep)
        else
          add_corner(io, outdents, dep)
        end
      end

      def add_tee(io, outdents, dep)
        io.puts line(outdents, TEE, dep.name)
        list_dependencies(io, dep, (outdents + [BAR]))
      end

      def add_corner(io, outdents, dep)
        io.puts line(outdents, CORNER, dep.name)
        list_dependencies(io, dep, (outdents + [SPACE]))
      end

      def line(outdents, sym, name)
        (outdents + [sym] + [name]).join
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
cobra_commander-0.6.1	lib/cobra_commander/output.rb
cobra_commander-0.6.0	lib/cobra_commander/output.rb
cobra_commander-0.5.1	lib/cobra_commander/output.rb
cobra_commander-0.5.0	lib/cobra_commander/output.rb