erde 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ceaf6b839dab986b6926445fe3881ab0eb969549
-  data.tar.gz: 1b2b7ca815801782137c01d55ad4752fbb26a4da
+  metadata.gz: 7c80277ab7aff1fa8b988fc0414cc96f4c2619f8
+  data.tar.gz: d19998855f85ade4eadb28ee1c04e3317d282e7a
 SHA512:
-  metadata.gz: 648f7365ef3d165e51f5ad20f835aa6f8d2003e3a600077b10f72c4de45b14dc18531884abeb65c6133e451214ba997866c2152d7faabf6ca40ccfdb13785097
-  data.tar.gz: b6a5b10fb2191b9ceea5be765fbbbf751e15de56a7af315a137102feeca4683b159aa00cd20f12c47cecf3360116d8388482cfbc8e9b55943fe70bdbd8717d92
+  metadata.gz: 37a82e7f74a09ed9d9c13f7bce3c72bf5b7effc4a338898c70e885c55ccd0f358c435bc12fb68064d2574d8316ee3de19503720a4f177f674aacb1b7bad2c7f5
+  data.tar.gz: 419c5ad4e0e179a14ef52a791c650d733c34c5a0e27ce83a5838fe75e164cec4f79b0f5b1a560e3bbabbf289a1175c2b7707f7bca3f940394b9d3fc365e87f53

data/.gitignore

@@ -7,3 +7,4 @@
 /pkg/
 /spec/reports/
 /tmp/
+/docs/

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.3.0
+- Add basic style
+- Support database as schema source.
+
 ## 0.2.0
 - Support text schemas.
 - Output the diagram as PNG.

data/README.md

@@ -9,6 +9,10 @@ Make sure you have [Graphviz](http://graphviz.org/) installed and available in y
 bin/erde file docs/schema.txt docs/schema.png
 ~~~
 
+~~~txt
+bin/erde database postgres://user:password@localhost/your_database docs/schema.png
+~~~
+
 ## Text Schema Format
 ~~~txt
 [identities]

data/erde.gemspec

@@ -21,4 +21,6 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "bundler", "~> 1.12"
   spec.add_development_dependency "rake", "~> 10.0"
   spec.add_development_dependency "minitest", "~> 5.0"
+
+  spec.add_dependency "sequel", "~> 4.0"
 end

data/lib/erde/cli.rb

@@ -1,6 +1,8 @@
 require "erde"
 require "pathname"
 require "open3"
+require "sequel"
+require "erb"
 
 class Erde::CLI
   def self.start(*args)
@@ -13,6 +15,12 @@ class Erde::CLI
       hash_schema = text_transformer.to_hash
     end
 
+    if command == "database"
+      url = args.shift.strip
+      database_transformer = Erde::DatabaseTransformer.new(url)
+      hash_schema = database_transformer.to_hash
+    end
+
     hash_transformer = Erde::HashTransformer.new(hash_schema)
     dot_schema = hash_transformer.to_dot
 
@@ -28,12 +36,15 @@ class Erde::HashTransformer
   end
 
   def to_dot
+    template = File.read(File.expand_path("../template.dot.erb", __FILE__))
+
     schema_string = ""
     schema_string << "digraph tables {"
-    schema_string << "node [shape=Mrecord rankdir=LR];"
+    schema_string << "node [shape=plaintext rankdir=LR];"
 
     @hash.each_pair do |table_name, table_schema|
-      schema_string << "#{table_name} [label=\"{#{table_name}|#{table_schema['columns'].map {|c| "<#{c}>#{c}" }.join("|")}}\"];"
+      renderer = ERB.new(template)
+      schema_string << renderer.result(binding)
 
       table_schema['relations'].each_pair do |column, target|
         schema_string << "#{table_name}:#{column} -> #{target['table']}:#{target['column']};"
@@ -46,6 +57,35 @@ class Erde::HashTransformer
   end
 end
 
+class Erde::DatabaseTransformer
+  def initialize(url)
+    @url = url
+  end
+
+  def to_hash
+    generated_hash = {}
+
+    Sequel.connect(@url) do |db|
+      db.tables.each do |table|
+        generated_hash[table] = {}
+        generated_hash[table]['columns'] = []
+        generated_hash[table]['relations'] = {}
+
+        generated_hash[table]['columns'] = db.schema(table).map(&:first)
+
+        db.foreign_key_list(table).each do |foreign_key|
+          generated_hash[table]['relations'][foreign_key[:columns].first] = {
+            'table' => foreign_key[:table],
+            'column' => foreign_key[:key].first
+          }
+        end
+      end
+    end
+
+    generated_hash
+  end
+end
+
 class Erde::TextTransformer
   def initialize(text)
     @lines = text.lines

data/lib/erde/template.dot.erb

@@ -0,0 +1,24 @@
+  <%= table_name %> [label=<
+
+    <TABLE BORDER="1" CELLBORDER="0" CELLSPACING="0" CELLPADDING="2" STYLE="ROUNDED" ROWS="*">
+      <TR>
+        <TD BORDER="1" SIDES="B">
+          <FONT FACE="Helvetica Neue Bold" POINT-SIZE="14"><%= table_name %></FONT>
+        </TD>
+      </TR>
+      <TR>
+        <TD>
+          <TABLE BORDER="0" CELLBORDER="0" CELLSPACING="2" CELLPADDING="0">
+          <% for @column in table_schema['columns'] %>
+            <TR>
+<TD ALIGN="LEFT" PORT="<%= @column %>">
+<FONT FACE="Helvetica Neue" POINT-SIZE="12"><%= @column %></FONT>
+</TD>
+            </TR>
+          <% end %>
+          </TABLE>
+        </TD>
+      </TR>
+    </TABLE>
+
+  >];

data/lib/erde/version.rb

@@ -1,3 +1,3 @@
 module Erde
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: erde
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - David Strauß
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-06-15 00:00:00.000000000 Z
+date: 2016-06-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -52,6 +52,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '5.0'
+- !ruby/object:Gem::Dependency
+  name: sequel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.0'
 description: Generate good looking Entity-Relationship-Diagrams from text files or
   a PostgreSQL database.
 email:
@@ -74,6 +88,7 @@ files:
 - erde.gemspec
 - lib/erde.rb
 - lib/erde/cli.rb
+- lib/erde/template.dot.erb
 - lib/erde/version.rb
 homepage: https://github.com/edgycircle/erde
 licenses: