lib/onebox/preview.rb in onebox-1.3.4 vs lib/onebox/preview.rb in onebox-1.3.5

@@ -9,18 +9,18 @@
       @engine_class = Matcher.new(@url).oneboxed
     end
 
     def to_s
       return "" unless engine
-      engine.to_html || ""
+      process_html(engine_html)
     rescue *Onebox::Preview.web_exceptions
       ""
     end
 
     def placeholder_html
       return "" unless engine
-      engine.placeholder_html || ""
+      process_html(engine.placeholder_html)
     rescue *Onebox::Preview.web_exceptions
       ""
     end
 
     def options
@@ -30,9 +30,38 @@
     def self.web_exceptions
      [Net::HTTPServerException, OpenURI::HTTPError, Timeout::Error, Net::HTTPError, Errno::ECONNREFUSED]
     end
 
     private
+
+    def engine_html
+      engine.to_html
+    end
+
+    def process_html(html)
+      return "" unless html
+
+      if @options[:max_width]
+        doc = Nokogiri::HTML::fragment(html)
+        if doc
+          doc.css('[width]').each do |e|
+            width = e['width'].to_i
+
+            if width > @options[:max_width]
+              height = e['height'].to_i
+              if (height > 0)
+                ratio = (height.to_f / width.to_f)
+                e['height'] = (@options[:max_width] * ratio).floor
+              end
+              e['width'] = @options[:max_width]
+            end
+          end
+          return doc.to_html
+        end
+      end
+
+      html
+    end
 
     def engine
       return nil unless @engine_class
       @engine ||= @engine_class.new(@url, cache)
     end