avm-eac_postgresql_base0 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 21a34c355a48e71fa5a12b91b27238a98844495a5742b30ac79b277f7e56fec4
-  data.tar.gz: 9e5b87bbca77cb621b8cb207c1ed220321b6bbfddddf548025461af03fa25aec
+  metadata.gz: 5a75bf3a56aaa8d9940062215c2bb24c9b0b5e9c6953abaa1e1b6fa10a2a7102
+  data.tar.gz: acf8c3cd099779f780fa841dbf9b655cbb37149b6b3d5c72479de4d4b47b96d5
 SHA512:
-  metadata.gz: f8edfd9211f650e24d3e93503ceda6f6cce0637abd4111289a3bdf7c2c253d35417b95a497a26a4d3721db8ed711df5ad46ff751f6324b2bf29e34130965bcbc
-  data.tar.gz: e1ab05d6d11d6adbe238e902b787c0fb0994527d951ba2d44f79a686e4138291a9e2f1241caeb6933dc9edfee3c40b41e65ff1322c1ff26e79a77142c32408d6
+  metadata.gz: 1c2fd992697a8da42c92724649bb26cea3f376becee96a7fae81881b41459c06ff7cd5c749082632d8c64020e56ef1b71701a28e8350dd1e3a70ce9d70069227
+  data.tar.gz: 59bfb018e82084fb4f532285d2abbb14ab1f2be0286d8b87c05f5246ac11542bfe60b1a2c8bdbe8489b7b8c67e9fba40ddfc782ff45cfa933e551d94dfd24f23

data/lib/avm/eac_postgresql_base0/instance/assert.rb

@@ -0,0 +1,77 @@
+# frozen_string_literal: true
+
+require 'eac_ruby_utils/core_ext'
+
+module Avm
+  module EacPostgresqlBase0
+    class Instance
+      class Assert
+        common_constructor :instance
+        delegate :name, :password, :user, to: :instance
+
+        def perform
+          create_user unless user_exist?
+          change_password unless password_ok?
+          create_database unless database_exist?
+          change_owner unless user_owner?
+        end
+
+        def change_owner
+          root_execute("ALTER DaTABASE \"#{name}\" OWNER TO \"#{user}\"")
+        end
+
+        def change_password
+          root_execute("ALTER USER \"#{user}\" WITH PASSWORD '#{password}'")
+        end
+
+        def create_user
+          root_execute("CREATE USER \"#{user}\" WITH PASSWORD '#{password}'")
+        end
+
+        def current_owner
+          root_query(<<~SQL
+            SELECT pg_catalog.pg_get_userbyid(datdba)
+            FROM pg_catalog.pg_database
+            WHERE datname = '#{name}'
+          SQL
+                    )
+        end
+
+        def database_exist?
+          root_boolean_query("FROM pg_database WHERE datname='#{name}'")
+        end
+
+        def password_ok?
+          instance.psql_command_command('SELECT 1', false)
+                  .execute!(exit_outputs: { 512 => 'login_failed' }).strip == '1'
+        end
+
+        def user_exist?
+          root_boolean_query("FROM pg_roles WHERE rolname='#{user}'")
+        end
+
+        def user_owner?
+          user == current_owner
+        end
+
+        def create_database
+          root_execute("CREATE DATABASE \"#{name}\" WITH OWNER \"#{user}\"")
+        end
+
+        private
+
+        def root_boolean_query(sql_after_projection)
+          root_query("SELECT 1 #{sql_after_projection}") == '1'
+        end
+
+        def root_execute(sql)
+          instance.root_psql_command(sql).execute!
+        end
+
+        def root_query(sql)
+          root_execute(sql).strip
+        end
+      end
+    end
+  end
+end

data/lib/avm/eac_postgresql_base0/instance.rb

@@ -1,13 +1,21 @@
 # frozen_string_literal: true
 
 require 'avm/eac_postgresql_base0/instance/data_unit'
+require 'eac_ruby_utils/core_ext'
 
 module Avm
   module EacPostgresqlBase0
     class Instance
-      def initialize(env, connection_params)
-        @env = env
-        @connection_params = connection_params.with_indifferent_access
+      require_sub __FILE__
+
+      MAINTENANCE_DATABASE = 'postgres'
+
+      common_constructor :env, :connection_params do
+        self.connection_params = connection_params.with_indifferent_access
+      end
+
+      def assert
+        ::Avm::EacPostgresqlBase0::Instance::Assert.new(self).perform
       end
 
       def data_unit
@@ -23,20 +31,23 @@ module Avm
         dump_command.append(['@ESC_|', 'gzip', '-9', '-c'])
       end
 
-      def psql_command
-        env.command("@ESC_PGPASSWORD=#{password}", 'psql', *common_command_args)
+      def psql_command(database = true)
+        env.command("@ESC_PGPASSWORD=#{password}", 'psql', *common_command_args(database))
       end
 
-      def psql_command_command(sql)
-        psql_command.append(['--quiet', '--tuples-only', '--command', sql])
+      def psql_command_command(sql, database = true)
+        psql_command(database).append(['--quiet', '--tuples-only', '--command', sql])
       end
 
-      private
-
-      attr_reader :env, :connection_params
+      def root_psql_command(sql = nil)
+        args = ['sudo', '-u', 'postgres', 'psql']
+        args += ['--quiet', '--tuples-only', '--command', sql] if sql.present?
+        env.command(*args)
+      end
 
-      def common_command_args
-        ['--host', host, '--username', user, '--port', port, name]
+      def common_command_args(database = true)
+        ['--host', host, '--username', user, '--port', port,
+         (database ? name : MAINTENANCE_DATABASE)]
       end
 
       def host

data/lib/avm/eac_postgresql_base0/version.rb

@@ -2,6 +2,6 @@
 
 module Avm
   module EacPostgresqlBase0
-    VERSION = '0.1.0'
+    VERSION = '0.2.0'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: avm-eac_postgresql_base0
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Put here the authors
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-12-11 00:00:00.000000000 Z
+date: 2022-04-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: avm
@@ -66,6 +66,7 @@ extra_rdoc_files: []
 files:
 - lib/avm/eac_postgresql_base0.rb
 - lib/avm/eac_postgresql_base0/instance.rb
+- lib/avm/eac_postgresql_base0/instance/assert.rb
 - lib/avm/eac_postgresql_base0/instance/data_unit.rb
 - lib/avm/eac_postgresql_base0/instance_with.rb
 - lib/avm/eac_postgresql_base0/version.rb