telemetry-snmp 0.1.0

data/lib/telemetry/snmp/controllers/devices.rb

@@ -0,0 +1,94 @@
+require 'sinatra/extension'
+
+module Telemetry
+  module Snmp
+    module Controller
+      module Devices
+        extend Sinatra::Extension
+
+        get '' do
+          devices = {}
+          Telemetry::Snmp::Data::Model::Device.all.each do |device|
+            devices[device.values[:id]] = device.values
+          end
+
+          devices
+        end
+
+        get '/:id' do
+          device = Telemetry::Snmp::Data::Model::Device[params[:id]]
+          status 404 if device.nil?
+          { error: true, message: "#{params[:id]} not found" } if device.nil?
+
+          device.values
+        end
+
+        post '' do
+          body = MultiJson.load(request.body.read, symbolize_keys: true)
+
+          if body[:hostname].nil?
+            status 400
+            { error: true, message: 'hostname cannot be null for new record' }
+          end
+
+          {
+            id: Telemetry::Snmp::Data::Model::User.insert(
+              hostname: body[:hostname],
+              ip_address: body[:ip_address],
+              active: body[:active] || 1,
+              port: body[:port] || 161,
+              snmp_version: body[:snmp_version] || 3,
+              device_cred_id: body[:device_cred_id],
+              frequency: body[:frequency] || 60,
+              environment: body[:environment] || 'production',
+              datacenter: body[:datacenter],
+              zone: body[:zone],
+              created: Sequel::CURRENT_TIMESTAMP
+            ),
+            hostname: body[:hostname]
+          }
+        end
+
+        put '/:id' do
+          device = Telemetry::Snmp::Data::Model::Device[params[:id]]
+          status 404 if device.nil?
+          { error: true, message: "#{params[:id]} not found" } if device.nil?
+
+          status 405
+          { error: true, message: 'puts not supported', id: params[:id] }
+        end
+
+        patch '/:id' do
+          # id, hostname, ip_address, active, port, snmp_version, device_cred_id,
+          # frequency, environment, datacenter, zone, last_polled, created, updated
+          device = Telemetry::Snmp::Data::Model::Device[params[:id]]
+          status 404 if device.nil?
+          { error: true, message: "#{params[:id]} not found" } if device.nil?
+
+          body = MultiJson.load(request.body.read, symbolize_keys: true)
+          updates = {}
+          fields = %i[hostname ip_address active port snmp_version device_cred_id frequency environment
+                      datacenter zone]
+          fields.each { |field| updates[field] = body[field] if body.key? field }
+
+          if updates.empty?
+            status 400
+            return { error: true, message: 'no valid fields to update' }
+          end
+
+          updates[:updated] = Sequel::CURRENT_TIMESTAMP
+          device.update(**updates)
+          { error: false, updated_fields: updates.keys, id: params[:id] }
+        end
+
+        delete '/:id' do
+          device = Telemetry::Snmp::Data::Model::Device[params[:id]]
+          status 404 if device.nil?
+          { error: true, message: "#{params[:id]} not found" } if device.nil?
+
+          { error: !device.delete, id: params[:id] }
+        end
+      end
+    end
+  end
+end

data/lib/telemetry/snmp/controllers/oid_groups.rb

@@ -0,0 +1,71 @@
+require 'sinatra/extension'
+
+module Telemetry
+  module Snmp
+    module Controller
+      module OIDGroups
+        extend Sinatra::Extension
+
+        get '' do
+          groups = {}
+          Telemetry::Snmp::Data::Model::OIDGroup.all.each do |group|
+            groups[group.values[:id]] = group.values
+          end
+
+          groups
+        end
+
+        get '/:id' do
+          group = Telemetry::Snmp::Data::Model::OIDGroup[params[:id]]
+          status 404 if group.nil?
+          { error: true, message: "#{params[:id]} not found" } if group.nil?
+
+          group.values
+        end
+
+        post '' do
+          body = MultiJson.load(request.body.read, symbolize_keys: true)
+          if body[:name].nil?
+            status 400
+            { error: true, message: 'name cannot be null for new record' }
+          end
+
+          {
+            id: Telemetry::Snmp::Data::Model.OIDGroup.insert(
+              name: body[:name],
+              active: body[:active] || 1,
+              created: Sequel::CURRENT_TIMESTAMP
+            ),
+            name: body[:name]
+          }
+        end
+
+        put '/:id' do
+          group = Telemetry::Snmp::Data::Model::OIDGroup[params[:id]]
+          status 404 if group.nil?
+          { error: true, message: "#{params[:id]} not found" } if group.nil?
+
+          status 405
+          { error: true, message: 'puts not supported', id: params[:id] }
+        end
+
+        patch '/:id' do
+          group = Telemetry::Snmp::Data::Model::OIDGroup[params[:id]]
+          status 404 if group.nil?
+          { error: true, message: "#{params[:id]} not found" } if group.nil?
+
+          status 405
+          { error: true, message: 'patch not supported', id: params[:id] }
+        end
+
+        delete '/:id' do
+          group = Telemetry::Snmp::Data::Model::OIDGroup[params[:id]]
+          status 404 if group.nil?
+          { error: true, message: "#{params[:id]} not found" } if group.nil?
+
+          { success: !group.delete, id: params[:id] }
+        end
+      end
+    end
+  end
+end

data/lib/telemetry/snmp/controllers/oids.rb

@@ -0,0 +1,80 @@
+require 'sinatra/extension'
+
+module Telemetry
+  module Snmp
+    module Controller
+      module OIDs
+        extend Sinatra::Extension
+
+        get '' do
+          oids = {}
+
+          Telemetry::Snmp::Data::MOdel::OID.all.each do |oid|
+            oids[oid.values[:id]] = oid.values
+          end
+
+          oids
+        end
+
+        get '/:id' do
+          oid = Telemetry::Snmp::Data::Model::OID[params[:id]]
+          status 404 if oid.nil?
+          return { error: true, message: "#{params[:id]} not found" } if oid.nil?
+
+          oid.values
+        end
+
+        post '' do
+          body = MultiJson.load(request.body.read, symbolize_keys: true)
+          status 400 if body[:oid].nil?
+          { error: true, message: 'oid is a required field' } if body[:oid].nil?
+
+          insert = {
+            oid: body[:oid],
+            name: body[:name],
+            description: body[:description],
+            created: Sequel::CURRENT_TIMESTAMP
+          }
+
+          { id: Telemetry::Snmp::Data::Model::OID.insert(**insert), oid: body[:oid] }
+        end
+
+        put '/:id' do
+          oid = Telemetry::Snmp::Data::Model::OID[params[:id]]
+          status 404 if oid.nil?
+          return { error: true, message: "#{params[:id]} not found" } if oid.nil?
+
+          status 405
+          { error: true, message: 'puts not supported', id: params[:id] }
+        end
+
+        patch '/:id' do
+          oid = Telemetry::Snmp::Data::Model::OID[params[:id]]
+          status 404 if oid.nil?
+          return { error: true, message: "#{params[:id]} not found" } if oid.nil?
+
+          body = MultiJson.load(request.body.read, symbolize_keys: true)
+          updates = {}
+          fields = %i[oid name description]
+          fields.each { |field| updates[field] = body[field] if body.key? field }
+          if updates.empty?
+            status 400
+            return { error: true, message: 'no valid fields to update' }
+          end
+
+          updates[:updated] = Sequel::CURRENT_TIMESTAMP
+          oid.update(**updates)
+          { error: false, updated_fields: updates.keys, id: params[:id] }
+        end
+
+        delete '/:id' do
+          oid = Telemetry::Snmp::Data::Model::OID[params[:id]]
+          status 404 if oid.nil?
+          return { error: true, message: "#{params[:id]} not found" } if oid.nil?
+
+          { error: !oid.delete, id: params[:id] }
+        end
+      end
+    end
+  end
+end

data/lib/telemetry/snmp/controllers/users.rb

@@ -0,0 +1,81 @@
+require 'sinatra/extension'
+
+module Telemetry
+  module Snmp
+    module Controller
+      module Users
+        extend Sinatra::Extension
+        get '' do
+          users = {}
+          Telemetry::Snmp::Data::Model::User.all.each do |user|
+            users[user.values[:id]] = user.values
+          end
+
+          users
+        end
+
+        get '/:id' do
+          user = Telemetry::Snmp::Data::Model::User[params[:id]]
+          status 404 if user.nil?
+          return { error: true, message: "#{params[:id]} not found" } if user.nil?
+
+          user.values
+        end
+
+        post '' do
+          body = MultiJson.load(request.body.read, symbolize_keys: true)
+          insert = {
+            username: body[:username],
+            email: body[:email],
+            active: body[:active] || 1,
+            admin: body[:admin] || 0,
+            created: Sequel::CURRENT_TIMESTAMP
+          }
+
+          unless insert.values.include? nil
+            status 400
+            { error: true, message: 'missing required fields' }
+          end
+
+          { id: Telemetry::Snmp::Data::Model::User.insert(**insert), username: body[:username] }
+        end
+
+        put '/:id' do
+          user = Telemetry::Snmp::Data::Model::User[params[:id]]
+          status 404 if user.nil?
+          return { error: true, message: "#{params[:id]} not found" } if user.nil?
+
+          status 405
+          { error: true, message: 'puts not supported', id: params[:id] }
+        end
+
+        patch '/:id' do
+          user = Telemetry::Snmp::Data::Model::User[params[:id]]
+          status 404 if user.nil?
+          return { error: true, message: "#{params[:id]} not found" } if user.nil?
+
+          body = MultiJson.load(request.body.read, symbolize_keys: true)
+          updates = {}
+          fields = %i[username email active admin]
+          fields.each { |field| updates[field] = body[field] if body.key? field }
+          if updates.empty?
+            status 400
+            return { error: true, message: 'no valid fields to update' }
+          end
+
+          updates[:updated] = Sequel::CURRENT_TIMESTAMP
+          user.update(**updates)
+          { error: false, updated_fields: updates.keys, id: params[:id] }
+        end
+
+        delete '/:id' do
+          user = Telemetry::Snmp::Data::Model::User[params[:id]]
+          status 404 if user.nil?
+          return { error: true, message: "#{params[:id]} not found" } if user.nil?
+
+          { error: !user.delete, id: params[:id] }
+        end
+      end
+    end
+  end
+end

data/lib/telemetry/snmp/controllers/walks.rb

@@ -0,0 +1,89 @@
+require 'sinatra/extension'
+require 'oj'
+require 'multi_json'
+
+module Telemetry
+  module Snmp
+    module Controller
+      module Walks
+        extend Sinatra::Extension
+        # id, oid_index, oid_walk, active, measurement_name, created, updated
+
+        get '' do
+          results = {}
+          Telemetry::Snmp::Data::Model::OIDWalks.all.each do |walk|
+            results[walk.values[:id]] = walk.values
+          end
+
+          results
+        end
+
+        get '/:id' do
+          result = Telemetry::Snmp::Data::Model::OIDWalks[params[:id]]
+          return {} if result.nil?
+
+          result.values
+        end
+
+        post '' do
+          body = MultiJson.load(request.body.read, symbolize_keys: true)
+          insert = {
+            oid_index: body[:oid_index],
+            oid_walk: body[:oid_walk],
+            active: body[:active] || 1,
+            measurement_name: body[:measurement_name] || 'snmp',
+            created: Sequel::CURRENT_TIMESTAMP
+          }
+
+          if insert[:oid_index].nil? || insert[:oid_walk].nil?
+            status 400
+            { error: true, missing_oid_index: insert[:oid_index].nil?, missing_oid_walk: insert[:oid_walk].nil? }
+          end
+
+          { id: Telemetry::Snmp::Data::Model::OIDWalks.insert(**insert), error: false }
+        end
+
+        patch '/:id' do
+          result = Telemetry::Snmp::Data::Model::OIDWalks[params[:id]]
+          if result.nil?
+            status 404
+            return { error: true, message: "#{param[:id]} not found" }
+          end
+
+          update = {}
+          fields = %i[oid_index oid_walk active measurement_name]
+          body = MultiJson.load(request.body.read, symbolize_keys: true)
+          fields.each do |field|
+            next unless body.key? field
+
+            update[field] = body[field]
+          end
+
+          if update.empty?
+            status 400
+            return { error: true, message: 'no valid fields to update' }
+          end
+
+          update[:updated] = Sequel::CURRENT_TIMESTAMP
+          result.update(**update)
+          { error: false, updated_field: update, id: params[:id] }
+        end
+
+        put '/:id' do
+          status 405
+          { error: true, message: 'puts not supported', id: params[:id] }
+        end
+
+        delete '/:id' do
+          result = Telemetry::Snmp::Data::Model::OIDWalks[params[:id]]
+          if result.nil?
+            status 404
+            return { error: true, message: "#{param[:id]} not found" }
+          end
+
+          { error: result.delete, id: params[:id] }
+        end
+      end
+    end
+  end
+end

data/lib/telemetry/snmp/data.rb

@@ -0,0 +1,69 @@
+require 'sequel'
+require 'sequel/extensions/migration'
+require 'sequel/plugins/serialization'
+
+require 'telemetry/snmp/data/default_opts'
+require 'oj'
+
+module Telemetry
+  module Snmp
+    module Data
+      extend Telemetry::Snmp::Data::DefaultOpts
+
+      class << self
+        def migration_path
+          "#{__dir__}/data/migrations"
+        end
+
+        def migration
+          @migration_version = Sequel::Migrator.run(connection, migration_path, use_transactions: true)
+        end
+
+        def migration_version
+          @migration_version || migration
+        end
+
+        def migrations_up_to_date?
+          Sequel::Migrator.check_current(connection, migration_path)
+          true
+        rescue Sequel::Migrator::NotCurrentError
+          false
+        end
+
+        def load_models(*models_array)
+          models_array = models if models_array.empty?
+          models_array.each do |model|
+            require "telemetry/snmp/data/models/#{model}.rb"
+          end
+        end
+
+        def models
+          %w[user device_cred device oid oid_group oid_oid_groups oid_walk user_audit_log]
+        end
+
+        def connection(**opts)
+          @connection ||= Sequel.connect(**opts.merge(default_credentials))
+        end
+
+        def connected?
+          connection.test_connection
+        end
+
+        def setup_oj_serializer
+          Sequel::Plugins::Serialization.register_format(:oj_json,
+                                                         ->(value) { Oj.dump(value) },
+                                                         ->(value) { Oj.load(value.nil? ? 'null' : value) })
+        end
+
+        def start!
+          raise 'failed to start db connection' unless connected?
+
+          migration
+          setup_oj_serializer
+
+          load_models
+        end
+      end
+    end
+  end
+end