huginn_ruby_agent 0.1 → 0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 46786fbef6ebd0f956499b17915fa8970f4285795073e3f55a9f6d026e8f9597
-  data.tar.gz: 21cf1e531eb5652e34eecf0a8b0151458902f4bee65fe69b5c4110cfe5b2c0f5
+  metadata.gz: 602117200b2aadebec8c4c6e959c9ab19c4bb585df0a9c72b7926f8b0adc8638
+  data.tar.gz: 65d559d7e90b569f74c8cd890786ac830d469fd58923c75f50399af64c914b1e
 SHA512:
-  metadata.gz: 2a5c0ebee329f780bdc08637a03803de430cf16753f27f14456380c4cedd598b85a8398a4eb94c1c4355e75a9fc095279e0be90c4032872d8c3159ba934f879b
-  data.tar.gz: 2ecc2518b29ef5f6edfc432325670f8ad09d7f29bb70cd25bc0d09329a87dae0cf9f8053714fd1a335772ce06fd3e106533d9d273ac29034f607aa6399ed433d
+  metadata.gz: 540dd8062b8fed871c15a5f784de8eece0317480a6e59025d4345a86267409660b187f37ca4f03ce48e89b0d8b685b196fc9c095e7ca6c20cb7335b15d42960c
+  data.tar.gz: c68d5ef8c376d7af0121378ffc4323cae0bd51a78802716990c0f9b62dd6d11f7646697d74b73934735cc182a6ba12032682b00caaf61a652cd2dcf6dd4f175e

data/lib/huginn_ruby_agent/ruby_agent.rb

@@ -1,29 +1,238 @@
 # frozen_string_literal: true
 
+require 'date'
+require 'cgi'
+require 'tempfile'
+require 'base64'
+
+# https://stackoverflow.com/questions/23884526/is-there-a-safe-way-to-eval-in-ruby-or-a-better-way-to-do-this
 module Agents
   class RubyAgent < Agent
-    default_schedule '12h'
+    include FormConfigurable
+
+    can_dry_run!
+
+    default_schedule "never"
+
+    gem_dependency_check { defined?(MiniRacer) }
 
     description <<-MD
-      Add a Agent description here
+      The Ruby Agent allows you to write code in Ruby that can create and receive events.  If other Agents aren't meeting your needs, try this one!
+
+      You should put code in the `code` option.
+
+      You can implement `Agent.check` and `Agent.receive` as you see fit.  The following methods will be available on Agent:
+
+      * `createEvent(payload)`
+      * `incomingEvents()` (the returned event objects will each have a `payload` property)
+      * `memory()`
+      * `memory(key)`
+      * `memory(keyToSet, valueToSet)`
+      * `setMemory(object)` (replaces the Agent's memory with the provided object)
+      * `deleteKey(key)` (deletes a key from memory and returns the value)
+      * `credential(name)`
+      * `credential(name, valueToSet)`
+      * `options()`
+      * `options(key)`
+      * `log(message)`
+      * `error(message)`
     MD
 
+    form_configurable :code, type: :text, ace: true
+    form_configurable :expected_receive_period_in_days
+    form_configurable :expected_update_period_in_days
+
+    def validate_options
+      errors.add(:base, "The 'code' option is required") unless options['code'].present?
+    end
+
+    def working?
+      return false if recent_error_logs?
+
+      if interpolated['expected_update_period_in_days'].present?
+        return false unless event_created_within?(interpolated['expected_update_period_in_days'])
+      end
+
+      if interpolated['expected_receive_period_in_days'].present?
+        return false unless last_receive_at && last_receive_at > interpolated['expected_receive_period_in_days'].to_i.days.ago
+      end
+
+      true
+    end
+
+    def check
+      log_errors do
+        execute_check
+      end
+    end
+
+    def receive(events)
+      log_errors do
+        execute_receive(events)
+      end
+    end
+
     def default_options
-      {}
+      code = <<~CODE
+
+        require "bundler/inline"
+
+        gemfile do
+          source "https://rubygems.org"
+
+          # gem "mechanize"
+        end
+
+        class Agent
+          def initialize(api)
+            @api = api
+          end
+
+          def check
+            @api.create_event({ message: 'I made an event!' })
+          end
+
+          def receive(incoming_events)
+            incoming_events.each do |event|
+              @api.create_event({ message: 'new event', event_was: event[:payload] })
+            end
+          end
+        end
+      CODE
+
+      {
+        'code' => code,
+        'expected_receive_period_in_days' => '2',
+        'expected_update_period_in_days' => '2'
+      }
     end
 
-    def validate_options; end
+    private
 
-    def working?
-      # Implement me! Maybe one of these next two lines would be a good fit?
-      # checked_without_error?
-      # received_event_without_error?
+    def execute_check
+      Bundler.with_original_env do
+        Open3.popen3("ruby", chdir: '/') do |input, output, err, thread|
+          input.write sdk_code
+          input.write code
+          input.write <<~CODE
+
+          Agent.new(Huginn::API.new).check
+
+          CODE
+          input.close
+
+
+          output.readlines.map { |line| JSON.parse(line, symbolize_names: true) }.each do |data|
+            case data[:action]
+            when 'create_event'
+              create_event(payload: data[:payload])
+            when 'log'
+              log data[:payload]
+            when 'error'
+              error data[:payload]
+            end
+          end
+
+          errors = err.read
+
+          error err.read
+          log "thread #{thread.value}"
+        end
+      end
+    end
+
+    def execute_receive(events)
+      Bundler.with_original_env do
+        Open3.popen3("ruby", chdir: '/') do |input, output, err, thread|
+          input.write sdk_code
+          input.write code
+          input.write <<~CODE
+
+          api = Huginn::API.new
+          begin
+            Agent.new(api).receive(
+              JSON.parse(
+                Base64.decode64(
+                  "#{Base64.encode64(events.to_json)}"
+                ),
+                symbolize_names: true
+              )
+            )
+          rescue StandardError => ex
+            api.error ex
+          end
+
+          CODE
+          input.close
+
+
+          output.readlines.map { |line| JSON.parse(line, symbolize_names: true) }.each do |data|
+            case data[:action]
+            when 'create_event'
+              create_event(payload: data[:payload])
+            when 'log'
+              log data[:payload]
+            when 'error'
+              error data[:payload]
+            end
+          end
+
+          errors = err.read
+
+          error err.read
+          log "thread #{thread.value}"
+        end
+      end
+    end
+
+    def code
+      interpolated['code']
     end
 
-    #    def check
-    #    end
+    def sdk_code
+      <<~CODE
+        require 'json'
+        require 'base64'
+
+        module Huginn
+          class API
+            def create_event(payload)
+              puts(
+                {
+                  action: :create_event,
+                  payload: payload
+                }.to_json
+              )
+            end
+
+            def log(message)
+              puts(
+                {
+                  action: :log,
+                  payload: message
+                }.to_json
+              )
+            end
 
-    #    def receive(incoming_events)
-    #    end
+            def error(message)
+              puts(
+                {
+                  action: :error,
+                  payload: message
+                }.to_json
+              )
+            end
+          end
+        end
+      CODE
+    end
+
+    def log_errors
+      begin
+        yield
+      rescue StandardError => e
+        error "Runtime error: #{e.message}"
+      end
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: huginn_ruby_agent
 version: !ruby/object:Gem::Version
-  version: '0.1'
+  version: '0.2'
 platform: ruby
 authors:
 - Sergei O. Udalov
@@ -52,6 +52,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: huginn_agent
   requirement: !ruby/object:Gem::Requirement
@@ -94,7 +108,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.3.7
+rubygems_version: 3.1.6
 signing_key:
 specification_version: 4
 summary: Ruby Agent for Huginn automation platform