erd_map 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ed4d072a4862423c842760702dbb6d1e51d749e3dcb53968a53f34a7780962f6
-  data.tar.gz: 5da740ea6a8613083bcb5dcd4274087482189f9cbc0ec8e56e8c4032c4c9e3be
+  metadata.gz: ff2d016ce1d70f303ba81a480185f13058d35cb1f0cd98c224637eaadaac481b
+  data.tar.gz: a48d58de84ed40da26d8327aac5fb2241c51ddf1333eca9365398ce72fb79628
 SHA512:
-  metadata.gz: c887c21acfb720eb0ae8a3a2b5d6aa1caeca816ccf68781384e29a901a2e8dafb30bb6bde4524b3c46319da7ad10e53eb384535c5f01e8733d304cdadd9e86bc
-  data.tar.gz: 0ae4645e25e70ad6b709c37ab2525285bf421ede860b9eed22e0ef307fafcfe29461da9fbbe81430030877a269396b7fca2eea43aed0a39aed8f85760daad8d0
+  metadata.gz: 0aad4e151277b66e1996d5953552de91f9798e2e566f5b01869cad123c6c7e83cd69f35960cc6741ecda719d77da118b172e63e1b2efc168512663e23153752c
+  data.tar.gz: bdf8341667f56d3ed3ed20a9c3204a485b078a81ee2288942c83e9b6821e22e4e4db989afdbaf677de8529c19e3a0a895930bc3e3c7aac8cd7a4c48486687275

data/lib/erd_map/graph.rb

@@ -99,6 +99,15 @@ module ErdMap
       end
     end
 
+    def nodes_with_i18n_labels
+      return @nodes_with_i18n_labels if @nodes_with_i18n_labels
+      @nodes_with_i18n_labels = {}
+      whole_models.each do |model|
+        @nodes_with_i18n_labels[model.name] = model.model_name.human(default: "")
+      end
+      @nodes_with_i18n_labels
+    end
+
     def association_columns
       return @association_columns if @association_columns
 

data/lib/erd_map/graph_renderer.rb

@@ -42,11 +42,11 @@ module ErdMap
         layoutsByChunkData: graph.layouts_by_chunk.to_json,
         chunkedNodesData: graph.chunked_nodes.to_json,
         nodeWithCommunityIndexData: graph.node_with_community_index.to_json,
-        searchBox: plot.button_set[:search_box],
-        selectingNodeLabel: plot.button_set[:selecting_node_label],
-        zoomModeToggle: plot.button_set[:zoom_mode_toggle],
-        tapModeToggle: plot.button_set[:tap_mode_toggle],
-        displayTitleModeToggle: plot.button_set[:display_title_mode_toggle],
+        searchBox: plot.search_box,
+        selectingNodeLabel: plot.selecting_node_label,
+        zoomModeToggle: plot.zoom_mode_toggle,
+        tapModeToggle: plot.tap_mode_toggle,
+        displayTitleModeToggle: plot.display_title_mode_toggle,
         nodeLabels: {
           titleModelLabel: title_model_label,
           foreignModelLabel: foreign_model_label,
@@ -81,12 +81,14 @@ module ErdMap
 
       columns_label = []
       title_label = []
+      i18n_label = []
       rect_heights = []
       graph.node_names.map do |node_name|
         title_text = format_text([node_name], title: true)
         columns_text = [*title_text.scan("\n"), "\n", format_text(graph.association_columns[node_name])].join
         columns_label << columns_text
         title_label << [title_text, "\n", *columns_text.scan("\n")].join
+        i18n_label << [node_name, graph.nodes_with_i18n_labels[node_name]].compact_blank.join("\n")
 
         padding = 36
         line_count = columns_text.scan("\n").size + 1
@@ -102,6 +104,7 @@ module ErdMap
           radius: graph.node_radius,
           original_radius: graph.node_radius,
           rect_height: rect_heights,
+          i18n_label: i18n_label,
           title_label: title_label,
           columns_label: columns_label,
           fill_color: graph.node_colors,
@@ -181,7 +184,7 @@ module ErdMap
       bokeh_models.LabelSet.new(
         x: "x",
         y: "y",
-        text: "index",
+        text: { field: "i18n_label" },
         source: graph_renderer.node_renderer.data_source,
         text_font_size: "12pt",
         text_color: { field: "text_color" },

data/lib/erd_map/map_builder.rb

@@ -33,19 +33,7 @@ module ErdMap
       bokeh_models.Column.new(
         children: [
           bokeh_models.Row.new(
-            children: [
-              plot.button_set[:left_spacer],
-              plot.button_set[:selecting_node_label],
-              plot.button_set[:search_box],
-              plot.button_set[:zoom_mode_toggle],
-              plot.button_set[:tap_mode_toggle],
-              plot.button_set[:display_title_mode_toggle],
-              plot.button_set[:re_layout_button],
-              plot.button_set[:zoom_in_button],
-              plot.button_set[:zoom_out_button],
-              plot.button_set[:re_compute_button],
-              plot.button_set[:right_spacer],
-            ],
+            children: plot.header,
             sizing_mode: "stretch_width",
           ),
           plot.plot,

data/lib/erd_map/plot.rb

@@ -26,8 +26,8 @@ module ErdMap
       end
     end
 
-    def button_set
-      @button_set ||= ButtonSet.new
+    def header
+      @header.to_a
     end
 
     private
@@ -36,6 +36,7 @@ module ErdMap
 
     def initialize(graph)
       @graph = graph
+      @header = Header.new
       register_callback
     end
 
@@ -70,14 +71,34 @@ module ErdMap
       JS
     end
 
-    class ButtonSet
+    class Header
       extend Forwardable
-      def_delegators :@button_set, :[]
+      ACCESSIBLE_PARTS = [:search_box, :selecting_node_label, :zoom_mode_toggle, :tap_mode_toggle, :display_title_mode_toggle]
+
+      def to_a
+        [
+          :left_spacer,
+          :selecting_node_label,
+          :search_box,
+          :zoom_mode_toggle,
+          :tap_mode_toggle,
+          :display_title_mode_toggle,
+          :re_layout_button,
+          :zoom_in_button,
+          :zoom_out_button,
+          :re_compute_button,
+          :right_spacer,
+        ].map { |name| @header[name] }
+      end
 
       private
 
+      ACCESSIBLE_PARTS.each do |name|
+        define_method(name) { @header[name] }
+      end
+
       def initialize
-        @button_set = {
+        @header = {
           left_spacer: left_spacer,
           selecting_node_label: selecting_node_label,
           search_box: search_box,
@@ -189,5 +210,9 @@ module ErdMap
         )
       end
     end
+
+    Header::ACCESSIBLE_PARTS.each do |name|
+      public define_method(name) { @header.__send__(name) }
+    end
   end
 end

data/lib/erd_map/version.rb

@@ -1,3 +1,3 @@
 module ErdMap
-  VERSION = "0.1.4"
+  VERSION = "0.1.5"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: erd_map
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - makicamel