# This set of tasks helps automate the workflow as described on
# http://reductivelabs.com/trac/puppet/wiki/Development/DevelopmentLifecycle


def find_start(start)
# This is a case statement, as we might want to map certain
# git tags to starting points that are not currently in git.
	case start
		when nil?:
		when @next_release: return "master"
		else return start
	end
end

desc "Set up git for working with Puppet"
task :git_setup do
	# This should be changed as new versions get released
	@next_release = '0.26.x'
	@remote = {}
	default_remote = {}
	default_remote[:url] = 'git://github.com/reductivelabs/puppet'
	default_remote[:name] = 'origin'
	@remote[:name] = %x{git config puppet.defaultremote}.chomp
	@remote[:name] = @remote[:name].empty? ? default_remote[:name] : @remote[:name]
	@remote[:url] = default_remote[:url] if @remote[:name] == default_remote[:name]
	default_fetch = '+refs/heads/*:refs/remotes/puppet/*'
	@remote[:fetch] = %x{git config puppet.#{@remote[:name]}.fetch}.chomp
	@remote[:fetch] = @remote[:fetch].empty? ?  default_fetch : @remote[:fetch]
end

desc "Start work on a feature"
task :start_feature, [:feature,:remote,:branch] => :git_setup do |t, args|
	args.with_defaults(:remote => @remote[:name])
	args.with_defaults(:branch => @next_release)
	start_at = find_start(args.branch)
	branch = "feature/#{start_at}/#{args.feature}"
	sh "git checkout -b #{branch} #{start_at}" do |ok, res|
		if ! ok
			raise <<EOS
Was not able to create branch for #{args.feature} on branch #{args.branch}, starting at #{start_at}: error code was: #{res.exitstatus}
EOS
		end
	end
	sh "git config branch.#{branch}.remote #{args.remote}" do |ok, res|
		raise "Could not set remote: #{$?}" unless ok
	end

	sh "git config branch.#{branch}.merge refs/heads/#{branch}" do |ok, res|
		raise "Could not configure merge: #{$?}" unless ok
	end
end

desc "Do git prep to start work on a Redmine ticket"
task :start_ticket, [:ticket, :remote, :branch] => :git_setup do |t, args|
	args.with_defaults(:remote => @remote[:name])
	args.with_defaults(:branch => @next_release)
	start_at = find_start(args.branch)
	branch = "tickets/#{start_at}/#{args.ticket}"
	sh "git checkout -b #{branch} #{start_at}" do |ok, res|
		unless ok
			raise <<EOS
Was not able to create branch for ticket #{args.ticket} on branch #{args.branch}, starting at #{start_at}: error code was: #{$?}
Git command used was: #{command}
EOS
		end
	end
    sh "git config branch.#{branch}.remote #{args.remote}" do |ok, res|
    	raise "Could not set remote: #{$?}" unless ok
	end

    sh "git config branch.#{branch}.merge refs/heads/#{branch}" do |ok, res|
    	raise "Could not configure merge: #{$?}" unless ok
	end
end

# This isn't very useful by itself, but we might enhance it later, or use it
# in a dependency for a more complex task.
desc "Push out changes"
task :push_changes, [:remote] do |t, arg|
	branch = %x{git branch | grep "^" | awk '{print $2}'}
	sh "git push #{arg.remote} #{branch}" do |ok, res|
		raise "Unable to push to #{arg.remote}" unless ok
	end
end

desc "Send patch information to the puppet-dev list"
task :mail_patches do
    if Dir.glob("00*.patch").length > 0
        raise "Patches already exist matching '00*.patch'; clean up first"
    end

    unless %x{git status} =~ /On branch (.+)/
        raise "Could not get branch from 'git status'"
    end
    branch = $1

    unless branch =~ %r{^([^\/]+)/([^\/]+)/([^\/]+)$}
        raise "Branch name does not follow <type>/<parent>/<name> model; cannot autodetect parent branch"
    end

    type, parent, name = $1, $2, $3

    # Create all of the patches
    sh "git format-patch -C -M -s -n --subject-prefix='PATCH/puppet' #{parent}..HEAD"

    # And then mail them out.

    # If we've got more than one patch, add --compose
    if Dir.glob("00*.patch").length > 1
        compose = "--compose"
    else
        compose = ""
    end

    # Now send the mail.
    sh "git send-email #{compose} --no-signed-off-by-cc --suppress-from --to puppet-dev@googlegroups.com 00*.patch"

    # Finally, clean up the patches
    sh "rm 00*.patch"
end