yml2erd 0.9.7 → 0.9.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 30861d60e3b7ec898d77d7651b115801e4a2411e
-  data.tar.gz: c89d8ff74ca28c0ffd9b8289e5c403acf90867f9
+  metadata.gz: f2559f6e066eeec7dec8fa5b69954f31ac61e6fb
+  data.tar.gz: ea165d8d635601f4b932f6b44c93a6aa58f12263
 SHA512:
-  metadata.gz: 0d63ce75a0dc3b53a456ef574c238ac2da6f9369931783e6f9fbaea0126425994f8f92e3212298d486429bf7df100e32316eca3d1310c3336aea5ca03a749493
-  data.tar.gz: 0629c73061313f60e203098664648d8f7117bd5f07c59dba1894d4f5fe978352397a2f2ab2beb0b8203eea7edbe6b34ee9c3e0458d0f5ec5cb6b09d21b543c2b
+  metadata.gz: b42de50a6a3e97539534a550963fae46fdc8278ce2199fc807269b6c94ba9e8aa3933b0bb1b545f5f722c7f561058315b23f42eb23c7c5393d9f11ecad6c3487
+  data.tar.gz: cb1a0a5595feaaae8b6edbb2e19bea23972f4b4729cc4d56fcd82e5d3ef5ff0eef04585adf18b52633bb46c630d5c78b1ce9122f3a8b7d097ad55c3151344bef

data/lib/yml2erd/diagram.rb

@@ -20,7 +20,8 @@ module Yml2erd
           end
 
           index = schema_structure.index(table_name)
-          label = build_label(table_name, columns, index)
+          comment = schema_structure.comment(table_name)
+          label = build_label(table_name, columns, index, comment)
           g.add_nodes(table_name, shape: "record", label: label, style: "rounded")
         end
 
@@ -40,12 +41,15 @@ module Yml2erd
         end
       end
 
-      def build_label(table_name, columns, index)
-        if index
-          "<{<FONT POINT-SIZE='15'>#{table_name}</FONT> | #{columns} | indexed: #{index.to_s}}>"
-        else
-          "<{<FONT POINT-SIZE='15'>#{table_name}</FONT> | #{columns}}>"
-        end
+      def build_label(table_name, columns, index, comment)
+        partial ||= "| indexed: #{index.to_s}" if index
+        partial += "| #{comment}" if comment
+        base_label(table_name, columns, partial)
+      end
+
+      # TODO: Contemplates method name
+      def base_label(table_name, columns, partial = nil)
+        base = "<{<FONT POINT-SIZE='15'>#{table_name}</FONT> | #{columns}#{partial}}>"
       end
 
       def build_column(name, type)

data/lib/yml2erd/schema_structure.rb

@@ -53,5 +53,11 @@ module Yml2erd
     def shared_columns
       structure_yml['shared_columns']
     end
+
+    # returns a string like below
+    # => "This is users table"
+    def comment(table_name)
+      structure_yml[table_name]['comment']
+    end
   end
 end

data/lib/yml2erd/version.rb

@@ -1,3 +1,3 @@
 module Yml2erd
-  VERSION = "0.9.7"
+  VERSION = "0.9.8"
 end

data/sample.yml

@@ -10,6 +10,7 @@ companies:
     has_many:
       - users
 users:
+  comment: This is users table.
   columns:
     - crypted_password: string
     - email: string

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: yml2erd
 version: !ruby/object:Gem::Version
-  version: 0.9.7
+  version: 0.9.8
 platform: ruby
 authors:
 - asmsuechan
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-12-28 00:00:00.000000000 Z
+date: 2017-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-graphviz