umbrellio-sequel-plugins 0.14.0.192 → 0.15.0.198

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 62df42e1d4a608c7a2f6f4d3b33956b0e8d7c962de1f4ba970b5ddb3e858c9df
-  data.tar.gz: cc6dadf3aaea66e8af9e694996144daee03ca8af13893d8e930fb8b28f20b254
+  metadata.gz: b159522b1cb01dea723411ba97da921347c22c98999f23da63df0c1a0a205154
+  data.tar.gz: 7bd434016e9d2e5aa93a271e7540f88cee8b2943575739b79a8491a653bcb0c4
 SHA512:
-  metadata.gz: '0961d5fce4dd8f7d0aaba3152577c22e097bff92b66cf7b7a716c39693272b020fcfc0fb91f4bca530617559e7cfa4c2ed97cbd7c7807084c0eae458910a6a77'
-  data.tar.gz: e31c2fc3285cb90b2780d2718f6e854fd8bee2c7d54a9e988df766348ecd3838cb92243c1de210f9b5fc4ff2ef4d3acc53edc5271d8cd66cc65cf5d91547c681
+  metadata.gz: e37d16419fd9356d677ee852013947061f3749b3e117fcd7c897b62f2b8dcbd65d0dc4517bde5c87bb781d34780556f899199eaa6dd618d4a2ebc93e1ff56155
+  data.tar.gz: 610c27188c8fe14536e143b181455d8a558acd48c7e3b99b6699c61ef26601ea0d264493d50aa75ffc7e38991c225f4e2181eac91c0300b71f56357f4852acb5

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    umbrellio-sequel-plugins (0.14.0)
+    umbrellio-sequel-plugins (0.15.0)
       sequel
       symbiont-ruby
 

data/lib/umbrellio_sequel_plugins/rails_db_command.rb

@@ -9,45 +9,86 @@ class Rails::Command::DbconsoleCommand < Rails::Command::Base
   def perform
     require "rake"
     Rake.with_application(&:load_rakefile) # Needed to initialize Rails.application
-    Rails::DBConsole.start(options)
+    start!
   end
-end
-
-class Rails::DBConsole
-  DBConfig = Struct.new(:configuration_hash, :adapter, :database)
 
   private
 
-  def db_config
-    @db_config ||= DBConfig.new(configuration_hash, adapter, database)
+  # See ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.dbconsole
+  def start!
+    ENV["PGUSER"] = pg_config[:username] if pg_config[:username]
+    ENV["PGHOST"] = pg_config[:host] if pg_config[:host]
+    ENV["PGPORT"] = pg_config[:port].to_s if pg_config[:port]
+
+    if pg_config[:password] && options[:include_password]
+      ENV["PGPASSWORD"] = pg_config[:password].to_s
+    end
+
+    ENV["PGSSLMODE"] = pg_config[:sslmode].to_s if pg_config[:sslmode]
+    ENV["PGSSLCERT"] = pg_config[:sslcert].to_s if pg_config[:sslcert]
+    ENV["PGSSLKEY"] = pg_config[:sslkey].to_s if pg_config[:sslkey]
+    ENV["PGSSLROOTCERT"] = pg_config[:sslrootcert].to_s if pg_config[:sslrootcert]
+
+    if pg_config[:variables]
+      ENV["PGOPTIONS"] = pg_config[:variables].filter_map do |name, value|
+        "-c #{name}=#{value.to_s.gsub(/[ \\]/, '\\\\\0')}" unless value.in?([":default", :default])
+      end.join(" ")
+    end
+
+    find_cmd_and_exec("psql", database)
   end
 
-  def configuration_hash
-    return @configuration_hash if defined?(@configuration_hash)
+  def pg_config
+    @pg_config ||= begin
+      rails_db_config = Rails.application.config.database_configuration
 
-    rails_db_config = Rails.application.config.database_configuration
+      sequel_configuration = SequelRails::Configuration.new
+      SequelRails.configuration = sequel_configuration.merge!(raw: rails_db_config)
 
-    sequel_configuration = SequelRails::Configuration.new
-    SequelRails.configuration = sequel_configuration.merge!(raw: rails_db_config)
+      storage = SequelRails::Storage.adapter_for(Rails.env)
+      config = storage.config.with_indifferent_access
 
-    storage = SequelRails::Storage.adapter_for(Rails.env)
-    config = storage.config.with_indifferent_access
+      if @options[:server]
+        server_config = config.fetch(:servers).fetch(@options[:server])
+        config.merge!(server_config)
+      end
 
-    if @options[:server]
-      server_config = config.fetch(:servers).fetch(@options[:server])
-      config.merge!(server_config)
+      config
     end
-
-    @configuration_hash = config
   end
 
-  def adapter
-    mapping = SequelRails::DbConfig::ADAPTER_MAPPING.invert
-    value = configuration_hash.fetch(:adapter)
-    mapping[value] || value
+  # See ActiveRecord::ConnectionAdapters::AbstractAdapter.find_cmd_and_exec
+  def find_cmd_and_exec(commands, *args) # rubocop:disable Metrics/MethodLength
+    commands = Array(commands)
+
+    dirs_on_path = ENV["PATH"].to_s.split(File::PATH_SEPARATOR)
+    unless (ext = RbConfig::CONFIG["EXEEXT"]).empty?
+      commands = commands.map { |cmd| "#{cmd}#{ext}" }
+    end
+
+    full_path_command = nil
+    found = commands.detect do |cmd|
+      dirs_on_path.detect do |path|
+        full_path_command = File.join(path, cmd)
+        begin
+          stat = File.stat(full_path_command)
+        rescue SystemCallError
+        else
+          stat.file? && stat.executable?
+        end
+      end
+    end
+
+    if found
+      exec(*[full_path_command, *args].compact)
+    else
+      abort(
+        "Couldn't find database client: #{commands.join(', ')}. Check your $PATH and try again.",
+      )
+    end
   end
 
   def database
-    @options[:database] || configuration_hash.fetch(:database)
+    options[:database] || pg_config.fetch(:database)
   end
 end

data/umbrellio-sequel-plugins.gemspec

@@ -4,7 +4,7 @@ lib = File.expand_path("lib", __dir__)
 $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 
 Gem::Specification.new do |spec|
-  gem_version = "0.14.0"
+  gem_version = "0.15.0"
 
   if ENV.fetch("PUBLISH_JOB", nil)
     release_version = "#{gem_version}.#{ENV.fetch("GITHUB_RUN_NUMBER")}"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: umbrellio-sequel-plugins
 version: !ruby/object:Gem::Version
-  version: 0.14.0.192
+  version: 0.15.0.198
 platform: ruby
 authors:
 - Team Umbrellio
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-12-15 00:00:00.000000000 Z
+date: 2024-05-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sequel