pocky 2.3.0 → 2.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: bf6db66f143fca23ba7dd902fc5af98d3d664972
-  data.tar.gz: a7acc183fe0d502d6a4a48b5866617a335c4b6ba
+  metadata.gz: ec4633fc47cd55e4bb54ea0378567ebaa792ed20
+  data.tar.gz: 0ff1ee4a69d65d586440ed2287958370ff2edc7f
 SHA512:
-  metadata.gz: 05e7cc845b387c26c5a65c2d3503d5080b807ee61328538cd267680546d7b32cbdbbebb14327560bdbad2a9927c4552ce5281b75f105e564922a89b6c0f105a8
-  data.tar.gz: d2aebfb4e4f70e8e5b90925825d082acf5ade98b85e8d5cf81c46d4b5d3733845bdf5c2364c5a44fe196e8cb6146acb400cfc652241d3746c727ec5e83331cee
+  metadata.gz: 17398615fbe7d3b8ac88ac5fd3a36844b417f998d7aea02a6d8f361f5af8402dac03284e091d1a8e2a2776f48a89b1e5f1c8fbef7fefe51681034ebfa82e1331
+  data.tar.gz: 94517f49a5813913ecc8d29dcdbf8f6e6a1847a741dcb622d4b7ae10c6d14565a8da62ff8b16072de3f6018be13a57c7794cd23ed8c5f920bfc1dd6b82d52604

data/lib/pocky/packwerk.rb

@@ -8,19 +8,32 @@ module Pocky
   class InvalidRootPathError < StandardError
   end
 
+  class RubyFileSize
+    def self.compute(directory)
+      # Sum up all ruby source except for specs
+      package_size = Dir[File.join(directory, '**', '*.rb').to_s].reduce(0) do |size, filename|
+        size += File.size(filename) unless filename.match(/spec\.rb$/)
+        size
+      end
+
+      # to kB
+      (package_size / 1024).ceil
+    end
+  end
+
   class Packwerk
     DEPENDENCIES_FILENAME = 'package.yml'
     DEPRECATED_REFERENCES_FILENAME = 'deprecated_references.yml'
     MAX_EDGE_WIDTH = 5
 
-    def self.generate(params)
+    def self.generate(params = {})
       new(**params).generate
     end
 
     private_class_method :new
     def initialize(
       package_path: nil,
-      default_package: 'app',
+      default_package: 'root',
       filename: 'packwerk.png',
       dpi: 100,
       package_color: '#5CC8FF',
@@ -45,8 +58,9 @@ module Pocky
         color: package_color,
         height: 1.0,
         style: 'filled, rounded',
-        shape: 'box',
+        shape: 'box'
       }
+
       @dependency_edge_options = {
         color: dependency_edge
       }
@@ -65,6 +79,27 @@ module Pocky
 
     private
 
+    def node_overrides(file_size)
+      if file_size < 10
+        { fontsize: 26 }
+      elsif file_size < 100
+        { fontsize: 26 * 4, margin: 0.2 }
+      elsif file_size < 1000
+        { fontsize: 26 * 8, margin: 0.4 }
+      elsif file_size < 10_000
+        { fontsize: 26 * 16, margin: 0.8 }
+      else
+        { fontsize: 26 * 32, margin: 1.0 }
+      end
+    end
+
+    def draw_node(package)
+      package_name = package_name_for_dependency(package)
+      path = package == '.' ? @root_path : @root_path.join(package)
+      file_size = RubyFileSize.compute(path.to_s)
+      @graph.add_nodes(package_name, **@node_options.merge(node_overrides(file_size)))
+    end
+
     def build_directed_graph
       @graph = GraphViz.new(:G, type: :digraph, dpi: @dpi)
       draw_dependencies
@@ -74,14 +109,13 @@ module Pocky
 
     def draw_dependencies
       @package_dependencies.each do |package, file|
-        @nodes[package] ||= @graph.add_nodes(package, **@node_options)
+        @nodes[package] ||= draw_node(package)
         file.each do |provider|
-          provider_package = package_name_for_dependency(provider)
-          @nodes[provider_package] ||= @graph.add_nodes(provider_package, **@node_options)
+          @nodes[provider] ||= draw_node(provider)
 
           @graph.add_edges(
             @nodes[package],
-            @nodes[provider_package],
+            @nodes[provider],
             **@dependency_edge_options
           )
         end
@@ -90,14 +124,13 @@ module Pocky
 
     def draw_deprecated_references
       @deprecated_references.each do |package, references|
-        @nodes[package] ||= @graph.add_nodes(package, **@node_options)
+        @nodes[package] ||= draw_node(package)
         references.each do |provider, invocations|
-          provider_package = package_name_for_dependency(provider)
-          @nodes[provider_package] ||= @graph.add_nodes(provider_package, **@node_options)
+          @nodes[provider] ||= draw_node(provider)
 
           @graph.add_edges(
             @nodes[package],
-            @nodes[provider_package],
+            @nodes[provider],
             **@deprecated_references_edge_options.merge(
               penwidth: edge_width(invocations.length),
             ),

data/lib/pocky/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Pocky
-  VERSION = "2.3.0"
+  VERSION = "2.4.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pocky
 version: !ruby/object:Gem::Version
-  version: 2.3.0
+  version: 2.4.0
 platform: ruby
 authors:
 - Quan Nguyen
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2021-01-26 00:00:00.000000000 Z
+date: 2021-02-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-graphviz