class PG::Connection

def self.conndefaults_hash
	return self.conndefaults.each_with_object({}) do |info, hash|
		hash[ info[:keyword].to_sym ] = info[:val]
	end
end

def self::parse_connect_args( *args )
	return '' if args.empty?
	hash_arg = args.last.is_a?( Hash ) ? args.pop : {}
	option_string = ''
	options = {}
	# Parameter 'fallback_application_name' was introduced in PostgreSQL 9.0
	# together with PQescapeLiteral().
	if PG::Connection.instance_methods.find {|m| m.to_sym == :escape_literal }
		options[:fallback_application_name] = $0.sub( /^(.{30}).{4,}(.{30})$/ ){ $1+"..."+$2 }
	end
	if args.length == 1
		case args.first
		when URI, /\A#{URI.regexp}\z/
			uri = URI(args.first)
			options.merge!( Hash[URI.decode_www_form( uri.query )] ) if uri.query
		when /=/
			# Option string style
			option_string = args.first.to_s
		else
			# Positional parameters
			options[CONNECT_ARGUMENT_ORDER.first.to_sym] = args.first
		end
	else
		max = CONNECT_ARGUMENT_ORDER.length
		raise ArgumentError,
			"Extra positional parameter %d: %p" % [ max + 1, args[max] ] if args.length > max
		CONNECT_ARGUMENT_ORDER.zip( args ) do |(k,v)|
			options[ k.to_sym ] = v if v
		end
	end
	options.merge!( hash_arg )
	if uri
		uri.host     = nil if options[:host]
		uri.port     = nil if options[:port]
		uri.user     = nil if options[:user]
		uri.password = nil if options[:password]
		uri.path     = '' if options[:dbname]
		uri.query    = URI.encode_www_form( options )
		return uri.to_s.sub( /^#{uri.scheme}:(?!\/\/)/, "#{uri.scheme}://" )
	else
		option_string += ' ' unless option_string.empty? && options.empty?
		return option_string + options.map { |k,v| "#{k}=#{quote_connstr(v)}" }.join( ' ' )
	end
end

def self::quote_connstr( value )
	return "'" + value.to_s.gsub( /[\\']/ ) {|m| '\\' + m } + "'"
end

def conndefaults
	return self.class.conndefaults
end

def conndefaults_hash
	return self.class.conndefaults_hash
end

def conninfo_hash
	return self.conninfo.each_with_object({}) do |info, hash|
		hash[ info[:keyword].to_sym ] = info[:val]
	end
end

def copy_data( sql, coder=nil )
	res = exec( sql )
	case res.result_status
	when PGRES_COPY_IN
		begin
			if coder
				old_coder = self.encoder_for_put_copy_data
				self.encoder_for_put_copy_data = coder
			end
			yield res
		rescue Exception => err
			errmsg = "%s while copy data: %s" % [ err.class.name, err.message ]
			put_copy_end( errmsg )
			get_result
			raise
		else
			put_copy_end
			get_last_result
		ensure
			self.encoder_for_put_copy_data = old_coder if coder
		end
	when PGRES_COPY_OUT
		begin
			if coder
				old_coder = self.decoder_for_get_copy_data
				self.decoder_for_get_copy_data = coder
			end
			yield res
		rescue Exception => err
			cancel
			while get_copy_data
			end
			while get_result
			end
			raise
		else
			res = get_last_result
			if !res || res.result_status != PGRES_COMMAND_OK
				while get_copy_data
				end
				while get_result
				end
				raise PG::NotAllCopyDataRetrieved, "Not all COPY data retrieved"
			end
			res
		ensure
			self.decoder_for_get_copy_data = old_coder if coder
		end
	else
		raise ArgumentError, "SQL command is no COPY statement: #{sql}"
	end
end

def ssl_attributes
	ssl_attribute_names.each.with_object({}) do |n,h|
		h[n] = ssl_attribute(n)
	end
end