rookout 0.1.0

data/lib/rookout/processor/paths/canopy/markers.rb

@@ -0,0 +1,186 @@
+module Rookout
+  module Processor
+    module Paths
+      module Canopy
+        require_relative "consts"
+
+        class Marker; end
+
+        class Operation < Marker
+          def read _namespace, _create
+            raise NotImplementedError
+          end
+
+          def write _namespace, _value
+            raise Exception.RookOperationReadOnly, self.class.to_s
+          end
+        end
+
+        class FunctionOperation < Operation
+          def initialize name, args
+            @name = name
+            @args = args
+          end
+
+          def read namespace, _create
+            namespace.call_method @name, @args
+          end
+        end
+
+        class AttributeOperation < Operation
+          def initialize name
+            @name = name
+          end
+
+          def read namespace, create
+            namespace.read_attribute @name
+          rescue Exceptions::RookAttributeNotFound
+            raise unless create
+            namespace.write_attribute name, ContainerNamespace.new
+          end
+
+          def write namespace, value
+            namespace.write_attribute @name, value
+          end
+        end
+
+        class LookupOperation < Operation
+          def initialize text
+            if (text.start_with?("'") && text.end_with?("'")) || (text.start_with?("\"") && text.end_with?("\""))
+              @key = text[1...-1]
+            else
+              @key = text.to_i
+            end
+          end
+
+          def read namepsace, _create
+            namepsace.read_key @key
+          end
+        end
+
+        class ObjectMarker < Marker
+          def initialize text, obj
+            @text = text
+            @obj = obj
+          end
+
+          attr_reader :text, :obj
+
+          def to_s
+            @obj.to_s
+          end
+        end
+
+        class ToolExceptionMarker < ObjectMarker
+          def initialize exc
+            super exc.to_s, exc
+          end
+        end
+
+        class NamespaceResult < ObjectMarker
+          def initialize namespace, text
+            super text, namespace
+          end
+        end
+
+        class Text < ObjectMarker
+          def initialize text
+            super text, text
+          end
+        end
+
+        class TextResult < ObjectMarker
+          def initialize text
+            super text, text
+          end
+        end
+
+        class List < ObjectMarker
+          def initialize list, text
+            super text, list
+          end
+        end
+
+        class FloatNumber < ObjectMarker
+          def initialize text
+            super text, text.to_f
+          end
+        end
+
+        class Number < ObjectMarker
+          def initialize text
+            super text, text.to_i
+          end
+        end
+
+        class Char < ObjectMarker
+          def initialize text
+            super "'@{text}'", text
+          end
+        end
+
+        class Bool < ObjectMarker
+          def initialize text
+            super text, %w[true True].include?(text)
+          end
+        end
+
+        class Nil < ObjectMarker
+          def initialize
+            super text, obj
+          end
+        end
+
+        class Opt < Marker
+          def initialize opt
+            @opt = opt
+            @level = nil
+
+            # Some ops have alternatives we convert back to standard
+            @opt = OPS_ALTERNATIVES[@opt.upcase] if OPS_ALTERNATIVES.key? @opt.upcase
+
+            ALL_LEVELS.each_with_index do |ops, index|
+              if ops.include? @opt
+                @level = index
+                break
+              end
+            end
+
+            raise Exceptions::RookInvalidArithmeticPath, @opt if @level.nil?
+          end
+
+          attr_reader :level
+
+          def execute_operation first, second
+            # Remove  wrapping of objects as needed
+            first_obj = decapsulate_item first
+            second_obj = decapsulate_item second
+
+            result = nil
+            begin
+              result = OPS_FUNCTIONS[@opt].call first_obj, second_obj
+            rescue StandardError => e
+              raise Exceptions::RookExceptionEvaluationFailed.new "", e
+            end
+
+            # If we don't have a result
+            if result.nil?
+              # Verify objects are primitives
+              raise Exceptions::RookNonPrimitiveObjectType, a.text unless PRIMITIVES.include? first_obj.class
+              raise Exceptions::RookNonPrimitiveObjectType, b.text unless PRIMITIVES.include? second_obj.class
+            end
+
+            ObjectMarker.new result.to_s, result
+          end
+
+          private
+
+          def decapsulate_item item
+            return item.obj.obj if item.is_a? NamespaceResult
+            item.obj
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rookout/processor/paths/path.rb

@@ -0,0 +1,15 @@
+module Rookout
+  module Processor
+    module Paths
+      class Path
+        def read_from _namespace
+          raise NotImplementedError
+        end
+
+        def write_to _namespace, _value
+          raise NotImplementedError
+        end
+      end
+    end
+  end
+end

data/lib/rookout/processor/processor.rb

@@ -0,0 +1,34 @@
+module Rookout
+  module Processor
+    require_relative "rook_error"
+    require_relative "../logger"
+    require_relative "../user_warnings"
+
+    class Processor
+      def initialize configuration, factory
+        @operations = []
+
+        configuration.each do |it|
+          operation = factory.create_operation it
+          @operations.push operation
+        end
+      end
+
+      def process namespace
+        @operations.each do |operation|
+          result = nil
+          begin
+            result = operation.execute namespace
+          rescue StandardError => e
+            message = "Error in operation"
+            Logger.instance.exception message, e
+            warning = RookError.new e, message
+            UserWarnings.notify_warning warning
+          end
+
+          return result unless result.nil?
+        end
+      end
+    end
+  end
+end

data/lib/rookout/processor/processor_factory.rb

@@ -0,0 +1,23 @@
+module Rookout
+  module Processor
+    require_relative "processor"
+    require_relative "operations/set_operation"
+    require_relative "paths/arithmetic_path"
+
+    class ProcessorFactory
+      def initialize; end
+
+      def create_operation configuration
+        Operations::SetOperation.new configuration, self
+      end
+
+      def create_path configuration
+        Paths::ArithmeticPath.new configuration
+      end
+
+      def create_processor configuration
+        Processor.new configuration, self
+      end
+    end
+  end
+end

data/lib/rookout/processor/rook_error.rb

@@ -0,0 +1,45 @@
+module Rookout
+  module Processor
+    require_relative "../protobuf/variant_pb"
+
+    require_relative "namespace_serializer"
+    require_relative "namespaces/ruby_utils_namespace"
+
+    require_relative "../exceptions"
+
+    class RookError
+      def initialize exception, message = ""
+        @exception = exception
+        @traceback = exception.backtrace
+
+        if @exception.is_a? Exceptions::ToolException
+          raw_type = @exception.class.to_s
+          @type = raw_type[raw_type.rindex(":") + 1..raw_type.length]
+          @message = @exception.message
+          @parameters = @exception.parameters
+        else
+          @type = "Unknown"
+          @message = message
+          @parameters = nil
+        end
+      end
+
+      attr_reader :message
+
+      def dumps
+        parameters = NamespaceSerializer.dump Namespaces::RubyObjectNamespace.new(@parameters), false
+        exception = NamespaceSerializer.dump Namespaces::RubyObjectNamespace.new(@exception), false
+
+        backtrace_string = @exception.backtrace.join "\n\t"
+        backtrace_object = Namespaces::RubyObjectNamespace.new(backtrace_string).tailor_limits!
+        traceback = NamespaceSerializer.dump backtrace_object, false
+
+        Com::Rookout::Error.new message: @message,
+                                type: @type,
+                                parameters: parameters,
+                                exc: exception,
+                                traceback: traceback
+      end
+    end
+  end
+end

data/lib/rookout/protobuf/agent_info_pb.rb

@@ -0,0 +1,68 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: agent_info.proto
+
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("agent_info.proto", :syntax => :proto3) do
+    add_message "com.rookout.SCMInformation" do
+      optional :commit, :string, 1
+      optional :origin, :string, 2
+      repeated :sources, :message, 4, "com.rookout.SCMInformation.SourceInfo"
+    end
+    add_message "com.rookout.SCMInformation.SourceInfo" do
+      optional :remoteOriginUrl, :string, 1
+      optional :commit, :string, 2
+    end
+    add_message "com.rookout.NetworkInformation" do
+      optional :ip_addr, :string, 1
+      optional :network, :string, 2
+      optional :external_ip, :string, 3
+    end
+    add_message "com.rookout.SystemInformation" do
+      optional :hostname, :string, 1
+      optional :os, :string, 2
+      optional :os_version, :string, 3
+      optional :distro, :string, 4
+      optional :arch, :string, 5
+    end
+    add_message "com.rookout.VersionInformation" do
+      optional :version, :string, 1
+      optional :commit, :string, 2
+    end
+    add_message "com.rookout.PlatformInformation" do
+      optional :platform, :string, 1
+      optional :version, :string, 2
+      optional :variant, :string, 3
+    end
+    add_message "com.rookout.AgentInformation" do
+      optional :agent_id, :string, 1
+      optional :version, :message, 2, "com.rookout.VersionInformation"
+      optional :network, :message, 3, "com.rookout.NetworkInformation"
+      optional :system, :message, 4, "com.rookout.SystemInformation"
+      optional :platform, :message, 5, "com.rookout.PlatformInformation"
+      optional :executable, :string, 6
+      repeated :command_arguments, :string, 7
+      optional :process_id, :uint32, 8
+      map :labels, :string, :string, 9
+      optional :scm, :message, 10, "com.rookout.SCMInformation"
+      repeated :tags, :string, 11
+    end
+    add_message "com.rookout.AgentInformationArray" do
+      repeated :agents, :message, 1, "com.rookout.AgentInformation"
+    end
+  end
+end
+
+module Com
+  module Rookout
+    SCMInformation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.SCMInformation").msgclass
+    SCMInformation::SourceInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.SCMInformation.SourceInfo").msgclass
+    NetworkInformation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.NetworkInformation").msgclass
+    SystemInformation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.SystemInformation").msgclass
+    VersionInformation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.VersionInformation").msgclass
+    PlatformInformation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.PlatformInformation").msgclass
+    AgentInformation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.AgentInformation").msgclass
+    AgentInformationArray = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.AgentInformationArray").msgclass
+  end
+end

data/lib/rookout/protobuf/controller_info_pb.rb

@@ -0,0 +1,29 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: controller_info.proto
+
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("controller_info.proto", :syntax => :proto3) do
+    add_message "com.rookout.ControllerInformation" do
+      optional :id, :string, 1
+      optional :commit, :string, 2
+      optional :version, :string, 3
+      repeated :tags, :string, 4
+      optional :hostname, :string, 5
+      optional :machine_type, :string, 6
+      optional :network, :string, 7
+      optional :ip, :string, 8
+      optional :platform, :string, 9
+      optional :platform_release, :string, 10
+      optional :platform_version, :string, 11
+      optional :platform_string, :string, 12
+    end
+  end
+end
+
+module Com
+  module Rookout
+    ControllerInformation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.ControllerInformation").msgclass
+  end
+end

data/lib/rookout/protobuf/envelope_pb.rb

@@ -0,0 +1,21 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: envelope.proto
+
+require 'google/protobuf'
+
+require 'google/protobuf/timestamp_pb'
+require 'google/protobuf/any_pb'
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("envelope.proto", :syntax => :proto3) do
+    add_message "com.rookout.Envelope" do
+      optional :timestamp, :message, 1, "google.protobuf.Timestamp"
+      optional :msg, :message, 2, "google.protobuf.Any"
+    end
+  end
+end
+
+module Com
+  module Rookout
+    Envelope = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.Envelope").msgclass
+  end
+end

data/lib/rookout/protobuf/messages_pb.rb

@@ -0,0 +1,189 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: messages.proto
+
+require 'google/protobuf'
+
+require_relative 'variant_pb'
+require_relative 'agent_info_pb'
+require_relative 'controller_info_pb'
+require 'google/protobuf/any_pb'
+require 'google/protobuf/timestamp_pb'
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("messages.proto", :syntax => :proto3) do
+    add_message "com.rookout.NewAgentMessage" do
+      optional :agent_info, :message, 1, "com.rookout.AgentInformation"
+    end
+    add_message "com.rookout.StackFrame" do
+      optional :filename, :string, 1
+      optional :line, :uint32, 2
+      optional :function, :string, 3
+    end
+    add_message "com.rookout.StackTrace" do
+      repeated :frames, :message, 1, "com.rookout.StackFrame"
+    end
+    add_message "com.rookout.Exception" do
+      optional :type, :string, 1
+      optional :message, :string, 2
+      optional :instance, :message, 3, "com.rookout.Variant"
+      optional :traceback, :message, 4, "com.rookout.StackTrace"
+    end
+    add_message "com.rookout.LogMessage" do
+      optional :timestamp, :message, 1, "google.protobuf.Timestamp"
+      optional :agent_id, :string, 2
+      optional :level, :enum, 3, "com.rookout.LogMessage.LogLevel"
+      optional :filename, :string, 4
+      optional :line, :uint32, 5
+      optional :text, :string, 6
+      repeated :arguments, :message, 7, "com.rookout.Variant"
+      optional :exception, :message, 8, "com.rookout.Exception"
+      optional :formatted_message, :string, 9
+      optional :legacy_arguments, :message, 10, "com.rookout.Variant"
+      optional :class_name, :string, 11
+      optional :method_name, :string, 12
+      optional :controller_id, :string, 13
+    end
+    add_enum "com.rookout.LogMessage.LogLevel" do
+      value :TRACE, 0
+      value :DEBUG, 1
+      value :INFO, 2
+      value :WARNING, 3
+      value :ERROR, 4
+      value :FATAL, 5
+    end
+    add_message "com.rookout.ControllerLogMessage" do
+      optional :timestamp, :message, 1, "google.protobuf.Timestamp"
+      optional :level, :enum, 2, "com.rookout.ControllerLogMessage.LogLevel"
+      optional :filename, :string, 3
+      optional :line, :uint32, 4
+      optional :text, :string, 5
+      optional :class_name, :string, 6
+      optional :method_name, :string, 7
+      optional :controller_id, :string, 8
+      optional :error, :message, 9, "com.rookout.ControllerLogMessage.Error"
+      map :metadata, :string, :string, 10
+      map :metadataJSONValues, :string, :bytes, 11
+    end
+    add_message "com.rookout.ControllerLogMessage.Error" do
+      optional :type, :string, 1
+      optional :message, :string, 2
+    end
+    add_enum "com.rookout.ControllerLogMessage.LogLevel" do
+      value :TRACE, 0
+      value :DEBUG, 1
+      value :INFO, 2
+      value :WARNING, 3
+      value :ERROR, 4
+      value :FATAL, 5
+    end
+    add_message "com.rookout.RuleStatusMessage" do
+      optional :agent_id, :string, 1
+      optional :rule_id, :string, 2
+      optional :active, :string, 3
+      optional :error, :message, 4, "com.rookout.Error"
+      optional :controller_id, :string, 5
+    end
+    add_message "com.rookout.AugReportMessage" do
+      optional :agent_id, :string, 1
+      optional :aug_id, :string, 2
+      optional :arguments, :message, 3, "com.rookout.Variant"
+      optional :report_id, :string, 4
+    end
+    add_message "com.rookout.UserMsg" do
+      optional :aug_report, :message, 1, "com.rookout.AugReportMessage"
+      optional :report_id, :string, 2
+      optional :workspace_id, :string, 3
+    end
+    add_message "com.rookout.AddAugCommand" do
+      optional :aug_json, :string, 1
+    end
+    add_message "com.rookout.InitialAugsCommand" do
+      repeated :augs, :string, 1
+    end
+    add_message "com.rookout.RemoveAugCommand" do
+      optional :aug_id, :string, 1
+    end
+    add_message "com.rookout.InitFinishedMessage" do
+    end
+    add_message "com.rookout.ClearAugsCommand" do
+    end
+    add_message "com.rookout.PingMessage" do
+      optional :value, :int32, 1
+    end
+    add_message "com.rookout.ErrorMessage" do
+      optional :message, :string, 1
+    end
+    add_message "com.rookout.C2CInitMessage" do
+      optional :controllerInfo, :message, 1, "com.rookout.ControllerInformation"
+      repeated :connectedAgents, :string, 2
+    end
+    add_message "com.rookout.DataOnPremTokenRequest" do
+    end
+    add_message "com.rookout.DataOnPremTokenResponse" do
+      optional :token, :bytes, 1
+    end
+    add_message "com.rookout.AgentsList" do
+      repeated :agents_ids, :string, 1
+    end
+    add_message "com.rookout.AssertAgents" do
+      repeated :agent_ids, :string, 1
+    end
+    add_message "com.rookout.InvalidateAgentsCommand" do
+      repeated :agents_ids, :string, 1
+    end
+    add_message "com.rookout.ConfirmAgentsCommand" do
+      repeated :agents_ids, :string, 1
+    end
+    add_message "com.rookout.AgentIdHitCountPair" do
+      optional :agent_id, :string, 1
+      optional :hit_count, :uint32, 2
+    end
+    add_message "com.rookout.HitCountUpdateMessage" do
+      optional :aug_id, :string, 1
+      repeated :hitCountInformation, :message, 2, "com.rookout.AgentIdHitCountPair"
+    end
+    add_message "com.rookout.C2cRpcRequest" do
+      optional :request_uuid, :string, 1
+      optional :message, :message, 2, "google.protobuf.Any"
+    end
+    add_message "com.rookout.C2cRpcResponse" do
+      optional :request_uuid, :string, 1
+      optional :message, :message, 2, "google.protobuf.Any"
+      optional :error, :string, 3
+    end
+  end
+end
+
+module Com
+  module Rookout
+    NewAgentMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.NewAgentMessage").msgclass
+    StackFrame = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.StackFrame").msgclass
+    StackTrace = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.StackTrace").msgclass
+    Exception = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.Exception").msgclass
+    LogMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.LogMessage").msgclass
+    LogMessage::LogLevel = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.LogMessage.LogLevel").enummodule
+    ControllerLogMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.ControllerLogMessage").msgclass
+    ControllerLogMessage::Error = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.ControllerLogMessage.Error").msgclass
+    ControllerLogMessage::LogLevel = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.ControllerLogMessage.LogLevel").enummodule
+    RuleStatusMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.RuleStatusMessage").msgclass
+    AugReportMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.AugReportMessage").msgclass
+    UserMsg = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.UserMsg").msgclass
+    AddAugCommand = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.AddAugCommand").msgclass
+    InitialAugsCommand = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.InitialAugsCommand").msgclass
+    RemoveAugCommand = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.RemoveAugCommand").msgclass
+    InitFinishedMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.InitFinishedMessage").msgclass
+    ClearAugsCommand = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.ClearAugsCommand").msgclass
+    PingMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.PingMessage").msgclass
+    ErrorMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.ErrorMessage").msgclass
+    C2CInitMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.C2CInitMessage").msgclass
+    DataOnPremTokenRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.DataOnPremTokenRequest").msgclass
+    DataOnPremTokenResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.DataOnPremTokenResponse").msgclass
+    AgentsList = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.AgentsList").msgclass
+    AssertAgents = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.AssertAgents").msgclass
+    InvalidateAgentsCommand = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.InvalidateAgentsCommand").msgclass
+    ConfirmAgentsCommand = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.ConfirmAgentsCommand").msgclass
+    AgentIdHitCountPair = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.AgentIdHitCountPair").msgclass
+    HitCountUpdateMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.HitCountUpdateMessage").msgclass
+    C2cRpcRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.C2cRpcRequest").msgclass
+    C2cRpcResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("com.rookout.C2cRpcResponse").msgclass
+  end
+end