module WpUsernames

# File lib/wpscan/modules/wp_usernames.rb, line 102
def author_url(author_id)
  @uri.merge("?author=#{author_id}").to_s
end

# File lib/wpscan/modules/wp_usernames.rb, line 77
def extract_nickname_from_body(body)
  body[%r{<title>([^<]*)</title>}, 1]
end

# File lib/wpscan/modules/wp_usernames.rb, line 69
def get_nickname_from_response(resp)
  nickname = nil
  if resp.code == 200
    nickname = extract_nickname_from_body(resp.body)
  end
  nickname
end

# File lib/wpscan/modules/wp_usernames.rb, line 60
def get_nickname_from_url(url)
  resp = Browser.instance.get(url, { follow_location: true, max_redirects: 2 })
  nickname = nil
  if resp.code == 200
    nickname = extract_nickname_from_body(resp.body)
  end
  nickname
end

# File lib/wpscan/modules/wp_usernames.rb, line 81
def remove_junk_from_nickname(usernames)
  unless usernames.kind_of? Array
    raise('Need an array as input')
  end
  nicknames = []
  usernames.each do |u|
    unless u.kind_of? WpUser
      raise('Items must be of type WpUser')
    end
    nickname = u.nickname
    unless nickname == 'empty'
      nicknames << nickname
    end
  end
  junk = get_equal_string_end(nicknames)
  usernames.each do |u|
    u.nickname = u.nickname.sub(%r#{Regexp.escape(junk)}$/, '')
  end
  usernames
end

# File lib/wpscan/modules/wp_usernames.rb, line 28
def usernames(options = {})
  range = options[:range] || (1..10)
  browser = Browser.instance
  usernames = []

  range.each do |author_id|
    url = author_url(author_id)
    response = browser.get(url)

    username = nil
    nickname = nil
    if response.code == 301 # username in location?
      username = response.headers_hash['location'][%r{/author/([^/\b]+)/?}, 1]
      # Get the real name from the redirect site
      nickname = get_nickname_from_url(url)
    elsif response.code == 200 # username in body?
      username = response.body[%r{posts by (.*) feed}, 1]
      nickname = get_nickname_from_response(response)
    end

    unless username == nil and nickname == nil
      usernames << WpUser.new(username, author_id, nickname)
    end
  end
  usernames = remove_junk_from_nickname(usernames)

  # clean the array, remove nils and possible duplicates
  usernames.flatten!
  usernames.compact!
  usernames.uniq
end