Fix Polars method signature rendering via Nokogiri HTML5 parser

Author: mcagriardic

lib/docs/scrapers/polars.rb

@@ -30,5 +30,29 @@ class Polars < UrlScraper
     def get_latest_version(opts)
       get_latest_github_release('pola-rs', 'polars', opts).sub(/\Apy-/, '')
     end
+
+    private
+
+    def parse(response)
+      if response.body.include?('class="sig')
+        doc = Nokogiri::HTML5(response.body)
+        doc.css('.sig').each do |node|
+          node.css('.headerlink').remove
+          node.css('.reference.external').each { |a| a.remove if a.text.strip == '[source]' }
+          sig = node.text.gsub(/\s+/, ' ').strip
+          if (m = sig.match(/\A(.+?\()\s*(.+?)\s*(\).*)\z/m))
+            head, params, tail = m[1], m[2], m[3]
+            split_params = params.split(/,\s+/).map { |p| p.sub(/,\z/, '') }.reject(&:empty?)
+            sig = "#{head}\n    #{split_params.join(",\n    ")},\n#{tail}" unless split_params.empty?
+          end
+          pre = Nokogiri::XML::Node.new('pre', doc)
+          pre['data-language'] = 'python'
+          pre.content = sig
+          node.replace(pre)
+        end
+        response.body.replace(doc.to_html)
+      end
+      super
+    end
   end
 end