unimatrix-cli 2.2.0

data/lib/unimatrix_cli/cli.rb

@@ -0,0 +1,30 @@
+module UnimatrixCLI
+  class CLI < Command
+      
+    synopsis Unimatrix::VERSION + 
+    "\n\nA CLI for interacting with Unimatrix apps\n\nAvailable commands:\n\n" +
+    available_commands +
+    "\n\nType 'unimatrix <command> --help' to see options\n"
+
+    educate_if_empty
+    stop_on_unknown
+
+    def parse_command
+      command = ARGV.shift
+
+      if command == "login" || command == "logout"
+        sub_command = "#{ command.capitalize }Command"
+        "#{ sub_command }"
+      else
+        unimatrix_app, resource, sub_command = command.split( '::' ).
+          map { | word | camelize( word ) }
+        sub_command = "#{ sub_command }Command"
+        "#{ unimatrix_app }::#{ resource }::#{ sub_command }"
+      end
+    end
+
+    private; def camelize( word )
+      word.split( '_' ).map( &:capitalize ).join
+    end
+  end
+end

data/lib/unimatrix_cli/command.rb

@@ -0,0 +1,138 @@
+require 'net/http'
+require 'json'
+
+module UnimatrixCLI
+  class Command
+    include UnimatrixParser
+
+    def initialize
+      @options = self.class.options
+    end
+
+    #----------------------------------------------------------------------------
+    # Helpers
+
+    # TODO add color options
+    def write( options )
+      if options[ :error ]
+        $stderr.puts options[ :message ]
+      else
+        $stdout.puts options[ :message ]
+      end
+    end
+
+    def validate( object, type )
+      object.present? &&
+      object.is_a?( "Unimatrix::#{ type }".constantize ) &&
+      object.errors.blank? rescue nil
+    end
+
+    def validate_collection( objects, type )
+      if objects.present?
+        objects.all? do | object |
+          validate( object, type )
+        end
+      else
+        false
+      end
+    end
+
+    #----------------------------------------------------------------------------
+    # File Reading
+
+    def read_file( path )
+      begin
+        file = File.open( path, 'r' )
+        extension = path.split( '.' ).last.downcase
+        contents = file_to_hash( file, extension )
+      rescue
+        write( 
+          message: "There was a problem accessing and reading #{ path }",
+          error: true
+        )
+      ensure
+        file.close
+      end
+    end
+
+    private; def file_to_hash( file, extension )
+      contents = file.read
+      if extension == 'yaml'
+        contents = YAML.load( contents )
+      elsif extension == 'json'
+        contents = JSON.parse( contents )
+      else
+        write( 
+          message: "The file extension #{ extension } is not supported",
+          error: true
+        )
+      end
+      contents
+    end
+
+    #----------------------------------------------------------------------------
+    # Requests
+
+    def operation( endpoint, options = {} )
+      unless self.class.to_s.include?( 'Zephyrus' )
+        options.merge!( access_token: Configuration.access_token )
+      end
+      Unimatrix::Operation.new( endpoint, options )
+    end
+
+    def parse_object( object )
+      object.fields.map do | field |
+        key = field.first
+        value = object.send( key )
+        "#{ key }: #{ value }\n"
+      end
+    end
+
+    def make_request( endpoint, http_method, body = nil )
+      uri = URI.parse( endpoint )
+      http = Net::HTTP.new( uri.host, uri.port )
+      http.use_ssl = ( uri.scheme == 'https' )
+
+      http_request = Module.const_get( "Net::HTTP::#{ http_method }" )
+      
+      response = nil
+      begin
+        request = http_request.new( endpoint, 'Content-Type' => 'application/json' )
+        request.body = body.to_json if body
+        response = http.request( request )
+      rescue Timeout::Error
+        response = nil
+      end
+
+      parse_response( response.body ) if response
+    end
+
+    private; def parse_response( body )
+      JSON.parse( body ) rescue nil
+    end
+
+    #----------------------------------------------------------------------------
+    # Class Methods
+
+    class << self
+      def descendants
+        ObjectSpace.each_object( Class ).select { | klass | klass < self }
+      end
+
+      def available_commands
+        commands = Command.descendants
+        commands.delete( UnimatrixCLI::CLI )
+        commands.delete( UnimatrixCLI::Citadel::CitadelCommand )
+
+        parsed_commands = commands.map do | command |
+          command_words = command.to_s.split( '::' )
+          command_words.shift
+          command_words.last.slice!( 'Command' )
+          command_words.map( &:underscore ).join( '::' )
+        end
+
+        parsed_commands.sort.join( "\n" )
+      end
+    end
+  end
+end

data/lib/unimatrix_cli/config/acceptance.yml

@@ -0,0 +1,23 @@
+us-east-1:
+  keymaker_url: "http://keymaker.acceptance.boxxspring.com"
+  archivist_url: ""
+  rendition_profile_uuids:
+    - "011e6f444a1e8948f246b45c798e5a16"
+    - "c22924af95ba00825b9863c6b184e7b5"
+    - "2dd9cbab22ed92bb9d2ccc9837df9a3c"
+    - "fef49a4734f873246388384ce1c7e200"
+    - "17ed2ae6c2f091c4c7b1753a17a698b7"
+  zephyrus_url: "http://zephyrus.acceptance.boxxspring.com"
+  unimatrix_api_url: "http://api.acceptance.unimatrix.io"
+
+us-west-2:
+  keymaker_url: "http://us-west-2.keymaker.acceptance.boxxspring.net"
+  archivist_url: "http://us-west-2.archivist.acceptance.boxxspring.net"
+  rendition_profile_uuids:
+    - "011e6f444a1e8948f246b45c798e5a16"
+    - "c22924af95ba00825b9863c6b184e7b5"
+    - "2dd9cbab22ed92bb9d2ccc9837df9a3c"
+    - "fef49a4734f873246388384ce1c7e200"
+    - "17ed2ae6c2f091c4c7b1753a17a698b7"
+  zephyrus_url: "http://zephyrus.acceptance.boxxspring.com"
+  unimatrix_api_url: "http://us-west-2.api.acceptance.unimatrix.io"

data/lib/unimatrix_cli/config/configuration.rb

@@ -0,0 +1,66 @@
+require 'yaml'
+
+module UnimatrixCLI
+  class Configuration
+
+    class << self
+
+      def default_config
+        @default_config ||= begin
+          default_config_path = File.join( 
+            File.dirname( __FILE__ ), "./#{ environment }.yml"
+          )
+          if File.file?( default_config_path )
+            default_config_file = YAML.load_file( default_config_path )
+            default_config_file[ region ]
+          else
+            {}
+          end
+        end
+      end
+
+      def environment
+        @environment ||= begin
+          File.read( environment_file_path ).chomp if File.file?( environment_file_path )
+        end
+      end
+
+      def access_token
+        @access_token ||= begin
+          File.read( access_token_file_path ).chomp if File.file?( access_token_file_path )
+        end
+      end
+
+      def region
+        @region ||= begin
+          File.read( region_file_path ).chomp if File.file?( region_file_path ) || 
+          ENV[ 'AWS_REGION' ]
+        end
+      end
+
+      def unimatrix_directory_path
+        @unimatrix_directory_path ||= File.join( Dir.home, ".unimatrix" )
+      end
+
+      def environment_file_path
+        @environment_file_path ||= begin
+          File.join( unimatrix_directory_path, "/.environment" )
+        end
+      end
+
+      def access_token_file_path
+        @access_token_file_path ||= begin
+          File.join( unimatrix_directory_path, "/.access_token" )
+        end
+      end
+
+      def region_file_path
+        @region_file_path ||= begin
+          File.join( unimatrix_directory_path, "/.region" )
+        end
+      end
+
+    end
+
+  end
+end

data/lib/unimatrix_cli/config/production.yml

@@ -0,0 +1,15 @@
+us-east-1:
+  keymaker_url: "http://keymaker.boxxspring.com"
+  archivist_url: ""
+  rendition_profile_uuids:
+    - ""
+  zephyrus_url: ""
+  unimatrix_api_url: ""
+
+us-west-2:
+  keymaker_url: ""
+  archivist_url: ""
+  rendition_profile_uuids:
+    - ""
+  zephyrus_url: ""
+  unimatrix_api_url: ""

data/lib/unimatrix_cli/config/staging.yml

@@ -0,0 +1,15 @@
+us-east-1:
+  keymaker_url: "http://keymaker.staging.boxxspring.com"
+  archivist_url: ""
+  rendition_profile_uuids:
+    - ""
+  zephyrus_url: "http://zephyrus.staging.boxxspring.com"
+  unimatrix_api_url: "http://api.staging.unimatrix.io"
+
+us-west-2:
+  keymaker_url: "http://us-west-2.keymaker.staging.boxxspring.net"
+  archivist_url: "http://us-west-2.archivist.staging.boxxspring.net"
+  rendition_profile_uuids:
+    - ""
+  zephyrus_url: "http://zephyrus.staging.boxxspring.com"
+  unimatrix_api_url: "http://us-west-2.api.staging.unimatrix.io"

data/lib/unimatrix_cli/config/unimatrix.rb

@@ -0,0 +1,4 @@
+Unimatrix.configuration do
+  url   UnimatrixCLI::Configuration.default_config[ 'unimatrix_api_url' ] || 
+        ENV[ 'UNIMATRIX_API_URL' ]
+end

data/lib/unimatrix_cli/iris/stream/create_command.rb

@@ -0,0 +1,43 @@
+module UnimatrixCLI
+  module Iris
+    module Stream
+      class CreateCommand < Command
+
+        option :realm, "The realm uuid",                 type: :string, required: true
+        option :name,  "The name of the stream",         type: :string, required: true
+        option :type,  "The type (event or continuous)", type: :string, default:  "event"
+
+        synopsis "Create a new stream"
+
+        def execute
+          stream = create_stream
+
+          if validate( stream, 'Iris::Stream' )
+            write( 
+              message: "Stream uuid: #{ stream.uuid }" 
+            )
+          else
+            write( 
+              message: "Error creating stream: #{ stream.inspect }",
+              error: true
+            )
+          end
+        end
+
+        #----------------------------------------------------------------------------
+
+        private; def create_stream
+        stream = Unimatrix::Iris::Stream.new(
+          name: @options[ :name ],
+          stream_type: @options[ :type ]
+        )
+
+          operation( 
+            '/streams',
+            realm_uuid: @options[ :realm ]
+          ).write( 'streams', stream ).first rescue nil
+        end
+      end
+    end
+  end
+end

data/lib/unimatrix_cli/iris/stream/describe_command.rb

@@ -0,0 +1,35 @@
+module UnimatrixCLI
+  module Iris
+    module Stream
+      class DescribeCommand < Command
+
+        option :stream, "The stream uuid", type: :string, required: true
+
+        synopsis "Describe a stream"
+
+        def execute
+          stream = find_stream
+
+          if validate( stream, 'Iris::Stream' )
+            parse_object( stream ).each do | response | 
+              write( message: response )
+            end
+          else
+            write( 
+              message: "Stream could not be found: #{ stream.inspect }",
+              error: true
+            )
+          end
+        end
+
+        #----------------------------------------------------------------------------
+
+        private; def find_stream
+          endpoint = "/streams/#{ @options[ :stream ] }"
+
+          operation( endpoint ).read.first
+        end
+      end
+    end
+  end
+end

data/lib/unimatrix_cli/iris/stream_encoder/create_command.rb

@@ -0,0 +1,43 @@
+module UnimatrixCLI
+  module Iris
+    module StreamEncoder
+      class CreateCommand < Command
+
+        option :realm,  "The realm uuid",              type: :string, required: true
+        option :stream, "The stream_encoder's stream", type: :string, required: true
+        option :region, "The region of the stream",    type: :string, default: "us-east-1"
+
+        synopsis "Create a new stream_encoder"
+
+        def execute
+          stream_encoder = create_stream_encoder
+
+          if validate( stream_encoder, 'Iris::StreamEncoder' )
+            write( message: "StreamEncoder uuid: #{ stream_encoder.uuid }" )
+          else
+            write( 
+              message: "Error creating stream_encoder: #{ stream_encoder.inspect }", 
+              error: true 
+            )
+          end
+        end
+
+        #----------------------------------------------------------------------------
+
+        private; def create_stream_encoder
+          endpoint = "/streams/#{ @options[ :stream ] }/stream_encoders"
+
+          stream_encoder = Unimatrix::Iris::StreamEncoder.new(
+            region: @options[ :region ],
+            state: 'default'
+          )
+
+          operation( 
+            endpoint, 
+            realm_uuid: @options[ :realm ] 
+          ).write( 'stream_encoders', stream_encoder ).first rescue nil
+        end
+      end
+    end
+  end
+end

data/lib/unimatrix_cli/iris/stream_encoder/describe_command.rb

@@ -0,0 +1,37 @@
+module UnimatrixCLI
+  module Iris
+    module StreamEncoder
+      class DescribeCommand < Command
+
+        option :stream_encoder, "The stream_encoder uuid", type: :string, required: true, short: :e
+        option :stream,         "The uuid of its stream",  type: :string, required: true
+
+        synopsis "Describe a stream_encoder"
+
+        def execute
+          stream_encoder = find_stream_encoder
+
+          if validate( stream_encoder, 'Iris::StreamEncoder' )
+            parse_object( stream_encoder ).each do | response | 
+              write( message: response )
+            end
+          else
+            write( 
+              message: "StreamEncoder could not be found: #{ stream_encoder.inspect }", 
+              error: true 
+            )
+          end
+        end
+
+        #----------------------------------------------------------------------------
+
+        private; def find_stream_encoder
+          endpoint = "/streams/#{ @options[ :stream ] }/stream_encoders/" +
+            "#{ @options[ :stream_encoder ] }"
+
+          operation( endpoint ).read.first
+        end
+      end
+    end
+  end
+end