robohash_client 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b672467cd0e2035a9a8cf6e26210c03be0d2433f
-  data.tar.gz: a0f3583e7eddeb2d48b8de4ab064d751e29e23ac
+  metadata.gz: b682fec95cae8ea1d5e02b5cb5c5123987991939
+  data.tar.gz: 384b0c778ee03fa48a7978996339007e5c8419ae
 SHA512:
-  metadata.gz: c11c421a3fccdff22b8771aa412421ffaac447bd86028ebb6b03601dd6f780a9f64c72ec33761699b56741b8a22fc770a54644295f48df098fcb08fdbdabb12d
-  data.tar.gz: cc8b0f7e27cea4bfebdd2273fec41dc8e1cb6c12f09e968aeb80c5891c88a6950bdbba1b041d6b251b87c5d84b1f4ddea4db2c1721d872920c574186f4e0adff
+  metadata.gz: 39077a6b2d276813133c23e6e80dfb4a2d8d0c9a4edd05fe56992dbd1fa50bd79f3db54c8fecaedcca8fa37432390bcd2d18196161677069dca056f693f3af33
+  data.tar.gz: 057e7c00cf58fb452d348c9755b2cf0c07f3a59e8bd25255ad33720b51b4c9d1e0146682a277a3a2557cc63ba442fe229c213cf5b85e22e07ac9caba01d778a7

data/lib/robohash_client.rb

@@ -2,35 +2,77 @@ require 'net/http'
 require 'json'
 
 class Robohash
-	BASE_URL = 'https://robohash.org'
-	DEFAULT_DIRECTORY = 'robohash_images'
-	
-	def self.get(name, options = '')
-		uri = URI("#{BASE_URL}/#{name}.png?#{options}")
-		response = Net::HTTP.get_response(uri)
-
-		if response.is_a?(Net::HTTPSuccess)
-			save(response, name)
-			puts "Image #{name}.png saved with success!"
-    else
-    	puts "Error obtaining image from #{uri}."
-    end
-	end
+	class << self
+		BASE_URL = 'https://robohash.org'
+		DEFAULT_DIRECTORY = 'robohash_images'
+		BGSET = 'bgset='
+		SIZE = 'size='
+		SET = 'set='
+		OPTIONS = { 
+			set: { 
+				classic: "#{SET}set1", human: "#{SET}set2", heads: "#{SET}set3", cats: "#{SET}set4", any: "#{SET}any" 
+			},
+			size: {
+				small: "#{SIZE}100x100", medium: "#{SIZE}250x250", large: "#{SIZE}500x500", extra: "#{SIZE}900x900"
+			},
+			bgset: {
+				one: "#{BGSET}bg1", two: "#{BGSET}bg2"
+			}
+		}
 
-	def self.get_many(names, options = '')
-		names.reject! do |name| 
-			!name.is_a?(String) || name == ''
+		def get(name, options = {})
+			valid_options = extract_valid_options(options)
+			query_string = build_query_string(valid_options)
+			make_request(name, query_string)
+		end
+
+		def get_many(names, options = {})
+			names.reject! { |name| !name.is_a?(String) || name == '' }
+			valid_options = extract_valid_options(options)
+			query_string = build_query_string(valid_options)
+			names.each { |name| make_request(name, query_string) }
 		end
-		names.each { |name| self.get(name, options) }
-	end
 
-	private 
-	
-	def self.save(image, name)
-    Dir.mkdir(DEFAULT_DIRECTORY) unless Dir.exists? DEFAULT_DIRECTORY
+		private 
+
+		def extract_valid_options(options)
+			valid_options = []
+
+			if options.is_a?(Hash)
+				options.each_pair do |key, value|
+					inner_hash = OPTIONS.fetch(key, nil)
+					option = inner_hash.fetch(value, nil) if inner_hash.is_a?(Hash)
+					valid_options.push(option) if option
+				end
+			end
+
+			valid_options
+		end
 
-    open("robohash_images/#{name}.png", 'wb') do |file|
-      file.write(image.body)
-    end
+		def make_request(name, query_string)
+			uri = URI("#{BASE_URL}/#{name}#{query_string}")
+			response = Net::HTTP.get_response(uri)
+
+			if response.is_a?(Net::HTTPSuccess)
+				save(response, name)
+				puts "Image #{name}.png saved successfully!"
+	    else
+	    	puts "Error obtaining image from #{uri}."
+	    end
+		end
+
+		def build_query_string(valid_options)
+			query_string = '?'
+			valid_options.each { |option| query_string += "#{option}&" }
+			query_string
+		end
+
+		def save(image, name)
+	    Dir.mkdir(DEFAULT_DIRECTORY) unless Dir.exists? DEFAULT_DIRECTORY
+
+	    open("robohash_images/#{name}.png", 'wb') do |file|
+	      file.write(image.body)
+	    end
+		end
 	end
-end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: robohash_client
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Danilo Barion Nogueira
@@ -10,7 +10,7 @@ bindir: bin
 cert_chain: []
 date: 2017-11-04 00:00:00.000000000 Z
 dependencies: []
-description: A simple client for fetch robohash avatars from Robohash.org
+description: A simple client to get robohash avatar's from Robohash.org.
 email: danilo.barion1986@live.com
 executables: []
 extensions: []
@@ -40,5 +40,5 @@ rubyforge_project:
 rubygems_version: 2.6.13
 signing_key: 
 specification_version: 4
-summary: CLient for Robohash.org
+summary: Client for Robohash.org
 test_files: []