ubalo 0.0.1 → 0.0.2

data/bin/ubalo

@@ -0,0 +1,332 @@
+#!/usr/bin/env ruby
+
+require 'gli'
+require 'yaml'
+
+require 'highline'
+require 'highline/import'
+hl = HighLine.new
+
+require 'ubalo'
+include GLI
+
+use_openstruct true
+
+class Ubalo
+  class << self
+    def config
+      if @config
+        @config
+      elsif File.exists?(config_path)
+        @config = YAML.load_file(config_path)
+      else
+        {}
+      end
+    end
+
+    def write_config new_config
+      h = config.merge(new_config)
+      File.open(config_path, 'w') do |f|
+        YAML.dump(h, f)
+      end
+      config
+    end
+
+    private
+    def config_path
+      File.expand_path("~/.ubalorc")
+    end
+  end
+end
+
+def ubalo
+  @ubalo
+end
+
+program_desc 'Command-line access to ubalo.com.'
+version "0.0.1"
+
+desc "Change the connect url"
+flag 'connect-url'
+
+desc 'Display authentication information'
+command :whoami do |c|
+  c.action do |global_options,options,args|
+    puts ubalo.whoami
+  end
+end
+
+desc 'Display pods'
+command :pods do |c|
+  c.desc 'show all pods you can snap (default)'
+  c.switch :all
+
+  c.desc 'show only your pods'
+  c.switch :mine
+
+  c.desc 'show only the pods you can edit and snap'
+  c.switch :editable
+
+  c.action do |global_options,options,args|
+    unless options.all or options.edit_only
+      # Default to showing just your pods.
+      options.mine = true
+    end
+
+    your_pods, editable_pods, snappable_pods = ubalo.pods
+
+    # Always show your pods.
+    your_pods.each do |pod|
+      puts Ubalo.format_pod(pod)
+    end
+
+    if options.all or options.edit_only
+      editable_pods.each do |pod|
+        puts Ubalo.format_pod(pod)
+      end
+    end
+
+    if options.all
+      snappable_pods.each do |pod|
+        puts Ubalo.format_pod(pod)
+      end
+    end
+  end
+end
+
+desc 'Display tasks'
+command :tasks do |c|
+  c.desc 'filter by task state (pending, complete, etc)'
+  c.long_desc 'possibilities are waiting,running,pending,complete,failed,all'
+  c.default_value 'all'
+  c.flag 'state'
+
+  c.desc 'maximum number of tasks to display'
+  c.default_value 20
+  c.flag 'max'
+
+  c.action do |global_options,options,args|
+    unless options.pending or options.failed
+      # Default to showing just your pods.
+      options.all = true
+    end
+
+    ubalo.tasks(options.state, options.max).each do |task|
+      puts Ubalo.format_task(task)
+    end
+  end
+end
+
+desc 'Add an ssh key (defaults to ~/.ssh/id_rsa.pub)'
+arg_name '<file containing ssh key>'
+command :add_key do |c|
+  c.action do |global_options,options,args|
+
+    public_key_file, = *args
+
+    public_key_file ||= '~/.ssh/id_rsa.pub'
+    public_key_file = File.expand_path(public_key_file)
+
+    puts ubalo.upload_key(File.read(public_key_file))
+  end
+end
+
+desc 'Clear all ssh keys'
+command :clear_keys do |c|
+  c.action do |global_options,options,args|
+    puts ubalo.clear_keys
+  end
+end
+
+desc 'Modify a pod'
+arg_name '<pod name>'
+command :edit do |c|
+  c.action do |global_options,options,args|
+    pod_name = args.shift
+    unless pod_name
+      raise "please specify a pod"
+    end
+
+    ubalo.ssh(pod_name)
+  end
+end
+
+desc 'Show the code in a pod'
+arg_name '<pod name>'
+command :cat do |c|
+  c.action do |global_options,options,args|
+    pod_name = args.shift
+    unless pod_name
+      raise "please specify a pod"
+    end
+
+    puts ubalo.cat_code(pod_name)
+  end
+end
+
+desc 'Set the code in a pod'
+arg_name '<pod_name> <code_file>'
+command :code do |c|
+  c.action do |global_options,options,args|
+    pod_name = args.shift
+    unless pod_name
+      raise "please specify a pod"
+    end
+
+    file_name = args.shift
+    if file_name
+      code = File.read(file_name)
+    elsif $stdin.tty?
+      raise "please give the new code as a filename or on stdin"
+    else
+      code = $stdin.read
+    end
+
+    puts ubalo.upload_code(pod_name, code)
+  end
+end
+
+desc 'Show the files in a pod environ'
+arg_name '<pod name>'
+command :ls do |c|
+  c.desc 'list in long format'
+  c.switch :l
+
+  c.desc 'list in human-readable form'
+  c.switch :h
+
+  c.action do |global_options,options,args|
+    opts = ""
+    opts << " -l" if options.l
+    opts << " -h" if options.h
+
+    pod_name = args.shift
+    unless pod_name
+      raise "please specify a pod"
+    end
+    puts ubalo.ls(pod_name, opts)
+  end
+end
+
+desc 'Copy files into and out of a pod environ'
+arg_name '<pod name>'
+command :cp do |c|
+  c.desc 'copy recursively'
+  c.switch :r
+
+  c.action do |global_options,options,args|
+    opts = ""
+    opts << " -r" if options.r
+
+    scp_args = args.join(" ").gsub(/[\w-]+(?=:)/) do |pod_name|
+      ssh_path, environ = ubalo.ssh_path(pod_name)
+      "#{ssh_path}:#{environ}"
+    end
+
+    ubalo.scp(opts, scp_args)
+  end
+end
+
+desc 'Snap a pod'
+arg_name '<pod name>'
+command :snap do |c|
+  c.desc 'detach, leaving task running in the background; return an id'
+  c.switch [:detach, :d]
+
+  c.desc 'explicitly show the complete result'
+  c.switch [:verbose, :v]
+
+  c.desc 'outputs as json to stdout; stdout > stderr'
+  c.switch [:json, :j]
+
+  c.action do |global_options,options,args|
+    pod_name = args.shift
+    unless pod_name
+      raise "please specify a pod"
+    end
+
+    h = ubalo.submit_task(pod_name, args.join(" "))
+    task_id = h['id']
+
+    if options.detach
+      puts task_id
+    else
+      result = ubalo.check_task(task_id, true)
+      ubalo.show_result(result, options.verbose, options.json)
+    end
+  end
+end
+
+desc 'Check or get the result of a task'
+arg_name '<task id>'
+command :check do |c|
+  c.desc 'explicitly show the complete result'
+  c.switch [:verbose, :v]
+
+  c.desc 'wait for the result if the task is pending'
+  c.switch [:wait, :w]
+
+  c.desc 'outputs as json to stdout; stdout > stderr'
+  c.switch [:json, :j]
+
+  c.action do |global_options,options,args|
+    task_id = args.shift
+    unless task_id
+      raise "please specify a task"
+    end
+
+    result = ubalo.check_task(task_id, true)
+    ubalo.show_result(result, options.verbose, options.json)
+  end
+end
+
+desc 'Retrieve an API token'
+command :authorize do |c|
+  c.action do |global_options,options,args|
+    puts "Please enter your Ubalo username/email and password to unlock command-line access."
+    login    = hl.ask("     login: "){|q| q.echo = true}
+    password = hl.ask("  password: "){|q| q.echo = false}
+
+    ubalo.token = ubalo.get_api_token(login, password)
+    Ubalo.write_config('token' => ubalo.token)
+    puts "Access details saved to ~/.ubalorc."
+    puts ubalo.whoami
+  end
+end
+
+desc 'Clear any stored API tokens'
+command :deauthorize do |c|
+  c.action do |global_options,options,args|
+    Ubalo.write_config('token' => nil)
+    puts "Access details deleted."
+  end
+end
+
+pre do |global,command,options,args|
+  config = Ubalo.config
+  token = config['token']
+
+  unless token or (command && command.name == :authorize)
+    raise "No credentials found. Please run 'ubalo authorize'."
+  end
+
+  connect_url = global['connect-url'] || config['connect-url']
+  @ubalo ||= Ubalo.authorize(token, connect_url)
+
+  true
+end
+
+on_error do |exception|
+  case exception
+  when RestClient::BadRequest
+    $stderr.puts exception.inspect.sub('400 Bad Request', 'Error')
+    false
+  when UbaloExit
+    # Normal exit, preserving the correct exit code.
+    exit exception.message
+  else
+    true
+  end
+end
+
+exit GLI.run(ARGV)

data/lib/ubalo.rb

@@ -0,0 +1,210 @@
+require 'rest-client'
+require 'json'
+
+class String
+  def lfit max_length
+    if length - 3 <= max_length
+      ljust(max_length)
+    else
+      "#{self[0,max_length-3]}..."
+    end
+  end
+
+  def rfit max_length
+    if length - 3 <= max_length
+      rjust(max_length)
+    else
+      "#{self[0,max_length-3]}..."
+    end
+  end
+
+  def indent amount=2
+    each_line.map do |l|
+      " "*amount + l
+    end.join
+  end
+end
+
+class NilClass
+  def lfit max_length
+    "".lfit(max_length)
+  end
+
+  def rfit max_length
+    "".rfit(max_length)
+  end
+
+  def indent amount
+    "".indent(amount)
+  end
+end
+
+class UbaloExit < StandardError
+
+end
+
+class Ubalo
+  class << self
+    def format_task task
+      "#{task['id'].to_s[0,6]}  #{task['state'].lfit(8)}  #{task['pod_name'].lfit(25)}  #{task['arg'].to_s.lfit(40)}"
+    end
+
+    def format_pod pod
+      "#{pod['environ'].lfit(18)}  #{pod['code_lines'].to_s.rjust 3} lines  #{pod['user'].rfit(10)} / #{pod['name']}"
+    end
+  end
+
+  attr_reader :base_url
+  attr_accessor :token
+
+  def self.authorize token, base_url=nil
+    new(token, base_url)
+  end
+
+  def initialize token, base_url
+    @token = token
+    @base_url = base_url || "https://ubalo.com/api"
+  end
+
+  def get(action, params={})
+    url = "#{base_url}/#{action}"
+    if token
+      params.merge!({:auth_token => token})
+    end
+    response = RestClient.get url, :params => params
+    response
+  end
+
+  def post(action, params={})
+    url = "#{base_url}/#{action}"
+    RestClient.post url, {:auth_token => token}.merge(params)
+  end
+
+  def username
+    username, role = whoami
+    username
+  end
+
+  def whoami
+    get(:whoami)
+  end
+
+  def tasks state, count
+    parse(get(:tasks, :count => count, :state => state))
+  end
+
+  def submit_task(pod_name, arg)
+    parse(post(:submit_task, {:pod_name => pod_name, :arg => arg}))
+  end
+
+  def check_task(id, blocking=false)
+    if blocking
+      10.times do
+        h = check_task_once(id)
+        return(h) if %w{complete failed}.include?(h['state'])
+        sleep 0.5
+      end
+      raise "timed-out waiting for task"
+    else
+      check_task_once(id)
+    end
+  end
+
+  def upload_key(key)
+    post(:upload_key, :key => key)
+  end
+
+  def clear_keys
+    post(:clear_keys)
+  end
+
+  def cat_code(name)
+    get(:cat_code, :pod_name => name)
+  end
+
+  def upload_code(name, code)
+    post(:upload_code, {:pod_name => name, :code => code})
+  end
+
+  def ssh_path(name)
+    h = parse(get(:ssh_path, :pod_name => name))
+    [h['ssh_path'], h['environ'], h['environ_fullname']]
+  end
+
+  def pods
+    h = parse(get(:pods))
+    [h['your_pods'], h['editable_pods'], h['snappable_pods']]
+  end
+
+  def ssh(pod_name)
+    ssh_path, environ = ssh_path(pod_name)
+    puts "Opening an ssh connection to edit the '#{environ}' environment for the '#{pod_name}' pod."
+    ssh_command = "ssh -tq #{ssh_path} #{environ}"
+    Process.exec({'TERM' => 'xterm'}, ssh_command)
+  end
+
+  def scp(opts, args)
+    Process.exec("scp#{opts} #{args}")
+  end
+
+  def ls(pod_name, options)
+    ssh_path, environ = ssh_path(pod_name)
+    ssh_command = "ssh -tq #{ssh_path} #{environ}"
+    `#{ssh_command} ls#{options}`
+  end
+
+  def show_result(result, verbose=false, json=false)
+    if result['outputs']
+      outputs = JSON.dump(result['outputs'])
+    end
+
+    if json
+      puts outputs
+    else
+      if verbose
+        puts "    id: #{result['id']}"
+        puts "  name: #{result['pod_name']}"
+        puts " state: #{result['state']}"
+        puts "status: #{result['status']}"
+        if result['stdout']
+          puts "stdout:"
+          puts result['stdout'].indent
+        end
+        if result['stderr']
+          puts "stderr:"
+          puts result['stderr'].indent
+        end
+      else
+        if result['stdout']
+          print result['stdout']
+        end
+        if result['stderr']
+          $stderr.print result['stderr']
+        end
+
+        exit_code = Integer(result['status'] || 0)
+        unless exit_code.zero?
+          raise UbaloExit, exit_code
+        end
+      end
+
+      if outputs
+        $stderr.puts "outputs:"
+        $stderr.puts outputs.indent
+      end
+    end
+  end
+
+  def get_api_token login, password
+    parse(post(:get_api_token, :user => {:login => login, :password => password}))['api_token']
+  end
+
+  private
+  def check_task_once id
+    parse(get(:check_task, :id => id))
+  end
+
+  def parse(result)
+    JSON.load(result)
+  end
+end

metadata

@@ -1,23 +1,59 @@
 --- !ruby/object:Gem::Specification
 name: ubalo
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
-- TBD
+- Ubalo Crew
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-09-13 00:00:00.000000000Z
-dependencies: []
-description: i am waiting to upload this
-email: thief@example.org
-executables: []
+date: 2011-11-22 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: gli
+  requirement: &2153389080 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2153389080
+- !ruby/object:Gem::Dependency
+  name: highline
+  requirement: &2153388480 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2153388480
+- !ruby/object:Gem::Dependency
+  name: rest-client
+  requirement: &2153387840 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.6.3
+  type: :runtime
+  prerelease: false
+  version_requirements: *2153387840
+description: CLI and API client for Ubalo
+email: dev@ubalo.com
+executables:
+- ubalo
 extensions: []
 extra_rdoc_files: []
-files: []
-homepage: 
+files:
+- lib/ubalo.rb
+- bin/ubalo
+homepage: http://ubalo.com/
 licenses: []
 post_install_message: 
 rdoc_options: []
@@ -37,8 +73,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.7
+rubygems_version: 1.8.6
 signing_key: 
 specification_version: 3
-summary: i am waiting to upload this
+summary: CLI and API client for Ubalo
 test_files: []