gadget 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ece84105a691b5880948bbeac5dcb229ba654aa6
-  data.tar.gz: 4fc8cc56f871fdcb5c471a50a8a9a39295375b7d
+  metadata.gz: 974283a8788c4a40b2a75f8d8350f6c316836ff5
+  data.tar.gz: 66abb23005b88002c0fae59554f7695b6f452bf0
 SHA512:
-  metadata.gz: b18de42940505918186b4622d7e1cac13cd0e55b03b0cd687a634dc6267c2899ee6a3032593c998f3890565be16d94d7989daf6cdefbb19dc2ee6ff6ee3666e6
-  data.tar.gz: 92381cb642b34f08e841335f4c7bf4fcbcdc9607a811f7dc02385f12e61948bfb4a5b34ed8d6bb07e5273f28ef82d9b1fd50bcf914f61f8ca55e51f45f0dd6a9
+  metadata.gz: 95e4383e87f251b97eb861e406d6452c3a7ca684d42e384f30311ff6e85a53b2a1ce72bcbd673191fb6fabfb8dd4d6512cd97b1e3372e2c680649b5e163df165
+  data.tar.gz: 167996bc9821904dd2e036de143b02d522aae6d2415536a8508b8059e5bb3c2fdbb209c209bb37d228ba8aaf97faa65697dd3e53178004c1cd7b76cedd6a0c82

data/README.md

@@ -32,6 +32,14 @@ If `tablename` is given, returns the columns in only that table.
 Returns a list of all foreign keys in the schema reachable through `conn`.
 If `tablename` is given, returns the foreign keys in only that table.
 
+`#functions(conn)`
+
+Returns a list of all functions in the schema reachable through `conn`.
+
+`#triggers(conn)`
+
+Returns a list of all triggers in the schema reachable through `conn`.
+
 `#dependencies(conn)`
 
 Returns a structure representing the dependencies between tables in the schema reachable through `conn`.

data/bin/gadget

@@ -4,9 +4,23 @@ require 'gadget'
 
 require 'pg'
 
-dbname = ARGV.shift
+(command, dbname) = ARGV
 
-conn = PG::Connection.open(:dbname => dbname)
-tables = Gadget.tables_in_dependency_order(conn)
+conn = PG::Connection.open(:dbname => dbname, :host => 'localhost')
+result = nil
+case command
+when 'dependencies'
+  result = Gadget.dependencies(conn)
+when 'dependency-graph'
+  result = Gadget.dependency_graph(conn)
+when 'dependency-order'
+  result = Gadget.tables_in_dependency_order(conn)
+when 'functions'
+  result = Gadget.functions(conn)
+when 'tables'
+  result = Gadget.tables(conn)
+when 'triggers'
+  result = Gadget.triggers(conn)
+end
 conn.close
-p tables
+p result

data/lib/gadget.rb

@@ -25,9 +25,9 @@ module Gadget
   def self.columns(conn, tablename = nil)
     sql = <<-END_OF_SQL
 SELECT t.tablename, a.attname
-FROM pg_attribute a
-INNER JOIN pg_class c ON a.attrelid=c.oid
-INNER JOIN pg_tables t ON c.relname=t.tablename
+FROM pg_catalog.pg_attribute a
+INNER JOIN pg_catalog.pg_class c ON a.attrelid=c.oid
+INNER JOIN pg_catalog.pg_tables t ON c.relname=t.tablename
 WHERE a.attnum >= 0
     END_OF_SQL
     if tablename.nil?
@@ -44,11 +44,11 @@ WHERE a.attnum >= 0
   def self.foreign_keys(conn, tablename = nil)
     sql = <<-END_OF_SQL
 SELECT t1.tablename AS tablename, pg_constraint.conkey as cols, t2.tablename AS refname, pg_constraint.confkey as refcols
-FROM pg_constraint
-INNER JOIN pg_class c1 ON pg_constraint.conrelid=c1.oid
-INNER JOIN pg_tables t1 ON c1.relname=t1.tablename
-INNER JOIN pg_class c2 ON pg_constraint.confrelid=c2.oid
-INNER JOIN pg_tables t2 ON c2.relname=t2.tablename
+FROM pg_catalog.pg_constraint
+INNER JOIN pg_catalog.pg_class c1 ON pg_constraint.conrelid=c1.oid
+INNER JOIN pg_catalog.pg_tables t1 ON c1.relname=t1.tablename
+INNER JOIN pg_catalog.pg_class c2 ON pg_constraint.confrelid=c2.oid
+INNER JOIN pg_catalog.pg_tables t2 ON c2.relname=t2.tablename
 WHERE t1.schemaname='public'
 AND t2.schemaname='public'
 AND pg_constraint.contype='f'
@@ -105,4 +105,31 @@ AND pg_constraint.contype='f'
     puts "}"
   end
 
+  def self.functions(conn)
+    rs = conn.exec(<<-END_OF_SQL)
+SELECT p.oid, p.proname
+FROM pg_catalog.pg_proc p
+INNER JOIN pg_catalog.pg_namespace n ON p.pronamespace = n.oid
+WHERE n.nspname = 'public'
+    END_OF_SQL
+
+    tuples = rs.reduce({}) { | h, row | h[row['proname']] = { :oid => row['oid'] }; h }
+    rs.clear
+    tuples
+  end
+
+  def self.triggers(conn)
+    rs = conn.exec(<<-END_OF_SQL)
+SELECT tg.oid, tg.tgname, t.tablename
+FROM pg_catalog.pg_trigger tg
+INNER JOIN pg_catalog.pg_class c ON tg.tgrelid = c.oid
+INNER JOIN pg_catalog.pg_tables t ON c.relname = t.tablename
+WHERE tg.tgconstrrelid = 0
+    END_OF_SQL
+
+    tuples = rs.reduce({}) { | h, row | h[row['tgname']] = { :oid => row['oid'], :tablename => row['tablename'] }; h }
+    rs.clear
+    tuples
+  end
+
 end

data/lib/gadget/version.rb

@@ -1,3 +1,3 @@
 module Gadget
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: gadget
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Craig S. Cottingham
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-01-28 00:00:00.000000000 Z
+date: 2014-02-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pg
@@ -87,6 +87,14 @@ description: |
   Returns a list of all foreign keys in the schema reachable through `conn`.
   If `tablename` is given, returns the foreign keys in only that table.
 
+  `#functions(conn)`
+
+  Returns a list of all functions in the schema reachable through `conn`.
+
+  `#triggers(conn)`
+
+  Returns a list of all triggers in the schema reachable through `conn`.
+
   `#dependencies(conn)`
 
   Returns a structure representing the dependencies between tables in the schema reachable through `conn`.