snmp2mkr 0.1.0

data/lib/snmp2mkr/config_types/meta_definition.rb

@@ -0,0 +1,28 @@
+require 'snmp2mkr/config_types/base'
+require 'snmp2mkr/config_types/oid'
+require 'snmp2mkr/config_types/value_definition'
+
+module Snmp2mkr
+  module ConfigTypes
+    class MetaDefinition < Base
+      def setup(h)
+        raise TypeError, "#{self.class} must be given a Hash" unless h.kind_of?(Hash)
+        raise ArgumentError, "#{self.class} hash must have 'keys' (Hash)" unless h['keys'].kind_of?(Hash)
+        raise ArgumentError, "#{self.class} hash must have 'values' (Hash)" unless h['values'].kind_of?(Hash)
+
+        @keys = h['keys'].map do |k, v|
+          [k, Oid.new(v)]
+        end.to_h
+        @values = h['values'].map do |k, v|
+          [k, ValueDefinition.new([k,v])]
+        end.to_h
+      end
+
+      attr_reader :keys, :values
+
+      def collect_children
+        [*@keys.values, *@values.values]
+      end
+    end
+  end
+end

data/lib/snmp2mkr/config_types/metric_definition.rb

@@ -0,0 +1,34 @@
+require 'snmp2mkr/config_types/base'
+require 'snmp2mkr/config_types/oid'
+require 'snmp2mkr/config_types/template_string'
+require 'snmp2mkr/config_types/transformation'
+
+module Snmp2mkr
+  module ConfigTypes
+    class MetricDefinition < Base
+      def setup(kv)
+        raise TypeError, "#{self.class} must be given an Array (BUG?)" unless kv.kind_of?(Array) && kv.size == 2
+        k, v = kv
+
+        raise ArgumentError, "#{self.class} key must be String" unless k.kind_of?(String)
+        raise ArgumentError, "#{self.class} value must be String or Hash" unless v.kind_of?(String) || v.kind_of?(Hash)
+
+        if v.kind_of?(String)
+          v = {'oid' => v}
+        end
+
+        @name = TemplateString.new(k)
+        @oid = Oid.new(TemplateString.new(v.fetch('oid')))
+        ts = v['transformations'] || v['transform'] || []
+        raise TypeError, "#{self.class} transformations must be an Array" unless ts.kind_of?(Array)
+        @transformations = ts.map { |_| Transformation.new(_) }
+      end
+
+      attr_reader :name, :oid, :transformations
+
+      def collect_children
+        [@name, @oid, *@transformations]
+      end
+    end
+  end
+end

data/lib/snmp2mkr/config_types/metric_definition_collection.rb

@@ -0,0 +1,26 @@
+require 'snmp2mkr/config_types/base'
+require 'snmp2mkr/config_types/metric_definition'
+
+module Snmp2mkr
+  module ConfigTypes
+    class MetricDefinitionCollection < Base
+      def setup(hash)
+        if !hash.kind_of?(Hash) || hash.any? { |k,v| !k.kind_of?(String) }
+          raise TypeError, "#{self.class} must be given a Hash<String, Object)>"
+        end
+
+        @value = hash.map { |k, v| [k, MetricDefinition.new([k,v])] }.to_h
+      end
+
+      def [](k)
+        @value[k]
+      end
+
+      def collect_children
+        @value.values
+      end
+
+      attr_reader :value
+    end
+  end
+end

data/lib/snmp2mkr/config_types/metric_discovery_rule.rb

@@ -0,0 +1,32 @@
+require 'snmp2mkr/config_types/base'
+require 'snmp2mkr/config_types/metric_definition_collection'
+
+module Snmp2mkr
+  module ConfigTypes
+    class MetricDiscoveryRule < Base
+      def setup(hash)
+        metrics_hash = hash.fetch('metrics')
+        if !metrics_hash.kind_of?(Hash) || metrics_hash.any? { |k,v| !k.kind_of?(String) }
+          raise TypeError, "#{self.class} metrics must be a Hash<String, Object)>"
+        end
+
+        keys_hash = hash.fetch('keys')
+        if !keys_hash.kind_of?(Hash) || keys_hash.any? { |k,v| !k.kind_of?(String) }
+          raise TypeError, "#{self.class} keys must be a Hash<String, Object>"
+        end
+
+        @metrics = MetricDefinitionCollection.new(metrics_hash)
+        @keys = keys_hash.map { |k, v| [k, Oid.new(v)] }.to_h
+      end
+
+      attr_reader :metrics, :keys
+
+      def collect_children
+        [
+          @metrics,
+          *@keys.values,
+        ]
+      end
+    end
+  end
+end

data/lib/snmp2mkr/config_types/metric_discovery_rule_collection.rb

@@ -0,0 +1,26 @@
+require 'snmp2mkr/config_types/base'
+require 'snmp2mkr/config_types/metric_discovery_rule'
+
+module Snmp2mkr
+  module ConfigTypes
+    class MetricDiscoveryRuleCollection < Base
+      def setup(hash)
+        if !hash.kind_of?(Hash) || hash.any? { |k,v| !k.kind_of?(String) }
+          raise TypeError, "#{self.class} must be given a Hash<String, Object)>"
+        end
+
+        @value = hash.map { |k, v| [k, MetricDiscoveryRule.new(v)] }.to_h
+      end
+
+      def [](k)
+        @value[k]
+      end
+
+      def collect_children
+        @value.values
+      end
+
+      attr_reader :value
+    end
+  end
+end

data/lib/snmp2mkr/config_types/oid.rb

@@ -0,0 +1,24 @@
+require 'snmp2mkr/config_types/base'
+require 'snmp2mkr/config_types/template_string'
+require 'snmp2mkr/config_types/raw_string'
+
+module Snmp2mkr
+  module ConfigTypes
+    class Oid < Base
+      def setup(str)
+        raise TypeError, 'Oid must be given a String or a TemplateString' unless str.kind_of?(String) || str.kind_of?(TemplateString)
+        @value = str.is_a?(String) ? RawString.new(str) : str
+      end
+
+      attr_reader :value
+
+      def collect_children
+        [@value]
+      end
+
+      def evaluate(context: binded_context, previous: nil)
+        value.evaluate(context: context, previous: previous)
+      end
+    end
+  end
+end

data/lib/snmp2mkr/config_types/raw_string.rb

@@ -0,0 +1,14 @@
+require 'snmp2mkr/config_types/base'
+
+module Snmp2mkr
+  module ConfigTypes
+    class RawString < Base
+      def setup(str)
+        raise TypeError, "#{self.class} must be given a String" unless str.kind_of?(String)
+        @value = str
+      end
+
+      attr_reader :value
+    end
+  end
+end

data/lib/snmp2mkr/config_types/root.rb

@@ -0,0 +1,38 @@
+require 'snmp2mkr/config_types/base'
+
+require 'snmp2mkr/config_types/host_collection'
+require 'snmp2mkr/config_types/template_collection'
+require 'snmp2mkr/config_types/env_string'
+
+module Snmp2mkr
+  module ConfigTypes
+    class Root < Base
+      def setup(hash)
+        raise TypeError, "#{self.class} must be given a Hash" unless hash.kind_of?(Hash)
+
+        @mib_no_default_modules = hash.fetch('mib_no_default_modules', false)
+
+        @mib_load_path = hash.fetch('mib_load_path', [])
+        raise TypeError, "mib_load_path should be an Array" unless @mib_load_path.kind_of?(Array)
+        @mib_modules = hash.fetch('mib_load_path', [])
+        raise TypeError, "mib_modules should be an Array" unless @mib_modules.kind_of?(Array)
+
+        @hosts = HostCollection.new(hash.fetch('hosts'))
+        @templates = TemplateCollection.new(hash.fetch('templates'))
+        @persist_file = EnvString.new(hash.fetch('persist_file'))
+        @api_key = EnvString.new(hash.fetch('api_key'))
+      end
+
+      def collect_children
+        [
+          @hosts,
+          @templates,
+          @persist_file,
+          @api_key,
+        ].compact
+      end
+
+      attr_reader :hosts, :templates, :persist_file, :api_key, :mib_load_path, :mib_modules, :mib_no_default_modules
+    end
+  end
+end

data/lib/snmp2mkr/config_types/template.rb

@@ -0,0 +1,41 @@
+require 'snmp2mkr/config_types/base'
+
+require 'snmp2mkr/config_types/templates_list'
+require 'snmp2mkr/config_types/metric_definition_collection'
+require 'snmp2mkr/config_types/meta_definition'
+require 'snmp2mkr/config_types/interfaces_definition'
+require 'snmp2mkr/config_types/metric_discovery_rule_collection'
+require 'snmp2mkr/config_types/vhost_discovery_rule_collection'
+
+module Snmp2mkr
+  module ConfigTypes
+    class Template < Base
+      def setup(kv)
+        raise TypeError, "#{self.class} must be given an Array (BUG?)" unless kv.kind_of?(Array) && kv.size == 2
+        k, hash = kv
+        @name = k
+        raise TypeError, "#{self.class} must be given a Hash" unless hash.kind_of?(Hash)
+
+        @templates = hash.key?('templates') ? TemplatesList.new(hash['templates']) : nil
+        @metrics = hash.key?('metrics') ? MetricDefinitionCollection.new(hash['metrics']) : nil
+        @meta = hash.key?('meta') ? MetaDefinition.new(hash['meta']) : nil
+        @interfaces = hash.key?('interfaces') ? InterfacesDefinition.new(hash['interfaces']) : nil
+        @metric_discoveries = hash.key?('metric_discoveries') ? MetricDiscoveryRuleCollection.new(hash['metric_discoveries']) : nil
+        @vhost_discoveries = hash.key?('vhost_discoveries') ? VhostDiscoveryRuleCollection.new(hash['vhost_discoveries']) : nil
+      end
+
+      def collect_children
+        [
+          @templates,
+          @metrics,
+          @meta,
+          @interfaces,
+          @metric_discoveries,
+          @vhost_discoveries,
+        ].compact
+      end
+
+      attr_reader :name, :templates, :metrics, :meta, :interfaces, :metric_discoveries, :vhost_discoveries
+    end
+  end
+end

data/lib/snmp2mkr/config_types/template_collection.rb

@@ -0,0 +1,30 @@
+require 'snmp2mkr/config_types/base'
+require 'snmp2mkr/config_types/template'
+
+module Snmp2mkr
+  module ConfigTypes
+    class TemplateCollection < Base
+      def setup(hash)
+        if !hash.kind_of?(Hash) || hash.any? { |k,v| !k.kind_of?(String) }
+          raise TypeError, "#{self.class} must be given a Hash<String, Object)>"
+        end
+
+        @value = hash.map { |k, v| [k, Template.new([k,v])] }.to_h
+      end
+
+      def [](k)
+        @value[k]
+      end
+
+      def fetch(k)
+        @value.fetch(k)
+      end
+
+      def collect_children
+        @value.values
+      end
+
+      attr_reader :value
+    end
+  end
+end

data/lib/snmp2mkr/config_types/template_string.rb

@@ -0,0 +1,22 @@
+require 'snmp2mkr/config_types/base'
+
+module Snmp2mkr
+  module ConfigTypes
+    class TemplateString < Base
+      class MissingContextVariable < StandardError; end
+
+      def setup(str)
+        raise TypeError, 'TemplateString must be given a String' unless str.kind_of?(String)
+        @value = str
+      end
+
+      attr_reader :value
+
+      def evaluate(context: binded_context, previous: nil)
+        @value.gsub(/\#{(.+?)}/) do |_|
+          context[$1 || $1.to_sym] or raise MissingContextVariable.new("variable #{$1.inspect} is missing from context")
+        end
+      end
+    end
+  end
+end

data/lib/snmp2mkr/config_types/templates_list.rb

@@ -0,0 +1,26 @@
+require 'snmp2mkr/config_types/base'
+require 'snmp2mkr/config_types/template_string'
+
+module Snmp2mkr
+  module ConfigTypes
+    class TemplatesList < Base
+      def setup(ary)
+        if !ary.kind_of?(Array) || ary.any? { |_| !_.kind_of?(String) }
+          raise TypeError, "#{self.class} must be given an Array<String>"
+        end
+
+        @value = ary.map { |_| TemplateString.new(_) }
+      end
+
+      def collect_children
+        @value
+      end
+
+      attr_reader :value
+
+      def evaluate(context: binded_context, previous: nil)
+        @value.map { |_| _.evaluate(context: context, previous: previous) }
+      end
+    end
+  end
+end

data/lib/snmp2mkr/config_types/transformation.rb

@@ -0,0 +1,26 @@
+require 'snmp2mkr/config_types/base'
+require 'snmp2mkr/config_types/transformation'
+
+module Snmp2mkr
+  module ConfigTypes
+    class Transformation < Base
+      def setup(h)
+        raise TypeError, "#{self.class} must be given an Hash" unless h.kind_of?(Hash)
+
+        @type = h.fetch('type')
+        raise TypeError, "#{self.class} type must be a String"  unless @type.kind_of?(String)
+
+        case @type
+        when 'persec'
+          # do nothing
+        else
+          raise ArgumentError, "#{self.class} doesn't know type #{type.inspect}"
+        end
+      end
+
+      attr_reader :type
+    end
+  end
+end
+
+

data/lib/snmp2mkr/config_types/value_definition.rb

@@ -0,0 +1,50 @@
+require 'snmp2mkr/config_types/base'
+require 'snmp2mkr/config_types/oid'
+require 'snmp2mkr/config_types/template_string'
+require 'snmp2mkr/config_types/raw_string'
+require 'snmp2mkr/config_types/transformation'
+
+module Snmp2mkr
+  module ConfigTypes
+    class ValueDefinition < Base
+      def setup(kv)
+        raise TypeError, "#{self.class} must be given an Array (BUG?)" unless kv.kind_of?(Array) && kv.size == 2
+        k, v = kv
+        @name = k
+
+        raise ArgumentError, "#{self.class} key must be String" unless k.kind_of?(String)
+        raise ArgumentError, "#{self.class} value must be String or Hash" unless v.kind_of?(String) || v.kind_of?(Hash)
+
+        if v.kind_of?(String)
+          v = {'value' => TemplateString.new(v), 'type' => 'string'}
+        end
+
+        @type = v.fetch('type')
+        raise TypeError, "#{self.class} value, key 'type' must be a String" unless @type.kind_of?(String)
+
+        case @type
+        when 'array_append', 'string'
+          @value = v['value']
+          @value = TemplateString.new(v['value']) if @value.kind_of?(String)
+        else
+          raise ArgumentError, "#{self.class} doesn't know type #{@type.inspect}"
+        end
+      end
+
+      attr_reader :name, :type, :value
+
+      def evaluate(context: binded_context, previous: nil)
+        case @type
+        when 'string'
+          @value.evaluate
+        when 'array_append'
+          [*previous, @value.evaluate]
+        end
+      end
+
+      def collect_children
+        [@value]
+      end
+    end
+  end
+end

data/lib/snmp2mkr/config_types/vhost_discovery_rule.rb

@@ -0,0 +1,43 @@
+require 'snmp2mkr/config_types/base'
+require 'snmp2mkr/config_types/template_string'
+require 'snmp2mkr/config_types/metric_definition_collection'
+require 'snmp2mkr/config_types/oid'
+
+module Snmp2mkr
+  module ConfigTypes
+    class VhostDiscoveryRule < Base
+      def setup(hash)
+        metrics_hash = hash.fetch('metrics')
+        if !metrics_hash.kind_of?(Hash) || metrics_hash.any? { |k,v| !k.kind_of?(String) }
+          raise TypeError, "#{self.class} metrics must be a Hash<String, Object)>"
+        end
+
+        keys_hash = hash.fetch('keys')
+        if !keys_hash.kind_of?(Hash) || keys_hash.any? { |k,v| !k.kind_of?(String) }
+          raise TypeError, "#{self.class} keys must be a Hash<String, Object>"
+        end
+
+        roles = hash.fetch('roles', [])
+        if !roles.kind_of?(Array) || roles.any? { |k,v| !k.kind_of?(String) }
+          raise TypeError, "#{self.class} roles must be a Array<String>"
+        end
+
+        @keys = keys_hash.map { |k, v| [k, Oid.new(v)] }.to_h
+        @name = TemplateString.new(hash.fetch('name'))
+        @roles = roles.map { |v| TemplateString.new(v) }
+        @metrics = MetricDefinitionCollection.new(metrics_hash)
+      end
+
+      attr_reader :metrics, :keys, :roles, :name
+
+      def collect_children
+        [
+          *@metrics,
+          *@keys.values,
+          *@roles,
+          @name,
+        ]
+      end
+    end
+  end
+end