pg-aws_rds_iam 0.1.0

data/lib/pg/aws_rds_iam/connection_info/keyword_value_string.rb

@@ -0,0 +1,114 @@
+# frozen_string_literal: true
+
+module PG
+  module AWS_RDS_IAM
+    module ConnectionInfo
+      class KeywordValueString
+        attr_reader :auth_token_generator_name
+
+        def initialize(connection_string)
+          @params = Parser.new(connection_string).parse
+          @auth_token_generator_name = @params.delete(:aws_rds_iam_auth_token_generator)
+        end
+
+        def user
+          @params[:user]
+        end
+
+        def host
+          @params[:host]
+        end
+
+        def port
+          @params[:port]
+        end
+
+        def password=(value)
+          @params[:password] = value
+        end
+
+        def to_s
+          @params.map { |key, value| "#{key}=#{PG::Connection.quote_connstr(value)}" }.join(" ")
+        end
+
+        def to_h
+          @params.dup
+        end
+
+        class Parser
+          def initialize(connection_string)
+            @buffer = StringScanner.new(connection_string)
+          end
+
+          # @see https://github.com/postgres/postgres/blob/REL_12_1/src/interfaces/libpq/fe-connect.c#L5344-L5457
+          def parse
+            result = {}
+
+            skip_whitespace
+            until @buffer.eos?
+              key = parse_key
+              skip_whitespace
+              assert_followed_by_equals_sign key
+              skip_whitespace
+              value = parse_value
+              skip_whitespace
+
+              result[key] = value
+            end
+
+            @buffer.reset
+            result
+          end
+
+          private
+
+          def skip_whitespace
+            @buffer.skip(/\s+/)
+          end
+
+          def parse_key
+            key = @buffer.scan(/[^=\s]+/)
+            raise ParseError, %(Missing parameter name before "#{@buffer.rest}") unless key
+
+            key.to_sym
+          end
+
+          def assert_followed_by_equals_sign(key)
+            raise ParseError, %(Missing "=" after "#{key}") unless @buffer.getch == "="
+          end
+
+          def parse_value
+            if @buffer.peek(1) == "'"
+              parse_quoted_value
+            else
+              parse_unquoted_value
+            end
+          end
+
+          def parse_quoted_value
+            value = +""
+            @buffer.skip(/'/)
+            loop do
+              value << (@buffer.scan(/[^'\\]+/) || "")
+              case @buffer.getch
+              when "'" then return value
+              when "\\" then value << (@buffer.getch || "")
+              else raise ParseError, "Unterminated quoted value"
+              end
+            end
+          end
+
+          def parse_unquoted_value
+            value = +""
+            loop do
+              value << (@buffer.scan(/[^\s\\]+/) || "")
+              return value unless @buffer.getch == "\\"
+
+              value << (@buffer.getch || "")
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/pg/aws_rds_iam/connection_info/parse_error.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module PG
+  module AWS_RDS_IAM
+    module ConnectionInfo
+      ParseError = Class.new(StandardError)
+    end
+  end
+end

data/lib/pg/aws_rds_iam/connection_info/uri.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+module PG
+  module AWS_RDS_IAM
+    module ConnectionInfo
+      class URI
+        def self.match?(connection_string)
+          /\A#{::URI::ABS_URI_REF}\z/.match?(connection_string)
+        end
+
+        attr_reader :auth_token_generator_name
+
+        def initialize(connection_string)
+          @uri = ::URI.parse(connection_string)
+          @query = ::URI.decode_www_form(@uri.query).to_h
+          @auth_token_generator_name = @query.delete("aws_rds_iam_auth_token_generator")
+        end
+
+        def user
+          @uri.user || @query["user"]
+        end
+
+        def host
+          @uri.host || @query["host"]
+        end
+
+        def port
+          @uri.port || @query["port"]
+        end
+
+        def password=(value)
+          @uri.password = nil
+          @query["password"] = value
+        end
+
+        def to_s
+          @uri.query = ::URI.encode_www_form(@query)
+
+          @uri.to_s.sub(%r{^#{@uri.scheme}:(?!//)}, "#{@uri.scheme}://")
+        end
+      end
+    end
+  end
+end

data/lib/pg/aws_rds_iam/version.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+module PG
+  module AWS_RDS_IAM
+    # The current version of the gem.
+    VERSION = "0.1.0"
+  end
+end

data/pg-aws_rds_iam.gemspec

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+lib = File.expand_path("lib", __dir__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "pg/aws_rds_iam/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "pg-aws_rds_iam"
+  spec.version = PG::AWS_RDS_IAM::VERSION
+  spec.authors = ["Andrew Haines"]
+  spec.email = ["andrew@haines.org.nz"]
+
+  spec.summary = "IAM authentication for PostgreSQL on Amazon RDS"
+  spec.description = "PG::AWS_RDS_IAM is a plugin for the pg gem that adds support for IAM authentication when connecting to PostgreSQL databases hosted in Amazon RDS."
+  spec.homepage = "https://github.com/haines/pg-aws_rds_iam"
+  spec.license = "MIT"
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = spec.homepage
+  spec.metadata["changelog_uri"] = "#{spec.homepage}/blob/master/CHANGELOG.md"
+
+  spec.files = Dir.chdir(__dir__) { `git ls-files -z`.split("\x0").reject { |path| path.start_with?("test/") } }
+
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "aws-sdk-rds", "~> 1.0"
+  spec.add_dependency "pg", ">= 0.18", "< 2.0"
+
+  spec.add_development_dependency "activerecord", ">= 5.2", "< 7.0"
+  spec.add_development_dependency "aws-sdk-ec2", "~> 1.137"
+  spec.add_development_dependency "bundler", "~> 2.0"
+  spec.add_development_dependency "minitest", "~> 5.14"
+  spec.add_development_dependency "minitest-reporters", "~> 1.4"
+  spec.add_development_dependency "pry", "~> 0.12"
+  spec.add_development_dependency "rake", "~> 13.0"
+  spec.add_development_dependency "rubocop", "~> 0.79"
+  spec.add_development_dependency "timecop", "~> 0.9"
+  spec.add_development_dependency "yard", "~> 0.9"
+end

metadata

@@ -0,0 +1,256 @@
+--- !ruby/object:Gem::Specification
+name: pg-aws_rds_iam
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Andrew Haines
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2020-02-05 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-rds
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0.18'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0.18'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '5.2'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '7.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '5.2'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '7.0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-ec2
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.137'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.137'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.14'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.14'
+- !ruby/object:Gem::Dependency
+  name: minitest-reporters
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.4'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.4'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.12'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.12'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.79'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.79'
+- !ruby/object:Gem::Dependency
+  name: timecop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+- !ruby/object:Gem::Dependency
+  name: yard
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+description: PG::AWS_RDS_IAM is a plugin for the pg gem that adds support for IAM
+  authentication when connecting to PostgreSQL databases hosted in Amazon RDS.
+email:
+- andrew@haines.org.nz
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".github/workflows/pull-request.yml"
+- ".gitignore"
+- ".rubocop.yml"
+- ".ruby-version"
+- ".yardopts"
+- CHANGELOG.md
+- CODE_OF_CONDUCT.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/bundle
+- bin/console
+- bin/install-bundler
+- bin/rake
+- bin/setup
+- bin/yard
+- lib/pg/aws_rds_iam.rb
+- lib/pg/aws_rds_iam/auth_token_generator.rb
+- lib/pg/aws_rds_iam/auth_token_generator_registry.rb
+- lib/pg/aws_rds_iam/auth_token_injector.rb
+- lib/pg/aws_rds_iam/connection.rb
+- lib/pg/aws_rds_iam/connection_info.rb
+- lib/pg/aws_rds_iam/connection_info/keyword_value_string.rb
+- lib/pg/aws_rds_iam/connection_info/parse_error.rb
+- lib/pg/aws_rds_iam/connection_info/uri.rb
+- lib/pg/aws_rds_iam/version.rb
+- pg-aws_rds_iam.gemspec
+homepage: https://github.com/haines/pg-aws_rds_iam
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/haines/pg-aws_rds_iam
+  source_code_uri: https://github.com/haines/pg-aws_rds_iam
+  changelog_uri: https://github.com/haines/pg-aws_rds_iam/blob/master/CHANGELOG.md
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.1.2
+signing_key: 
+specification_version: 4
+summary: IAM authentication for PostgreSQL on Amazon RDS
+test_files: []