add psql command

Author: will

dev_setup.fish

@@ -1,4 +1,4 @@
-  set -gp fish_user_paths bin/
-  alias scb src/cli.cr
-  complete --command cb --arguments '(cb --_completion (commandline -cp))' --no-files
-  complete --command scb --arguments '(scb --_completion (commandline -cp))' --no-files
+set -gp fish_user_paths bin/
+alias scb src/cli.cr
+complete --command cb --arguments '(cb --_completion (commandline -cp))' --no-files
+complete --command scb --arguments '(scb --_completion (commandline -cp))' --no-files

src/cb/client.cr

@@ -106,6 +106,13 @@ class CB::Client
     raise Program::Error.new "cluster #{id.colorize.t_id} does not exist, or you do not have access to it"
   end
 
+  jrecord Role, name : String, password : String, uri : URI
+
+  def get_cluster_default_role(id)
+    resp = get "clusters/#{id}/roles/default"
+    Role.from_json resp.body
+  end
+
   def create_cluster(cc)
     body = {
       ha:            cc.ha,

src/cb/completion.cr

@@ -19,16 +19,15 @@ class CB::Completion
       case args.first
       when "info"
         return info
+      when "psql"
+        return info
       when "destroy"
         return info
       when "create"
         return create
       when "firewall"
         return firewall
       else
-        STDERR.puts
-        STDERR.puts commandline.inspect
-        STDERR.puts args.inspect
         [] of String
       end
     end
@@ -45,6 +44,7 @@ class CB::Completion
       "create\tProvision a new cluster",
       "destroy\tDestroy a cluster",
       "firewall\tManage firewall rules",
+      "psql\tInteractive psql console",
     ]
   end
 

src/cb/program.cr

@@ -126,11 +126,16 @@ class CB::Program
     end
   end
 
-  def info(id)
-    c = client.get_cluster id
+  private def print_team_slash_cluster(c)
     team_name = team_name_for_cluster c
     output << team_name << "/" if team_name
     output << c.name.colorize.t_name << "\n"
+    team_name
+  end
+
+  def info(id)
+    c = client.get_cluster id
+    print_team_slash_cluster c
 
     details = {
       "state"    => c.state,
@@ -158,6 +163,37 @@ class CB::Program
     firewall_rules.each { |fr| output << " "*(pad + 4) << fr.rule << "\n" }
   end
 
+  def psql(id)
+    c = client.get_cluster id
+    role = client.get_cluster_default_role id
+    uri = role.uri
+
+    output << "connecting to "
+    team_name = print_team_slash_cluster c
+
+    psqlpromptname = String.build do |s|
+      s << "%[%033[32m%]#{team_name}%[%033m%]" << "/" if team_name
+      s << "%[%033[36m%]#{c.name}%[%033m%]"
+    end
+
+    psqlrc = File.tempfile(c.id, "pslrc")
+    File.copy("~/.psqlrc", psqlrc.path) if File.exists?("~/.psqlrc")
+    File.open(psqlrc.path, "a") do |f|
+      f.puts "\\set ON_ERROR_ROLLBACK interactive"
+      f.puts "\\set x auto"
+      f.puts "\\set PROMPT1 '#{psqlpromptname}/%[%033[33;1m%]%x%x%x%[%033[0m%]%[%033[1m%]%/%[%033[0m%]%R%# '"
+    end
+
+    Process.exec("psql", env: {
+      "PGHOST"     => uri.hostname,
+      "PGUSER"     => uri.user,
+      "PGPASSWORD" => uri.password,
+      "PGDATABASE" => uri.path.lchop('/'),
+      "PGPORT"     => uri.port.to_s,
+      "PSQLRC"     => psqlrc.path.to_s,
+    })
+  end
+
   private def team_name_for_cluster(c)
     # no way to look up a single team yet
     client.get_teams.find { |t| t.id == c.team_id }.try &.name.colorize.t_alt

src/cli.cr

@@ -65,6 +65,14 @@ op = OptionParser.new do |parser|
     end
   end
 
+  parser.on("psql", "connect to the dabase using `psql`") do
+    parser.banner = "Usage: cb psql <cluster id>"
+    parser.unknown_args do |args|
+      id = get_id_arg.call(args)
+      action = ->{ PROG.psql id }
+    end
+  end
+
   parser.on("firewall", "manage firewall rules") do
     action = manage = CB::ManageFirewall.new(PROG.client)
     parser.banner = "Usage: cb firewall <--cluster> [--add] [--remove]"

src/stdlib_ext.cr

@@ -39,3 +39,9 @@ macro jrecord(name, *properties)
     {{yield}}
   end
 end
+
+class URI
+  def self.new(pull : JSON::PullParser)
+    parse pull.read_string
+  end
+end