class Inspec::Resources::AptRepository

def determine_ppa_url(ppa_url)
  # verify if we have the url already, then just return
  return ppa_url if ppa_url =~ HTTP_URL_RE
  # otherwise start generating the ppa url
  # special care if the name stats with :
  ppa_url = ppa_url.split(':')[1] if ppa_url.start_with?('ppa:')
  # parse ppa owner and repo
  ppa_owner, ppa_repo = ppa_url.split('/')
  ppa_repo = 'ppa' if ppa_repo.nil?
  # construct new ppa url and return it
  format('http://ppa.launchpad.net/%s/%s/ubuntu', ppa_owner, ppa_repo)
end

def enabled?
  return false if find_repo.count == 0
  actives = find_repo.map { |repo| repo[:active] }
  actives = actives.uniq
  actives.size == 1 && actives[0] = true
end

def exists?
  find_repo.count > 0
end

def find_repo
  read_debs.select { |repo| repo[:url] == @deb_url && repo[:type] == 'deb' }
end

def initialize(ppa_name)
  @deb_url = nil
  # check if the os is ubuntu or debian
  if inspec.os.debian?
    @deb_url = determine_ppa_url(ppa_name)
  else
    # this resource is only supported on ubuntu and debian
    skip_resource 'The `apt` resource is not supported on your OS yet.'
  end
end

def read_debs
  return @repo_cache if defined?(@repo_cache)
  # load all lists
  cmd = inspec.command("find /etc/apt/ -name \*.list -exec sh -c 'cat {} || echo -n' \\;")
  # @see https://help.ubuntu.com/community/Repositories/CommandLine#Explanation_of_the_Repository_Format
  @repo_cache = cmd.stdout.chomp.split("\n").each_with_object([]) do |raw_line, lines|
    active = true
    # detect if the repo is commented out
    line = raw_line.gsub(/^(#\s*)*/, '')
    active = false if raw_line != line
    # eg.: deb http://archive.ubuntu.com/ubuntu/ wily main restricted
    parse_repo = /^\s*(\S+)\s+"?([^ "\t\r\n\f]+)"?\s+(\S+)\s+(.*)$/.match(line)
    # check if we got any result and the second param is an url
    next if parse_repo.nil? || !parse_repo[2] =~ HTTP_URL_RE
    # map data
    repo = {
      type: parse_repo[1],
      url: parse_repo[2],
      distro: parse_repo[3],
      components: parse_repo[4].chomp.split(' '),
      active: active,
    }
    next unless ['deb', 'deb-src'].include? repo[:type]
    lines.push(repo)
  end
end

def to_s
  "Apt Repository #{@deb_url}"
end