gadget 0.2.1 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a9eefefd9392371161169e66ec7c6ecfc9e855dd
-  data.tar.gz: e1b1b8b2cd5be52ca74dfad670e833176f41ac16
+  metadata.gz: b7f6c737f72e0f2069beade77ada0e12a8fcc27c
+  data.tar.gz: 1cfcbfe2d33e0962ae52d45093f37722aee588eb
 SHA512:
-  metadata.gz: 86a0ddd76729246d2e3e81043d1e47af8fefd760ba358f2a11b716870bb68468dbd1d1765f516275fbd0746df99ebf9188ee9aff9c9b22f478ae2fecc4ca3a31
-  data.tar.gz: bf10d81863eded6472abbb34f16b5284a5a99acbcf19142278d1852e8083b700dd7e7a565ecba94dfe69620762003e8fd426580d932db2eff2d9981644d0432e
+  metadata.gz: eeea42b422fdde212fcc28ff3231828a8193a3da60e04d462e6a4d8d306e6ca2f28a82dc67f304c6b663289d735e2c359a250a3c12d41bd406d4be5738d5f4b9
+  data.tar.gz: 272b24bb7655d230860f7037012cc88760813576e40a636fe5841c1dbc820cd586f6e042bbf268664bfaafad2b4b3bc48ae4e6e0b7494916d83451593f746e33

data/bin/gadget

@@ -11,6 +11,8 @@ result = nil
 case command
 when 'columns'
   result = Gadget.columns(conn, cmdargs.shift)
+when 'constraints'
+  result = Gadget.columns(conn, cmdargs.shift)
 when 'dependencies'
   result = Gadget.dependencies(conn)
 when 'dependency-graph'

data/lib/gadget.rb

@@ -79,6 +79,51 @@ AND pg_constraint.contype = 'f'
     tuples
   end
 
+  def self.constraints(conn, tablename = nil)
+    sql = <<-END_OF_SQL
+SELECT pg_constraint.conname AS name,
+       pg_constraint.contype AS constrainttype,
+       t.tablename AS tablename
+FROM pg_catalog.pg_constraint
+INNER JOIN pg_catalog.pg_class c ON pg_constraint.conrelid = c.oid
+INNER JOIN pg_catalog.pg_tables t ON c.relname = t.tablename
+WHERE t.schemaname = 'public'
+    END_OF_SQL
+    if tablename.nil?
+      rs = conn.exec(sql)
+    else
+      sql += " AND t.tablename = $1"
+      rs = conn.exec_params(sql, [ tablename ])
+    end
+    tuples = rs.reduce({}) do | h, row |
+      name = row['tablename']
+      h[name] ||= { :refs => [] }
+      new_ref = {
+        :name => row['name'],
+        :kind => case row['constrainttype']
+                 when 'c'
+                   'check'
+                 when 'f'
+                   'foreign key'
+                 when 'p'
+                   'primary key'
+                 when 't'
+                   'trigger'
+                 when 'u'
+                   'unique'
+                 when 'x'
+                   'exclusion'
+                 else
+                   "*** unknown: '#{row['constrainttype']}'"
+                 end,
+      }
+      h[name][:refs] << new_ref
+      h
+    end
+    rs.clear
+    tuples
+  end
+
   def self.dependencies(conn)
     tables = self.tables(conn)
     foreign_keys = self.foreign_keys(conn)

data/lib/gadget/version.rb

@@ -1,3 +1,3 @@
 module Gadget
-  VERSION = "0.2.1"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: gadget
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.3.0
 platform: ruby
 authors:
 - Craig S. Cottingham
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-02-19 00:00:00.000000000 Z
+date: 2014-02-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pg