#!/usr/bin/env ruby

#

# A 'smarter' jexec in ruby.

#

# Run a command in multiple jails in parallel.

# Jails can be selected via host, ip, or jid.

#

#   "Run 'ls' in all jails that match the hostname 'trac':

#       ./jexec2.rb trac ls

#

BEGIN {

        require 'pathname'

        basedir = Pathname.new( __FILE__ ).dirname.parent

        $LOAD_PATH.unshift basedir + "ext" unless 

                $LOAD_PATH.include? basedir + "ext"

}

require 'jail'

jarg = ARGV.shift

args = ARGV

jails = BSD::Jail.find_all do |j|

    j.jid     == jarg.to_i ||

    j.ip.to_s == jarg      ||

    j.host    =~ /#{jarg}/

end or raise "Unable to find jails that match '#{jarg}'."

jails.each do |j|

    $deferr.puts "Parent #{Process.pid} about to attach to #{j.host} in a block!"

    childpid = j.attach do

        $deferr.puts "Child #{Process.pid} exec()ing:", "  " + args.join(" ")

        exec( *args )

    end

    $deferr.puts "Parent: waiting on imprisoned child #{childpid}"

    Process.waitpid( childpid )

    $deferr.puts "Child exited with exit code: %d" % [ $?.exitstatus ]

end