acts_as_link 0.0.1 → 0.0.2

data/lib/acts_as_link/link.rb

@@ -3,6 +3,8 @@ require 'uri'
 
 class Link
   
+  IMAGE_EXTENSIONS = %w(.bmp .gif .ico .jpeg .jpg .png .psd .svg .tif .tiff .xcf)
+  
   def initialize(url)
     @uri = URI.parse(url)
     raise 'Link is not an url' if @uri.class != URI::HTTP
@@ -13,6 +15,11 @@ class Link
     response_code_is_not_valid?(response.code)
   end
   
+  def is_an_image?
+    extension = get_extension(@uri.to_s)
+    verify_if_extension_is_an_image(extension)
+  end
+  
 private 
 
   def response_code_is_not_valid?(code) 
@@ -34,5 +41,21 @@ private
   def code_is_an_error_code(code)
     code[0..0] == "4" || code[0..0] == "5"
   end
+  
+  def get_extension(uri)
+    point_position = uri.rindex('.')
+    uri_length = uri.length
+    point_position > (uri_length - 6) ? uri[point_position..uri_length] : nil
+  end
+  
+  def verify_if_extension_is_an_image(extension)
+    return false if extension.nil?
+    is_an_image = false
+    IMAGE_EXTENSIONS.each do |ext| 
+      is_an_image = (extension == ext)
+      break if is_an_image
+    end
+    is_an_image
+  end
     
 end

data/spec/acts_as_link/link_spec.rb

@@ -13,8 +13,8 @@ describe Link do
       lambda{Link.new('')}.should raise_error(RuntimeError,/Link is not an url/)
     end
     
-    it 'should return an exception when try to initialize with a number' do
-      lambda{Link.new('123')}.should raise_error(RuntimeError,/Link is not an url/)
+    it 'should return an exception when try to initialize with a string is not a link' do
+      lambda{Link.new('test.com.br')}.should raise_error(RuntimeError,/Link is not an url/)
     end
     
   end
@@ -32,4 +32,35 @@ describe Link do
     end
     
   end
+  
+  describe 'is_an_image?' do
+    
+    it 'should return true when url links to an image' do
+      link = Link.new('http://www.google.com/images/srpr/nav_logo35.png')
+      link.is_an_image?.should be_true
+    end
+    
+    it 'should return false when url does not link to an image' do
+      link = Link.new('http://www.google.com')
+      link.is_an_image?.should be_false
+    end
+    
+    it 'should return false when url links to a pdf file' do
+      link = Link.new('http://qualidadebr.files.wordpress.com/2009/06/livro-qualidadebr.pdf')
+      link.is_an_image?.should be_false
+    end
+    
+    it 'should return false when url extension image is not in the end' do
+      link = Link.new('http://www.google.com/images/srpr/nav_.pnglogo35')
+      link.is_an_image?.should be_false
+    end
+    
+    it 'should return true for all image extensions defined' do
+      Link::IMAGE_EXTENSIONS.each do |ext|
+        link = Link.new("http://www.google.com/images/srpr/nav_logo35#{ext}")
+        link.is_an_image?.should be_true
+      end
+    end
+    
+  end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: acts_as_link
 version: !ruby/object:Gem::Version 
-  hash: 29
+  hash: 27
   prerelease: false
   segments: 
   - 0
   - 0
-  - 1
-  version: 0.0.1
+  - 2
+  version: 0.0.2
 platform: ruby
 authors: 
 - "Fabr\xC3\xADcio Ferrari de Campos"
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-02-12 00:00:00 -02:00
+date: 2011-02-13 00:00:00 -02:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -34,7 +34,7 @@ dependencies:
         version: 1.3.0
   type: :development
   version_requirements: *id001
-description: For now only verification if link is broken
+description: For now only verification if link is broken and if it is to an image
 email: fabricio@vizir.com.br
 executables: []