lib/sdr_client/redesigned_client/cli.rb
# frozen_string_literal: true
require 'byebug'
require 'launchy'
require 'thor'
require_relative 'cli/config'
require_relative 'cli/credentials'
require_relative 'cli/update'
module SdrClient
class RedesignedClient
# The SDR command-line interface
class CLI < Thor
include Thor::Actions
# Make sure Thor commands preserve exit statuses
# @see https://github.com/rails/thor/wiki/Making-An-Executable
def self.exit_on_failure?
true
end
# Print out help and exit with error code if command not found
def self.handle_no_command_error(command)
puts "Command '#{command}' not found, displaying help:"
puts
puts help
exit(1)
end
def self.default_url
'https://sdr-api-prod.stanford.edu'
end
package_name 'sdr'
class_option :url, desc: 'URL of SDR API endpoint', type: :string, default: default_url
desc 'get DRUID', 'Retrieve an object from the SDR'
def get(druid)
say client.find(object_id: druid)
end
desc 'version', 'Display the SDR CLI version'
def version
say VERSION
end
desc 'update DRUID', 'Update an object in the SDR'
option :skip_polling, type: :boolean, default: false, aliases: '-s',
desc: 'Print out job ID instead of polling for result'
option :apo, desc: 'Druid identifier of the admin policy object', aliases: '--admin-policy'
option :collection, desc: 'Druid identifier of the collection object'
option :copyright, desc: 'Copyright statement'
option :use_and_reproduction, desc: 'Use and reproduction statement'
option :license, desc: 'License URI'
option :view, enum: %w[world stanford location-based citation-only dark], desc: 'Access view level for the object'
option :download, enum: %w[world stanford location-based none], desc: 'Access download level for the object'
option :location, enum: %w[spec music ars art hoover m&m], desc: 'Access location for the object'
option :cdl, type: :boolean, default: false, desc: 'Controlled digital lending'
option :cocina_file, desc: 'Path to a file containing Cocina JSON'
option :cocina_pipe, type: :boolean, default: false, desc: 'Indicate Cocina JSON is being piped in'
option :basepath, default: Dir.getwd, desc: 'Base path for the files'
def update(druid)
validate_druid!(druid)
# Make sure client is configured
client
job_id = Update.run(druid, **options)
if options[:skip_polling]
say "job ID #{job_id} queued (not polling because `-s` flag was supplied)"
return
end
# the extra args to `say` prevent appending a newline
say('SDR is processing your request.', nil, false)
job_status = client.job_status(job_id: job_id)
job_status.wait_until_complete { say('.', nil, false) }
if job_status.complete?
if job_status.errors
say_error " errored! #{job_status.errors}"
else
say " success! (druid: #{job_status.druid})"
end
else
say_error " job #{job_id} did not complete\n#{job_status.result.inspect}"
end
end
desc 'deposit OPTIONAL_FILES', 'Deposit (accession) an object into the SDR'
option :skip_polling, type: :boolean, default: false, aliases: '-s',
desc: 'Print out job ID instead of polling for result'
option :apo, required: true, desc: 'Druid identifier of the admin policy object', aliases: '--admin-policy'
option :source_id, required: true, desc: 'Source ID for this object'
option :label, desc: 'Object label'
option :type, enum: %w[image book document map manuscript media three_dimensional object collection admin_policy],
desc: 'The object type'
option :collection, desc: 'Druid identifier of the collection object'
option :catkey, desc: 'Symphony catkey for this item'
option :folio_instance_hrid, desc: 'Folio instance HRID for this item'
option :copyright, desc: 'Copyright statement'
option :use_and_reproduction, desc: 'Use and reproduction statement'
option :viewing_direction, enum: %w[left-to-right right-to-left], desc: 'Viewing direction (if a book)'
option :view, enum: %w[world stanford location-based citation-only dark], desc: 'Access view level for the object'
option :download, enum: %w[world stanford location-based none], desc: 'Access download level for the object'
option :location, enum: %w[spec music ars art hoover m&m], desc: 'Access location for the object'
option :files_metadata, desc: 'JSON string representing per-file metadata'
option :grouping_strategy, enum: %w[default filename], desc: 'Strategy for grouping files into filesets'
option :basepath, default: Dir.getwd, desc: 'Base path for the files'
def deposit(*files)
register_or_deposit(files: files, accession: true)
end
desc 'register OPTIONAL_FILES', 'Create a draft object in the SDR and retrieve a Druid identifier'
option :skip_polling, type: :boolean, default: false, aliases: '-s',
desc: 'Print out job ID instead of polling for result'
option :apo, required: true, desc: 'Druid identifier of the admin policy object', aliases: '--admin-policy'
option :source_id, required: true, desc: 'Source ID for this object'
option :label, desc: 'Object label'
option :type, enum: %w[image book document map manuscript media three_dimensional object collection admin_policy],
desc: 'The object type'
option :collection, desc: 'Druid identifier of the collection object'
option :catkey, desc: 'Symphony catkey for this item'
option :folio_instance_hrid, desc: 'Folio instance HRID for this item'
option :copyright, desc: 'Copyright statement'
option :use_and_reproduction, desc: 'Use and reproduction statement'
option :viewing_direction, enum: %w[left-to-right right-to-left], desc: 'Viewing direction (if a book)'
option :view, enum: %w[world stanford location-based citation-only dark], desc: 'Access view level for the object'
option :download, enum: %w[world stanford location-based none], desc: 'Access download level for the object'
option :location, enum: %w[spec music ars art hoover m&m], desc: 'Access location for the object'
option :files_metadata, desc: 'JSON string representing per-file metadata'
option :grouping_strategy, enum: %w[default filename], desc: 'Strategy for grouping files into filesets'
option :basepath, default: Dir.getwd, desc: 'Base path for the files'
def register(*files)
register_or_deposit(files: files, accession: false)
end
private
def client
SdrClient::RedesignedClient.configure(
url: options[:url],
token: Credentials.read || SdrClient::RedesignedClient.default_token,
token_refresher: -> { login_via_proxy }
)
end
def login_via_proxy
say 'Opened the configured authentication proxy in your browser. ' \
'Once there, generate a new token and copy the entire value.'
Launchy.open(authentication_proxy_url)
# Some CLI environments will pop up a message about opening the URL in
# an existing browse. Since this is OS-dependency, and not something
# we can control via Launchy, just wait a bit before rendering the
# `ask` prompt so it's clearer to the user what's happening
sleep 0.5
token_string = ask('Paste token here:')
expiry = JSON.parse(token_string)['exp']
Credentials.write(token_string)
say "You are now authenticated for #{options[:url]} until #{expiry}"
token_string
end
def authentication_proxy_url
Settings.authentication_proxy_url[options[:url]]
end
def register_or_deposit(files:, accession:)
opts = munge_options(options, files, accession)
job_id = client.build_and_deposit(apo: options[:apo], source_id: options[:source_id], **opts)
if opts.delete(:skip_polling)
say "job ID #{job_id} queued (not polling because `-s` flag was supplied)"
return
end
job_status = client.job_status(job_id: job_id)
if job_status.wait_until_complete
say "success! (druid: #{job_status.druid})"
else
say_error "errored! #{job_status.errors}"
end
end
def munge_options(options, files, accession)
options.to_h.symbolize_keys.tap do |opts|
opts[:access] = accession
opts[:type] = Cocina::Models::ObjectType.public_send(options[:type]) if options[:type]
opts[:files] = expand_files(files) if files.present?
opts[:files_metadata] = JSON.parse(options[:files_metadata]) if options[:files_metadata]
opts.delete(:apo)
opts.delete(:source_id)
end
end
def expand_files(files)
files.flat_map do |file|
next file unless Dir.exist?(file)
Dir.glob("#{file}/**/*").select { |f| File.file?(f) }
end
end
def validate_druid!(druid)
return if druid.present?
say_error "Not a valid druid: #{druid.inspect}"
exit(1)
end
end
end
end