jah 0.0.3

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,7 @@
+*.sw?
+*.log
+.DS_Store
+coverage
+rdoc
+pkg
+jah.yaml

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Marcos Piccinini
+
+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.

data/README.md

@@ -0,0 +1,115 @@
+# ![Jah](http://fireho.com/system/jahlogo.png)
+
+Talk to your machines. Like a god.
+
+## Install
+
+Just run it for the first time to create a config file.
+Use `jah install` to force the wizard again.
+
+Gems needed for XMPP mode:
+
+* EventMachine (gem install eventmachine)
+* Blather      (gem install blather)
+
+
+Jah will try to find a `jah.yaml` file on ~/.jah or /etc.
+
+
+## Modes
+
+### XMPP
+
+Chat with your server with your favorite XMPP client (pidgin, adium..)
+Receive reports and failures realtime!
+Send commands and
+
+
+### POST
+
+The old school way. Set up a Jah Web Server, and give Jah it`s address.
+Jah will periodically post info. You can send commands through SSH.
+
+
+### DUMP
+
+AkA: Security freak. Jah just writes to tmp/ or whatever a dump file,
+Jah Web securely connects (scp) and downloads the data to parse.
+
+
+## Use God?
+
+
+## XMPP Bot
+
+
+Execute sh commands:
+
+    pwd
+    $> /home/jah
+
+
+Executing ruby statements:
+
+    ! 2 + 2
+    => 4
+
+    ! def foo; "hi"; end
+    ! foo
+    => "hi"
+
+
+Execute Jah commands:
+
+    cpu?
+    msweet: 0.14, 0.25, 0.14
+
+    mem?
+    msweet: 53%
+
+    net?
+    7 connections
+    [ips...]
+
+    ok?
+    [Personalized cool phrase...]
+
+
+Group session:
+
+    me: ok?
+    msweet: I'm fine, thanks..
+    ssaint: Kinda busy right now..
+    naomi: I need you! Now!
+
+
+Execute God statements:
+
+Start, stop, restart, monitor and unmonitor words on the beginning
+of a phrase will make Jah add a "god" on the front of it, making it
+trivial to work with your services:
+
+
+    start nginx
+    restart nanites
+    unmonitor postgresql
+
+
+More to come...
+
+
+
+## Note on Patches/Pull Requests
+
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but
+   bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+## Copyright
+
+Copyright (c) 2009 Marcos Piccinini. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,75 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "jah"
+    gem.summary = "be omnipresent..."
+    gem.description = "talk to your machines"
+    gem.email = "x@nofxx.com"
+    gem.homepage = "http://github.com/nofxx/jah"
+    gem.authors = ["Marcos Piccinini"]
+    gem.add_dependency "blather"
+    gem.add_dependency "i18n"
+    gem.add_development_dependency "rspec"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
+end
+
+require 'spec/rake/spectask'
+Spec::Rake::SpecTask.new(:spec) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.spec_files = FileList['spec/**/*_spec.rb']
+end
+
+Spec::Rake::SpecTask.new(:rcov) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.rcov = true
+end
+
+task :spec => :check_dependencies
+task :default => :spec
+
+begin
+  require 'reek/rake_task'
+  Reek::RakeTask.new do |t|
+    t.fail_on_error = true
+    t.verbose = false
+    t.source_files = 'lib/**/*.rb'
+  end
+rescue LoadError
+  task :reek do
+    abort "Reek is not available. In order to run reek, you must: sudo gem install reek"
+  end
+end
+
+begin
+  require 'roodi'
+  require 'roodi_task'
+  RoodiTask.new do |t|
+    t.verbose = false
+  end
+rescue LoadError
+  task :roodi do
+    abort "Roodi is not available. In order to run roodi, you must: sudo gem install roodi"
+  end
+end
+
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  if File.exist?('VERSION')
+    version = File.read('VERSION')
+  else
+    version = ""
+  end
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "jah #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.0.3

data/bin/jah

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+
+$VERBOSE = true # -w
+$KCODE = "u" # -Ku
+
+$LOAD_PATH << File.join(File.dirname(__FILE__), *%w[.. lib])
+require "jah"
+
+Jah::Cli.dispatch(ARGV)

data/jah.gemspec

@@ -0,0 +1,89 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE
+# Instead, edit Jeweler::Tasks in Rakefile, and run `rake gemspec`
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{jah}
+  s.version = "0.0.3"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Marcos Piccinini"]
+  s.date = %q{2009-09-29}
+  s.default_executable = %q{jah}
+  s.description = %q{talk to your machines}
+  s.email = %q{x@nofxx.com}
+  s.executables = ["jah"]
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.md"
+  ]
+  s.files = [
+    ".document",
+     ".gitignore",
+     "LICENSE",
+     "README.md",
+     "Rakefile",
+     "VERSION",
+     "bin/jah",
+     "jah.gemspec",
+     "lib/jah.rb",
+     "lib/jah.yaml.template",
+     "lib/jah/agent.rb",
+     "lib/jah/agents/dump.rb",
+     "lib/jah/agents/post.rb",
+     "lib/jah/agents/xmpp.rb",
+     "lib/jah/cli.rb",
+     "lib/jah/collector.rb",
+     "lib/jah/collectors/cpu.rb",
+     "lib/jah/collectors/disk.rb",
+     "lib/jah/collectors/mem.rb",
+     "lib/jah/collectors/net.rb",
+     "lib/jah/collectors/prok.rb",
+     "lib/jah/collectors/services.rb",
+     "lib/jah/collectors/who.rb",
+     "lib/jah/command.rb",
+     "lib/jah/commands/admin.rb",
+     "lib/jah/commands/extra.rb",
+     "lib/jah/commands/pub.rb",
+     "lib/jah/commands/status.rb",
+     "lib/jah/god.rb",
+     "lib/jah/install.rb",
+     "lib/locales/en_us.yml",
+     "lib/locales/pt_br.yml",
+     "lib/locales/pt_br_gostosa.yml",
+     "lib/locales/pt_br_mano.yml",
+     "lib/locales/pt_br_mineiro.yml",
+     "spec/console",
+     "spec/jah_spec.rb",
+     "spec/spec_helper.rb"
+  ]
+  s.homepage = %q{http://github.com/nofxx/jah}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{be omnipresent...}
+  s.test_files = [
+    "spec/jah_spec.rb",
+     "spec/spec_helper.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<blather>, [">= 0"])
+      s.add_runtime_dependency(%q<i18n>, [">= 0"])
+      s.add_development_dependency(%q<rspec>, [">= 0"])
+    else
+      s.add_dependency(%q<blather>, [">= 0"])
+      s.add_dependency(%q<i18n>, [">= 0"])
+      s.add_dependency(%q<rspec>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<blather>, [">= 0"])
+    s.add_dependency(%q<i18n>, [">= 0"])
+    s.add_dependency(%q<rspec>, [">= 0"])
+  end
+end

data/lib/jah/agent.rb

@@ -0,0 +1,128 @@
+require 'socket'
+require 'net/https'
+require 'net/http'
+require 'zlib'
+require 'stringio'
+# require 'eventmachine'
+
+module Jah
+  class Agent
+    PID_FILE = File.join("/tmp", "jah.pid")
+    autoload :XmppAgent, "jah/agents/xmpp"
+    autoload :PostAgent, "jah/agents/xmpp"
+    autoload :DumpAgent, "jah/agents/xmpp"
+
+    def initialize(options=nil)#, config)
+      if Jah.daemon?
+        puts "Jah starting in background.."
+        fork do
+          daemonize
+          run_agent_run
+        end
+        exit
+      end
+      run_agent_run
+    end
+
+    def pid_file
+      @pidfile ||= PID_FILE
+    end
+
+    def run_agent_run
+      case Jah.mode # @mode
+      when "xmpp" then  XmppAgent.new.run
+      when "post" then  PostAgent.new.run
+      else
+        DumpAgent.new
+      end
+    end
+
+    def self.start
+      new
+    end
+
+    def self.stop
+      if File.exists?(PID_FILE)
+        pid = File.read(PID_FILE)
+        puts "Stopping #{pid}"
+        `kill #{pid}`
+      else
+        puts "Jah not running..."
+        exit
+      end
+    end
+
+    def self.restart
+      stop
+      start
+    end
+
+    def self.install
+      Install.new
+    end
+
+    def daemonize
+      begin
+        File.open(pid_file, File::CREAT|File::EXCL|File::WRONLY) do |pid|
+          pid.puts $$
+        end
+        at_exit do
+          begin
+            File.unlink(pid_file)
+          rescue
+            Log.error "Unable to unlink pid file: #{$!.message}"
+          end
+        end
+      rescue
+        pid = File.read(pid_file).strip.to_i rescue "unknown"
+        running = true
+        begin
+          Process.kill(0, pid)
+          if stat = File.stat(pid_file)
+            if mtime = stat.mtime
+              if Time.now - mtime > 25 * 60 # assume process is hung after 25m
+                Log.info "Trying to KILL an old process..."
+                Process.kill("KILL", pid)
+                running = false
+              end
+            end
+          end
+        rescue Errno::ESRCH
+          running = false
+        rescue
+          # do nothing, we didn't have permission to check the running process
+        end
+        if running
+          if pid == "unknown"
+            Log.warn "Could not create or read PID file. " +
+                     "You may need to the path to the config directory. " +
+                     "See: http://scoutapp.com/help#data_file"
+          else
+            Log.warn "Process #{pid} was already running"
+          end
+          exit
+        else
+          Log.info "Stale PID file found. Clearing it and reloading..."
+          File.unlink(pid_file) rescue nil
+          retry
+        end
+      end
+
+      self
+    end
+  end
+
+
+    # def do_something!
+    #   if @mode == :online
+    #     puts "online"
+    #     post(@config[:server], nil, format_out)
+    #   else
+    #     puts "Writing log.."
+    #     f = open("/tmp/jah.txt", "a")
+    #     f.write(format_out)
+    #     f.close
+    #   end
+    # end
+
+end

data/lib/jah/agents/dump.rb

@@ -0,0 +1,5 @@
+module Jah
+  class DumpAgent
+
+  end
+end

data/lib/jah/agents/post.rb

@@ -0,0 +1,18 @@
+
+module Jah
+  class PostAgent
+
+
+        def post(url, error, body, headers = Hash.new, &response_handler)
+      return unless url
+      request(url, response_handler, error) do |connection|
+        post = Net::HTTP::Post.new( url.path +
+                                    (url.query ? ('?' + url.query) : ''),
+                                    HTTP_HEADERS.merge(headers) )
+        post.body = body
+        connection.request(post)
+      end
+    end
+
+  end
+end

data/lib/jah/agents/xmpp.rb

@@ -0,0 +1,194 @@
+require "eventmachine"#autoload :Client,
+require "blather/client/client"
+
+module Jah
+ # include Blather::DSL
+  class XmppAgent
+    ROSTER = []
+    attr_reader :client
+
+
+    def initialize #(jid, key, server, port=5222, debug=false, report=0)
+      Blather.logger.level = Logger::DEBUG if Jah.debug
+      @report = Jah.report
+      @client = Blather::Client.setup Jah.jid, Jah.key, Jah.server, Jah.port
+      setup
+      self
+    end
+
+    def process_message(to, msg, type = :chat)
+      if comm = Jah::Command.find(msg)
+        puts "Commmand => #{comm} | #{msg}"
+        args = msg.split(" ")
+        body = comm[2].send(comm[0], *args[1..-1])
+      else
+        keywords = %w{ start stop restart monitor unmonitor }
+        kind = :god if msg =~ /#{keywords.join("|")}/
+        kind ||= msg =~ /^\!/ ? :ruby : :sh
+        body = case kind
+          when :ruby
+          puts "Executing ruby command => #{msg}"
+          "=> #{execute_ruby(msg)}"
+          when :sh
+          puts "Executing *sh command => #{msg}"
+          "$> #{execute_sh(msg)}"
+          when :god
+          puts "Executing god command => #{msg}"
+          "G> #{execute_sh('god ' + msg)}"
+          else "dunno what to do...."
+        end
+      end
+      #beautify(body)
+     [ Blather::Stanza::Message.new(to, body, type)]
+      rescue => e
+      "Something is wrong.. #{e}\n#{e.backtrace.join("\n")}"
+    end
+
+    # Naive.... to be improved
+    def execute_ruby(code)
+      keywords = / class | module | def | do | while | for /
+      if code =~ keywords
+        return "Unmatched ends" if code.scan(keywords).length != code.scan(/end(\s|$)/).length
+      end
+      return "Unmatched.." if code.scan(/\{|\(|\[/).length != code.scan(/\}|\)|\]/).length
+      return "Unmatched quotes.." if code.scan(/\"|\'/).length % 2 != 0
+      begin
+        eval(code[1..-1]) || "nil"
+      rescue => e
+        "Fail => #{e}"
+      end
+    end
+
+    def execute_sh(code)
+      return I18n.t(:no_sudo) if code =~ /sudo/
+      res = `#{code}`.chomp
+      ex = $?.exitstatus
+      if ex != 0
+        res << I18n.t(:exit_code, :ex => ex.to_s)
+      else
+        res = "\n" + res unless res.split("\n").length < 2
+      end
+      res
+    end
+
+    #TODO: need to write raw....
+    def beautify(txt)
+      #txt.gsub!(/\*(.*)\*/, "<span style=\"font-weight: bold;\">\\1</span>")
+      #txt.gsub!(/\/(.*)\//, "<em>\\1</em>") # Italic
+      #txt.gsub!(/\_(.*)\_/, "<span style=\"font-decoration: underline;\">\\1</span>")
+      #txt.gsub!(/\-(.*)\-/, "<span style=\"font-decoration: line-through;\">\\1</span>")
+      #   <span style="font-size: large;">ok?</span>
+      #   <span style="color: #FF0000;">ok?</span>
+      txt
+    end
+
+    # [:iq, :query, :roster, :disco_info, :disco_items, :message, :presence,
+    #  :status, :subscription, :pubsub_node, :pubsub_affiliations,
+    #  :pubsub_create, :pubsub_event, :pubsub_items, :pubsub_publish,
+    #  :pubsub_retract, :pubsub_subscribe, :pubsub_subscription,
+    #  :pubsub_subscriptions, :pubsub_unsubscribe, :pubsub_owner,
+    #  :pubsub_delete, :pubsub_purge]
+
+
+    def setup
+      #  return if client && client.setup?
+      client.register_handler(:ready) do
+        puts "Connected!"
+        ROSTER << [client.roster.items.keys, Jah.groups].flatten.uniq
+        ROSTER.flatten!
+        ROSTER.select { |j| j =~ /\@conference\./ }.each do |c|
+          presence = Blather::Stanza::Presence.new
+          presence.to = "#{c}/#{Jah.hostname}"
+          client.write presence
+        end
+      end
+
+      client.register_handler :subscription, :request? do |s|
+        if ROSTER.include?(s.from.stripped.to_s)
+          puts "[REQUEST] Approve #{s}"
+          client.write s.approve!
+        else
+          puts "[REQUEST] Refuse #{s}"
+          client.write s.refuse!
+        end
+      end
+
+      # client.register_handler :message, :chat?, :body => 'exit' do |m|
+      #   client.write Blather::Stanza::Message.new(m.from, 'Exiting...')
+      #   client.close
+      # end
+      #client.register_handler :roster, [],
+      #client.register_handler :message, :error?, :body do |m|
+      #client.register_handler :message, :headline?, :body do |m|
+      #client.register_handler :message, :normal?, :body do |m|
+
+      client.register_handler :pubsub_event, :items do |m|
+        puts "[PUBSUB] => #{m.inspect}"
+        client.write m.body
+      end
+
+      client.register_handler :message, :groupchat? do |m|
+        if m.body =~ /^!|^>|^\\/ && m.to_s !~ /x.*:delay/ #delay.nil?
+          puts "[GROUP] => #{m.inspect}"
+          for msg in process_message(m.from.stripped, m.body, :groupchat)
+            client.write msg
+          end
+        end
+      end
+
+      client.register_handler :message, :chat?, :body do |m|
+        if ROSTER.include?(m.from.stripped.to_s)
+          puts "[PVT] => #{m.inspect}"
+          for msg in process_message(m.from, m.body)
+            client.write msg
+          end
+        end
+      end
+
+    end
+
+
+    def run
+      puts "Starting Jah Client...#{client.jid.stripped}"
+      trap(:INT) { EM.stop }
+      trap(:TERM) { EM.stop }
+
+      EM.run do
+        client.run
+        if @report != 0
+          puts "will report..."
+          EM.add_periodic_timer(@report) do
+            puts " I'm ok.. "
+            client.write Blather::Stanza::Message.new(
+                "nofxx@Jah.host.com", Jah::Collector.quick_results)
+          end
+        end
+
+      end
+
+
+    end
+
+  end
+
+end
+
+
+# # when_ready do
+# #   roster.grouped.each do |group, items|
+# #     puts "#{'*'*3} #{group || 'Ungrouped'} #{'*'*3}"
+# #     items.each { |item| puts "- #{item.name} (#{item.jid})" }
+# #     puts
+# #   end
+# #   shutdown
+# # end
+
+# # message :chat?, :body do |m|
+# #   begin
+# #     say m.from, eval(m.body)
+# #   rescue => e
+# #     say m.from, e.inspect
+# #   end
+
+
+#   end