Contents

require 'net/http'
require 'uri'
require_relative '../application'
require_relative '../logger'

# Search URL images on Internet
# Methods:
# * +load+ - Accept String or an Array with the desired search
# * +sanitize_string+ - Clean URL string
# * +sanitize_array+ - Clean URL Array
module ImageUrlLoader
  # Search "input" images on Google and return URL
  def self.load(input = [])
    filters = []
    if input.class == String
      filters += sanitize_string(input.clone)
    elsif input.class == Array
      filters = sanitize_array(input.clone)
    else
      raise "[ERROR] ImageUrlLoader: Unkown type #{input.class}"
    end
    # Search Image URLs from Google site, selected by <filters>
    search_url = 'https://www.google.es/search?q='
    search_url << filters.flatten.join('+').to_s
    search_url << '&source=lnms&tbm=isch&sa=X&ved=2ahUKEwj_g8Wfst7nAhWpzoUKHf_wDbsQ_AUoAnoECBMQBA&biw=1280&bih=591'
    image_urls = []
    begin
      uri = URI.parse(search_url)
      response = Net::HTTP.get_response(uri)

      r = response.body.split(' ')
      r.each do |line|
        if line.include? 'src="https'
          image_urls << line.delete('"')[4, line.size]
        end
      end
    rescue
      Logger.verboseln '[ERROR] ImageUrlLoader'
      Logger.verboseln " => #{search_url}"
      Logger.verboseln ' => Check Internet connections'
      Logger.verboseln ' => Ensure URL is well formed'
    end
    image_urls
  end

  def self.sanitize_string(input)
    r = [ ['á', 'a'], ['é', 'e'], ['í', 'i'], ['ó', 'o'], ['ú', 'u'], ['ñ', 'n'], ['Á', 'A'], ['É', 'E'], ['Í', 'I'], ['Ó', 'O'], ['Ú', 'U'], ['Ñ', 'N']]
    r.each { |item| input.gsub!(item[0], item[1]) }
    r = ['-', '_', ',', '"']
    r.each { |item| input.gsub!(item, ' ') }
    input.split(' ')
  end

  def self.sanitize_array(input)
    input.map { |i| sanitize_string(i) }
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
asker-tool-2.1.7	lib/asker/loader/image_url_loader.rb
asker-tool-2.1.6	lib/asker/loader/image_url_loader.rb