seira 0.1.6 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 2139b5d967293724e307e500bd87c3a77998f563
-  data.tar.gz: 5eb1767a8f7c23a6dad5357823fd651d5089e21d
+  metadata.gz: ff2eb0fe8f04aa7734da72c7f0542c7d5b556b84
+  data.tar.gz: 662cee51d65fc44ad8010b4dd3dfdc29b644de6b
 SHA512:
-  metadata.gz: 38121663c35640ef9a8f6c2180965197f088b46ed609a0ea122b301907653b69ae93c7e13d164868f1d4c21ac71964e4f281323a916352f34a3186e63950c63b
-  data.tar.gz: d167aa0f6e90e6130e42721b0240867ed03d9cdcbaa13c245eb86ca29a45e4baf3dc7845e076152a178238b43e5d4e74cc408ad8674c506f51685f61040c720b
+  metadata.gz: dc8e4d479761c4c4f3e09b2b56d0d2e70aca3295fceab18316f56751030b0cee4291a26c5dc9eeec00fcf1d2d1abdba5f47e945c06fdd440e3f861f069eacd14
+  data.tar.gz: 47e734d06b5456df2f54497fb9efe7f02f44f02c7f54de12dc7ce2e8287c69a512cda8f3c96d3c68ea92fd4042149736292a650faafec4bb5c45ededd4ab7995

data/lib/seira.rb

@@ -1,5 +1,6 @@
 require 'json'
 require 'highline/import'
+require 'colorize'
 
 require "seira/version"
 require 'seira/app'

data/lib/seira/db.rb

@@ -1,10 +1,10 @@
 require 'securerandom'
 
-require_relative 'db/create.rb'
+require_relative 'db/create'
 
 module Seira
   class Db
-    VALID_ACTIONS = %w[help create delete list].freeze
+    VALID_ACTIONS = %w[help create delete list restart connect].freeze
     SUMMARY = "Manage your Cloud SQL Postgres databases.".freeze
 
     attr_reader :app, :action, :args, :context
@@ -26,11 +26,27 @@ module Seira
         run_delete
       when 'list'
         run_list
+      when 'restart'
+        run_restart
+      when 'connect'
+        run_connect
       else
         fail "Unknown command encountered"
       end
     end
 
+    # NOTE: Relies on the pgbouncer instance being named based on the db name, as is done in create command
+    def primary_instance
+      database_url = Secrets.new(app: app, action: 'get', args: [], context: context).get('DATABASE_URL')
+      return nil unless database_url
+
+      primary_uri = URI.parse(database_url)
+      host = primary_uri.host
+
+      # Convert handshake-onyx-burmese-pgbouncer-service to handshake-onyx-burmese
+      host.gsub('-pgbouncer-service', '')
+    end
+
     private
 
     def run_help
@@ -39,6 +55,8 @@ module Seira
       puts "create: Create a new postgres instance in cloud sql. Supports creating replicas and other numerous flags."
       puts "delete: Delete a postgres instance from cloud sql. Use with caution, and remove all kubernetes configs first."
       puts "list: List all postgres instances."
+      puts "restart: Fully restart the database."
+      puts "connect: Open a psql command prompt. You will be shown the password needed before the prompt opens."
     end
 
     def run_create
@@ -61,6 +79,23 @@ module Seira
       puts existing_instances
     end
 
+    def run_restart
+      name = "#{app}-#{args[0]}"
+      if system("gcloud sql instances restart #{name}")
+        puts "Successfully restarted sql instance #{name}"
+      else
+        puts "Failed to restart sql instance #{name}"
+      end
+    end
+
+    def run_connect
+      name = args[0] || primary_instance
+      puts "Connecting to #{name}..."
+      root_password = Secrets.new(app: app, action: 'get', args: [], context: context).get("#{name.tr('-', '_').upcase}_ROOT_PASSWORD") || "Not found in secrets"
+      puts "Your root password for 'postgres' user is: #{root_password}"
+      system("gcloud sql connect #{name}")
+    end
+
     def existing_instances
       `gcloud sql instances list --uri`.split("\n").map { |uri| uri.split('/').last }.select { |name| name.start_with? "#{app}-" }.map { |name| name.gsub(/^#{app}-/, '') }
     end

data/lib/seira/db/create.rb

@@ -58,7 +58,7 @@ module Seira
           elsif arg.start_with? '--storage='
             @storage = arg.split('=')[1]
           elsif arg.start_with? '--primary='
-            @replica_for = arg.split('=')[1] # TODO: Read secret to get it automatically
+            @replica_for = arg.split('=')[1] # TODO: Read secret to get it automatically, but allow for fallback
           elsif arg.start_with? '--highly-available'
             @make_highly_available = true
           elsif /^--[\w\-]+=.+$/.match? arg

data/lib/seira/secrets.rb

@@ -103,7 +103,7 @@ module Seira
       if value.nil?
         puts "Secret '#{key}' not found"
       else
-        puts "#{key}: #{value}"
+        puts "#{key.green}: #{value}"
       end
     end
 
@@ -123,7 +123,7 @@ module Seira
       secrets = fetch_current_secrets
       puts "Base64 encoded keys for #{app}:"
       secrets['data'].each do |k, v|
-        puts "#{k}: #{v}"
+        puts "#{k.green}: #{v}"
       end
     end
 
@@ -131,7 +131,7 @@ module Seira
       secrets = fetch_current_secrets
       puts "Decoded (raw) keys for #{app}:"
       secrets['data'].each do |k, v|
-        puts "#{k}: #{Base64.decode64(v)}"
+        puts "#{k.green}: #{Base64.decode64(v)}"
       end
     end
 

data/lib/seira/version.rb

@@ -1,3 +1,3 @@
 module Seira
-  VERSION = "0.1.6".freeze
+  VERSION = "0.1.7".freeze
 end

data/seira.gemspec

@@ -21,6 +21,7 @@ Gem::Specification.new do |spec|
   spec.require_paths = ["lib"]
 
   spec.add_runtime_dependency "highline"
+  spec.add_runtime_dependency "colorize"
 
   spec.add_development_dependency "bundler", "~> 1.14"
   spec.add_development_dependency "rake", "~> 10.0"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: seira
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.7
 platform: ruby
 authors:
 - Scott Ringwelski
@@ -24,6 +24,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement