18-jira_update_status.rb

# frozen_string_literal: true

load './lib/common.rb'

SPACE_NAME = ENV['JIRA_API_PROJECT_NAME']
JIRA_PROJECT_NAME = SPACE_NAME + (@debug ? ' TEST' : '')

JIRA_API_STATUSES = ENV['JIRA_API_STATUSES']

@assembla_status_to_jira = {}
JIRA_API_STATUSES.split(',').each do |status|
  from, to = status.split(':')
  @assembla_status_to_jira[from.downcase] = to || from
end

space = get_space(SPACE_NAME)
dirname_assembla = get_output_dirname(space, 'assembla')

# Assembla tickets
tickets_csv = "#{dirname_assembla}/tickets.csv"
@tickets_assembla = csv_to_array(tickets_csv)

# --- Filter by date if TICKET_CREATED_ON is defined --- #
tickets_created_on = get_tickets_created_on

if tickets_created_on
  puts "\nFilter newer than: #{tickets_created_on}"
  @tickets_assembla.select! { |item| item_newer_than?(item, tickets_created_on) }
end

# Collect ticket statuses
@assembla_statuses = {}
@extra_summary_types = {}
@tickets_assembla.each do |ticket|
  status = ticket['status']
  summary = ticket['summary']
  if @assembla_statuses[status].nil?
    @assembla_statuses[status] = 0
  else
    @assembla_statuses[status] += 1
  end
  if summary.match(/^([A-Z]*):/)
    t = summary.sub(/:.*$/, '\1')
    @extra_summary_types[t] = true if !ASSEMBLA_TYPES_IN_SUMMARY.include?(t.downcase) && @extra_summary_types[t].nil?
  end
end

@total_assembla_tickets = @tickets_assembla.length
puts "\nTotal Assembla tickets: #{@total_assembla_tickets}"

puts "\nAssembla ticket statuses:"
@assembla_statuses.keys.each do |key|
  puts "* #{key}: #{@assembla_statuses[key]}"
end

if @extra_summary_types.length.positive?
  puts "\nExtra statuses detected in the summary (ignored): #{@extra_summary_types.length}"
  @extra_summary_types.keys.sort.each do |type|
    puts "* #{type}"
  end
end

# Sanity check just in case
@missing_statuses = []
@assembla_statuses.keys.each do |key|
  @missing_statuses << key unless @assembla_status_to_jira[key.downcase]
end

if @missing_statuses.length.positive?
  puts "\nSanity check => NOK"
  puts "The following statuses are missing:"
  @missing_statuses.each do |status|
    puts "* #{status}"
  end
  goodbye("Update JIRA_API_STATUSES in .env file and create JIRA statuses if needed")
end
puts "Sanity check => OK"

# Jira tickets
resolutions_jira_csv = "#{OUTPUT_DIR_JIRA}/jira-resolutions.csv"
statuses_jira_csv = "#{OUTPUT_DIR_JIRA}/jira-statuses.csv"
tickets_jira_csv = "#{OUTPUT_DIR_JIRA}/jira-tickets-all.csv"

@resolutions_jira = csv_to_array(resolutions_jira_csv)
@statuses_jira = csv_to_array(statuses_jira_csv)
@tickets_jira = csv_to_array(tickets_jira_csv)

@jira_resolution_name_to_id = {}
puts "\nJira ticket resolutions:"
@resolutions_jira.each do |resolution|
  @jira_resolution_name_to_id[resolution['name'].downcase] = resolution['id']
  puts "* #{resolution['id']}: #{resolution['name']}"
end

@jira_status_name_to_id = {}
puts "\nJira ticket statuses:"
@statuses_jira.each do |status|
  @jira_status_name_to_id[status['name'].downcase] = status['id']
  puts "* #{status['id']}: #{status['name']}"
end

# Convert assembla_ticket_id to jira_ticket
@assembla_id_to_jira = {}
@tickets_jira.each do |ticket|
  jira_id = ticket['jira_ticket_id']
  assembla_id = ticket['assembla_ticket_id']
  @assembla_id_to_jira[assembla_id] = jira_id
end

# GET /rest/api/2/issue/{issueIdOrKey}/transitions
def jira_get_transitions(issue_id)
  result = nil
  url = "#{URL_JIRA_ISSUES}/#{issue_id}/transitions"
  begin
    response = RestClient::Request.execute(method: :get, url: url, headers: JIRA_HEADERS)
    result = JSON.parse(response.body)
    puts "\nGET #{url} => OK"
  rescue RestClient::ExceptionWithResponse => e
    rest_client_exception(e, 'GET', url)
  rescue => e
    puts "\nGET #{url} => NOK (#{e.message})"
  end
  if result.nil?
    nil
  else
    transitions = result['transitions']
    puts "\nJira ticket transitions:"
    transitions.each do |transition|
      puts "* #{transition['id']} '#{transition['name']}' =>  #{transition['to']['id']} '#{transition['to']['name']}'"
    end
    puts
    transitions
  end
end

# POST /rest/api/2/issue/{issueIdOrKey}/transitions
def jira_update_status(issue_id, status, counter)
  if status.casecmp('done').zero? || status.casecmp('invalid').zero?
    payload = {
      update: {},
      transition: {
        id: @transition_target_name_to_id['done'].to_i
      }
    }.to_json
    transition = {
      from: {
        id: @jira_status_name_to_id['to do'],
        name: 'to do'
      },
      to: {
        id: @jira_status_name_to_id['done'],
        name: 'done'
      }
    }
  elsif status.casecmp('new').zero?
    # Do nothing
    transition = {
      from: {
        id: @jira_status_name_to_id['to do'],
        name: 'to do'
      },
      to: {
        id: @jira_status_name_to_id['to do'],
        name: 'to do'
      }
    }
    return { transition: transition }
  elsif status.casecmp('in progress').zero?
    payload = {
      update: {},
      transition: {
        id: @transition_target_name_to_id['in progress'].to_i
      }
    }.to_json
    transition = {
      from: {
        id: @jira_status_name_to_id['to do'],
        name: 'to do'
      },
      to: {
        id: @jira_status_name_to_id['in progress'],
        name: 'in progress'
      }
    }
  else
    # TODO: Figure out how to deal with the other statuses: testable, blocked, ready for acceptance, etc.
    # For now just set them to 'in progress'
    payload = {
      update: {},
      transition: {
        id: @transition_target_name_to_id['in progress'].to_i
      }
    }.to_json
    transition = {
      from: {
        id: @jira_status_name_to_id['to do'],
        name: 'to do'
      },
      to: {
        id: @jira_status_name_to_id['in progress'],
        name: 'in progress'
      }
    }
  end

  result = nil
  url = "#{URL_JIRA_ISSUES}/#{issue_id}/transitions"
  begin
    percentage = ((counter * 100) / @total_assembla_tickets).round.to_s.rjust(3)
    RestClient::Request.execute(method: :post, url: url, payload: payload, headers: JIRA_HEADERS)
    puts "#{percentage}% [#{counter}|#{@total_assembla_tickets}] POST #{url} '#{transition[:from][:name]}' to '#{transition[:to][:name]}' => OK"
    result = { transition: transition }
  rescue RestClient::ExceptionWithResponse => e
    rest_client_exception(e, 'POST', url, payload)
  rescue => e
    puts "#{percentage}% [#{counter}|#{@total_assembla_tickets}] POST #{url} #{payload.inspect} => NOK (#{e.message})"
  end
  if result
    # If the issue has been closed (done) we set the resolution to the appropriate value
    if status.casecmp('done').zero? || status.casecmp('invalid').zero?
      resolution_name = status.casecmp('invalid').zero? ? "Won't do" : 'Done'
      resolution_id = @jira_resolution_name_to_id[resolution_name.downcase].to_i
      payload = {
        update: {},
        fields: {
          resolution: {
            id: "#{resolution_id}"
          }
        }
      }.to_json
      url = "#{URL_JIRA_ISSUES}/#{issue_id}"
      begin
        RestClient::Request.execute(method: :put, url: url, payload: payload, headers: JIRA_HEADERS)
      rescue RestClient::ExceptionWithResponse => e
        rest_client_exception(e, 'PUT', url, payload)
      rescue => e
        puts "PUT #{url} resolution='#{resolution_name}' => NOK (#{e.message})"
      end
    end
  end
  result
end

@transitions = jira_get_transitions(@assembla_id_to_jira[@tickets_assembla.first['id']])

@transition_target_name_to_id = {}
@transitions.each do |transition|
  @transition_target_name_to_id[transition['to']['name'].downcase] = transition['id'].to_i
end

@jira_updates_tickets = []

@tickets_assembla.each_with_index do |ticket, index|
  assembla_ticket_id = ticket['id']
  assembla_ticket_status = ticket['status']
  jira_ticket_id = @assembla_id_to_jira[ticket['id']]
  result = jira_update_status(jira_ticket_id, assembla_ticket_status, index + 1)
  @jira_updates_tickets << {
    result: result.nil? ? 'NOK' : 'OK',
    assembla_ticket_id: assembla_ticket_id,
    assembla_ticket_status: assembla_ticket_status,
    jira_ticket_id: jira_ticket_id,
    jira_transition_from_id: result.nil? ? 0 : result[:transition][:from][:id],
    jira_transition_from_name: result.nil? ? 0 : result[:transition][:from][:name],
    jira_transition_to_id: result.nil? ? 0 : result[:transition][:to][:id],
    jira_transition_to_name: result.nil? ? 0 : result[:transition][:to][:name]
  }
end

puts "\nTotal updates: #{@jira_updates_tickets.length}"
updates_tickets_jira_csv = "#{OUTPUT_DIR_JIRA}/jira-tickets-status-updates.csv"
write_csv_file(updates_tickets_jira_csv, @jira_updates_tickets)