fix: add an error to catch

Add ArgumentError to #doc because it has a possibility to throw the error
ninoseki · Apr 3, 2019 · 19e7044 · 19e7044
1 parent eb914a9
commit 19e7044
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/lib/miteru/website.rb b/lib/miteru/website.rb
@@ -54,11 +54,13 @@ def get
     end
 
     def doc
-      @doc ||= [].tap do |out|
-        out << Oga.parse_html(response.body.to_s)
-               rescue LL::ParserError => _
-                 out << nil
-      end.first
+      @doc ||= parse_html(response.body.to_s)
+    end
+
+    def parse_html(html)
+      Oga.parse_html(html)
+    rescue ArgumentError, LL::ParserError => _
+      nil
     end
 
     def links