watchmonkey_cli 1.4

data/lib/watchmonkey_cli/helpers.rb

@@ -0,0 +1,19 @@
+module WatchmonkeyCli
+  module Helpers
+    BYTE_UNITS = %W(TiB GiB MiB KiB B).freeze
+
+    def human_filesize(s)
+      s = s.to_f
+      i = BYTE_UNITS.length - 1
+      while s > 512 && i > 0
+        i -= 1
+        s /= 1024
+      end
+      ((s > 9 || s.modulo(1) < 0.1 ? '%d' : '%.1f') % s) + ' ' + BYTE_UNITS[i]
+    end
+
+    def human_number(n)
+      n.to_s.reverse.gsub(/...(?=.)/,'\&,').reverse
+    end
+  end
+end

data/lib/watchmonkey_cli/hooks/platypus.rb

@@ -0,0 +1,38 @@
+module WatchmonkeyCli
+  class Platypus
+    def self.hook!(app, opts = {})
+      opts = opts.reverse_merge(notifications: true)
+      app.instance_eval do
+        # send errors via notification center
+        hook :result_dump do |robj, args, checker|
+          if robj.error?
+            robj.messages.each do |m|
+              msg  = "#{robj.args[0].try(:name) || robj.args[0].presence || "?"}: #{m}"
+
+              case opts[:notifications]
+              when 1
+                # makes no sound
+                fmsg = msg.gsub('"', '\"').gsub("'", %{'"'"'})
+                `osascript -e 'display notification "#{fmsg}" with title "WatchMonkey"'`
+              when 2
+                # makes a sound
+                sync { puts "NOTIFICATION:#{msg}" }
+              end
+            end
+          end
+        end if opts[:notifications]
+
+        hook :wm_work_start, :wm_work_end do
+          # mastermind calculation I swear :D (<-- no idea what I did here)
+          # sync { puts "PROGRESS:#{((@threads.length-@threads.select{|t| t[:working] }.length.to_d) / @threads.length * 100).to_i}" }
+          sync do
+            active = @threads.select{|t| t[:working] }.length
+            total  = @threads.select{|t| t[:working] }.length + @queue.length
+            perc   = total.zero? ? 100 : (active.to_d / total * 100).to_i
+            puts "PROGRESS:#{perc}"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/watchmonkey_cli/hooks/requeue.rb

@@ -0,0 +1,106 @@
+module WatchmonkeyCli
+  class Requeue
+    def self.hook!(app)
+      app.instance_eval do
+        # app options
+        @opts[:loop_forever] = true
+        @opts[:logfile] = logger_filename # enable logging
+
+        # module options
+        @opts[:default_requeue]                   = 60
+        # @opts[:default_requeue_ftp_availability]  = 60
+        @opts[:default_requeue_mysql_replication] = 30
+        @opts[:default_requeue_ssl_expiration]    = 1.hour
+        @opts[:default_requeue_unix_defaults]     = false
+        # @opts[:default_requeue_unix_df]           = 60
+        # @opts[:default_requeue_unix_file_exists]  = 60
+        # @opts[:default_requeue_unix_load]         = 60
+        @opts[:default_requeue_unix_mdadm]        = 5.minutes
+        # @opts[:default_requeue_unix_memory]       = 60
+        @opts[:default_requeue_www_availability]  = 30
+
+        # Requeue threads
+        @requeue = []
+
+
+        # =================
+        # = Status thread =
+        # =================
+        @platypus_status_thread = Thread.new do
+          Thread.current.abort_on_exception = true
+          while STDIN.gets
+            sync do
+              puts "==========  STATUS  =========="
+              puts "     Queue: #{@queue.length}"
+              puts "   Requeue: #{@requeue.length}"
+              puts "   Workers: #{@threads.select{|t| t[:working] }.length}/#{@threads.length} working (#{@threads.select(&:alive?).length} alive)"
+              puts "   Threads: #{Thread.list.length}"
+              # puts "            #{@threads.select(&:alive?).length} alive"
+              # puts "            #{@threads.select{|t| t.status == "run" }.length} running"
+              # puts "            #{@threads.select{|t| t.status == "sleep" }.length} sleeping"
+              puts " Processed: #{@processed}"
+              puts "========== //STATUS =========="
+            end
+          end
+        end
+
+
+        # =========
+        # = Hooks =
+        # =========
+        hook :dequeue do |checker, args|
+          opts = args.extract_options!
+          retry_in = opts[:every] if opts[:every].is_a?(Fixnum)
+          retry_in = @opts[:"default_requeue_#{checker.class.checker_name}"] if retry_in.nil?
+          retry_in = @opts[:default_requeue] if retry_in.nil?
+          if retry_in
+            debug "Requeuing #{checker} in #{retry_in} seconds"
+            requeue checker, args + [opts], retry_in
+          end
+        end
+
+        hook :wm_shutdown do
+          sync do
+            debug "[ReQ] #{@requeue.length} items in requeue..."
+            unless @requeue.empty?
+              @requeue.each(&:kill).each(&:join).select!(&:alive?)
+              debug "[ReQ] #{@requeue.length} items in requeue..."
+            end
+            @requeue_status_thread.try(:kill).try(:join)
+          end
+        end
+
+
+        # ===========
+        # = Methods =
+        # ===========
+        def requeue checker, args, delay = 10
+          return if $wm_runtime_exiting
+          sync do
+            @requeue << Thread.new {
+              Thread.current.abort_on_exception = true
+              sleep(delay)
+              checker.enqueue(*args)
+              sync { @requeue.delete Thread.current }
+            }
+          end
+        end
+      end
+    end
+  end
+end
+
+
+__END__
+
+
+log "checking...", false
+log "PROGRESS:100", false
+$threads.select!(&:alive?)
+GC.start
+sleep 3
+log "sleeping...", false
+20.times do |i|
+  log "PROGRESS:#{(100-(i*3/60.0*100)).round(0)}", false
+  sleep 3
+end

data/lib/watchmonkey_cli/loopback_connection.rb

@@ -0,0 +1,36 @@
+module WatchmonkeyCli
+  class LoopbackConnection
+    def initialize(id, opts = {}, &initializer)
+      @id = id
+      @opts = {}.merge(opts)
+      # @mutex = Monitor.new
+      initializer.try(:call, @opts)
+    end
+
+    def to_s
+      "#<WatchmonkeyCli::LoopbackConnection:#{@id}>"
+    end
+
+    def name
+      "lo:#{@id}"
+    end
+
+    def sync &block
+      # @mutex.synchronize(&block)
+      block.try(:call)
+    end
+
+    def exec cmd, chomp = true
+      _stdin, _stdouterr, _thread = Open3.popen2e(cmd)
+      _thread.join
+      res = _stdouterr.read
+      chomp ? res.chomp : res
+    ensure
+      _stdin.close rescue false
+      _stdouterr.close rescue false
+    end
+
+    def close!
+    end
+  end
+end

data/lib/watchmonkey_cli/ssh_connection.rb

@@ -0,0 +1,50 @@
+module WatchmonkeyCli
+  class SshConnection
+    def initialize(id, opts = {}, &initializer)
+      @id = id
+
+      if opts.is_a?(String)
+        u, h = opts.split("@", 2)
+        opts = { user: u, host_name: h }
+      elsif opts[:host].is_a?(String)
+        u, h = opts[:host].split("@", 2)
+        opts = opts.merge(user: u, host_name: h)
+        opts.delete(:host)
+      end
+
+      # net/ssh options
+      @opts = {
+        config: false,
+      }.merge(opts)
+      @mutex = Monitor.new
+      initializer.try(:call, @opts)
+    end
+
+    def to_s
+      "#<WatchmonkeyCli::SshConnection:#{@id}>"
+    end
+
+    def name
+      "ssh:#{@id}"
+    end
+
+    def sync &block
+      @mutex.synchronize(&block)
+    end
+
+    def exec cmd, chomp = true
+      sync do
+        res = connection.exec!(cmd)
+        chomp ? res.chomp : res
+      end
+    end
+
+    def connection
+      sync { @ssh ||= Net::SSH.start(nil, nil, @opts) }
+    end
+
+    def close!
+      @ssh.try(:close) rescue false
+    end
+  end
+end

data/lib/watchmonkey_cli/version.rb

@@ -0,0 +1,4 @@
+module WatchmonkeyCli
+  VERSION = "1.4"
+  UPDATE_URL = "https://raw.githubusercontent.com/2called-chaos/watchmonkey_cli/master/VERSION"
+end

data/lib/watchmonkey_cli.rb

@@ -0,0 +1,41 @@
+STDOUT.sync = true
+
+# stdlib
+require "benchmark"
+require "fileutils"
+require "thread"
+require "monitor"
+require "optparse"
+require "open3"
+require "net/ftp"
+require 'net/https'
+
+# 3rd party
+require "active_support"
+require "active_support/core_ext"
+begin ; require "pry" ; rescue LoadError ; end
+require "httparty"
+require 'net/ssh'
+
+# lib
+require "watchmonkey_cli/version"
+require "watchmonkey_cli/loopback_connection"
+require "watchmonkey_cli/ssh_connection"
+require "watchmonkey_cli/helpers"
+require "watchmonkey_cli/checker"
+require "watchmonkey_cli/application/colorize"
+require "watchmonkey_cli/application/configuration"
+require "watchmonkey_cli/application/dispatch"
+require "watchmonkey_cli/application"
+
+# require buildin checkers
+require "watchmonkey_cli/checkers/ftp_availability"
+require "watchmonkey_cli/checkers/mysql_replication"
+require "watchmonkey_cli/checkers/ssl_expiration"
+require "watchmonkey_cli/checkers/unix_defaults"
+require "watchmonkey_cli/checkers/unix_df"
+require "watchmonkey_cli/checkers/unix_file_exists"
+require "watchmonkey_cli/checkers/unix_load"
+require "watchmonkey_cli/checkers/unix_mdadm"
+require "watchmonkey_cli/checkers/unix_memory"
+require "watchmonkey_cli/checkers/www_availability"

data/watchmonkey_cli.gemspec

@@ -0,0 +1,27 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'watchmonkey_cli/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "watchmonkey_cli"
+  spec.version       = WatchmonkeyCli::VERSION
+  spec.authors       = ["Sven Pachnit"]
+  spec.email         = ["sven@bmonkeys.net"]
+  spec.summary       = %q{Watchmonkey CLI - dead simple agentless monitoring via SSH, HTTP, FTP, etc.}
+  spec.description   = %q{If you want an easy way to monitor services without the need of installing agents let a monkey do the job by polling status information via transport protocols.}
+  spec.homepage      = "https://github.com/2called-chaos/watchmonkey_cli"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0")
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "activesupport"
+  spec.add_dependency "httparty"
+  spec.add_dependency "net-ssh"
+  spec.add_development_dependency "pry"
+  spec.add_development_dependency "bundler"
+  spec.add_development_dependency "rake"
+end

metadata

@@ -0,0 +1,163 @@
+--- !ruby/object:Gem::Specification
+name: watchmonkey_cli
+version: !ruby/object:Gem::Version
+  version: '1.4'
+platform: ruby
+authors:
+- Sven Pachnit
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-10-18 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: httparty
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  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: bundler
+  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: rake
+  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'
+description: If you want an easy way to monitor services without the need of installing
+  agents let a monkey do the job by polling status information via transport protocols.
+email:
+- sven@bmonkeys.net
+executables:
+- watchmonkey
+- watchmonkey.sh
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- VERSION
+- bin/watchmonkey
+- bin/watchmonkey.sh
+- lib/watchmonkey_cli.rb
+- lib/watchmonkey_cli/application.rb
+- lib/watchmonkey_cli/application/colorize.rb
+- lib/watchmonkey_cli/application/configuration.rb
+- lib/watchmonkey_cli/application/configuration.tpl
+- lib/watchmonkey_cli/application/dispatch.rb
+- lib/watchmonkey_cli/checker.rb
+- lib/watchmonkey_cli/checkers/ftp_availability.rb
+- lib/watchmonkey_cli/checkers/mysql_replication.rb
+- lib/watchmonkey_cli/checkers/ssl_expiration.rb
+- lib/watchmonkey_cli/checkers/unix_defaults.rb
+- lib/watchmonkey_cli/checkers/unix_df.rb
+- lib/watchmonkey_cli/checkers/unix_file_exists.rb
+- lib/watchmonkey_cli/checkers/unix_load.rb
+- lib/watchmonkey_cli/checkers/unix_mdadm.rb
+- lib/watchmonkey_cli/checkers/unix_memory.rb
+- lib/watchmonkey_cli/checkers/www_availability.rb
+- lib/watchmonkey_cli/helpers.rb
+- lib/watchmonkey_cli/hooks/platypus.rb
+- lib/watchmonkey_cli/hooks/requeue.rb
+- lib/watchmonkey_cli/loopback_connection.rb
+- lib/watchmonkey_cli/ssh_connection.rb
+- lib/watchmonkey_cli/version.rb
+- watchmonkey_cli.gemspec
+homepage: https://github.com/2called-chaos/watchmonkey_cli
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.6.6
+signing_key: 
+specification_version: 4
+summary: Watchmonkey CLI - dead simple agentless monitoring via SSH, HTTP, FTP, etc.
+test_files: []