#!/usr/bin/env ruby

require 'pathname'
require 'shellwords'
require 'stringio'
require 'ipaddr'
require_relative '../lib/dencli'

cli = DenCli.new :example, "This is an example for generate a DenCli-API"

class Capture
	def initialize cli, verbose:
		@cli, @counter, @verbose = cli, 0, verbose
	end

	def capture
		@args = NilClass
		@counter += 1
		stdout, stderr = $stdout, $stderr
		$stdout = $stderr = StringIO.new
		begin
			yield stdout, stderr
		ensure
			$stderr, $stdout = stderr, stdout
		end
	end

	def args= args
		@args = args
	end

	def logstart command
		STDERR.printf "[% 4d] \e[1;35m? \e[0m %s tests %s\r", @counter, $0.shellescape, command.shelljoin
	end

	def logok info, command
		STDERR.printf "[% 4d] \e[1;32mok\e[0m %s | %s tests %s\e[J\n", @counter, info, $0.shellescape, command.shelljoin
	end

	def logfail command
		STDERR.printf "[% 4d] \e[1;31mer\e[0m %s tests %s\e[J\n", @counter, $0.shellescape, command.shelljoin
	end

	def logexception prefix, exception
		loginfo "#{prefix} (#{exception.class.name}) #{exception}"
		exception.backtrace[0...-Kernel.caller.length].each {|l| loginfo "  #{l}" }
	end

	def loginfo text
		STDERR.printf "           %s\n", text
	end

	def should_ok expect, *command
		logstart command
		$capture.capture { @cli.call 'tests', *command }
		if expect === @args
			logok @args, command
		else
			logfail command
			loginfo "expected args: #{expect.inspect}"
			loginfo "given args: #{@args.inspect}"
			STDERR.puts
		end
	rescue SystemExit
		if 0 == $!.status
			logok @args, command
		else
			logfail command
		end
	rescue Object
		logfail command
		logexception "unexpected raise:", $!
		STDERR.puts
	end

	def should_fail exception, message, *command
		logstart command
		$capture.capture { @cli.call 'tests', *command }
		logfail command
	rescue exception
		if message === $!.message
			logok exception, command
			if @verbose
				logexception "raised:", $!
				STDERR.puts
			end
		else
			logexception "unexpected message:", $!
			STDERR.puts
		end
	rescue Object
		logfail command
		logexception "unexpected raised:", $!
		STDERR.puts
	end
end

cli.sub :tests, "Some tests", noshortaliases: true do |tcli|
	OptionParser.accept IPAddr do |arg|
		begin
			IPAddr.new arg
		rescue IPAddr::InvalidAddressError
			raise OptionParser::InvalidArgument, "#{$!.message}: #{arg}"
		end
	end

	tcli.cmd( :'-', "No arguments no options expected", &lambda {|| $capture.args = [] })
	tcli.cmd( :'arg', "", &lambda {|one| $capture.args = [one] })
	tcli.cmd( :'oar', "", &lambda {|one=nil| $capture.args = [one] })
	tcli.cmd( :'arg-arg', "", &lambda {|one, two| $capture.args = [one, two] })
	tcli.cmd( :'oar-oar', "", &lambda {|one=nil, two=nil| $capture.args = [one, two] })
	tcli.cmd( :'arg-oar', "expected", &lambda {|one, two=nil| $capture.args = [one, two] })
	tcli.cmd( :'oar-arg', "expected", &lambda {|one=nil, two| $capture.args = [one, two] })
	tcli.cmd( :'bool', '', &lambda {|a:| $capture.args = [a] }).opt(:a, '-a', '')
	tcli.cmd( :'optbool', '', &lambda {|a: nil| $capture.args = [a] }).opt(:a, '-a', '')
	tcli.cmd( :'defbool', '', &lambda {|a:| $capture.args = [a] }).opt(:a, '-a', '', default: 'default')
	tcli.cmd( :'str', '', &lambda {|a:| $capture.args = [a] }).opt(:a, '-a=STR', '')
	tcli.cmd( :'lstr', '', &lambda {|a:| $capture.args = [a] }).opt(:a, '--astr=STR', '')
	tcli.cmd( :'bstr', '', &lambda {|a:| $capture.args = [a] }).opt(:a, '-a', '--astr=STR', '')
	tcli.cmd( :'ipaddr', '', &lambda {|a:| $capture&.args = [a] }).opt(:a, '-a', '--addr=ADDR', IPAddr, '')

	tcli.cmd( :run, "Run all tests") do |verbose:|
		$capture = Capture.new cli, verbose: verbose

		$capture.should_fail DenCli::UnknownCommand, //, 'unknown-command'

		$capture.should_ok [], '-'
		$capture.should_fail DenCli::UsageError, //, '-', 'unexpected'

		$capture.should_ok %w[first], 'arg', 'first'
		$capture.should_fail DenCli::UsageError, //, 'arg'
		$capture.should_fail DenCli::UsageError, //, 'arg', 'first', 'unexpected'

		$capture.should_ok %w[first], 'oar', 'first'
		$capture.should_ok [nil], 'oar'
		$capture.should_fail DenCli::UsageError, //, 'oar', 'first', 'unexpected'

		$capture.should_ok %w[first two], 'oar-oar', 'first', 'two'
		$capture.should_ok ['first', nil], 'oar-oar', 'first'
		$capture.should_ok [nil,nil], 'oar-oar'
		$capture.should_fail DenCli::UsageError, //, 'oar-oar', 'first', 'two', 'unexpected'

		$capture.should_ok %w[first two], 'arg-oar', 'first', 'two'
		$capture.should_ok ['first', nil], 'arg-oar', 'first'
		$capture.should_fail DenCli::UsageError, //, 'arg-oar'

		$capture.should_ok [nil, 'first'], 'oar-arg', 'first'
		$capture.should_ok ['first', 'second'], 'oar-arg', 'first', 'second'
		$capture.should_fail DenCli::UsageError, //, 'oar-arg'
		$capture.should_fail DenCli::UsageError, //, 'oar-arg', 'first', 'two', 'unexpected'

		$capture.should_ok [true], 'bool', '-a'
		$capture.should_fail DenCli::UsageError, //, 'bool'
		$capture.should_fail DenCli::UsageError, //, 'bool', '-a', 'unexpected'
		$capture.should_fail OptionParser::InvalidOption, //, 'bool', '-b'
		$capture.should_fail OptionParser::InvalidOption, //, 'bool', '--unexpected'
		$capture.should_fail DenCli::UsageError, //, 'bool', 'unexpected'

		$capture.should_ok [true], 'optbool', '-a'
		$capture.should_ok [nil], 'optbool'
		$capture.should_fail DenCli::UsageError, //, 'optbool', '-a', 'unexpected'
		$capture.should_fail OptionParser::InvalidOption, //, 'optbool', '-b'
		$capture.should_fail OptionParser::InvalidOption, //, 'optbool', '--unexpected'
		$capture.should_fail DenCli::UsageError, //, 'optbool', 'unexpected'

		$capture.should_ok [true], 'defbool', '-a'
		$capture.should_ok ['default'], 'defbool'
		$capture.should_fail DenCli::UsageError, //, 'defbool', '-a', 'unexpected'
		$capture.should_fail OptionParser::InvalidOption, //, 'defbool', '-b'
		$capture.should_fail OptionParser::InvalidOption, //, 'defbool', '--unexpected'
		$capture.should_fail DenCli::UsageError, //, 'defbool', 'unexpected'

		$capture.should_ok %w[first], 'str', '-a', 'first'
		$capture.should_ok %w[first], 'str', '-afirst'
		$capture.should_fail OptionParser::MissingArgument, //, 'str', '-a'
		$capture.should_fail DenCli::UsageError, //, 'str'
		$capture.should_fail OptionParser::InvalidOption, //, 'str', '-b'
		$capture.should_fail OptionParser::InvalidOption, //, 'str', '--unexpected'
		$capture.should_fail DenCli::UsageError, //, 'str', 'unexpected'

		$capture.should_ok %w[first], 'lstr', '--astr', 'first'
		$capture.should_ok %w[first], 'lstr', '--astr=first'
		$capture.should_fail OptionParser::MissingArgument, //, 'lstr', '--astr'
		$capture.should_fail DenCli::UsageError, //, 'lstr'
		$capture.should_fail OptionParser::InvalidOption, //, 'lstr', '-b'
		$capture.should_fail OptionParser::InvalidOption, //, 'lstr', '--unexpected'
		$capture.should_fail DenCli::UsageError, //, 'lstr', 'unexpected'

		$capture.should_ok %w[first], 'bstr', '-a', 'first'
		$capture.should_ok %w[first], 'bstr', '-afirst'
		$capture.should_ok %w[first], 'bstr', '--astr', 'first'
		$capture.should_ok %w[first], 'bstr', '--astr=first'
		$capture.should_fail OptionParser::MissingArgument, //, 'bstr', '--astr'
		$capture.should_fail OptionParser::MissingArgument, //, 'bstr', '-a'
		$capture.should_fail DenCli::UsageError, //, 'bstr'
		$capture.should_fail OptionParser::InvalidOption, //, 'bstr', '-b'
		$capture.should_fail OptionParser::InvalidOption, //, 'bstr', '--unexpected'
		$capture.should_fail DenCli::UsageError, //, 'bstr', 'unexpected'

		$capture.should_ok [IPAddr.new('1.2.3.4')], 'ipaddr', '-a', '1.2.3.4'
		$capture.should_fail OptionParser::InvalidArgument, /invalid address/, 'ipaddr', '-a', '1.2.3.400'
	end.opt( :verbose, '-v', 'Prints additional information per test', default: false)
end

cli.cmd( :args, "Expects and prints given arguments",
	&lambda {|a, b, c:, d:, e:, f:, g:|
		p a: a, b: b, c: c, d: d, e: e
	}).
	opt( :c, '-c=ForC', "Option c").
	opt( :d, '-dForD', "Option d", default: "something").
	opt( :e, '-e', "Toggle e", default: false).
	opt( :f, '--[no-]f', "Toggle f", default: false).
	opt( :g, '--long-option=sth', "Long option, no short option", default: "nothing").
	opt( :h, '-hsth', "No long option, only short option", default: "nothing")

cli.cmd( :example, "I have an example command") { $stderr.puts "This is an example" }
cli.cmd( :help, "An example for help", aliases: [nil, '-h', '--help'], &lambda {|*commands, full:|
	if full
		cli.help_full *commands, output: $stderr
	else
		cli.help *commands, output: $stderr
	end
}).
	opt( :full, '-f', '--[no-]full', "Print all commands and sub-commands.", default: false)

cli.sub( :more, "Sub-Commands are also possible with a new cli") do |sub|
	sub.cmd( :help, "", aliases: [nil, '-h', '--help']) {|*args| $stderr.puts sub.help(*args) }
	sub.cmd( :example, "Here is an example, too") { $stderr.puts "This is an other example" }
	sub.cmd( :foo, "BAR") { $stderr.puts "FOO bar"}

	sub.cmd( :args, "Expects and prints given arguments", &lambda {|a, b=1, c:, d: 5, e:|
			p a: a, b: b, c: c, d: d, e: e
		}).
		opt( :c, '-c=ForC', "Option c").
		opt( :d, '-d=ForD', "Option d (implicit default)").
		opt( :e, '-e', "Toggle e")

	sub.sub( :deeper, "You want to have Sub-Sub-Commands?") do |sub2|
		sub2.cmd( :help, "", aliases: [nil, '-h', '--help'], &lambda {|*commands| sub2.help( *commands, output: $stderr) })
		sub2.cmd( :last, "The last example", &lambda { $stderr.puts "The last example" })

		sub2.sub( :'sub-commands', "Endless Sub-Sub- ...") do |sub3|
			sub3.cmd( :help, "") {|*args| $stderr.puts sub3.help( sub3, *args) }
			sub3.cmd( :hehe, "The real last example", min: 2) { $stderr.puts "Trust me!" }
		end
	end
end

begin
	cli.call *ARGV
rescue DenCli::UsageError
	$stderr.puts $!
	exit 1
end