lib/onebox/engine/pastebin_onebox.rb in onebox-1.5.34 vs lib/onebox/engine/pastebin_onebox.rb in onebox-1.5.35

@@ -1,13 +1,40 @@
 module Onebox
   module Engine
     class PastebinOnebox
       include Engine
-      include StandardEmbed
+      include LayoutSupport
 
+      MAX_LINES = 10
+
       matches_regexp(/^http?:\/\/pastebin\.com/)
 
+      private
+
+      def data
+        @data ||= {
+          title: 'pastebin.com',
+          link: link,
+          content: content,
+          truncated?: truncated?
+        }
+      end
+
+      def content
+        lines.take(MAX_LINES).join("\n")
+      end
+
+      def truncated?
+        lines.size > MAX_LINES
+      end
+
+      def lines
+        return @lines if @lines
+        response = Onebox::Helpers.fetch_response("http://pastebin.com/raw/#{paste_key}", 1)
+        @lines = response.body.split("\n")
+      end
+
       def paste_key
         if uri.path =~ /\/raw\//
           match = uri.path.match(/\/raw\/([^\/]+)/)
           return match[1] if match && match[1]
         elsif uri.path =~ /\/download\//
@@ -22,16 +49,9 @@
         end
 
         nil
       rescue
         return nil
-      end
-
-      def to_html
-        return nil unless paste_key
-        response = Onebox::Helpers.fetch_response("http://pastebin.com/raw/#{paste_key}", 1)
-        return nil unless response && response.code.to_i == 200
-        return "<iframe src='//pastebin.com/embed_iframe/#{paste_key}' style='border:none;width:100%;max-height:100px;'></iframe>"
       end
     end
   end
 end