polyseerio 0.0.1

data/lib/agent/enum.rb

@@ -0,0 +1,35 @@
+module Polyseerio
+  module Agent
+    module Enum
+      module Expectation
+        IS_ALIVE = :is_alive
+      end
+
+      module Event
+        START = :start
+        STOP  = :stop
+      end
+
+      module Fact
+        PID  = :pid
+        GID  = :gid
+      end
+
+      module Signal
+        SIGHUP  = :HUP
+        SIGINT  = :INT
+        SIGTERM = :TERM
+      end
+
+      module Process
+        EXIT = :exit
+      end
+
+      module Metric
+        MEMORY = :memory
+        CPU    = :cpu
+        UPTIME = :uptime
+      end
+    end
+  end
+end

data/lib/agent/executor.rb

@@ -0,0 +1,62 @@
+require 'concurrent'
+require 'helper'
+require 'resource/routine'
+require 'agent/handler/index'
+require 'agent/helper'
+require 'agent/default_config'
+
+module Polyseerio
+  module Agent
+    # Agent executor functions.
+    module Executor
+      # Setsup a client's agent.
+      def self.setup(client, options = {})
+        options = Polyseerio::Helper.defaults(options, DEFAULT_CONFIG)
+
+        Concurrent::Promise.new do
+          if options[:attach]
+            name = Helper.resolve_name(options)
+
+            # Resolve the instance.
+            # TODO: better handle failure to upsert.
+            instance = Resource::Routine.upsert(
+              client.Instance,
+              name: name
+            ).execute.value
+
+            # Set the client's instance.
+            client.instance = instance
+
+            # Create a setup handler.
+            handler_options = Helper.extract_handler_options options
+            setup_handler = Helper.setup_with_handler(
+              Polyseerio::Agent::Handler::MAP,
+              client,
+              handler_options
+            )
+
+            # Gather setup operations.
+            setups = handler_options.map { |key, _| setup_handler.call(key) }
+
+            # Perform setups.
+            Concurrent::Promise.zip(*setups).execute.value
+
+            # Start monitoring.
+            instance.attach.execute.value
+
+            instance
+          end
+        end
+      end
+
+      # Tearsdown a client's agent.
+      def self.teardown(client, instance)
+        Concurrent::Promise.new do
+          teardown_handler('event', client)
+
+          instance.detach.excecute.value
+        end
+      end
+    end
+  end
+end

data/lib/agent/handler/event.rb

@@ -0,0 +1,32 @@
+require 'agent/enum'
+
+module Polyseerio
+  module Agent
+    # Introduce event handlers.
+    module Handler
+      include Agent::Enum
+
+      def self.event
+        {
+          Event::START => proc do |_config, client|
+            client.Event.create(
+              name: "#{client.instance.name} agent has started.",
+              color: Polyseerio::Enum::Color::GREEN,
+              icon: Polyseerio::Enum::Icon::CHAIN
+            )
+          end,
+
+          Event::STOP => {
+            Interface::TEARDOWN => proc do |_config, client|
+              client.Event.create(
+                name: "#{client.instance.name} agent has stopped.",
+                color: Polyseerio::Enum::Color::ORANGE,
+                icon: Polyseerio::Enum::Icon::CHAIN_BROKER
+              )
+            end
+          }
+        }
+      end
+    end
+  end
+end

data/lib/agent/handler/expectation.rb

@@ -0,0 +1,10 @@
+module Polyseerio
+  module Agent
+    # Introduce expectation handlers.
+    module Handler
+      def self.expectation
+        {}
+      end
+    end
+  end
+end

data/lib/agent/handler/fact.rb

@@ -0,0 +1,18 @@
+module Polyseerio
+  module Agent
+    # Introduce fact handlers.
+    module Handler
+      def self.fact
+        {
+          Fact::PID => proc do |_config, client|
+            client.instance.set_fact('pid', ::Process.pid)
+          end,
+
+          Fact::GID => proc do |_config, client|
+            client.instance.set_fact('gid', ::Process.gid)
+          end
+        }
+      end
+    end
+  end
+end

data/lib/agent/handler/index.rb

@@ -0,0 +1,15 @@
+require 'helper'
+
+module Polyseerio
+  module Agent
+    # Contains handlers and handler map.
+    module Handler
+      MAP = Polyseerio::Helper.dir_proc_map(File.dirname(__FILE__), self)
+
+      # Extracts handler type options from agent options.
+      def self.extract_options(options)
+        options.select { |key, _| MAP.key? key }
+      end
+    end
+  end
+end

data/lib/agent/handler/interface.rb

@@ -0,0 +1,11 @@
+module Polyseerio
+  module Agent
+    module Handler
+      # Interface used for handlers.
+      module Interface
+        SETUP    = :setup
+        TEARDOWN = :teardown
+      end
+    end
+  end
+end

data/lib/agent/handler/metric.rb

@@ -0,0 +1,23 @@
+module Polyseerio
+  module Agent
+    # Introduce metric handlers.
+    module Handler
+      def self.metric
+        {
+          Metric::MEMORY => proc do |_config, client|
+            client.instance.add_gauge 'memory', 0
+          end,
+
+          Metric::CPU => proc do |_config, client|
+            client.instance.add_gauge 'cpu_user', 0
+            client.instance.add_gauge 'cpu_system', 0
+          end,
+
+          Metric::UPTIME => proc do |_config, client|
+            client.instance.add_gauge 'uptime', 0
+          end
+        }
+      end
+    end
+  end
+end

data/lib/agent/handler/process.rb

@@ -0,0 +1,10 @@
+module Polyseerio
+  module Agent
+    # Introduce process handlers.
+    module Handler
+      def self.process
+        {}
+      end
+    end
+  end
+end

data/lib/agent/helper.rb

@@ -0,0 +1,137 @@
+require 'functional'
+require 'concurrent'
+require 'agent/handler/index'
+require 'agent/handler/interface'
+
+module Polyseerio
+  module Agent
+    # Helper functions for Agent
+    module Helper
+      # Returns an iterator for a handler subtype.
+      def self.create_subtype_iterator(handlers, iteratee, *args)
+        proc do |key, value|
+          if handlers.key? key.to_sym # TODO: unit-test to_sym
+            iteratee.call(handlers, key, value, *args)
+          else
+            Concurrent::Promise.reject('Could not find a handler subtype: ' \
+              "#{key}.")
+          end
+        end
+      end
+
+      # Returns a handler function.
+      def self.create_handler(iteratee)
+        proc do |map, client, config, type, *args|
+          if map.key? type.to_sym
+            if config.key? type
+              # TODO: unit-test passed args
+              work = config.fetch(type)
+                           .map(&iteratee.call(map.fetch(type), client, *args))
+
+              Concurrent::Promise.zip(*work)
+            else
+              Concurrent::Promise.fulfill []
+            end
+          else
+            Concurrent::Promise.reject('Could not find a handler type: ' \
+              "#{type}.")
+          end
+        end
+      end
+
+      # Returns a setup handler function.
+      @setup = proc do |handlers_outer, *args_outer|
+        setup_iterator = proc do |handlers, key, value, *args|
+          handler = handlers.fetch(key)
+
+          if handler.respond_to? :call
+            handler.call(value, *args) # TODO: unit-test passed args
+          elsif handler.key? Handler::Interface::SETUP
+            # TODO: unit-test passed args
+            handler[Handler::Interface::SETUP].call(value, *args)
+          else
+            Concurrent::Promise.fulfill
+          end
+        end
+
+        create_subtype_iterator(handlers_outer, setup_iterator, *args_outer)
+      end
+
+      @teardown = proc do |handlers_outer, *args_outer|
+        teardown_iterator = proc do |handlers, key, _value, *args|
+          if handlers.fetch(key).key? Handler::Interface::TEARDOWN
+            handlers.fetch(key).fetch(Handler::Interface::TEARDOWN).call(*args)
+          else
+            Concurrent::Promise.fulfill
+          end
+        end
+
+        create_subtype_iterator(handlers_outer, teardown_iterator, *args_outer)
+      end
+
+      # Sets up a handler type.
+      def self.setup_with_handler(*args)
+        create_handler(@setup).curry.call(*args)
+      end
+
+      # Tears down a handler type.
+      def self.teardown_with_handler(*args)
+        create_handler(@teardown).curry.call(*args)
+      end
+
+      # Returns a unique name.
+      def self.generate_name
+        'ruby-instance'
+      end
+
+      # Reduce handler options based on if they are enabled.
+      def self.reduce_handler_option
+        proc do |(name, config), acc|
+          acc[name] = config if handle? config
+
+          acc
+        end
+      end
+
+      # Given handler options, enabled subtype options are returned.
+      def self.filter_enabled_handler_options(options)
+        options.each_with_object({}, &reduce_handler_option)
+      end
+
+      # Given agent options, handlers options are returned.
+      def self.filter_handlers(options)
+        options.each_with_object({}) do |(name, config), acc|
+          acc[name] = filter_enabled_handler_options config
+
+          acc
+        end
+      end
+
+      # Extracts handler options and then filters them based on if they
+      # are enabled or not. TODO: unit-test or integration-test
+      def self.extract_handler_options(options)
+        options = Handler.extract_options options
+
+        filter_handlers options
+      end
+
+      # Determines if a handler configuration should be handled.
+      def self.handle?(value)
+        return value if value == true
+
+        if Functional::TypeCheck::Type?(value, Hash) && (value.key? :enabled)
+          return value[:enabled]
+        end
+
+        false
+      end
+
+      # Given an agent config a name will be returned.
+      def self.resolve_name(config)
+        return config[:name] if config.key?(:name) && !config[:name].nil?
+
+        generate_name
+      end
+    end
+  end
+end

data/lib/client.rb

@@ -0,0 +1,47 @@
+require 'agent/agent'
+require 'helper'
+
+module Polyseerio
+  # The Polyseer.io client.
+  class Client
+    INITIALIZE_DEFAULTS = {
+      agent_class: Agent::Agent,
+      request: nil
+    }.freeze
+
+    attr_accessor :cid, :agent, :instance
+    attr_writer :instance
+
+    def initialize(cid, options = {})
+      options = Helper.defaults(options, INITIALIZE_DEFAULTS)
+
+      if options[:request].nil?
+        raise ArgumentError, 'Cannot create an instance of Client without' \
+          'passing a request instance.'
+      end
+
+      if options.key? :resources
+        options[:resources].each(&Helper.attach_to_instance!(self))
+      end
+
+      @cid = cid
+      @agent = nil
+      @instance = nil
+      @request = options[:request]
+      @agent_class = options[:agent_class]
+    end
+
+    # Start an agent using this client.
+    def start_agent(*args)
+      raise 'Agent has already started.' unless @agent.nil?
+
+      @agent = @agent_class.new(self)
+
+      @agent.start(*args)
+    end
+
+    private
+
+    attr_accessor :request, :agent_class
+  end
+end

data/lib/constant.rb

@@ -0,0 +1,12 @@
+module Polyseerio
+  module Constant
+    ACCESS_TOKEN_HEADER  = 'X-AUTH-HEADER'.freeze
+    DEFAULT_API_BASE_URL = 'api.polyseer.io/polyseer'.freeze
+    DEFAULT_API_PROTOCOL = 'https://'.freeze
+    DEFAULT_API_VERSION  = 'v1'.freeze
+    DEFAULT_ENV          = 'RAILS_ENV'.freeze
+    DEFAULT_ENVIRONMENT  = 'development'.freeze
+    DEFAULT_TIMEOUT      = 10_000
+    DEFAULT_TOKEN_ENV    = 'POLYSEERIO_TOKEN'.freeze
+  end
+end

data/lib/enum.rb

@@ -0,0 +1,82 @@
+require 'helper'
+
+module Polyseerio
+  module Enum
+    # Contains resource routes.
+    module Resource
+      ALERT       = 'alerts'.freeze
+      CHANNEL     = 'channels'.freeze
+      ENVIRONMENT = 'environments'.freeze
+      EVENT       = 'events'.freeze
+      EXPECTATION = 'expectations'.freeze
+      INSTANCE    = 'instances'.freeze
+      LOGIC_BLOCK = 'logic-blocks'.freeze
+      MEMBER      = 'members'.freeze
+      MESSAGE     = 'messages'.freeze
+      SETTING     = 'settings'.freeze
+      TASK        = 'tasks'.freeze
+    end
+
+    # Contains type constants for use with API calls.
+    module Type
+      Resource.constants.each do |key|
+        value = Helper.resource_to_type Resource.const_get(key)
+
+        const_set(key, value)
+      end
+    end
+
+    module Determiner
+      ONE  = 'one'.freeze
+      SOME = 'some'.freeze
+    end
+
+    module Direction
+      INBOUND  = 'inbound'.freeze
+      OUTBOUND = 'outbound'.freeze
+    end
+
+    module Subtype
+      LONG_RUNNING = 'long_running'.freeze
+      PERIODIC     = 'periodic'.freeze
+    end
+
+    module Color
+      BLUE   = 'blue'.freeze
+      BROWN  = 'brown'.freeze
+      GREEN  = 'green'.freeze
+      ORANGE = 'orange'.freeze
+      PURPLE = 'purple'.freeze
+      RED    = 'red'.freeze
+      TEAL   = 'teal'.freeze
+      WHITE  = 'white'.freeze
+      YELLOW = 'yellow'.freeze
+      NONE   = nil
+    end
+
+    module Icon
+      THUMBS_UP    = 'thumbs-up'.freeze
+      CALENDAR     = 'calendar'.freeze
+      SERVER       = 'server'.freeze
+      SIGNAL       = 'wifi'.freeze
+      GIT          = 'git'.freeze
+      CODE         = 'code'.freeze
+      CHECK        = 'check'.freeze
+      ERROR        = 'exclamation-triangle'.freeze
+      PENCIL       = 'pencil'.freeze
+      CHAIN        = 'chain'.freeze
+      CHAIN_BROKEN = 'chain-broken'.freeze
+      NONE         = nil
+    end
+
+    module Protocol
+      SMTP = 'smtp'.freeze
+      SMS  = 'sms'.freeze
+    end
+
+    module Strategy
+      FALLBACK = :fallback
+      ID       = :id
+    end
+  end
+end