Rakefile

# coding: UTF-8
require 'octokit'
require 'yaml'
require 'stove/rake_task'
require 'stove/cookbook'

if ENV['GITHUB_API_TOKEN']
  @octokit = Octokit::Client.new(:access_token => ENV['GITHUB_API_TOKEN'])
else
  @octokit = Octokit::Client.new
end

Stove::RakeTask.new

VERSION_WITH_NAME_REGEX = /version\s*'\d+\.\d+\.\d+'/
VERSION_REGEX = /\d+\.\d+\.\d+/
REPO = "bbaugher/apache_zookeeper"

task :release do
  puts "Releasing the cookbook ..."

  version = cookbook_version

  # Update change log
  puts "Updating change log ..."
  update_change_log version
  puts "Change log updated!"

  # Publish the cookbook
  puts "Publishing the cookbook ..."

  # Run stove's rake task
  Rake::Task['publish'].invoke

  puts "Published cookbook!"

  # Bump version
  versions = version.split "."
  versions[1] = versions[1].to_i + 1

  # Reset bug number if available
  if versions.size == 3
    versions[2] = 0
  end

  new_version = versions.join "."

  # Update the cookbook version
  puts "Updating version from #{version} to #{new_version} ..."
  update_cookbook_version new_version
  puts "Version updated!"
end

task :build_change_log do
  closed_milestones = @octokit.milestones REPO, {:state => "closed"}

  version_to_milestone = Hash.new
  versions = Array.new

  closed_milestones.each do |milestone|
    version = Gem::Version.new(milestone.title)
    version_to_milestone.store version, milestone
    versions.push version
  end

  versions = versions.sort.reverse

  change_log = File.open('CHANGELOG.md', 'w')

  begin
    change_log.write "Change Log\n"
    change_log.write "==========\n"
    change_log.write "\n"

    versions.each do |version|
      milestone = version_to_milestone[version]
      change_log.write generate_milestone_markdown(milestone)
      change_log.write "\n"
    end
  ensure
    change_log.close
  end
end

def cookbook_version
  cookbook = Stove::Cookbook.new('.')
  cookbook.version
end

def update_cookbook_version new_version
  # Read in the metadata file
  metadata = IO.read(File.join(File.dirname(__FILE__), 'metadata.rb')).chomp
  version_with_name = VERSION_WITH_NAME_REGEX.match(metadata)[0]

  index_to_version_info = metadata.index VERSION_WITH_NAME_REGEX
  # Have to subtract 1 to not include the first '
  index_to_version = version_with_name.index(VERSION_REGEX) - 1

  File.open("metadata.rb", 'w') { |file|
    file.write metadata[0, index_to_version_info + index_to_version]
    file.write "'#{new_version}'\n"
  }

  run_command "git add metadata.rb"
  run_command "git commit -m 'Released #{cookbook_version} and bumped version to #{new_version}'"
  run_command "git push origin HEAD"
end

def update_change_log version
  change_log_lines = IO.read(File.join(File.dirname(__FILE__), 'CHANGELOG.md')).split("\n")

  change_log = File.open('CHANGELOG.md', 'w')

  begin

    # Keep change log title
    change_log.write change_log_lines.shift
    change_log.write "\n"
    change_log.write change_log_lines.shift
    change_log.write "\n"
    change_log.write "\n"

    # Write new milestone info
    change_log.write generate_milestone_markdown(milestone(version))

    # Add previous change log info
    change_log_lines.each do |line|
      change_log.write line
      change_log.write "\n"
    end

  ensure
    change_log.close
  end

  run_command "git add CHANGELOG.md"
  run_command "git commit -m 'Added #{version} to change log'"
  run_command "git push origin HEAD"
end

def generate_milestone_markdown milestone
  strings = Array.new

  title = "[#{milestone.title} - #{milestone.updated_at.strftime("%m-%d-%Y")}](https://github.com/#{REPO}/issues?milestone=#{milestone.number}&state=closed)"

  strings.push "#{title}"
  strings.push "-" * title.length
  strings.push ""

  issues = @octokit.issues REPO, {:milestone => milestone.number, :state => "closed"}

  issues.each do |issue|
    strings.push "  * [#{issue_type issue}] [Issue-#{issue.number}](https://github.com/#{REPO}/issues/#{issue.number}) : #{issue.title}"
  end

  strings.push ""

  strings.join "\n"
end

def milestone version
  closedMilestones = @octokit.milestones REPO, {:state => "closed"}

  closedMilestones.each do |milestone|
    if milestone["title"] == version
      return milestone
    end
  end

  openMilestones = @octokit.milestones REPO

  openMilestones.each do |milestone|
    if milestone["title"] == version
      return milestone
    end
  end

  raise "Unable to find milestone with title [#{version}]"
end

def issue_type issue
  labels = Array.new
  issue.labels.each do |label|
    labels.push label.name.capitalize
  end
  labels.join "/"
end

def run_command command
  output = `#{command}`
  unless $?.success?
    raise "Command : [#{command}] failed.\nOutput : \n#{output}"
  end
end