Add --database flag to scope and psql

These changes add the ability to specify a database name to connect when
using the `scope` and `psql` command.

The flag is setup to be optional. If not provided then it will default
to `postgres`.

Author: abrightwell

src/cb/psql.cr

@@ -2,11 +2,14 @@ require "./action"
 
 class CB::Psql < CB::Action
   property cluster_id : String?
+  property database : String?
 
   def run
     c = client.get_cluster cluster_id
     uri = client.get_cluster_default_role(cluster_id).uri
 
+    database.tap { |db| uri.path = db if db }
+
     output << "connecting to "
     team_name = print_team_slash_cluster c, output
 
@@ -32,6 +35,10 @@ class CB::Psql < CB::Action
     @cluster_id = str
   end
 
+  def database=(str : String)
+    @database = str
+  end
+
   private def ensure_cert(team_id) : String
     cert_dir = CB::Creds::CONFIG / "certs"
     path = cert_dir / "#{team_id}.pem"

src/cb/scope.cr

@@ -4,13 +4,18 @@ require "./scope_checks/*"
 class CB::Scope < CB::Action
   property cluster_id : String?
   property checks : Array(::Scope::Check.class) = [] of ::Scope::Check.class
+  property database : String?
   property suite : String?
 
   def run
     check_required_args { |missing| missing << "cluster" unless cluster_id }
 
     uri = client.get_cluster_default_role(cluster_id).uri
 
+    if database.presence
+      uri.path = database.to_s
+    end
+
     self.suite = "quick" if checks.empty? && suite.nil?
 
     case suite
@@ -32,3 +37,8 @@ class CB::Scope < CB::Action
     end
   end
 end
+
+def database=(str : String)
+  str = str.downcase
+  @database = str
+end

src/cli.cr

@@ -77,9 +77,11 @@ op = OptionParser.new do |parser|
   end
 
   parser.on("psql", "Connect to the database using `psql`") do
-    parser.banner = "Usage: cb psql <cluster id> [-- [args for psql such as -c or -f]]"
+    parser.banner = "Usage: cb psql <cluster id> [--database] [-- [args for psql such as -c or -f]]"
     action = psql = CB::Psql.new(PROG.client)
 
+    parser.on("--database NAME", "Database name (default: postgres)") { |arg| psql.database = arg }
+
     parser.unknown_args do |args|
       psql.cluster_id = get_id_arg.call(args)
     end
@@ -129,6 +131,7 @@ op = OptionParser.new do |parser|
     action = scope = CB::Scope.new PROG.client
     parser.on("--cluster ID", "Choose cluster") { |arg| scope.cluster_id = arg }
     parser.on("--suite <all|quick>", "Run suite of scopes (default: quick)") { |arg| scope.suite = arg }
+    parser.on("--database NAME", "Database name (default: postgres)") { |arg| scope.database = arg }
     Scope::Check.all.each do |ck|
       parser.on(ck.flag, ck.desc) { scope.checks << ck.type }
     end