cantemo-portal-agent 1.0.9

data/lib/envoi/mam/agent/cli/commands/vdms.rb

@@ -0,0 +1,41 @@
+#!/usr/bin/env ruby
+lib_path = __FILE__ == '(irb)' ? File.join(Dir.cwd, 'lib') : File.expand_path('../../../lib', __FILE__)
+$:.unshift(lib_path) unless $:.include?(lib_path) or !File.exists?(lib_path)
+require 'rubygems'
+
+require 'envoi/mam/agent/cli'
+require 'envoi/mam/vdms/agent'
+
+current_command = ARGV.shift
+ARGV << '--help' if ARGV.empty?
+
+args = {
+    :config_file_path => default_config_file_paths,
+    :dry_run => false,
+    :preserve_path => true,
+    :destination_path => File.expand_path('.'),
+}
+
+args[:config_service_app_id] = ENV['ENVOI_MAM_AGENT_APP_ID']
+args[:config_service_app_token] = ENV['ENVOI_MAM_AGENT_APP_TOKEN']
+args[:config_service_app_url] = ENV['ENVOI_MAM_AGENT_APP_URL']
+
+op = OptionParser.new
+op.on('-c', '--config-file-path FILEPATH', 'The path to the configuration file.') { |v| args[:config_file_path] = v }
+op.on('--agent-app-id ID', '') { |v| args[:config_service_app_id] = v }
+op.on('--agent-token TOKEN', '') { |v| args[:config_service_app_token] = v }
+op.on('--agent-service-api URL', '') { |v| args[:config_service_api_url] = v }
+
+op.on('-d', '--destination-path DIRPATH', 'The destination path to download or upload to.', "default: '#{args[:destination_path]}'") { |v| args[:destination_path] = v }
+op.on('-p','--[no-]preserve-path', "default: #{args[:preserve_path]}") { |v| args[:preserve_path] = v }
+
+# op.on('--dry-run', 'Run without executing the transfer.') { |v| args[:dry_run] = v }
+op.on('--help', 'Show this message.') { puts op; exit }
+op.load
+op.parse!
+
+args[:file_path] ||= ARGV.shift
+args[:destination_path] ||= ARGV.shift unless ARGV.empty?
+
+agent = Envoi::Mam::Vdms::Agent.load_config_and_init(args)
+agent.run_operation

data/lib/envoi/mam/agent/cli/commands/vidispine.rb

@@ -0,0 +1,57 @@
+#!/usr/bin/env ruby
+lib_path = __FILE__ == '(irb)' ? File.join(Dir.cwd, 'lib') : File.expand_path('../../../lib', __FILE__)
+$:.unshift(lib_path) unless $:.include?(lib_path) or !File.exists?(lib_path)
+require 'rubygems'
+require 'optparse'
+require 'open3'
+
+require 'envoi/mam/agent/cli'
+require 'envoi/mam/vidispine/agent'
+
+default_config_file_paths = Envoi::Mam::Agent::Cli::CONFIG_FILE_PATHS
+
+aspera_ascp_paths = Envoi::Mam::Agent::TransferClient::Aspera::ASCP_PATHS
+default_aspera_ascp_path = aspera_ascp_paths.find { |v| File.exist? v } || aspera_ascp_paths.first
+
+current_command = ARGV.shift
+ARGV << '--help' if ARGV.empty?
+
+args = {
+    :config_file_path  => default_config_file_paths,
+    :default_ascp_path => default_aspera_ascp_path,
+    :dry_run           => false,
+    :operation         => :download,
+    :preserve_path     => true,
+    :transfer_type     => '',
+}
+
+args[:config_service_app_id] = ENV['ENVOI_MAM_AGENT_APP_ID']
+args[:config_service_app_token] = ENV['ENVOI_MAM_AGENT_APP_TOKEN']
+args[:config_service_app_url] = ENV['ENVOI_MAM_AGENT_APP_URL']
+
+op = OptionParser.new
+op.on('-c', '--config-file-path FILEPATH', 'The path to the configuration file.', "Default Path(s): #{args[:config_file_path]}") { |v| args[:config_file_path] = v }
+op.on('--agent-app-id ID', '') { |v| args[:config_service_app_id] = v }
+op.on('--agent-token TOKEN', '') { |v| args[:config_service_app_token] = v }
+op.on('--agent-service-api URL', '') { |v| args[:config_service_api_url] = v }
+
+op.on('-i', '--item-id ITEMID', 'The Vidispine item id.') { |v| args[:item_id] = v }
+op.on('-s', '--shape-id SHAPEID', 'The Vidispine shape id.') { |v| args[:shape_id] = v }
+op.on('-t', '--shape-tag SHAPETAG', 'The Vidispine shape tag used to find the shape when a shape id is not supplied.') { |v| args[:shape_tag] = v }
+op.on('--storage-id STORAGEID', 'The Vidispine storage id to use for the transfer.') { |v| args[:storage_id] = v}
+
+op.on('-d', '--destination-path DIRPATH', 'The destination path to download or upload to.', "default: '.'") { |v| args[:destination_path] = v }
+op.on('-p','--[no-]preserve-path', "default: true") { |v| args[:preserve_path] = v }
+op.on('-p', '--operation OP', [ :download, :upload ], 'Transfer Direction.', "default: #{args[:operation]}" ) { |v| args[:operation] = v }
+op.on('--transfer-type TYPE', [ :aspera, :s3 ],'Force the type of transfer to use. Ex: aspera || s3') { |v| args[:transfer_type] = v }
+op.on('--transfer-token TOKEN', '') { |v| args[:transfer_token] = v }
+op.on('--dry-run', 'Run without executing the transfer.') { |v| args[:dry_run] = v }
+op.on('--help', 'Show this message.') { puts op; exit }
+op.load
+op.parse!
+
+args[:file_path] ||= ARGV.shift
+args[:destination_path] ||= ARGV.shift unless ARGV.empty?
+
+agent = Envoi::Mam::Vidispine::Agent.load_config_and_init(args)
+agent.run_operation

data/lib/envoi/mam/agent/cli/commands/wiredrive.rb

@@ -0,0 +1,223 @@
+require 'cgi'
+require 'open-uri'
+
+require 'ubiquity/wiredrive/presentation_utility'
+require 'ubiquity/envoi/api/utilities'
+
+require 'envoi/mam/agent/cli'
+require 'envoi/mam/wiredrive/agent'
+require 'envoi/mam/iconik/agent'
+
+current_command = ARGV.shift
+ARGV << '--help' if ARGV.empty?
+
+default_config_file_paths = Envoi::Mam::Agent::Cli::CONFIG_FILE_PATHS
+# test_args = %w(--download --send-to-envoi --destination-path /tmp)
+# ARGV.concat test_args
+args = {
+    :config_file_path => default_config_file_paths,
+    :dry_run => false,
+    :preserve_path => true,
+    :destination_path => File.expand_path('.'),
+    :should_download_presentation => false,
+    :should_add_presentation_assets_to_envoi => false
+}
+
+args[:config_service_app_id] = ENV['ENVOI_MAM_AGENT_APP_ID']
+args[:config_service_app_token] = ENV['ENVOI_MAM_AGENT_APP_TOKEN']
+args[:config_service_app_url] = ENV['ENVOI_MAM_AGENT_APP_URL']
+
+op = OptionParser.new
+op.on('-c', '--config-file-path FILEPATH', 'The path to the configuration file.') { |v| args[:config_file_path] = v }
+op.on('--agent-app-id ID', '') { |v| args[:config_service_app_id] = v }
+op.on('--agent-token TOKEN', '') { |v| args[:config_service_app_token] = v }
+op.on('--agent-service-api URL', '') { |v| args[:config_service_api_url] = v }
+
+op.on('--presentation-invitation-token TOKEN', 'Presentation invitation token.') { |v| args[:presentation_invitation_token] = v }
+op.on('--presentation-password PASSWORD', 'Presentation password') { |v| args[:presentation_password] = v }
+op.on('--presentation-url URL', 'A shared presentation URL.') { |v| args[:presentation_url] = v }
+# op.on('--presentation-id ID', '') { |v| args[:presentation_id] = v }
+#
+op.on('--[no-]download-presentation-assets', 'Determines if presentation asset files should be downloaded.') { |v| args[:should_download_presentation] = v }
+op.on('--[no-]add-assets-to-envoi', 'Determines if presentation assets should be added to Envoi.') { |v| args[:should_add_presentation_assets_to_envoi] = v }
+op.on('--[no-]add-assets-to-iconik', '') { |v| args[:should_add_presentation_assets_to_iconik] = v }
+op.on('--iconik-storage-id STORAGEID', '') { |v| args[:iconik_storage_id] = v }
+
+op.on('-d', '--destination-path DIRPATH', 'The destination path to download or upload to.', "default: '#{args[:destination_path]}'") { |v| args[:destination_path] = v }
+op.on('-p','--[no-]preserve-path', "default: #{args[:preserve_path]}") { |v| args[:preserve_path] = v }
+
+# op.on('--dry-run', 'Run without executing the transfer.') { |v| args[:dry_run] = v }
+op.on('--help', 'Show this message.') { puts op; exit }
+op.load
+op.parse!
+
+args[:destination_path] ||= ARGV.shift unless ARGV.empty?
+
+presentation_url ||= args[:presentation_url] || ARGV.shift
+if presentation_url
+  uri = URI(presentation_url)
+  presentation_invitation_token = uri.path
+  presentation_url_token = presentation_invitation_token[1..-1] if presentation_invitation_token.start_with?('/')
+
+  query_data = uri.query ? Hash[uri.query.split('&').map { |v| v.split('=') }] : { }
+  query_password = query_data['password']
+  presentation_url_password ||= query_password
+
+  args[:presentation_invitation_token] ||= presentation_url_token
+  args[:presentation_password] ||= presentation_url_password
+end
+
+class EnvoiImportUtility
+
+  attr_accessor :initial_args, :api_client
+
+  def initialize(args = { })
+    @initial_args = args
+    initialize_api_client(args)
+  end
+
+  def after_initialize
+    @default_original_storage_id = args[:default_original_storage_id]
+    @default_thumbnail_storage_id = args[:default_thumbnail_storage_id]
+  end
+
+  def initialize_api_client(args = { })
+    @api_client = Ubiquity::Envoi::API::Utilities.new(args)
+  end
+
+  def import_wiredrive_presentation_asset(asset, options = { })
+    title = asset['title'] || asset['label']
+    description = asset['description']
+    keywords = asset['keywords']
+
+    media_elements = asset['media']
+
+    media = media_elements.find { |me| me['type'] == 'original' }
+    raise 'No original media found.' unless media
+    url = media['url']
+    uri = URI(url)
+    mime_type = media['mimeType']
+    size = media['fileSize']
+
+
+    url_path = uri.path.split('?').first
+    filename = File.basename(url_path)
+
+    original_storage_id = options[:original_storage_id] || options[:storage_id] || @default_original_storage_id
+    create_args = {
+        :name => title,
+        :description => description,
+        :file => {
+            :mime => 'video/mp4', # mime_type,
+            :path => url,
+            :name => filename,
+            :storage_key => url_path,
+            :setting_id => original_storage_id,
+            'size' => size || 0
+        }
+    }
+    r = api_client.media_file_create(create_args)
+
+    media_file_id = r['id']
+    raise 'No id found in response' unless media_file_id
+
+    video_thumb = asset['thumbnails'].find { |e| e['category'] == 'max' }
+    if video_thumb
+      thumbnail_storage_id = options[:thumbnail_storage_id] || options[:storage_id] || @default_thumbnail_storage_id
+      video_thumb_url = video_thumb['url']
+      video_thumb_uri = URI(video_thumb_url)
+      video_thumb_path = video_thumb_uri.path.split('?').first
+      media_file_file_add_args = {
+          'media_file_id' => media_file_id,
+          'shape_type' => 'thumb',
+          'shape_label' => 'Thumbnail',
+          'name' => File.basename(video_thumb_path),
+          'mime' => 'image/jpeg',
+          'path' => video_thumb_url,
+          'size' => 0,
+          'setting_id' => thumbnail_storage_id,
+          'storage_key' => video_thumb_path
+      }
+      r = api_client.media_file_file_add(media_file_file_add_args)
+
+    end
+  end
+
+end
+
+@args = args
+def wdpu(args = @args); @wdpu ||= Ubiquity::Wiredrive::PresentationUtility.new(args) end
+
+def presentation_assets_get(args = @args)
+  presentation_invitation_token = args[:presentation_invitation_token]
+  presentation_password = args[:presentation_password]
+  wdpu.presentation_assets_get_using_token(presentation_invitation_token, presentation_password)
+end
+def presentation_assets; @presentation_assets ||= presentation_assets_get end
+
+if args[:should_download_presentation]
+  presentation_assets_download_results = wdpu.presentation_assets_download(assets: presentation_assets, destination_path: args[:destination_path], overwrite: args[:overwrite_on_download])
+end
+
+if args[:should_add_presentation_assets_to_envoi]
+
+  wda = Envoi::Mam::Wiredrive::Agent.load_config_and_init(args)
+  envoi_config = wda.config['envoi']
+  envoi_http_host_address = envoi_config['http_host_address']
+  envoi_username = envoi_config['username']
+  envoi_password = envoi_config['password']
+  envoi_original_storage_id = envoi_config['original_storage_id'] || envoi_config['default_storage_id']
+  envoi_thumbnail_storage_id = envoi_config['thumbnail_storage_id'] || envoi_config['default_storage_id']
+  envoi_args = { http_host_address: envoi_http_host_address, username: envoi_username, password: envoi_password }
+  envoi = EnvoiImportUtility.new(envoi_args)
+  presentation_assets.each do |a|
+    begin
+      envoi.import_wiredrive_presentation_asset(a, { original_storage_id: envoi_original_storage_id, thumbnail_storage_id: envoi_thumbnail_storage_id })
+    rescue => e
+      warn "#{e.message} #{$@.first}"
+    end
+
+  end
+end
+
+if args[:should_add_presentation_assets_to_iconik]
+  presentation_assets_download_results ||= wdpu.presentation_assets_download(assets: presentation_assets, destination_path: '/tmp')
+
+  # wda = Envoi::Mam::Wiredrive::Agent.load_config_and_init(args)
+  # iconik_config = wda.config['iconik']
+
+  iconik = Envoi::Mam::Iconik::Agent.load_config_and_init(args)
+  iconik_storage_id = args[:iconik_storage_id]
+  presentation_assets_download_results.each do |r|
+    # asset = r[:asset]
+    media_elements_downloaded = r[:media_element_download_responses]
+    media_element_downloaded = media_elements_downloaded.first
+    file_path = media_element_downloaded[:destination_file_path]
+    begin
+      # puts "WD FP: #{media_element_downloaded}"
+      iconik.upload(file_path: file_path, storage_id: iconik_storage_id, destination_path: '/', preserve_path: false)
+    rescue => e
+      warn "#{e.message} #{$@.first}"
+    end
+  end
+
+end
+
+if args[:should_add_presentation_assets_to_vidispine]
+  presentation_assets_download_results ||= wdpu.presentation_assets_download(assets: presentation_assets, destination_path: '/tmp')
+
+  vidispine = Envoi::Mam::Vidispine::Agent.load_config_and_init(args)
+  vidispine_storage_id = args[:vidispine_storage_id]
+  presentation_assets_download_results.each do |r|
+    # asset = r[:asset]
+    media_elements_downloaded = r[:media_element_download_responses]
+    media_element_downloaded = media_elements_downloaded.first
+    file_path = media_element_downloaded[:destination_file_path]
+    begin
+      vidispine.upload(file_path: file_path, storage_id: vidispine_storage_id, destination_path: '/', preserve_path: false)
+    rescue => e
+      warn "#{e.message} #{$@.first}"
+    end
+  end
+
+end

data/lib/envoi/mam/agent/config_service_client.rb

@@ -0,0 +1,60 @@
+require 'rubygems'
+require 'net/http'
+require 'json'
+
+module Envoi
+  module Mam
+    class Agent
+      class ConfigServiceClient
+
+        DEFAULT_API_URL = 'https://9nyhwwwjw7.execute-api.us-east-1.amazonaws.com/latest/'
+
+
+        def self.config_get(args = { })
+          app_id = args[:app_id]
+          token = args[:token]
+          api_url = args[:api_url] || DEFAULT_API_URL
+          request_url = "#{api_url}getuserrecord?token=#{token}&application_id=#{app_id}"
+          uri = URI(request_url)
+          http = Net::HTTP.new(uri.host, uri.port)
+          http.use_ssl = true
+          req = Net::HTTP::Get.new(uri.request_uri)
+          response = http.request(req)
+
+          code =  response.code
+          body_raw = response.body
+
+          unless code == '200'
+            puts uri.to_s
+            puts uri.request_uri
+            puts code
+            puts body_raw
+            return false
+          end
+
+          body_parsed = JSON.parse(body_raw)
+          result = body_parsed['result']
+          raise "Invalid response from configuration service. No result found. '#{body_raw}'" unless result
+
+          app_data = result.first
+          raise "Invalid response from configuration service. No app data found. '#{body_raw}'" unless app_data
+
+          agent_data_raw = app_data['agentdata']
+          raise "Invalid response from configuration service. No agentdata found. '#{body_raw}'" unless agent_data_raw
+
+          begin
+            agent_data = JSON.parse(agent_data_raw)
+          rescue => e
+            raise "Error parsing agentdata. #{e.message} '#{body_raw}'"
+          end
+
+          agent_data
+        end
+
+      end
+
+    end
+
+  end
+
+end

data/lib/envoi/mam/agent/transfer_client.rb

@@ -0,0 +1,37 @@
+
+require 'envoi/mam/agent'
+
+module Envoi
+  module Mam
+    class Agent
+      class TransferClient
+
+        attr_accessor :agent, :logger, :initial_args
+
+        def initialize(args = { })
+          @initial_args = args
+          @agent = args[:agent]
+          initialize_logger(args)
+
+          after_initialize
+        end
+
+        def after_initialize
+          # To be overridden by child class
+        end
+
+        def initialize_logger(args = { })
+          @logger = agent.logger if agent && agent.respond_to?(:logger) && agent.logger
+        end
+
+        # TransferClient
+      end
+
+      # Agent
+    end
+
+    # Mam
+  end
+
+  # Envoi
+end

data/lib/envoi/mam/agent/transfer_client/aspera.rb

@@ -0,0 +1,91 @@
+require 'envoi/mam/agent/transfer_client'
+
+module Envoi
+  module Mam
+    class Agent
+      class TransferClient
+
+        class Aspera < TransferClient
+
+          DEFAULT_DESTINATION_PATH = '.'
+          DEFAULT_ASCP_ARGS = '-v -k3 --overwrite=diff -P 33001'
+
+          ASCP_PATHS = [
+              '/usr/local/bin/ascp',
+              '/usr/bin/ascp',
+              '/Library/Aspera/bin/ascp',
+              '~/Applications/Aspera CLI/bin/ascp',
+              '~/Applications/Aspera Connect.app/Contents/Resources/ascp',
+              '/Applications/Aspera Connect.app/Contents/Resources/ascp'
+          ]
+          ASCP_PATHS.map { |v| File.expand_path(v) }
+
+          attr_accessor :default_ascp_path, :ascp_path, :default_ascp_args
+
+          def after_initialize(args = initial_args)
+            @default_ascp_path = args[:default_aspera_ascp_path]
+            @default_ascp_path ||= ASCP_PATHS.find { |v| File.exists? v } || ASCP_PATHS.first
+            @default_ascp_args = args[:default_ascp_args] || args[:default_aspera_ascp_args] || DEFAULT_ASCP_ARGS
+          end
+
+          def ascp_path_exists?
+            File.executable?(@ascp_path)
+          end
+
+          def build_ascp_command(config, mode, source_path, target_path)
+            aspera_host_address = config['host_address'] || config['host']
+            aspera_username = config['username']
+            aspera_password = config['password']
+            aspera_token = config['aspera_transfer_token'] || config['aspera_token'] || config['token'] || config['transfer_token']
+
+            @ascp_path = config['ascp_path'] || default_ascp_path
+            @ascp_path = File.expand_path(@ascp_path)
+
+            aspera_ascp_path = @ascp_path
+
+            ascp_args = config['ascp_args'] || default_ascp_args || agent.default_ascp_args
+            # command = %(export ASPERA_SCP_PASS="#{aspera_password}"; "#{aspera_ascp_path}" #{ascp_args} --host="#{aspera_host_address}" --user=#{aspera_username} --mode=#{mode} "#{source_path}" "#{target_path}" )
+            command = ''
+            command << %(export ASPERA_SCP_PASS="#{aspera_password}"; ) if aspera_password
+            command << %("#{aspera_ascp_path}" --mode=#{mode} --host="#{aspera_host_address}" --user="#{aspera_username}")
+            command << %( #{ascp_args}) if ascp_args && !ascp_args.empty?
+            command << %( -W "#{aspera_token}") if aspera_token && !aspera_token.empty?
+            command << %( "#{source_path.gsub('"', '\"')}" "#{target_path.gsub('"', '\"')}")
+          end
+
+          def download(config, path, destination_path = DEFAULT_DESTINATION_PATH)
+            aspera_base_path = config['base_path'] || ''
+            source_path = File.join(aspera_base_path, path)
+            command = build_ascp_command(config, 'recv', source_path, destination_path)
+
+            unless ascp_path_exists?
+              warn "ASCP not found. '#{ascp_path}'"
+              return false
+            end
+            agent.shell_execute(command)
+          end
+
+          def upload(config, path, destination_path = DEFAULT_DESTINATION_PATH)
+            aspera_base_path = config['base_path'] || ''
+            source_path = path
+            destination_path = aspera_base_path.empty? ? destination_path : File.join(aspera_base_path, destination_path)
+            command = build_ascp_command(config, 'send', source_path, destination_path)
+
+            unless ascp_path_exists?
+              warn "ASCP not found. '#{ascp_path}'"
+              return false
+            end
+            agent.shell_execute(command)
+          end
+
+          # AsperaTransferClient
+        end
+
+        # TransferClient
+      end
+      # Agent
+    end
+    # Mam
+  end
+  # Envoi
+end