require "main"
require "log4r"
require "activesupport"
require "piston/version"
require "piston/commands"
Main {
program "piston"
author "François Beausoleil <francois@teksol.info>"
version Piston::VERSION::STRING
mixin :standard_options do
option("verbose", "v") {
argument_optional
cast :integer
default 0
validate {|value| (0..5).include?(value)}
description "Verbosity level (0 to 5, 0 being the default)"
}
option("quiet", "q") { default false }
option("force") { default false }
option("dry-run") { default false }
end
mixin :revision_or_commit do
option "revision", "r" do
argument_required
default "HEAD"
description "The revision you wish to operate on"
end
option "commit" do
argument_required
default "HEAD"
description "The commit you wish to operate on"
end
def target_revision
case
when params["revision"].given?
params["revision"].value
when params["commit"].given?
params["commit"].value
else
:head
end
end
end
mode "import" do
mixin :standard_options
mixin :revision_or_commit
argument "repository" do
required
description "The repository you wish to Pistonize"
end
argument "directory" do
optional
default nil
description "Where to put the Pistonized repository"
end
option("lock") do
default false
description "Automatically lock down the revision/commit to protect against blanket updates"
end
option("repository-type") do
argument :required
default nil
description "Force a specific repository type, for when it's not possible to guess"
end
logger_level Logger::DEBUG
def run
configure_logging!
if params["revision"].given? && params["commit"].given? then
raise ArgumentError, "Only one of --revision or --commit can be given. Received both."
end
cmd = Piston::Commands::Import.new(:verbose => params["verbose"].value,
:quiet => params["quiet"].value,
:force => params["force"].value,
:dry_run => params["dry-run"].value,
:repository_type => params["repository-type"].value)
begin
cmd.run(params[:repository].value, self.target_revision, params[:directory].value)
rescue Piston::Repository::UnhandledUrl => e
supported_types = Piston::Repository.handlers.collect do |handler|
handler.repository_type
end
puts "Unsure how to handle:"
puts "\t#{params[:repository].value.inspect}."
puts "You should try using --repository-type. Supported types are:"
supported_types.each do |type|
puts "\t#{type}"
end
exit_failure!
end
# Lock the working copy, if the user asked for it
cmd = Piston::Commands::LockUnlock.new(:verbose => params["verbose"].value,
:quiet => params["quiet"].value,
:force => params["force"].value,
:dry_run => params["dry-run"].value)
if params["lock"].value
cmd.run(params["directory"].value, params["lock"].value)
else
cmd.run(params["directory"].value, false)
end
end
end
mode "lock" do
mixin :standard_options
argument "directory" do
argument_required
optional
description "Which directory to lock"
end
logger_level Logger::DEBUG
def run
configure_logging!
cmd = Piston::Commands::LockUnlock.new(:wcdir => params["directory"].value,
:verbose => params["verbose"].value,
:quiet => params["quiet"].value,
:force => params["force"].value)
begin
cmd.run(true)
puts "#{params["directory"].value} locked"
rescue Piston::WorkingCopy::NotWorkingCopy
puts "The #{params["directory"].value} is not Pistonized"
end
end
end
mode "unlock" do
mixin :standard_options
argument "directory" do
argument_required
optional
description "Which directory to lock"
end
logger_level Logger::DEBUG
def run
configure_logging!
cmd = Piston::Commands::LockUnlock.new(:wcdir => params["directory"].value,
:verbose => params["verbose"].value,
:quiet => params["quiet"].value,
:force => params["force"].value)
begin
cmd.run(false)
puts "#{params["directory"].value} unlocked"
rescue Piston::WorkingCopy::NotWorkingCopy
puts "The #{params["directory"].value} is not Pistonized"
end
end
end
mode "info" do
mixin :standard_options
argument "directory" do
argument_required
optional
description "Which directory to get info"
end
logger_level Logger::DEBUG
def run
configure_logging!
cmd = Piston::Commands::Info.new(:wcdir => params["directory"].value,
:verbose => params["verbose"].value,
:quiet => params["quiet"].value,
:force => params["force"].value)
begin
puts cmd.run(params["directory"].value)
rescue Piston::WorkingCopy::NotWorkingCopy
puts "The #{params["directory"].value} is not Pistonized"
end
end
end
mode "update" do
mixin :standard_options
mixin :revision_or_commit
argument("directory") { optional }
def run
configure_logging!
if params["revision"].given? && params["commit"].given? then
raise ArgumentError, "Only one of --revision or --commit can be given. Received both."
end
cmd = Piston::Commands::Update.new(:lock => params["lock"].value,
:verbose => params["verbose"].value,
:quiet => params["quiet"].value,
:force => params["force"].value,
:dry_run => params["dry-run"].value)
cmd.run(params["director"].value, target_revision)
end
end
option("version", "v")
def run
if params["version"].given? || ARGV.first == "version" then
puts Piston.version_message
exit_success!
elsif ARGV.empty?
puts Piston.version_message
puts "\nNo mode given. Call with help to find out the available options."
exit_failure!
else
puts "Unrecognized mode: #{ARGV.first.inspect}. Use the help mode to find the available options."
exit_warn!
end
end
def configure_logging!
Log4r::Logger.root.level = Log4r::INFO
case params["verbose"].value
when 0
main_level = Log4r::INFO
handler_level = Log4r::WARN
client_level = Log4r::WARN
client_out_level = Log4r::WARN
stdout_level = Log4r::INFO
when 1
main_level = Log4r::DEBUG
handler_level = Log4r::INFO
client_level = Log4r::WARN
client_out_level = Log4r::WARN
stdout_level = Log4r::DEBUG
when 2
main_level = Log4r::DEBUG
handler_level = Log4r::DEBUG
client_level = Log4r::INFO
client_out_level = Log4r::WARN
stdout_level = Log4r::DEBUG
when 3
main_level = Log4r::DEBUG
handler_level = Log4r::DEBUG
client_level = Log4r::DEBUG
client_out_level = Log4r::INFO
stdout_level = Log4r::DEBUG
when 4, 5
main_level = Log4r::DEBUG
handler_level = Log4r::DEBUG
client_level = Log4r::DEBUG
client_out_level = Log4r::DEBUG
stdout_level = Log4r::DEBUG
else
raise ArgumentError, "Did not expect verbosity to be outside 0..5: #{params["verbose"].value}"
end
Log4r::Logger.new("main", main_level)
Log4r::Logger.new("handler", handler_level)
Log4r::Logger.new("handler::client", client_level)
Log4r::Logger.new("handler::client::out", client_out_level)
Log4r::StdoutOutputter.new("stdout", :level => stdout_level)
Log4r::Logger["main"].add "stdout"
Log4r::Logger["handler"].add "stdout"
end
}
) is private.
