envoi-mam-agent 1.1.0

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

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

@@ -0,0 +1,70 @@
+require 'aws-sdk-s3'
+
+require 'envoi/mam/agent/transfer_client'
+
+module Envoi
+  module Mam
+    class Agent
+      class TransferClient
+
+        class S3 < TransferClient
+
+          def after_initialize(args = initial_args)
+
+          end
+
+          def initialize_s3_client(config)
+            region = config['region']
+            access_key_id = config['access_key_id']
+            secret_access_key = config['secret_access_key']
+
+            resource_args = { }
+            resource_args[:region] = region if region && !region.empty?
+            resource_args[:access_key_id] = access_key_id if access_key_id && !access_key_id.empty?
+            resource_args[:secret_access_key] = secret_access_key if secret_access_key && !secret_access_key.empty?
+            @s3 = Aws::S3::Resource.new(resource_args)
+          end
+
+          def download(config, path, destination_path = DEFAULT_DESTINATION_PATH)
+            s3 = initialize_s3_client(config)
+            bucket_name = config['bucket_name']
+            bucket = s3.bucket(bucket_name)
+            object_prefix = config['object_prefix']
+            object_path = object_prefix && !object_prefix.empty? ? File.join(object_prefix, path) : path
+            object_path = object_path[1..-1] if object_path.start_with?('/')
+            object = bucket.object(object_path)
+            # puts object.data
+            # puts "Path: '#{path}' DPath: '#{destination_path}'"
+            object.get(response_target: destination_path) if agent && !agent.dry_run?
+          end
+
+          def upload(config, path, destination_path)
+            s3 = initialize_s3_client(config)
+            bucket_name = config['bucket_name']
+            bucket = s3.bucket(bucket_name)
+            object_prefix = config['object_prefix']
+            object_path = object_prefix && !object_prefix.empty? ? File.join(object_prefix, destination_path) : path
+            object_path = object_path[1..-1] if object_path.start_with?('/')
+            object_path = object_path[0..-2] if object_path.end_with?('/')
+            object = bucket.object(object_path)
+            logger.debug { "Uploading File '#{path}' to '#{destination_path}' Object Path: '#{object_path}'" }
+
+            object.upload_file(path) if agent && !agent.dry_run?
+            object
+
+            # puts object.data
+            # puts "Path: '#{path}' DPath: '#{destination_path}'"
+            # abort
+          end
+
+          # S3TransferClient
+        end
+
+        # TransferClient
+      end
+      # Agent
+    end
+    # Mam
+  end
+  # Envoi
+end

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

@@ -0,0 +1,7 @@
+module Envoi
+  module Mam
+    class Agent
+      VERSION = '1.1.0'.freeze
+    end
+  end
+end

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

@@ -0,0 +1,195 @@
+require 'envoi/mam/agent/transfer_client/aspera'
+require 'envoi/mam/agent/transfer_client/s3'
+
+require 'ubiquity/iconik/api/utilities'
+
+module Envoi
+
+  module Mam
+
+    class Iconik
+
+      class Agent < Envoi::Mam::Agent
+
+        DEFAULT_FORMAT_NAME = 'ORIGINAL'
+        DEFAULT_DESTINATION_PATH = '.'
+
+        def after_initialize
+          @default_format_name = initial_args[:default_format_name] || DEFAULT_FORMAT_NAME
+        end
+
+        def iconik_config
+          config['iconik']
+        end
+
+        def initialize_api_client(args = { })
+          @api_client = args[:api_client] || begin
+            http_host_address = iconik_config['http_host_address']
+            application_id = iconik_config['application_id']
+            token = iconik_config['token']
+            client_args = { }
+            client_args[:http_host_address] = http_host_address if http_host_address && !http_host_address.empty?
+            client_args[:application_id] = application_id if application_id && !application_id.empty?
+            client_args[:token] = token if token && !token.empty?
+            Ubiquity::Iconik::API::Utilities.new(client_args)
+          end
+        end
+
+        def download(args = { })
+
+          asset_id = args[:asset_id]
+          file_id = args[:file_id]
+          if file_id && !file_id.empty?
+            asset_file = api_client.asset_file_get(asset_id: asset_id, file_id: file_id)
+          else
+            format_name = args[:format_name] || @default_format_name
+            format = api_client.asset_format_get_by_name(asset_id: asset_id, format_name: format_name.upcase)
+            format_id = format['id']
+
+            asset_files_get_response = api_client.asset_files_get(asset_id: asset_id, generate_signed_url: true)
+            asset_files = asset_files_get_response['objects']
+
+            asset_files_for_format = asset_files.find_all { |f| f['format_id'] == format_id }
+
+            asset_file = asset_files_for_format.first
+          end
+
+          # file = files.first
+          files = [ asset_file ] # just do the first file for now
+          files.each do |file|
+            begin
+              download_file(args, file)
+            rescue => e
+              logger.warn { "Exception: #{$!}. #{$@.first}" }
+            end
+          end
+          logger.info { 'DONE' }
+        end
+
+        def download_file(args, file)
+          logger.debug { "File: #{file}"}
+          transfer_type = args[:transfer_type] || ''
+
+          file_storage_id = file['storage_id']
+          file_directory_path = file['directory_path']
+          file_path = !file_directory_path || file_directory_path.empty? ? file['original_name'] : File.join(file_directory_path, file['original_name'])
+
+          file_storage_config = iconik_config['storages'][file_storage_id]
+
+          unless file_storage_config && !file_storage_config.empty?
+            raise Exception, "No configuration found for storage '#{file_storage_id}'"
+          end
+
+          logger.info { "Transferring File Path: '#{file_path}'" }
+          preserve_path = args.fetch(:preserve_path, file_storage_config.fetch('preserve_path', true))
+
+          destination_path = args[:destination_path] || file_storage_config['destination_path'] || DEFAULT_DESTINATION_PATH
+          relative_path = preserve_path ? file_directory_path : nil
+          relative_path = nil if relative_path == '.'
+
+          target_path = relative_path ? File.join(destination_path, relative_path) : destination_path
+          target_path = target_path[0..-1] if target_path.start_with?('/') && !destination_path.start_with?('/')
+
+          aspera_config = file_storage_config['aspera']
+          if (transfer_type.empty? || transfer_type == :aspera) && (aspera_config && !aspera_config.empty?)
+            client = Envoi::Mam::Agent::TransferClient::Aspera.new(agent: self)
+            return client.download(aspera_config, file_path, target_path)
+          end
+
+          s3_config = file_storage_config['s3']
+          if (transfer_type.empty? || transfer_type == :s3) && (s3_config && !s3_config.empty?)
+            target_path = File.expand_path(target_path) if target_path == '.'
+            target_path = File.join(target_path, File.basename(file_path))
+            client = Envoi::Mam::Agent::TransferClient::S3.new(agent: self)
+            return client.download(s3_config, file_path, target_path)
+          end
+
+          logger.warn { "No Supported TransferClient Configuration#{transfer_type && !transfer_type.empty? ? " for transfer type '#{transfer_type}' " : ''}Found in Storage Configuration." }
+        end
+
+        def upload(args = { })
+          file_path = args[:file_path]
+          if File.directory?(file_path)
+            file_paths = Dir.glob(File.join(file_path, '*.*'))
+            logger.debug { "File Paths: #{file_paths}"}
+            return file_paths.map { |fp| upload(args.merge(file_path: fp))}
+          end
+          logger.debug { "Preparing to upload '#{file_path}'" }
+
+          ingest_after_upload = args[:ingest_after_upload]
+          analyze_asset = args[:analyze_asset]
+
+          transfer_type = args[:transfer_type] || ''
+          storage_id = args[:storage_id]
+          path_on_storage = nil
+          iconik_storage_config = iconik_config['storages'][storage_id]
+
+          unless iconik_storage_config && !iconik_storage_config.empty?
+            raise Exception, "No configuration found for storage '#{storage_id}'"
+          end
+
+          preserve_path = args.fetch(:preserve_path, iconik_storage_config.fetch('preserve_path', true))
+
+          destination_path = args[:destination_path] || iconik_storage_config['destination_path'] || '/'
+          relative_path = preserve_path ? File.dirname(file_path) : nil
+          relative_path = File.expand_path(relative_path) if relative_path == '.'
+
+          target_path = relative_path ? File.join(destination_path, relative_path) : destination_path
+          target_path = target_path[1..-1] if target_path.start_with?('/') && !destination_path.start_with?('/')
+
+          # abort("FP: #{file_path} TP: #{target_path}")
+
+          transfer_response = begin
+            aspera_config = iconik_storage_config['aspera']
+            if (transfer_type.empty? || transfer_type == :aspera) && (aspera_config && !aspera_config.empty?)
+              client = Envoi::Mam::Agent::TransferClient::Aspera.new(agent: self)
+              resp = client.upload(aspera_config, file_path, target_path)
+            end
+
+            s3_config = iconik_storage_config['s3']
+            if (!resp) && (transfer_type.empty? || transfer_type == :s3) && (s3_config && !s3_config.empty?)
+              _target_path = target_path
+              _target_path = '' if target_path == '.'
+              _target_path = File.join(_target_path, File.basename(file_path))
+              client = Envoi::Mam::Agent::TransferClient::S3.new(agent: self)
+              resp = client.upload(s3_config, file_path, _target_path)
+              path_on_storage = _target_path
+            end
+
+            resp
+          end
+
+          if transfer_response.nil?
+            logger.warn { "No supported TransferClient configuration#{transfer_type && !transfer_type.empty? ? " for transfer type '#{transfer_type}' " : ''}found in storage configuration." }
+            return false
+          end
+
+          unless transfer_response
+            logger.warn { ""}
+            return false
+          end
+
+          file_size = File.exist?(file_path) ? File.size?(file_path) : 1024
+          path_on_storage ||= File.join(target_path, File.basename(file_path))
+
+          if ingest_after_upload
+            asset_create_response = api_client.asset_add_using_file_path(file_path: path_on_storage,
+                                                                         storage_id: storage_id,
+                                                                         file_size: file_size)
+            if analyze_asset
+              asset_id = asset_create_response[:asset]['id']
+              api_client.asset_analyze(:asset_id => asset_id)
+            end
+
+          end
+
+        end
+
+
+      end
+
+    end
+
+  end
+
+end