Merge pull request #1 from curoverse/master

[arvados.git] / sdk / cli / bin / arv

diff --git a/sdk/cli/bin/arv b/sdk/cli/bin/arv
index 0ecfb6514773189849428b0900be980b9c216a7e..36ec037bd80702b27137cb07824ca21cda641d99 100755 (executable)
--- a/sdk/cli/bin/arv
+++ b/sdk/cli/bin/arv
@@ -15,7 +15,7 @@ end
 begin
   require 'curb'
   require 'rubygems'
-  require 'google/api_client'
+  require 'arvados/google_api_client'
   require 'json'
   require 'pp'
   require 'trollop'
@@ -24,6 +24,7 @@ begin
   require 'active_support/inflector'
   require 'yaml'
   require 'tempfile'
+  require 'net/http'
 rescue LoadError
   abort <<-EOS
 
@@ -51,36 +52,6 @@ module Kernel
   end
 end
 
-class Google::APIClient
- def discovery_document(api, version)
-   api = api.to_s
-   discovery_uri = self.discovery_uri(api, version)
-   discovery_uri_hash = Digest::MD5.hexdigest(discovery_uri)
-   return @discovery_documents[discovery_uri_hash] ||=
-     begin
-       # fetch new API discovery doc if stale
-       cached_doc = File.expand_path "~/.cache/arvados/discovery-#{discovery_uri_hash}.json" rescue nil
-
-       if cached_doc.nil? or not File.exist?(cached_doc) or (Time.now - File.mtime(cached_doc)) > 86400
-         response = self.execute!(:http_method => :get,
-                                  :uri => discovery_uri,
-                                  :authenticated => false)
-
-         begin
-           FileUtils.makedirs(File.dirname cached_doc)
-           File.open(cached_doc, 'w') do |f|
-             f.puts response.body
-           end
-         rescue
-           return JSON.load response.body
-         end
-       end
-
-       File.open(cached_doc) { |f| JSON.load f }
-     end
- end
-end
-
 class ArvadosClient < Google::APIClient
   def execute(*args)
     if args.last.is_a? Hash
@@ -220,10 +191,10 @@ def edit_and_commit_object initial_obj, tmp_stem, global_opts, &block
         can_retry = true
         if e.is_a? Psych::SyntaxError
           this_error = "YAML error parsing your input: #{e}"
-        elsif e.is_a? JSON::ParserError
+        elsif e.is_a? JSON::ParserError or e.is_a? Oj::ParseError
           this_error = "JSON error parsing your input: #{e}"
-        elsif e.is_a? HttpResponse
-          this_error = "API responded with error #{e.status}"
+        elsif e.is_a? ArvadosAPIError
+          this_error = "API responded with error #{e}"
         else
           this_error = "#{e.class}: #{e}"
           can_retry = false
@@ -256,22 +227,20 @@ def edit_and_commit_object initial_obj, tmp_stem, global_opts, &block
   nil
 end
 
-class HttpResponse < RuntimeError
-  attr_reader :status
-  def initialize status
-    @status = status
-  end
+class ArvadosAPIError < RuntimeError
 end
 
 def check_response result
   begin
     results = JSON.parse result.body
-  rescue JSON::ParserError => e
+  rescue JSON::ParserError, Oj::ParseError => e
     raise "Failed to parse server response:\n" + e.to_s
   end
 
   if result.response.status != 200
-    raise HttpResponse.new "#{result.response.status}: " + (results['errors'] && results['errors'].join('\n') || "")
+    raise ArvadosAPIError.new("#{result.response.status}: #{
+                              ((results['errors'] && results['errors'].join('\n')) ||
+                                Net::HTTPResponse::CODE_TO_OBJ[status.to_s].to_s.sub(/^Net::HTTP/, '').titleize)}")
   end
 
   results
@@ -344,8 +313,8 @@ def arv_edit client, arvados, global_opts, remaining_opts
                      :headers => {
                        authorization: 'OAuth2 '+ENV['ARVADOS_API_TOKEN']
                      })
-      check_response result
-      puts "Updated object #{uuid}"
+      results = check_response result
+      puts "Updated object #{results['uuid']}"
     else
       puts "Object is unchanged, did not update."
     end