img_builder.rb in genit-1.0

- old
+ new

@@ -1,38 +1,24 @@
 # -*- encoding: utf-8 -*-
 
-require 'uri'
-
 module Genit
 
   # Modify img src.
-  class ImgBuilder < BuilderBase
+  class ImgBuilder < Relativizer
   
     # Public: Relativize image source.
     #
     # page_name - The string filename of the page.
     #
     # Returns the modified Nokogiri::XML::Document
     def build_for_page page_name
-      build page_name, get_links
+      build page_name, @document.css("body img")
     end
     
     private
     
-    def get_links
-      @document.css("body img")
-    end
-    
     def update link
-      @path = link['src']
-      return if not_an_internal_link?
-      nb = BuilderBase::get_number_of_base_dirs @page_name
-      make_relative nb
-      link['src'] = @path
-    end
-    
-    def not_an_internal_link?
-      @path.nil? or @path =~ URI::regexp
+      super link, "src"
     end
   
   end
   
 end