cl-magic 0.3.0

data/lib/cl/magic/cl-kube-restart

@@ -0,0 +1,84 @@
+#!/usr/bin/env ruby
+require 'optparse'
+require 'optparse/subcommand'
+require 'tty-command'
+require 'tty-prompt'
+
+require 'cl/magic/common/common_options.rb'
+require 'cl/magic/common/logging.rb'
+require 'cl/magic/common/gcloud.rb'
+require 'cl/magic/common/kubectl.rb'
+
+@logger = get_logger()
+@cl_cmd_name = File.basename(__FILE__).split('-').join(' ')
+
+#
+# Features
+#
+
+def do_work(options)
+  if TTY::Prompt.new.yes?("Are you sure you want to restart #{options[:deployment]}?")
+    cmd = "kubectl rollout restart deployment/#{options[:deployment]} --namespace=#{options[:namespace]}"
+    @logger.puts
+    @logger.wait cmd
+    exec(cmd)
+  end
+end
+
+#
+# Options
+#
+
+options = {}
+global_banner = <<DOC
+
+ Restart a deployment for a service to refresh all the containers
+
+ Usage: #{@cl_cmd_name} [options]
+
+DOC
+
+global = OptionParser.new do |g|
+  g.banner = global_banner
+  add_help_and_verbose(g)
+
+  g.on("-c", "--context CONTEXT", "a kubectl context name") do |v|
+    options[:kube_context] = v
+  end
+
+  g.on("-n", "--namespace NAMESPACE", "a kubectl namespace in the current context's cluster") do |v|
+    options[:namespace] = v
+  end
+
+  g.on("-d", "--deployment DEPLOYMENT", "a kubectl deployment") do |v|
+    options[:deployment] = v
+  end
+
+  g.on("-x", "use magic tools context") do |v|
+    options[:ktx] = v
+  end
+
+end
+
+#
+# Run
+#
+
+global.parse(ARGV)
+
+# set: context and namespace
+set_context_and_namespace(options)
+set_ktx(options)
+
+# set: deployment
+options[:deployment] = pick_kubectl_deployment(options).first
+
+# display full command
+@logger.puts ""
+write_history("""#{@cl_cmd_name} \\
+  --context=#{options[:kube_context]}
+  --namespace=#{options[:namespace]}
+  --deployment=#{options[:deployment]}
+""")
+
+do_work(options)

data/lib/cl/magic/cl-kube-search

@@ -0,0 +1,111 @@
+#!/usr/bin/env ruby
+require 'optparse'
+require 'optparse/subcommand'
+require 'tty-command'
+
+require 'cl/magic/common/common_options.rb'
+require 'cl/magic/common/logging.rb'
+require 'cl/magic/common/kubectl.rb'
+
+@logger = get_logger()
+@cl_cmd_name = File.basename(__FILE__).split('-').join(' ')
+
+#
+# Features
+#
+
+def do_search(options)
+
+  # k8 api-resources
+  unless options[:kube_service_name]
+    @logger.puts ""
+    cmd = "kubectl api-resources --verbs=list --namespaced -o name | grep -v events.events.k8s.io | grep -v events | sort | uniq"
+    @logger.wait cmd
+    api_resources = TTY::Command.new(:printer => :null).run(cmd).collect { |line| line.split() }
+    options[:kube_service_name] = pick_single_result(api_resources, "Pick a resource").first
+  end
+
+  # display full command
+  write_history("""#{@cl_cmd_name} \\
+  --context=#{options[:kube_context]} \\
+  --service=#{options[:kube_service_name]} \\
+  --namespace=#{options[:namespace]}
+""")
+
+  # get resources
+  @logger.puts ""
+  namespace = options[:namespace] ? "-n #{options[:namespace]}" : "--all-namespaces"
+  cmd = "kubectl get #{namespace} --ignore-not-found #{options[:kube_service_name]}"
+  cmd += " | grep #{options[:pattern]}" if options[:pattern]
+  @logger.wait cmd
+  result = TTY::Command.new(:printer => :null).run(cmd).out
+  only_one_result = options[:pattern] ? result.lines.count == 1 : result.lines.count == 2
+
+  # print resources
+  unless only_one_result
+    puts result
+    @logger.puts
+  end
+
+  # describe resource
+  if only_one_result or TTY::Prompt.new.yes?("Describe one of these resources?")
+    num_headers = options[:pattern] ? 0 : 1
+    resources_by_type = parse_table_results(cmd, "No resources found", num_headers=num_headers)
+    selection_index = 1
+    resource = pick_single_result(resources_by_type, "show resource?", nil, nil, selection_index)
+    cmd = "kubectl describe #{options[:kube_service_name]} #{resource[0]} --namespace=#{options[:namespace]}"
+    @logger.puts
+    @logger.wait cmd
+    puts TTY::Command.new(:printer => :null).run(cmd)
+  end
+end
+
+#
+# Options
+#
+
+options = {}
+global_banner = <<DOC
+
+ Search for k8 resources
+
+ Usage: #{@cl_cmd_name} [options]
+
+DOC
+
+global = OptionParser.new do |g|
+  g.banner = global_banner
+  add_help_and_verbose(g)
+
+  g.on("-c", "--context CONTEXT", "a kubectl context name") do |v|
+    options[:kube_context] = v
+  end
+
+  g.on("-s", "--service SERVICE_NAME", "a kubectl api-service name") do |v|
+    options[:kube_service_name] = v
+  end
+
+  g.on("-n", "--namespace NAMESPACE", "a kubectl namespace in the current context's cluster") do |v|
+    options[:namespace] = v
+  end
+
+  g.on("-p", "--pattern PATTERN", "pattern to grep for") do |v|
+    options[:pattern] = v
+  end
+
+  g.on("-x", "use magic tools context") do |v|
+    options[:ktx] = v
+  end
+end
+
+#
+# Run
+#
+
+global.parse(ARGV)
+
+# set: context
+set_context_and_namespace(options) if options[:ktx]
+options[:kube_context] = pick_kubectl_context(options[:kube_context])
+
+do_search(options)

data/lib/cl/magic/cl-kube-search-all

@@ -0,0 +1,97 @@
+#!/usr/bin/env ruby
+require 'optparse'
+require 'optparse/subcommand'
+require 'tty-command'
+require 'tty-prompt'
+require 'json'
+require 'pastel'
+
+require 'cl/magic/common/common_options.rb'
+require 'cl/magic/common/logging.rb'
+require 'cl/magic/common/kubectl.rb'
+
+@logger = get_logger()
+@cl_cmd_name = File.basename(__FILE__).split('-').join(' ')
+
+#
+# Features
+#
+
+def do_search(options)
+
+  # k8 api-resources
+  unless options[:kube_service_name]
+    @logger.puts ""
+    cmd = "kubectl api-resources --verbs=list --namespaced -o name | grep -v events.events.k8s.io | grep -v events | sort | uniq"
+    @logger.wait cmd
+    api_resources = TTY::Command.new(:printer => :null).run(cmd).collect { |line| line.split() }
+    options[:kube_service_name] = pick_single_result(api_resources, "Pick a resource").first
+  end
+
+  get_kubectl_contexts().each do |context|
+
+    context_name = context[1]
+
+    # set context
+    cmd = "kubectl config use-context #{context_name}"
+    TTY::Command.new(:printer => :null).run(cmd)
+
+    # searching in
+    puts "searching: #{context_name}"
+
+    # grep services
+    cmd = "kubectl get --all-namespaces --ignore-not-found #{options[:kube_service_name]}"
+    cmd += " | grep #{options[:pattern]}" if options[:pattern]
+
+    results = TTY::Command.new(:printer => :null).run!(cmd) do |out, err|
+      unless out.nil?
+        puts out
+        @logger.puts ""
+      end
+    end
+  end
+
+end
+
+#
+# Options
+#
+
+options = {}
+global_banner = <<DOC
+
+Grep across all clusters and namespaces
+
+Usage: #{@cl_cmd_name} [filter-name] [options]
+
+DOC
+
+global = OptionParser.new do |g|
+  g.banner = global_banner
+
+  g.on("-s", "--service SERVICE_NAME", "a kubectl api-service name; defaults to 'services'") do |v|
+    options[:kube_service_name] = v
+  end
+
+  g.on("-p", "--pattern PATTERN", "pattern to grep for") do |v|
+    options[:pattern] = v
+  end
+
+  g.on("-x", "use magic tools context") do |v|
+    options[:ktx] = v
+  end
+end
+
+#
+# Run
+#
+
+global.parse(ARGV)
+
+# display full command
+write_history("""#{@cl_cmd_name} \\
+  --grep=#{options[:pattern]} \\
+  --service=#{options[:kube_service_name]}
+""")
+
+do_search(options)

data/lib/cl/magic/cl-kube-ssh

@@ -0,0 +1,94 @@
+#!/usr/bin/env ruby
+require 'optparse'
+require 'optparse/subcommand'
+require 'tty-command'
+require 'tty-prompt'
+require 'json'
+require 'pastel'
+
+require 'cl/magic/common/common_options.rb'
+require 'cl/magic/common/logging.rb'
+require 'cl/magic/common/kubectl.rb'
+
+@logger = get_logger()
+@cl_cmd_name = File.basename(__FILE__).split('-').join(' ')
+
+#
+# Features
+#
+
+def visit_pod(options)
+  shell_command = options[:bash] ? "bash" : "sh"
+  cmd = "kubectl exec -it #{options[:pod_name]} --container #{options[:container_name]} --namespace #{options[:namespace]} -- #{shell_command}"
+  @logger.puts
+  @logger.wait cmd
+  exec(cmd)
+end
+
+#
+# Options
+#
+
+options = {}
+global_banner = <<DOC
+
+Ssh into a container
+
+Usage: #{@cl_cmd_name} [options]
+
+DOC
+
+global = OptionParser.new do |g|
+  g.banner = global_banner
+
+  g.on("-c", "--context CONTEXT", "a kubectl context name") do |v|
+    options[:kube_context] = v
+  end
+
+  g.on("-n", "--namespace NAMESPACE", "a kubectl namespace in the current context's cluster") do |v|
+    options[:namespace] = v
+  end
+
+  g.on("-d", "--deployment NAME", "a kubernetes deployment name") do |v|
+    options[:deployment_name] = v
+  end
+
+  g.on("-p", "--pod POD_NAME", "a kubernetes pod name") do |v|
+    options[:pod_name] = v
+  end
+
+  g.on("--container CONTAINER_NAME", "a kubernetes container you'd like to visit") do |v|
+    options[:container_name] = v
+  end
+
+  g.on("-x", "use magic tools context") do |v|
+    options[:ktx] = v
+  end
+
+  g.on("-b", "--bash", "bash instead of shell") do |v|
+    options[:bash] = v
+  end
+end
+
+#
+# Run
+#
+
+global.parse(ARGV)
+
+# set everything
+set_context_namespace_deployment_pod_container_options(options)
+set_ktx(options)
+
+# display full command
+write_history("""#{@cl_cmd_name} \\
+  --context=#{options[:kube_context]} \\
+  --namespace=#{options[:namespace]} \\
+  --deployment=#{options[:deployment_name]} \\
+  --pod=#{options[:pod_name]} \\
+  --container=#{options[:container_name]} \\
+  #{options[:bash] ? "--bash" : ""}
+""")
+
+# visit pod
+visit_pod(options)

data/lib/cl/magic/cl-poll

@@ -0,0 +1,51 @@
+#!/usr/bin/env ruby
+require 'optparse'
+require 'optparse/subcommand'
+require 'tty-command'
+require 'tty-prompt'
+
+require 'cl/magic/common/common_options.rb'
+require 'cl/magic/common/logging.rb'
+require 'cl/magic/common/gcloud.rb'
+require 'cl/magic/common/kubectl.rb'
+
+@logger = get_logger()
+@cl_cmd_name = File.basename(__FILE__).split('-').join(' ')
+
+#
+# Features
+#
+
+def do_work(options)
+  while true
+    tty_command = TTY::Command.new(printer: :null)
+    out, err = tty_command.run(ARGV.join(' '))
+    puts out
+    puts
+    sleep(1)
+  end
+end
+
+#
+# Options
+#
+
+options = {}
+global_banner = <<DOC
+
+ A sandbox to try things
+
+ Usage: #{@cl_cmd_name} [options]
+
+DOC
+
+global = OptionParser.new do |g|
+  g.banner = global_banner
+  add_help_and_verbose(g)
+end
+
+#
+# Run
+#
+
+do_work(options)

data/lib/cl/magic/cl-sandbox

@@ -0,0 +1,49 @@
+#!/usr/bin/env ruby
+require 'optparse'
+require 'optparse/subcommand'
+require 'tty-command'
+require 'tty-prompt'
+
+require 'cl/magic/common/common_options.rb'
+require 'cl/magic/common/logging.rb'
+require 'cl/magic/common/gcloud.rb'
+require 'cl/magic/common/kubectl.rb'
+
+@logger = get_logger()
+@cl_cmd_name = File.basename(__FILE__).split('-').join(' ')
+
+#
+# Features
+#
+
+def do_work(options)
+  write_history("""#{@cl_cmd_name}""")
+  @logger.success "Hello World"
+end
+
+#
+# Options
+#
+
+options = {}
+global_banner = <<DOC
+
+ A sandbox to try things
+
+ Usage: #{@cl_cmd_name} [options]
+
+DOC
+
+global = OptionParser.new do |g|
+  g.banner = global_banner
+  add_help_and_verbose(g)
+end
+
+#
+# Run
+#
+
+@working_dir = ENV['CL_WORKING_DIR'] # passed through cl-magic to here
+global.parse(ARGV)
+
+do_work(options)

data/lib/cl/magic/cl-vault

@@ -0,0 +1,84 @@
+#!/usr/bin/env ruby
+require 'optparse'
+require 'optparse/subcommand'
+require 'tty-command'
+require 'tty-prompt'
+
+require 'cl/magic/common/common_options.rb'
+require 'cl/magic/common/logging.rb'
+require 'cl/magic/common/gcloud.rb'
+require 'cl/magic/common/kubectl.rb'
+
+@logger = get_logger()
+@cl_cmd_name = File.basename(__FILE__).split('-').join(' ')
+
+#
+# Features
+#
+
+def transverse_vault_path(path)
+
+  # list the folder
+  cmd = "vault list #{path}"
+  results = parse_table_results(cmd, "no secrets", num_headers=2)
+  selected_path = pick_single_result(results, "").first
+  new_path = File.join(path, selected_path)
+
+  if new_path.end_with? '/'
+    transverse_vault_path(new_path) # another folder, recurse
+  else
+    return new_path # picked a file
+  end
+end
+
+def print_secret(secret_path)
+
+  write_history("""#{@cl_cmd_name} \\
+  --secret=#{secret_path}
+""")
+
+  cmd = "vault read #{secret_path}"
+  puts TTY::Command.new(:printer => :null).run(cmd).out
+end
+
+#
+# Options
+#
+
+options = {}
+global_banner = <<DOC
+
+ Browse secrets path and print a secret
+
+ Usage: #{@cl_cmd_name} [options]
+
+DOC
+
+global = OptionParser.new do |g|
+  g.banner = global_banner
+  add_help_and_verbose(g)
+
+  g.on("-p", "--path PATH", "a vault path to start from") do |v|
+    options[:vault_path] = v
+  end
+
+  g.on("-s", "--secret SECRET_PATH", "an exact secret to get") do |v|
+    options[:exact_secret_path] = v
+  end
+end
+
+#
+# Run
+#
+
+global.parse(ARGV)
+
+if options[:vault_path].nil?
+  options[:vault_path] = "secret"
+end
+
+if options[:exact_secret_path].nil?
+  options[:exact_secret_path] = transverse_vault_path(options[:vault_path])
+end
+
+print_secret(options[:exact_secret_path])

data/lib/cl/magic/common/common_options.rb

@@ -0,0 +1,16 @@
+
+def add_help_and_verbose(opts)
+  add_verbose(opts)
+  add_help(opts)
+end
+
+def add_verbose(opts)
+  opts.on("-v", "--[no-]verbose")
+end
+
+def add_help(opts)
+  opts.on("-h", "--help") do
+    puts opts
+    exit
+  end
+end

data/lib/cl/magic/common/gcloud.rb

@@ -0,0 +1,71 @@
+require 'json'
+require 'tty-command'
+require 'tty-prompt'
+require 'cl/magic/common/logging.rb'
+require 'cl/magic/common/parse_and_pick.rb'
+
+#
+# gcloud GET
+#
+
+def get_gcloud_projects(filter=nil)
+  filter_option = "--filter #{filter}" if filter
+  command = "gcloud projects list #{filter_option}"
+  error_message = "No gcloud projects?  Are you authenticated?"
+  return parse_table_results(command, error_message)
+end
+
+def get_current_gcloud_project()
+  command = "gcloud config get-value project"
+  return TTY::Command.new(:printer => :null).run(command).out.strip.chomp
+end
+
+def get_sql_instances()
+  command = "gcloud sql instances list"
+  return parse_table_results(command, 'No sql instances. :(')
+end
+
+def get_sql_users(instance)
+  command = "gcloud sql users list --instance=#{instance.first}"
+  return parse_table_results(command, 'No sql users. :(')
+end
+
+def get_sql_databases(instance)
+  command = "gcloud sql databases list --instance=#{instance.first}"
+  return parse_table_results(command, 'No sql databases. :(')
+end
+
+#
+# Misc
+#
+
+def pick_gcloud_project(filter=nil)
+
+  prev_project = get_current_gcloud_project()
+
+  # pick content
+  projects = get_gcloud_projects(filter)
+  selected_project = pick_single_result(projects, 'Pick a project', nil, prev_project)
+
+  # change context
+  if selected_project.first != prev_project
+    command = "gcloud config set project #{selected_project.first}"
+    @logger.wait command
+    TTY::Command.new(:printer => :null).run(command)
+  else
+    @logger.info "project already set to #{selected_project.first}"
+  end
+
+  return selected_project
+end
+
+def check_gcloud_project(project)
+  command = "gcloud info"
+  result = TTY::Command.new(:printer => :null).run(command)
+
+  current_project = result.select {|line| line.include? 'Project'}.first.split(': ').last
+  if current_project != "[#{current_project}]"
+    @logger.warn "Incorrect gcloud project: #{current_project}"
+    @logger.warn "Use `gcloud config set project #{project}` and try again."
+  end
+end