potluck-postgres 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 5350781163dc218410b42e790bedb5c2ca246572e1334672ea9471c81efb432c
+  data.tar.gz: fdd55c1e5b645f6f0c99df93b9761e8830776b52b32e5310cd0def002e5c4f51
+SHA512:
+  metadata.gz: 9e9475b41d894b36972aa8481eb06dfc6d1d0497d88a6841dfe92b5179bbb205db7f1cf48995df077aed00dbbc49371c553ccc5914154e15ca9c5684ed4bb9fe
+  data.tar.gz: d6df0284807374eafa9d9e48bebed6f79f00cbeb1398f6564831c2e1be703720ca059fbc5a09e0d84819f4f63f9fa5e8ff4519fc3f113a54f6696fd069c8fe86

data/LICENSE

@@ -0,0 +1,16 @@
+Copyright 2021 Nate Pickens
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
+documentation files (the "Software"), to deal in the Software without restriction, including without
+limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+the Software, and to permit persons to whom the Software is furnished to do so, subject to the following
+conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial
+portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT
+LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO
+EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN
+AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE
+OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,31 @@
+# Potluck - Postgres
+
+An extension to the Potluck gem that provides some basic utilities for setting up and connecting to Postgres
+databases, as well as control over the Postgres process.
+
+## Installation
+
+Add this line to your Gemfile:
+
+```ruby
+gem('potluck-postgres')
+```
+
+Or install manually on the command line:
+
+```bash
+gem install potluck-postgres
+```
+
+## Usage
+
+[Coming soon.]
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/npickens/potluck.
+
+## License
+
+The gem is available as open source under the terms of the
+[MIT License](https://opensource.org/licenses/MIT).

data/lib/potluck/postgres.rb

@@ -0,0 +1,132 @@
+# frozen_string_literal: true
+
+require('potluck')
+
+module Potluck
+  class Postgres < Dish
+    attr_reader(:database)
+
+    def initialize(config, **args)
+      super(**args)
+
+      @config = config
+    end
+
+    def connect
+      (tries ||= 0) && (tries += 1)
+      @database = Sequel.connect(@config, logger: @logger)
+    rescue Sequel::DatabaseConnectionError => e
+      if (dud = Sequel::DATABASES.last)
+        dud.disconnect
+        Sequel.synchronize { Sequel::DATABASES.delete(dud) }
+      end
+
+      if e.message =~ /role .* does not exist/ && tries == 1
+        create_database_role
+        create_database
+        retry
+      elsif e.message =~ /database .* does not exist/ && tries == 1
+        create_database
+        retry
+      elsif (@is_local && tries < 3) && (e.message.include?('could not connect') ||
+          e.message.include?('the database system is starting up'))
+        sleep(1)
+        retry
+      elsif e.message.include?('could not connect')
+        abort("#{e.class}: #{e.message.strip}")
+      else
+        abort("#{e.class}: #{e.message.strip}\n  #{e.backtrace.join("\n  ")}")
+      end
+    end
+
+    def disconnect
+      @database&.disconnect
+    end
+
+    def create_database_role
+      tmp_config = @config.dup
+      tmp_config[:database] = 'postgres'
+      tmp_config[:username] = ENV['USER']
+      tmp_config[:password] = nil
+
+      begin
+        Sequel.connect(tmp_config, logger: @logger) do |database|
+          database.execute("CREATE ROLE #{@config[:username]} WITH LOGIN CREATEDB REPLICATION PASSWORD "\
+            "'#{@config[:password]}'")
+        end
+      rescue => e
+        @logger.error("#{e.class}: #{e.message.strip}\n  #{e.backtrace.join("\n  ")}\n")
+        abort("Could not create role '#{@config[:username]}'. Make sure database user '#{ENV['USER']}' "\
+          'has permission to do so, or create it manually.')
+      end
+    end
+
+    def create_database
+      tmp_config = @config.dup
+      tmp_config[:database] = 'postgres'
+
+      begin
+        Sequel.connect(tmp_config, logger: @logger) do |database|
+          database.execute("CREATE DATABASE #{@config[:database]}")
+        end
+      rescue => e
+        @logger.error("#{e.class}: #{e.message.strip}\n  #{e.backtrace.join("\n  ")}\n")
+        abort("Could not create database '#{@config[:database]}'. Make sure database user "\
+          "'#{@config[:username]}' has permission to do so, or create it manually.")
+      end
+    end
+
+    def migrate(dir, steps = nil)
+      return unless File.directory?(dir)
+
+      Sequel.extension(:migration)
+
+      # Suppress Sequel schema migration table queries.
+      original_level = @logger.level
+      @logger.level = Logger::WARN
+
+      args = [Sequel::Model.db, dir, {allow_missing_migration_files: true}]
+      migrator = Sequel::TimestampMigrator.new(*args)
+
+      return if migrator.files.empty?
+
+      if steps
+        all = migrator.files.map { |f| File.basename(f) }
+        applied = migrator.applied_migrations
+        current = applied.last
+
+        return if applied.empty? && steps <= 0
+
+        index = [[0, (all.index(current) || -1) + steps].max, all.size].min
+        file = all[index]
+
+        args.last[:target] = migrator.send(:migration_version_from_file, file)
+      end
+
+      migrator = Sequel::TimestampMigrator.new(*args)
+      @logger.level = original_level
+      migrator.run
+    end
+
+    private
+
+    def self.plist
+      super(
+        <<~EOS
+          <key>ProgramArguments</key>
+          <array>
+            <string>/usr/local/opt/postgresql/bin/postgres</string>
+            <string>-D</string>
+            <string>/usr/local/var/postgres</string>
+          </array>
+          <key>WorkingDirectory</key>
+          <string>/usr/local</string>
+          <key>StandardOutPath</key>
+          <string>/usr/local/var/log/postgres.log</string>
+          <key>StandardErrorPath</key>
+          <string>/usr/local/var/log/postgres.log</string>
+        EOS
+      )
+    end
+  end
+end

metadata

@@ -0,0 +1,112 @@
+--- !ruby/object:Gem::Specification
+name: potluck-postgres
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Nate Pickens
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2021-03-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: potluck
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.0.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.0.1
+- !ruby/object:Gem::Dependency
+  name: sequel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.41'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.41'
+- !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.11.2
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 6.0.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 5.11.2
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 6.0.0
+description: An extension to the Potluck gem that provides some basic utilities for
+  setting up and connecting to Postgres databases, as well as control over the Postgres
+  process.
+email:
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- LICENSE
+- README.md
+- lib/potluck/postgres.rb
+homepage: https://github.com/npickens/potluck/tree/master/potluck-postgres
+licenses:
+- MIT
+metadata:
+  allowed_push_host: https://rubygems.org
+  homepage_uri: https://github.com/npickens/potluck/tree/master/potluck-postgres
+  source_code_uri: https://github.com/npickens/potluck/tree/master/potluck-postgres
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.5.8
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.2.3
+signing_key:
+specification_version: 4
+summary: A Ruby manager for Postgres.
+test_files: []