"arv tag remove --all\n"
end
-def tag_add(tag, obj_uuid)
- request_body = {}
+def api_call(method, parameters:{}, request_body:{})
request_body[:api_token] = ENV['ARVADOS_API_TOKEN']
+ result = $client.execute(:api_method => method,
+ :parameters => parameters,
+ :body => request_body,
+ :authenticated => false)
+
+ begin
+ results = JSON.parse result.body
+ rescue JSON::ParserError => e
+ abort "Failed to parse server response:\n" + e.to_s
+ end
- return $client.execute(:api_method => $arvados.links.create,
- :parameters => {
- :name => tag,
- :link_class => :tag,
- :head_uuid => obj_uuid,
- },
- :body => request_body,
- :authenticated => false)
+ if results["errors"]
+ abort "Error: #{results["errors"][0]}"
+ end
+
+ return results
end
-def tag_remove(tag, obj_uuid=nil)
- request_body = {}
- request_body[:api_token] = ENV['ARVADOS_API_TOKEN']
+def tag_add(tag, obj_uuid)
+ return api_call($arvados.links.create,
+ request_body: {
+ :link => {
+ :name => tag,
+ :link_class => :tag,
+ :head_uuid => obj_uuid,
+ }
+ })
+end
+
+def tag_remove(tag, obj_uuids=nil)
+ # If we got a list of objects to untag, look up the uuids for the
+ # links that need to be deleted.
+ link_uuids = []
+ if obj_uuids
+ obj_uuids.each do |uuid|
+ link = api_call($arvados.links.list,
+ request_body: {
+ :where => {
+ :link_class => :tag,
+ :name => tag,
+ :head_uuid => uuid,
+ }
+ })
+ if link['items_available'] > 0
+ link_uuids.push link['items'][0]['uuid']
+ end
+ end
+ else
+ all_tag_links = api_call($arvados.links.list,
+ request_body: {
+ :where => {
+ :link_class => :tag,
+ :name => tag,
+ }
+ })
+ link_uuids = all_tag_links['items'].map { |obj| obj['uuid'] }
+ end
- params = { :name => tag, :link_class => :tag }
- if obj_uuid then
- params[:head_uuid] = obj_uuid
+ results = []
+ if link_uuids
+ link_uuids.each do |uuid|
+ results.push api_call($arvados.links.delete, parameters:{ :uuid => uuid })
+ end
+ else
+ $stderr.puts "no tags found to remove"
end
- return $client.execute(:api_method => $arvados.links.destroy,
- :parameters => params,
- :body => request_body,
- :authenticated => false)
+ return results
end
if RUBY_VERSION < '1.9.3' then
require 'google/api_client'
require 'json'
require 'pp'
+ require 'oj'
require 'trollop'
rescue LoadError
abort <<-EOS
end
end
+global_opts = Trollop::options do
+ banner "arvados cli client"
+ opt :dry_run, "Don't actually do anything", :short => "-n"
+ opt :verbose, "Print some things on stderr", :short => "-v"
+ opt :uuid, "Return the UUIDs of the objects in the response, one per line (default)", :short => nil
+ opt :json, "Return the entire response received from the API server, as a JSON object", :short => "-j"
+ opt :human, "Return the response received from the API server, as a JSON object with whitespace added for human consumption", :short => "-h"
+ opt :pretty, "Synonym of --human", :short => nil
+ opt :yaml, "Return the response received from the API server, in YAML format", :short => "-y"
+ stop_on ['add', 'remove']
+end
+
p = Trollop::Parser.new do
opt(:all,
"Remove this tag from all objects under your ownership. Only valid with `tag remove'.",
results = []
cmd = ARGV.shift
+
+if ARGV.empty?
+ usage
+end
+
case cmd
when 'add'
ARGV.each do |tag|
when 'remove'
ARGV.each do |tag|
if $options[:all] then
- results.push(tag_remove(tag))
+ results.concat tag_remove(tag)
else
- $options[:object].each do |obj|
- results.push(tag_remove(tag, obj))
- end
+ results.concat tag_remove(tag, $options[:object])
end
end
else
puts results.to_yaml
elsif global_opts[:json] then
puts Oj.dump(results)
-elsif results["items"] and results["kind"].match /list$/i
- results['items'].each do |i| puts i['uuid'] end
-elsif results['uuid'].nil?
- abort("Response did not include a uuid:\n" +
- Oj.dump(results, :indent => 1) +
- "\n")
else
- results.each do |result|
- puts result['uuid']
+ results.each do |r|
+ if r['uuid'].nil?
+ abort("Response did not include a uuid:\n" +
+ Oj.dump(r, :indent => 1) +
+ "\n")
+ else
+ puts r['uuid']
+ end
end
end