ubalo 0.1 → 0.2

data/lib/ubalo.rb

@@ -4,326 +4,14 @@ require 'rest-client'
 require 'json'
 require 'pp'
 
-require 'zlib'
-require 'archive/tar/minitar'
-include Archive::Tar
-
-def indent str, amount=2
-  if str
-    str.each_line.map do |l|
-      " "*amount + l
-    end.join
-  else
-    ""
-  end
-end
-
-class UbaloOKExit < StandardError
-
-end
-
-class UbaloMessage < StandardError
-
-end
-
-class Ubalo
-  class UbaloJSON
-    def self.content_type
-      "application/json; schema=ubalo-data"
-    end
-
-    def self.dump(obj)
-      JSON.dump('data' => obj)
-    end
-
-    def self.load(data)
-      JSON.load(data)['data']
-    end
-  end
-
-  class PodError < StandardError
-  end
-
-  class << self
-    def make_tgz file_list, tgz_name
-      tgz = Zlib::GzipWriter.new(File.open(tgz_name, 'wb'))
-      Minitar.pack(file_list, tgz)
-      tgz_name
-    end
-
-    def extract_tgz destination_dir, tgz_name
-      Dir.chdir(destination_dir) do
-        tgz = Zlib::GzipReader.new(File.open(tgz_name, 'rb'))
-        Minitar.unpack(tgz, '.')
-      end
-    end
-
-    def print_content name, content
-      data = content.fetch('data')
-      content_type = content.fetch('content_type')
-      if content_type == UbaloJSON.content_type
-        s = UbaloJSON.load(data).pretty_inspect.strip
-        if s.include?("\n")
-          puts "#{name}:"
-          puts s
-        else
-          puts "#{name}: #{s}"
-        end
-      else
-        puts "#{name}: (#{content_type})"
-        puts indent(data)
-      end
-    end
-
-    def format_task task
-      "#{task['label']} (#{task['state']}) - #{task['pod_name']} (#{time_ago_in_words task['submitted_at']})"
-    end
-
-    def format_pod pod
-      "#{pod.fetch('fullname')} (updated #{time_ago_in_words pod.fetch('updated_at')})"
-    end
-
-    private
-    def pluralize count, word
-      if count == 1
-        "#{count} #{word}"
-      else
-        "#{count} #{word}s"
-      end
-    end
-
-    def time_ago_in_words time
-      seconds = Time.now - Time.parse(time)
-      case
-      when seconds < 60
-        "#{pluralize(seconds.floor, "second")} ago"
-      when seconds/60 < 60
-        "#{pluralize((seconds/60).floor, "minute")} ago"
-      when seconds/60/60 < 24
-        "#{pluralize((seconds/60/60).floor, "hour")} ago"
-      else
-        "#{pluralize((seconds/60/60/24).floor, "day")} ago"
-      end
-    end
-  end
-
-  attr_reader :base_url
-  attr_accessor :authorization
-
-  def self.login authorization, base_url
-    new(authorization, base_url)
-  end
-
-  def initialize authorization, base_url
-    @authorization = authorization
-    @base_url = base_url
-  end
-
-  def raw_request(method, url, params)
-    resource = RestClient::Resource.new url, timeout: 60
-    if method == :get
-      resource.get add_headers(:params => params, :accept => :json)
-    elsif method == :post
-      resource.post params, add_headers(:accept => :json)
-    elsif method == :put
-      resource.put params, add_headers(:accept => :json)
-    elsif method == :delete
-      resource.delete add_headers(:accept => :json)
-    else
-      raise "don't understand request method #{method.inspect}"
-    end
-  end
-
-  def request(method, url, params)
-    parse(raw_request(method, url, params))
-  end
-
-  def post(url, params={})
-    request(:post, url, params)
-  end
-
-  def get(url, params={})
-    request(:get, url, params)
-  end
-
-  def put(url, params={})
-    raw_request(:put, url, params)
-  end
-
-  def delete(url)
-    raw_request(:delete, url, {})
-  end
-
-  def whoami
-    response = get("#{base_url}/user")
-    "You are logged in as #{response['username'].inspect}."
-  end
-
-  def tasks count
-    get("#{base_url}/tasks", :count => count)
-  end
-
-  def submit_task pod_name, arg
-    params = {
-      :arg_content_type => UbaloJSON.content_type,
-      :arg => UbaloJSON.dump(arg)
-    }
-    post("#{base_url}/pods/#{pod_name}/tasks", params)
-  end
-
-  def poll_on(message)
-    600.times do
-      if result = yield
-        $stderr.puts " done."
-        return result
-      end
-      $stderr.print '.'
-      $stderr.flush
-      sleep 0.5
-    end
-
-    $stderr.puts
-    raise "timed-out polling on #{message}"
-  end
-
-  def wait_task(label)
-    poll_on("task #{label}") do
-      task = check_task(label)
-      task if task['state'] == 'exited'
-    end
-  end
-
-  def clone(pod_name)
-    get("#{base_url}/pods/#{pod_name}")
-  rescue RestClient::ResourceNotFound
-    raise UbaloMessage, "could not find #{pod_name}"
-  end
-
-  def pods
-    get("#{base_url}/pods")
-  end
-
-  def show_result(result)
-    puts " label: #{result['label']}"
-    puts "   pod: #{result['pod_name']}"
-    if arg = result['arg']
-      Ubalo.print_content('   arg', arg)
-    end
-
-    puts " state: #{result['state']}"
-
-    if container_process = result['container_process']
-      puts "status: #{container_process['exit_result']}"
-
-      if stdout = container_process['stdout'] and stdout.length > 0
-        puts "stdout:"
-        puts indent(stdout)
-      end
-
-      if stderr = container_process['stderr'] and stderr.length > 0
-        puts "stderr:"
-        puts indent(stderr)
-      end
-    end
-
-    if outputs = result['outputs']
-      Ubalo.print_content('output', outputs)
-    end
-  end
-
-  def get_authorization login, password
-    post("#{base_url}/user/authorization", :user => {:login => login, :password => password})
-  end
-
-  def create_or_update pod_url, params={}
-    put(pod_url, :pod => params)
-  end
-
-  def check_task label
-    get("#{base_url}/tasks/#{label}")
-  end
-
-  def upload_tgz pod_url, tgz_name
-    # Create the new archive in rails.
-    response = post("#{pod_url}/archives")
-    label = response.fetch('label')
-    put_url = response.fetch('put_url')
-    archive_location = response.fetch('url')
-
-    print "Uploading files... "
-    RestClient.put put_url, File.open(tgz_name, 'rb'), content_type: 'application/x-tar', content_encoding: 'x-gzip'
-    put("#{archive_location}/activate")
-
-    puts "done."
-  end
-
-  def push_files(pod_url)
-    ignore_filename = ".ubaloignore"
-    if File.exist?(ignore_filename)
-      ignore_patterns = File.read(ignore_filename).each_line.map(&:strip)
-    else
-      ignore_patterns = []
-    end
-
-    ignore_patterns += %w{. .. .ubalo}
-
-    filenames = []
-    Dir.glob('{.**,**}').each do |path|
-      next if ignore_patterns.any?{|pattern| File.fnmatch?(pattern, path, File::FNM_DOTMATCH)}
-
-      filenames << path
-    end
-
-    if filenames.empty?
-      raise UbaloMessage, "You do not have files to push. Cancelling!"
-    end
-
-    Dir.mktmpdir do |dir|
-      tar_path = File.join(dir, "files.tar.gz")
-      Ubalo.make_tgz(filenames, tar_path)
-      upload_tgz(pod_url, tar_path)
-    end
-
-    print "Waiting on pod compilation.."
-    poll_on("compilation of pod at #{pod_url}") do
-      pod = get(pod_url)
-      pod if pod['state'] == 'compiled'
-    end
-  end
-
-  def retrieve_files(destination_path, files_url)
-    Dir.mktmpdir do |dir|
-      tar_path = File.join(dir, "files.tar.gz")
-      RestClient.get(files_url) do |response|
-        if response.code == 200
-          File.open(tar_path, "wb") do |f|
-            f.write response
-          end
-          Ubalo.extract_tgz(destination_path, tar_path)
-        else
-          raise UbaloMessage, "failed to download pod files"
-        end
-      end
-    end
-  end
-
-  def delete_pod pod_name
-    delete("#{base_url}/pods/#{pod_name}")
-  end
-
-  private
-
-  def add_headers headers
-    # Add authorization headers.
-    if authorization
-      headers.merge!(:authorization => authorization)
-    end
-
-    # Add version header.
-    headers.merge!('X-Ubalo-Version' => Ubalo::VERSION)
-  end
-
-  def parse(result)
-    JSON.load(result)
+require "ubalo/util"
+require "ubalo/account"
+require "ubalo/pod"
+require "ubalo/pod_archive"
+require "ubalo/task"
+require "ubalo/pod_dir"
+
+module Ubalo
+  class Error < StandardError
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ubalo
 version: !ruby/object:Gem::Version
-  version: '0.1'
+  version: '0.2'
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-03-30 00:00:00.000000000 Z
+date: 2012-04-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: gli
-  requirement: &70153937178120 !ruby/object:Gem::Requirement
+  requirement: &70111153539360 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70153937178120
+  version_requirements: *70111153539360
 - !ruby/object:Gem::Dependency
   name: highline
-  requirement: &70153937175760 !ruby/object:Gem::Requirement
+  requirement: &70111153538720 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,10 +32,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70153937175760
+  version_requirements: *70111153538720
 - !ruby/object:Gem::Dependency
   name: json
-  requirement: &70153937174580 !ruby/object:Gem::Requirement
+  requirement: &70111153538200 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -43,10 +43,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70153937174580
+  version_requirements: *70111153538200
 - !ruby/object:Gem::Dependency
   name: rest-client
-  requirement: &70153937172960 !ruby/object:Gem::Requirement
+  requirement: &70111153537640 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -54,10 +54,10 @@ dependencies:
         version: 1.6.3
   type: :runtime
   prerelease: false
-  version_requirements: *70153937172960
+  version_requirements: *70111153537640
 - !ruby/object:Gem::Dependency
   name: archive-tar-minitar
-  requirement: &70153937171020 !ruby/object:Gem::Requirement
+  requirement: &70111153536860 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -65,7 +65,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70153937171020
+  version_requirements: *70111153536860
 description: CLI and API client for Ubalo
 email: dev@ubalo.com
 executables:
@@ -73,6 +73,12 @@ executables:
 extensions: []
 extra_rdoc_files: []
 files:
+- lib/ubalo/account.rb
+- lib/ubalo/pod.rb
+- lib/ubalo/pod_archive.rb
+- lib/ubalo/pod_dir.rb
+- lib/ubalo/task.rb
+- lib/ubalo/util.rb
 - lib/ubalo/version.rb
 - lib/ubalo.rb
 - bin/ubalo
@@ -88,12 +94,18 @@ required_ruby_version: !ruby/object:Gem::Requirement
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
+      segments:
+      - 0
+      hash: -3363828113033122894
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
+      segments:
+      - 0
+      hash: -3363828113033122894
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.11
@@ -101,4 +113,3 @@ signing_key:
 specification_version: 3
 summary: CLI and API client for Ubalo
 test_files: []
-has_rdoc: