Create the pgpass file if it does not exist.

wayneeseguin · Apr 24, 2012 · c7d27b3 · c7d27b3
1 parent e1a5dfa
commit c7d27b3
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/lib/psql-cm/database.rb b/lib/psql-cm/database.rb
@@ -36,19 +36,20 @@ def close!
     end
 
     def pgpass # Ensure a pgpass entry exists for this connection.
+      pgpass_file = File.join(ENV['HOME'], '.pgpass')
+      File.touch(pgpass_file) unless File.exists?(pgpass_file)
+
       pgpass_line = [ @config[:host], @config[:port], @config[:dbname],
                       @config[:user], @config[:password] ].join(':')
 
-      content = File.open(File.join(ENV['HOME'], '.pgpass'), 'r') do |file|
+      content = File.read(pgpass_file).split("\n")
-        file.read
-      end.split("\n")
 
       unless content.detect{ |line| line == pgpass_line }
-        File.open(File.join(ENV['HOME'], '.pgpass'), 'w') do |file|
+        File.open(pgpass_file, 'w') do |file|
           content << pgpass_line
           file.write(content.join("\n") + "\n")
         end
-        File.chmod(0600, File.join(ENV['HOME'], '.pgpass'))
+        File.chmod(0600, pgpass_file)
       end
       pgpass_line
     end # def pgpass