posgra 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: c2b2e3a65346e2b279888aaca122a42605633f1f
-  data.tar.gz: 940922471165b5cf650096699a2c18a5ed453f90
+  metadata.gz: 518a434dc5bec68e17e348b1786b914ae4ff0e35
+  data.tar.gz: 27dfbd44d5c896b6684f0442fd35889e1b2a3dc0
 SHA512:
-  metadata.gz: 5e10303ef482af44c9cac302a59c824c7de8754260fffb8fa3fea1c9bff20c5dc6fd5c7ad2de2b4c59548eb27091bf3de76018e2f7549aebc1f4e41e70c48c3d
-  data.tar.gz: 4490756c4187af2c9b2358036aa2543c7096c44d80a3df53396cc35c58a171a7024e2f042bd3570e1ead2728243f6e107b2aabb4d267cba41a8adf5301dcf5e1
+  metadata.gz: 7cf81139a91b2069a1ebd77f11443ecbfb5d4c4ce8d132dc04820986d9da0cd3a633bfca0ed66755bed67a40f1c62d6e0b00f4e1b985a706b44e5b861f8bfdf1
+  data.tar.gz: c0fbd3b0684e78b7e4b67adeb5c0a0394681c320f9ccdff8330e50a6f4e83eca53a298da926e6ecfb160db3e3418b3ada4012156d7fa37d7e5d4b9a0a9e85c91

data/lib/posgra/driver.rb

@@ -5,6 +5,10 @@ class Posgra::Driver
   DEFAULT_ACL_PRIVS = ENV['POSGRA_DEFAULT_ACL_PRIVS'] || 'arwdDxt'
   DEFAULT_ACL = "{%s=#{DEFAULT_ACL_PRIVS}/%s}"
 
+  DEFAULT_ACL_BY_KIND = {
+    'S' => '{%s=rwU/%s}'
+  }
+
   PRIVILEGE_TYPES = {
     'a' => 'INSERT',
     'r' => 'SELECT',
@@ -283,7 +287,8 @@ class Posgra::Driver
         pg_class.relname,
         pg_namespace.nspname,
         pg_class.relacl,
-        pg_user.usename
+        pg_user.usename,
+        pg_class.relkind
       FROM
         pg_class
         INNER JOIN pg_namespace ON pg_class.relnamespace = pg_namespace.oid
@@ -298,11 +303,12 @@ class Posgra::Driver
       nspname = row.fetch('nspname')
       relacl = row.fetch('relacl')
       usename = row.fetch('usename')
+      relkind = row.fetch('relkind')
 
       next unless matched?(relname, @options[:include_object], @options[:exclude_object])
       next unless matched?(nspname, @options[:include_schema], @options[:exclude_schema])
 
-      parse_aclitems(relacl, usename).each do |aclitem|
+      parse_aclitems(relacl, usename, relkind).each do |aclitem|
         role = aclitem.fetch('grantee')
         privs = aclitem.fetch('privileges')
         next unless matched?(role, @options[:include_role], @options[:exclude_role])
@@ -317,8 +323,9 @@ class Posgra::Driver
 
   private
 
-  def parse_aclitems(aclitems, owner)
-    aclitems ||= DEFAULT_ACL % [owner, owner]
+  def parse_aclitems(aclitems, owner, relkind)
+    aclitems_fmt = DEFAULT_ACL_BY_KIND.fetch(relkind, DEFAULT_ACL)
+    aclitems ||= aclitems_fmt % [owner, owner]
     aclitems = aclitems[1..-2].split(',')
 
     aclitems.map do |aclitem|

data/lib/posgra/version.rb

@@ -1,3 +1,3 @@
 module Posgra
-  VERSION = "0.1.4"
+  VERSION = "0.1.5"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: posgra
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - winebarrel