rboss 0.5.6 → 0.6.0

data/Gemfile

@@ -2,4 +2,4 @@ source "http://rubygems.org"
 
 gemspec
 
-gem 'yummi', '>=0.3'
+gem 'yummi'

data/bin/jboss-profile

@@ -2,7 +2,7 @@
 
 #                         The MIT License
 #
-# Copyright (c) 2011 Marcelo Guimarães <ataxexe@gmail.com>
+# Copyright (c) 2011-2012 Marcelo Guimarães <ataxexe@gmail.com>
 #
 # Permission is hereby granted, free of charge, to any person obtaining a copy
 # of this software and associated documentation files (the "Software"), to deal
@@ -83,6 +83,10 @@ end
 opts.on("-h", "--help", "Shows this help message") { puts opts; exit }
 opts.parse!(ARGV) rescue abort 'Invalid Option! Use --help or -h for usage help.'
 
+unless $stdout.isatty
+  require 'yummi/no_colors'
+end
+
 if params.empty?
   puts opts
   exit

data/bin/rboss-cli

@@ -0,0 +1,142 @@
+#!/usr/bin/env ruby
+
+#                         The MIT License
+#
+# Copyright (c) 2011-2012 Marcelo Guimarães <ataxexe@gmail.com>
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+require_relative '../lib/rboss'
+require 'optparse'
+require 'logger'
+require 'yaml'
+require 'fileutils'
+require 'yummi'
+
+params = {}
+@conf_dir = File.expand_path "~/.rboss"
+servers_file = "#@conf_dir/jboss-cli-servers.yaml"
+@components = {}
+@servers_file = File.expand_path(servers_file)
+save = false
+params[:jboss_home] = (ENV["JBOSS_HOME"] or ENV["RBOSS_CLI_JBOSS_HOME"] or Dir.pwd)
+params[:host] = '127.0.0.1'
+params[:port] = '9999'
+
+def load_yaml
+  return Hash::new unless File.exist?(@servers_file)
+  YAML::load_file(@servers_file)
+end
+
+opts = OptionParser::new
+opts.on('-j', '--jboss-home PATH', 'Defines the JBOSS_HOME variable') do |home|
+  params[:jboss_home] = home
+end
+opts.on('-s URL', 'Defines the JBoss server') do |server|
+  params[:server] = server
+end
+opts.on('--host HOST', 'Defines the JBoss host') do |host|
+  params[:host] = host
+end
+opts.on('--port PORT', 'Defines the JBoss admin port') do |port|
+  params[:port] = port
+end
+opts.on('-c', '--connect SERVER_NAME',
+        "Uses a configured server in #{servers_file}") do |server|
+  config = load_yaml[server]
+  abort "No configuration for #{server}" unless config
+  config.each do |key, value|
+    params[key.to_sym] = value
+  end
+end
+opts.on('--save SERVER_NAME', "Saves the server configuration in #{servers_file}") do |server|
+  save = server
+end
+opts.on('--loop INTERVAL', Float, 'Runs the mbean inside a loop') do |interval|
+  @loop = true
+  @interval = interval
+end
+
+opts.on('--no-color', 'Do not colorize output') do
+  require 'yummi/no_colors'
+end
+opts.on('-u', '--user USER', 'Defines the Admin User') do |user|
+  params[:user] = user
+end
+opts.on('-p', '--password PASSWORD', 'Defines the Admin Password') do |password|
+  params[:password] = password
+end
+opts.on('-v', '--verbose', 'Displays jboss-cli commands before execution') do
+  params[:log_level] = Logger::DEBUG
+end
+
+JBoss::Cli::Mappings.resource_mappings.each do |name, config|
+  if config[:scan]
+    opts.on("--#{name} [NAMES]", Array, config[:description]) do |resources|
+      @components[name] = resources || :all
+    end
+  else
+    opts.on("--#{name}", config[:description]) do
+      @components[name] = true
+    end
+  end
+end
+
+opts.on("-h", "--help", "Shows this help message") do
+  puts opts; exit
+end
+opts.parse!(ARGV) rescue abort 'Invalid Option! Use --help or -h for usage help.'
+
+@jboss_cli = JBoss::Cli::Invoker::new params
+
+unless $stdout.isatty
+  require 'yummi/no_colors'
+end
+
+if save
+  config = load_yaml
+  config[save] ||= {}
+  server_config = config[save]
+  params.delete :jboss_home
+  params.each do |key, value|
+    server_config[key.to_s] = value
+  end
+  FileUtils::mkdir_p File.expand_path(@conf_dir)
+  FileUtils::touch @servers_file
+  f = File.open(@servers_file, 'w')
+  YAML::dump(config, f)
+  f.close
+  puts "Configuration saved!"
+end
+
+puts opts if @components.empty?
+
+def execute_actions
+  content = @jboss_cli.content(@components)
+  system "clear" if @loop
+  puts Yummi.colorize(Time.now.strftime("At %H:%M:%S%n"), :white) if @loop
+  puts content
+end
+
+while @loop
+  execute_actions
+  sleep @interval
+end
+
+execute_actions

data/bin/twiddle

@@ -2,7 +2,7 @@
 
 #                         The MIT License
 #
-# Copyright (c) 2011 Marcelo Guimarães <ataxexe@gmail.com>
+# Copyright (c) 2011-2012 Marcelo Guimarães <ataxexe@gmail.com>
 #
 # Permission is hereby granted, free of charge, to any person obtaining a copy
 # of this software and associated documentation files (the "Software"), to deal
@@ -46,6 +46,10 @@ def load_yaml
   YAML::load_file(@servers_file)
 end
 
+unless $stdout.isatty
+  require 'yummi/no_colors'
+end
+
 opts = OptionParser::new
 opts.on('-j', '--jboss-home PATH', 'Defines the JBOSS_HOME variable') do |home|
   params[:jboss_home] = home
@@ -179,8 +183,8 @@ end
 puts opts if @mbeans.empty? and @commands.empty?
 
 @actions = JBoss::CommandActions::Twiddle::new @twiddle,
-  :no_details => no_details,
-  :mbeans => defaults
+                                               :no_details => no_details,
+                                               :mbeans => defaults
 
 def execute_actions
   buff = ""

data/lib/rboss.rb

@@ -1,6 +1,6 @@
 #                         The MIT License
 #
-# Copyright (c) 2011 Marcelo Guimarães <ataxexe@gmail.com>
+# Copyright (c) 2011-2012 Marcelo Guimarães <ataxexe@gmail.com>
 #
 # Permission is hereby granted, free of charge, to any person obtaining a copy
 # of this software and associated documentation files (the "Software"), to deal
@@ -23,5 +23,6 @@
 require_relative "rboss/version"
 require_relative "rboss/jboss_profile"
 require_relative "rboss/twiddle"
+require_relative "rboss/cli"
 require_relative "rboss/bin/command_actions"
 

data/lib/rboss/bin/command_actions.rb

@@ -1,6 +1,6 @@
 #                         The MIT License
 #
-# Copyright (c) 2011 Marcelo Guimarães <ataxexe@gmail.com>
+# Copyright (c) 2011-2012 Marcelo Guimarães <ataxexe@gmail.com>
 #
 # Permission is hereby granted, free of charge, to any person obtaining a copy
 # of this software and associated documentation files (the "Software"), to deal
@@ -20,8 +20,6 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 # THE SOFTWARE.
 
-require_relative '../table_builder'
-
 module JBoss
   module CommandActions
     class Twiddle
@@ -90,7 +88,7 @@ module JBoss
       def detail mbeans
         buff = ""
         mbeans.each do |mbean_id, resources|
-          table_builder = TableBuilder::new @opts[:mbeans][mbean_id]
+          builder = TableBuilder::new @opts[:mbeans][mbean_id]
           rows = []
           if resources.is_a? TrueClass
             row = []
@@ -99,7 +97,7 @@ module JBoss
             end
             rows << row
           elsif @opts[:no_details]
-            table_builder.no_details
+            builder.show_only_name
             @monitor.mbean(mbean_id).scan.each do |name|
               rows << [name]
             end
@@ -112,8 +110,9 @@ module JBoss
               rows << row
             end
           end
-          table_builder.data = rows
-          buff << table_builder.table.to_s
+          table = builder.build_table
+          table.data = rows
+          buff << table.to_s
         end
         buff
       end

data/lib/rboss/cli.rb

@@ -0,0 +1,22 @@
+require_relative 'cli/jboss_cli'
+
+if RUBY_PLATFORM['linux']
+  class JBoss::Cli::Invoker
+
+    def jboss_cli
+      "#@jboss_home/bin/jboss-cli.sh"
+    end
+
+  end
+elsif RUBY_PLATFORM['mingw'] #Windows
+  class JBoss::Cli::Invoker
+
+    def jboss_cli
+      "#@jboss_home/bin/jboss-cli.bat"
+    end
+
+  end
+else
+  puts "Platform #{RUBY_PLATFORM} not supported"
+end
+

data/lib/rboss/cli/colorizers.rb

@@ -0,0 +1,22 @@
+require 'yummi'
+
+module JBoss
+  module Cli
+    module Colorizers
+
+      def self.boolean params
+        lambda do |value|
+          return params[:if_true] if value
+          params[:if_false]
+        end
+      end
+
+      def self.with params
+        lambda do |value|
+          params[:color]
+        end
+      end
+
+    end
+  end
+end

data/lib/rboss/cli/formatters.rb

@@ -0,0 +1,17 @@
+require 'yummi'
+
+module JBoss
+  module Cli
+    module Formatters
+
+      def self.yes_or_no
+        Yummi::Formatters.yes_or_no
+      end
+
+      def self.byte params = {}
+        Yummi::Formatters.byte params
+      end
+
+    end
+  end
+end

data/lib/rboss/cli/health_checkers.rb

@@ -0,0 +1,43 @@
+require 'yummi'
+
+module JBoss
+  module Cli
+    module HealthCheckers
+
+      def self.percentage params
+        PercentageColorizer::new params
+      end
+
+      class PercentageColorizer
+
+        def initialize(params)
+          @max = params[:max]
+          @free = params[:free]
+          @using = params[:using]
+          @color = params[:color] || {
+            :bad => :red,
+            :warn => :brown,
+            :good => :green
+          }
+          @threshold = params[:threshold] || {
+            :warn => 0.15,
+            :bad => 0.30
+          }
+        end
+
+        def call(data)
+          max = data[@max.to_sym].to_f
+          free = @using ? max - data[@using.to_sym].to_f : data[@free.to_sym].to_f
+
+          percentage = free / max
+
+          return @color[:bad] if percentage <= @threshold[:bad]
+          return @color[:warn] if percentage <= @threshold[:warn]
+          @color[:good]
+        end
+
+      end
+
+    end
+  end
+end

data/lib/rboss/cli/jboss_cli.rb

@@ -0,0 +1,93 @@
+#                         The MIT License
+#
+# Copyright (c) 2011-2012 Marcelo Guimarães <ataxexe@gmail.com>
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+require_relative 'resource'
+require_relative 'mappings'
+
+require 'logger'
+
+module JBoss
+
+  module Cli
+
+    class Invoker
+      include JBoss::Cli::Mappings
+
+      attr_reader :server, :host, :port, :user, :password
+
+      def initialize(params = {})
+        params = {
+          :jboss_home => ENV["JBOSS_HOME"],
+        }.merge! params
+
+        @jboss_home = params[:jboss_home]
+
+        @server = params[:server]
+        @host = params[:host]
+        @port = params[:port]
+        @server ||= [@host, @port].join ':' if @host and @port
+
+        @user = params[:user]
+        @password = params[:password]
+
+        @logger = params[:logger]
+        unless @logger
+          @logger = Logger::new STDOUT
+          @logger.level = params[:log_level] || Logger::INFO
+          formatter = Yummi::Formatter::LogFormatter.new do |severity, message|
+            "#{severity} : #{message}"
+          end
+          @logger.formatter = formatter
+        end
+      end
+
+      def command
+        command = "#{jboss_cli} --connect"
+        command << " --controller=#@server" if @server
+        command << " --user='#@user'" if @user
+        command << " --password='#@password'" if @password
+        command
+      end
+
+      def content(components)
+        buff = ""
+        components.each do |key, resources|
+          if resource_mappings.has_key? key
+            mapping = resource_mappings[key]
+            component = JBoss::Cli::Resource::new(self, mapping)
+            buff << component.content(resources)
+          end
+        end
+        buff
+      end
+
+      def execute(*commands)
+        exec = "#{command} --commands=\"#{commands.join ','}\""
+        @logger.debug exec
+        `#{exec}`.chomp
+      end
+
+    end
+
+  end
+
+end