checkson 0.5 → 0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 52bcb98df683395347d3b160a5e406d585ec730fd7e88a70a103c0e807aa5062
-  data.tar.gz: 45ac1a795dc57c78cc2b5386ac30827eda21eab348bff61be5b7a5a44e3281ed
+  metadata.gz: 8884f708c7fbe9c64243ce90b8942d822ee0bf3b664f1198e97198c7af339b1d
+  data.tar.gz: 9e1289046c0f46788c918b3b611f6424a2c15fe632523487d7f6750faa2c3c06
 SHA512:
-  metadata.gz: 15a3f4f005b9c480168a42b0b24e05457b7e33c294bf41df2efd92f82aee50a19efe8086431e56ff20a54677cd716598305a44f864e3f0a1556bf45660f6956f
-  data.tar.gz: ca05c7df0e92b183cad87d9a67301e2827f5b47ff237720c23744244260a5816a996911521021a44432570b552f1979795e6991ea559f82cb91374795445c2e0
+  metadata.gz: aacb646e006f6af49070ad5f7e299bdb546c8cfea6c6b3d99fb92aea3d9f361f13b523f3f3546951d77b9a2ccba7b2a50ac19661b7326a061c05224aaa1e53e2
+  data.tar.gz: 76fde610194ed4e82fc4d653b18bed353dc0d269e1d99079faada485f29806c5e27531e2b601c360bb6af9145f65d4c70db9027a2509227ec397ced2de0f6d61

data/lib/checkson.rb

@@ -3,6 +3,7 @@
 module Checkson
   autoload :Config, 'checkson/config'
   autoload :Context, 'checkson/context'
+  autoload :APIClient, 'checkson/apiclient'
   autoload :UI, 'checkson/ui'
 
   module Check

data/lib/checkson/apiclient.rb

@@ -0,0 +1,75 @@
+# frozen_string_literal: true
+
+require 'net/http'
+require 'json'
+require 'uri'
+require 'erb'
+require 'socket'
+require 'digest'
+require 'logger'
+
+module Checkson
+  class APIClient
+    attr_reader :apiaddr
+    attr_reader :outfile
+    def initialize(apiaddr, outfile)
+      @apiaddr = apiaddr
+      @outfile = outfile
+    end
+
+    def getchecks
+      template = <<-TEMPLATE
+        <%@checks.each do |check| %>
+        check '<%=check["description"]%>' do
+          using <%=check["check"]%>
+          <%check["sets"].each do |key,value| %>
+          set :<%=key%>, "<%=value%>"
+          <%end%>
+        <%unless check["helps"].eql?(nil)%>
+          <%check["helps"].each do |help| %>
+          help '<%= help %>'
+          <%end%>
+        <%end%>
+        end
+        <% end %>
+      TEMPLATE
+      querydata = get
+      @checks = querydata['checks'] if querydata['status'].eql? 'ok'
+      begin
+        remove_empty_lines(ERB.new(template).result(binding))
+      rescue StandardError
+        warn querydata['error']
+      end
+    end
+
+    def writechecks
+      File.open(@outfile, 'w') do |f|
+        f.write(getchecks)
+      end
+    end
+
+    private
+
+    def remove_empty_lines(string)
+      regex = /^[\s]*$\n/
+      string.gsub(regex, '')
+    end
+
+    def get
+      token = ENV['CHECKSON_API_KEY']
+      token ||= 'empty'
+      hostname = Socket.gethostname
+      query("#{keyhash(token)}/get/#{hostname}")
+    end
+
+    def query(command)
+      @ep = "#{@apiaddr}/api/v1"
+      route = "#{@ep}/#{command}"
+      JSON.parse(Net::HTTP.get(URI.parse(route)))
+    end
+
+    def keyhash(token)
+      Digest::SHA512.hexdigest(token)
+    end
+  end
+end

data/lib/checkson/checks/packagemanagers/abstractpkgmgr.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module Checkson
+  module Check
+    module Pkganager
+      class AbstractPkgmanager
+        def self.descendants
+          ObjectSpace.each_object(Class).select { |klass| klass < self }
+        end
+      end
+
+      class Pkgmanager < AbstractPkgmanager
+        def self.adapters
+          AbstractPkgmanager.descendants - [self]
+        end
+
+        def self.adapter
+          @adapter ||= adapters.find { |adapter| adapter.supports_os? Facter['osfamily'].value.downcase }
+        end
+      end
+    end
+  end
+end

data/lib/checkson/checks/packages.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module Checkson
+  module Check
+    class Shell < Checkson::Check::Base
+      def initialize(opts = {})
+        @opts = (@opts || {}).merge(opts)
+        super()
+      end
+
+      def check
+        raise ArgumentError, 'No code given' unless @opts[:code]
+
+        execute(@opts[:code])
+        failed! unless $?.exitstatus.zero?
+      end
+
+      protected
+
+      def execute(command)
+        `sh -c "#{command.gsub(/"/, '\"')}"`
+      end
+    end
+  end
+end

data/lib/checkson/checks/process.rb

@@ -7,9 +7,7 @@ module Checkson
     class Process < Checkson::Check::Shell
       def initialize(opts = {})
         @opts = opts
-        unless @opts[:pidfile] || @opts[:name]
-          raise ArgumentError, 'Neither pidfile nor process name given'
-        end
+        raise ArgumentError, 'Neither pidfile nor process name given' unless @opts[:pidfile] || @opts[:name]
 
         super()
       end

data/lib/checkson/ui.rb

@@ -1,6 +1,8 @@
 # frozen_string_literal: true
 
 require 'optparse'
+require 'logger'
+require 'uri'
 
 module Checkson
   # Simple user interface for checkson
@@ -11,15 +13,15 @@ module Checkson
       start_checks
     end
 
-    protected
-
-    def winsize
-      require 'io/console'
-      IO.console.winsize
-    end
+    private
 
     def start_checks
-      @config = Checkson::Config.new(@opts[:config])
+      if @opts[:datasource].eql? :api
+        client = Checkson::APIClient.new(@opts[:endpoint], '/tmp/checkson.rb')
+        client.writechecks
+        @opts[:config_file] = client.outfile
+      end
+      @config = Checkson::Config.new(@opts[:config_file])
 
       @num_checks = @config.checks.length
       @num_ok = 0
@@ -35,8 +37,10 @@ module Checkson
 
     def set_default_options
       @opts = {
-        config: '/etc/checkson.rb',
-        verbose: false
+        datasource: :file,
+        verbose: false,
+        config_file: '/etc/checkson.rb',
+        endpoint: 'https://checkson.fsrv.services:8080'
       }
     end
 
@@ -46,14 +50,26 @@ module Checkson
         opts.on('-v', '--[no-]verbose', 'Run verbosely') do |v|
           @opts[:verbose] = v
         end
-        opts.on('-c', '--config FILE', 'Config file to use') do |file|
+        opts.on('-d', '--datasource [TYPE]', %i[file api], 'Set datasource type', '(file, api)') do |d|
+          @opts[:datasource] = d
+        end
+        opts.on('-f', '--file [FILE]', 'Config file to use') do |file|
           if File.exist?(file)
-            @opts[:config] = file
+            @opts[:config_file] = file
           else
             die("Config file #{file} does not exist")
           end
         end
+        opts.on('-e', '--endpoint [ADDR]', 'Config api to use') do |ep|
+          if ep =~ URI::DEFAULT_PARSER.make_regexp
+            @opts[:endpoint] = ep
+          else
+            die('URL of endpoint not valid')
+          end
+        end
       end.parse!
+      logger = Logger.new(STDOUT)
+      logger.info "Using datasource `#{@opts[:datasource]}`"
     end
 
     def output(sym, check)
@@ -69,6 +85,11 @@ module Checkson
       end
     end
 
+    def winsize
+      require 'io/console'
+      IO.console.winsize
+    end
+
     def die(msg)
       warn(msg)
       exit 1

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: checkson
 version: !ruby/object:Gem::Version
-  version: '0.5'
+  version: '0.6'
 platform: ruby
 authors:
 - Florian Bauer
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-03-19 00:00:00.000000000 Z
+date: 2020-04-13 00:00:00.000000000 Z
 dependencies: []
 description: A simple framework for checking node facts
 email: florian@fsrv.xyz
@@ -20,8 +20,11 @@ files:
 - README.md
 - bin/checkson
 - lib/checkson.rb
+- lib/checkson/apiclient.rb
 - lib/checkson/checks/base.rb
 - lib/checkson/checks/dns.rb
+- lib/checkson/checks/packagemanagers/abstractpkgmgr.rb
+- lib/checkson/checks/packages.rb
 - lib/checkson/checks/process.rb
 - lib/checkson/checks/shell.rb
 - lib/checkson/config.rb