truckle in cave
Denis Knauf
commit
12940cd521
|
|
@ -0,0 +1,234 @@
|
|||
#!/usr/bin/env ruby
|
||||
# -*- encoding : utf-8 -*-
|
||||
|
||||
require 'irb-pager'
|
||||
require 'shellwords'
|
||||
|
||||
class Commands < Hash
|
||||
class CommandError < Exception
|
||||
end
|
||||
class ExpectingCommandError < CommandError
|
||||
def initialize *args
|
||||
args = ['Command expected'] if args.empty?
|
||||
super *args
|
||||
end
|
||||
end
|
||||
class UnknownCommandError < CommandError
|
||||
def initialize *args
|
||||
args = ['This Command i do not know'] if args.empty?
|
||||
super *args
|
||||
end
|
||||
end
|
||||
|
||||
attr_accessor :exe, :prefix, :default_cmd
|
||||
|
||||
def self.arg_unshift arg, args
|
||||
arg = arg.is_a?(Array) ? arg : [arg]
|
||||
i = 1+args.index {|x|/^[^-]/=~x}
|
||||
args[0...i] + arg + args[i..-1]
|
||||
end
|
||||
|
||||
def self.new prefix, exe
|
||||
r = super()
|
||||
r.exe = exe
|
||||
r.prefix = prefix
|
||||
r.on {|cmd, *argv| raise UnknownCommandError, 'Unknown Command: %s' % cmd }
|
||||
r
|
||||
end
|
||||
|
||||
def on *names, &run
|
||||
if names.empty?
|
||||
@default_cmd = run
|
||||
else
|
||||
names.each {|name| self[name.to_s.to_sym] = run }
|
||||
end
|
||||
end
|
||||
|
||||
def cmd argv
|
||||
if @prefix == @exe
|
||||
raise ExpectingCommandError if argv.empty?
|
||||
[argv[0].to_sym, *argv[1..-1]]
|
||||
else
|
||||
@exe =~ /^(?:#{Regexp.escape @prefix}-)?(.*)$/
|
||||
[$1.to_sym, *argv]
|
||||
end
|
||||
end
|
||||
|
||||
def each all=nil, &block
|
||||
if not block_given?
|
||||
Enumerator.new self, :each, all
|
||||
elsif all
|
||||
super(&block)
|
||||
else
|
||||
super() {|k,v| yield k,v unless /^-/ =~ k.to_s }
|
||||
end
|
||||
end
|
||||
|
||||
def run *argv
|
||||
c, *argv = self.cmd( argv)
|
||||
(self[c] || @default_cmd).call c, *argv
|
||||
rescue CommandError
|
||||
STDERR.puts $!.message
|
||||
exit 1
|
||||
end
|
||||
alias call run
|
||||
|
||||
def to_proc
|
||||
method(:call).to_proc
|
||||
end
|
||||
end
|
||||
|
||||
class RunCave
|
||||
class CommandExpected <Exception
|
||||
end
|
||||
class ResumeFileExpected <Exception
|
||||
end
|
||||
|
||||
attr_accessor :preargs, :cmd, :args
|
||||
attr_reader :dummy, :resumable
|
||||
attr_writer :resume_file
|
||||
def initialize preargs = nil, cmd = nil, args = nil
|
||||
@resume_file, @dummy, @preargs, @cmd, @args = nil, nil, preargs || [], cmd || [], args || []
|
||||
end
|
||||
|
||||
def dummy?() @dummy end
|
||||
def dummy!() @dummy = true; self end
|
||||
def dummy=( v) @dummy = !!v; self end
|
||||
|
||||
def resumable?() @resumable end
|
||||
def resumable!() @resumable = true; self end
|
||||
def resumable=( v) @resumable = !!v; self end
|
||||
|
||||
def prepare_resume_file
|
||||
resumable? ? ['--resume-file', @resume_file || raise(ResumeFileExpected)] : []
|
||||
end
|
||||
|
||||
def prepare
|
||||
raise CommandExpected, "Set #{self}.cmd = yourcommand." if @cmd.nil? or @cmd.empty?
|
||||
[ :cave, @preargs, @cmd, prepare_resume_file, *@args ].flatten.select{|x|x}.map {|x| x.to_s }
|
||||
end
|
||||
|
||||
def run
|
||||
a = prepare
|
||||
if dummy?
|
||||
puts a.shelljoin
|
||||
0
|
||||
else
|
||||
Kernel.system *a
|
||||
$? ? $?.exitstatus : 130
|
||||
end
|
||||
end
|
||||
|
||||
def this cmd, *args
|
||||
@cmd, @args = cmd, args
|
||||
self
|
||||
end
|
||||
alias method_missing this
|
||||
|
||||
def resume_file f
|
||||
if f.nil?
|
||||
@resume_file
|
||||
else
|
||||
@resume_file = f
|
||||
self
|
||||
end
|
||||
end
|
||||
|
||||
def run_exit
|
||||
exit run
|
||||
end
|
||||
alias call run_exit
|
||||
end
|
||||
|
||||
def pager *args, &exe
|
||||
if $ttymode
|
||||
IRB::Pager::pager *args, &exe
|
||||
else
|
||||
exe.call
|
||||
end
|
||||
end
|
||||
|
||||
def pagered exe = nil, &block
|
||||
exe ||= block
|
||||
lambda {|*args| pager { exe.call *args } }
|
||||
end
|
||||
|
||||
ENV['LESS'] = "-FR #{ENV['LESS']}"
|
||||
|
||||
argv0 = $0
|
||||
cave = RunCave.new
|
||||
cmds = Commands.new 'truckle', File.basename(argv0)
|
||||
cmds.on {|*args| cave.this(*args).() }
|
||||
|
||||
cmds.on :help, '-h', '--help', &pagered { cmd=cmds.prefix; STDOUT.puts <<EOF }
|
||||
Usage
|
||||
=====
|
||||
|
||||
#{cmd} Command *ARGS
|
||||
|
||||
Resumable
|
||||
=========
|
||||
|
||||
You do not need to set a resume-file. #{cmd} will determine it automaticaly. First, you can give a first argument for tagging. Tag must be numerical!
|
||||
|
||||
#{cmd} 321 resolve netcat6
|
||||
#{cmd} 321 do
|
||||
|
||||
If you do not give a tag, #{cmd} will use the actual terminal-device-number. If it isn't possible to determine terminal, the parent-pid will be used.
|
||||
|
||||
Like cave but different
|
||||
=======================
|
||||
|
||||
«resolve» and «remove» will be displayed by a pager, so you can scroll up and down like
|
||||
|
||||
cave -cy resolve WHAT | less -R
|
||||
cave -cy uninstall WHAT | less -R
|
||||
|
||||
«install» and «uninstall» are not displayed by a pager, but will execute:
|
||||
|
||||
cave -cy resolve -x
|
||||
cave -cy uninstall -x
|
||||
|
||||
«do» and «resume» are special for:
|
||||
|
||||
cave resume
|
||||
EOF
|
||||
|
||||
cmds.on(:sync) { cave.sync.() }
|
||||
cmds.on :search, :show, &pagered {|*args| cave.this(*args).() }
|
||||
cmds.on :resolve, &pagered {|*args| cave.resumable!.this(*args).() }
|
||||
cmds.on 'fix-linkage', &pagered {|*args|
|
||||
cave.resumable!.this *args
|
||||
cave.args.push '--', cave.prepare_resume_file
|
||||
cave.resumable = false
|
||||
cave.()
|
||||
}
|
||||
cmds.on :remove, &pagered {|cmd, *args| cave.resumable!.uninstall(*args).() }
|
||||
cmds.on :upgrade, &pagered {|cmd, *args| cave.resumable!.resolve( '-c', :world, *args).() }
|
||||
cmds.on(:install) {|cmd, *argv| cave.resumable!.resolve( '-x', *argv).() }
|
||||
cmds.on(:uninstall) {|cmd, *argv| cave.resumable!.uninstall( '-x', *argv).() }
|
||||
|
||||
cmds.on(:do, :resume) {|cmd, *args| cave.resumable!.resume( *args).() }
|
||||
|
||||
cmds.on('--list-commands') { puts cmds.map{|k,v|k} }
|
||||
cmds.on '--install-commands' do
|
||||
dir, exe = File.split( argv0)
|
||||
cmds.each do |k,v|
|
||||
k = k.to_s
|
||||
k = File.join dir, "#{cmds.prefix}-#{k}"
|
||||
File.symlink exe, k unless File.exist? k
|
||||
end
|
||||
end
|
||||
|
||||
cave.dummy! if %w[1 true yes].include?( ENV['DUMMY'].to_s.downcase)
|
||||
$ttymode = STDOUT.tty?
|
||||
cave.preargs.push '-cy' if STDOUT.tty?
|
||||
resumefilesuffix = if /^\d+$/ =~ ARGV[0]
|
||||
"tag-#{ARGV.shift}"
|
||||
elsif $ttymode and STDIN.tty? and STDOUT.stat.rdev == STDIN.stat.rdev
|
||||
"dev-#{STDOUT.stat.rdev}"
|
||||
else
|
||||
"ppd-#{Process.ppid}"
|
||||
end
|
||||
cave.resume_file = "/tmp/truckle-resume-#{resumefilesuffix}"
|
||||
cmds.run *ARGV
|
||||
Loading…
Reference in New Issue