pg_query 0.2.7 → 0.2.8

checksums.yaml

@@ -1,7 +1,15 @@
 ---
-SHA1:
-  metadata.gz: 436c39c88606860708a62a007c94e3681946a179
-  data.tar.gz: 6ca769bb0fd9e9533b31f0c8729c4c91cfa2035d
+!binary "U0hBMQ==":
+  metadata.gz: !binary |-
+    YzUxOGE4ZjRmZjRhYTk3MGI5NzBkMzI5OWFiMWIzNWViZGIzZWJiOA==
+  data.tar.gz: !binary |-
+    ZmY3NmRhMDczMDYwODZlM2FmNzMwNmY1Nzc5YjM2YWU4MTQ1YTUxZg==
 SHA512:
-  metadata.gz: 38956a8035166579310f9e69ca40c71ce4dd648e298fb4f1bd2a90335e0e4b22e7d9d1c0461e2da7b86b3473f5139f2d64bd2f1451381d6bf33d4dcc1204caaf
-  data.tar.gz: 2ae516ad313a283e67d359dabcae0652d9e91cb0b86ac3d2ab7421d9305b501e6ccfc5895b74c6a5897106098f66c20419e1baada0845796600cbd32f85af6b3
+  metadata.gz: !binary |-
+    MWJjMzhhZWIyMGViNjY3ODJmZmRjYzYyNWZkMThhNjhhOWJhYTcxZWM3Yzlj
+    YTAwZGUyYWZmZTJhMThhMmE1Y2MzNjEzMTA2NDhhODI1MjU1Y2JlMTk5MzY3
+    ZGNhZDVkZGJiODlkZWUyZjI0MzcwZDBlYWM0OGU3ZDRlZmYzZGM=
+  data.tar.gz: !binary |-
+    MTdiN2M0NmFhMDBlZTA5Y2I5Yzc4ZWRiNGEzZGVlNjlmNDA0OGE3NmFmMzYx
+    MDY1MThjYWRjMDBiYWRlYTg2NWEyMWFhMTU5ZDMyNTBhOWY3NDBkZWJlZTkz
+    YzYxN2FlOWIxZDExMzdkNGZhYjE5MjNlMzIzODVmOGRjMDhjZDg=

data/lib/pg_query/parse.rb

@@ -27,4 +27,96 @@ class PgQuery
     @parsetree = parsetree
     @warnings = warnings
   end
+  
+  def tables
+    load_tables_and_aliases! if @tables.nil?
+    @tables
+  end
+  
+  def aliases
+    load_tables_and_aliases! if @aliases.nil?
+    @aliases
+  end
+  
+protected
+  def load_tables_and_aliases!
+    @tables = []
+    @aliases = {}
+  
+    statements = @parsetree.dup
+    from_clause_items = []
+    where_clause_items = []
+  
+    loop do
+      if statement = statements.shift
+        case statement.keys[0]
+        when "SELECT"
+          if statement["SELECT"]["op"] == 0
+            (statement["SELECT"]["fromClause"] || []).each do |item|
+              if item["RANGESUBSELECT"]
+                statements << item["RANGESUBSELECT"]["subquery"]
+              else
+                from_clause_items << item
+              end
+            end
+          elsif statement["SELECT"]["op"] == 1
+            statements << statement["SELECT"]["larg"] if statement["SELECT"]["larg"]
+            statements << statement["SELECT"]["rarg"] if statement["SELECT"]["rarg"]
+          end
+        when "INSERT INTO", "UPDATE", "DELETE FROM", "VACUUM", "COPY", "ALTER TABLE"
+          from_clause_items << statement.values[0]["relation"]
+        when "EXPLAIN"
+          statements << statement["EXPLAIN"]["query"]
+        when "CREATE TABLE AS"
+          from_clause_items << statement["CREATE TABLE AS"]["into"]["INTOCLAUSE"]["rel"] rescue nil
+        when "LOCK"
+          from_clause_items += statement["LOCK"]["relations"]
+        when "DROP"
+          object_type = statement["DROP"]["removeType"]
+          @tables += statement["DROP"]["objects"].map {|r| r.join('.') } if object_type == 26 # Table
+        end
+    
+        where_clause_items << statement.values[0]["whereClause"] if !statement.empty? && statement.values[0]["whereClause"]
+      end
+    
+      # Find subselects in WHERE clause
+      if next_item = where_clause_items.shift
+        case next_item.keys[0]
+        when /^AEXPR/, 'ANY'
+          ["lexpr", "rexpr"].each do |side|
+            next unless elem = next_item.values[0][side]
+            if elem.is_a?(Array)
+              where_clause_items += elem
+            else
+              where_clause_items << elem
+            end
+          end
+        when 'SUBLINK'
+          statements << next_item["SUBLINK"]["subselect"]
+        end
+      end
+    
+      break if where_clause_items.empty? && statements.empty?
+    end
+  
+    loop do
+      break unless next_item = from_clause_items.shift
+    
+      case next_item.keys[0]
+      when "JOINEXPR"
+        ["larg", "rarg"].each do |side|
+          from_clause_items << next_item["JOINEXPR"][side]
+        end
+      when "ROW"
+        from_clause_items += next_item["ROW"]["args"]
+      when "RANGEVAR"
+        rangevar = next_item["RANGEVAR"]
+        table = [rangevar["schemaname"], rangevar["relname"]].compact.join('.')
+        @tables << table
+        @aliases[rangevar["alias"]["ALIAS"]["aliasname"]] = table if rangevar["alias"]
+      end
+    end
+  
+    @tables.uniq!
+  end
 end

data/lib/pg_query/version.rb

@@ -1,3 +1,3 @@
 class PgQuery
-  VERSION = '0.2.7'
+  VERSION = '0.2.8'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pg_query
 version: !ruby/object:Gem::Version
-  version: 0.2.7
+  version: 0.2.8
 platform: ruby
 authors:
 - Lukas Fittl
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-05-31 00:00:00.000000000 Z
+date: 2014-06-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake-compiler
@@ -78,12 +78,12 @@ require_paths:
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
-  - - '>='
+  - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
-  - - '>='
+  - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []