helpers.rb

require 'net/http'
require 'json'
require 'digest/sha1'


def make_http_request(method, url, data={}, headers={})
  method.downcase!
  if not ['get', 'post'].include? method
    raise 'Bad method'
  end
  url = URI.parse url
  if method.eql? 'get'
    req = Net::HTTP::Get.new url.path, headers
  elsif method.eql? 'post'
    req = Net::HTTP::Post.new url.path, headers
    req.set_form_data data
  end
  req.basic_auth url.user, url.password

  http = Net::HTTP.new(url.host, url.port)
  http.use_ssl = true if url.scheme == 'https'
  return http.request(req)
end

def make_get_request(url, data={}, headers={})
  return make_http_request 'get', url, data, headers
end

def make_post_request(url, data={}, headers={})
  return make_http_request 'post', url, data, headers
end

def get_arc_settings(settings_file_name=nil)
  settings_file_name ||= File.expand_path "~/.arcrc"
  if not File.file? settings_file_name
    raise "No #{settings_file_name} file found"
  end
  return JSON.parse(File.read settings_file_name)
end

def get_username_from_arc_settings(settings_file_name=nil)
  settings = get_arc_settings settings_file_name
  return settings['hosts'].values[0]['user']
end

def get_host_from_arc_settings(settings_file_name=nil)
  settings = get_arc_settings settings_file_name
  return settings['hosts'].keys[0]
end

def check_arc_settings(settings_file_name=nil)
  # Checks if .arcrc has hosts. If there are more than one host, shows message.
  settings = get_arc_settings settings_file_name
  if not settings.include? 'hosts'
    raise "ERROR: Your .arcrc file has no info about host or doesn't exists. Install arc and run install-certificate."
  elsif settings['hosts'].keys.length > 1
    raise "ERROR: Your .arcrc file has more than one host. Comment out hosts you don't need"
  end
end

def generate_hash(token, cert)
  return Digest::SHA1.hexdigest token + cert
end

def get_phabricator_request_body(data={})
  return {
    'params' => data.to_json,
    'output' => 'json',
    'host' => 'http:/localhost',
  }
end

def get_auth_kwargs(username, settings_file_name)
  settings = get_arc_settings settings_file_name
  cert = settings['hosts'].values[0]['cert']
  token = "%.8f" % Time.now.to_f
  return {
    'authToken' => token,
    'authSignature' => generate_hash(token, cert),
    'user' => username,
  }
end

def get_result_commit_status commit_statuses
  if commit_statuses.nil?
    return 'in_progress'
  else
    if commit_statuses.include? 'concerned'
      return 'concerned'
    elsif commit_statuses.include? 'accepted'
      return 'accepted'
    else
      return 'in_progress'
   end
  end
end

def get_branches_from_raw_data branchres_response
  "Extracting branch names from phabricators commit/branches/ response"
  raw_data = branchres_response.split('for (;;);')[1]
  begin
    return JSON(raw_data)['payload'].scan(/>([^,]*?)</).flatten(1)
  rescue JSON::GeneratorError  # this happens when auth settings are incorrect and login page returned instead of JSON
    return []
  end
end

def remove_basic_auth_if_present url
  url = URI.parse url
  url.user = nil
  url.password = nil
  return url.to_s
end