Skip to content

Commit

Permalink
pass PGPASSWORD via env directly, not via shell
Browse files Browse the repository at this point in the history
  • Loading branch information
evgeni committed Oct 15, 2024
1 parent 060e2b8 commit a59e042
Showing 1 changed file with 19 additions and 14 deletions.
33 changes: 19 additions & 14 deletions lib/foreman_maintain/concerns/base_database.rb
Original file line number Diff line number Diff line change
Expand Up @@ -55,22 +55,25 @@ def query_csv(sql, config = configuration)

def psql(query, config = configuration)
if ping(config)
execute(psql_command(config),
cmd, env = psql_command(config)
execute(cmd,
:stdin => query,
:hidden_patterns => [config['password']])
:env => env)
else
raise_service_error
end
end

def ping(config = configuration)
execute?(psql_command(config),
cmd, env = psql_command(config)
execute?(cmd,
:stdin => 'SELECT 1 as ping',
:hidden_patterns => [config['password']])
:env => env)
end

def dump_db(file, config = configuration)
execute!(dump_command(config) + " > #{file}", :hidden_patterns => [config['password']])
cmd, env = dump_command(config)
execute!(cmd + " > #{file}", :env => env)
end

def restore_dump(file, localdb, config = configuration)
Expand All @@ -80,11 +83,10 @@ def restore_dump(file, localdb, config = configuration)
else
# TODO: figure out how to completely ignore errors. Currently this
# sometimes exits with 1 even though errors are ignored by pg_restore
dump_cmd = base_command(config, 'pg_restore') +
cmd, env = base_command(config, 'pg_restore') +
' --no-privileges --clean --disable-triggers -n public ' \
"-d #{config['database']} #{file}"
execute!(dump_cmd, :hidden_patterns => [config['password']],
:valid_exit_statuses => [0, 1])
execute!(cmd, :valid_exit_statuses => [0, 1], :env => env)
end
end

Expand Down Expand Up @@ -125,8 +127,8 @@ def dropdb(config = configuration)
def db_version(config = configuration)
if ping(config)
# Note - t removes headers, -A removes alignment whitespace
server_version_cmd = psql_command(config) + ' -c "SHOW server_version" -t -A'
version_string = execute!(server_version_cmd, :hidden_patterns => [config['password']])
cmd, env = psql_command(config) + ' -c "SHOW server_version" -t -A'
version_string = execute!(cmd, :env => env)
version(version_string)
else
raise_service_error
Expand All @@ -146,17 +148,20 @@ def raise_psql_missing_error
private

def base_command(config, command = 'psql')
"PGPASSWORD='#{config[%(password)]}' "\
"#{command} -h #{config['host'] || 'localhost'} "\
env = { 'PGPASSWORD' => config['password'] }
cmd = "#{command} -h #{config['host'] || 'localhost'} "\
" -p #{config['port'] || '5432'} -U #{config['username']}"
return cmd, env
end

def psql_command(config)
base_command(config, 'psql') + " -d #{config['database']}"
cmd, env = base_command(config, 'psql')
return cmd + " -d #{config['database']}", env
end

def dump_command(config)
base_command(config, 'pg_dump') + " -Fc #{config['database']}"
cmd, env = base_command(config, 'pg_dump')
return cmd + " -Fc #{config['database']}", env
end

def raise_service_error
Expand Down

0 comments on commit a59e042

Please sign in to comment.