module Docs
  class Sphinx
    class CleanHtmlFilter < Filter
      def call
        css('.headerlink', 'hr', '#contents .topic-title', '#topics .topic-title', 'colgroup', '.line-block', '.anchor-link').remove

        css('.contents > ul:first-child:last-child.simple > li:first-child:last-child').each do |node|
          node.parent.before(node.at_css('> ul')) if node.at_css('> ul')
          node.remove
        end

        css('em.xref', 'tt', 'cite').each do |node|
          node.name = 'code'
        end

        css('.toc-backref', '.toctree-wrapper', '.contents', 'span.pre', 'pre a > code', 'tbody', 'code > code', 'a > em').each do |node|
          node.before(node.children).remove
        end

        css('div[class*="highlight-"]', 'div[class*="hl-"]').each do |node|
          pre = node.at_css('pre')
          pre.content = pre.content
          lang = node['class'][/code (\w+) highlight/, 1] || node['class'][/highlight\-([\w\+]+)/, 1] || node['class'][/hl\-(\w+)/, 1]
          lang = 'php' if lang == 'ci'|| lang == 'html+php'
          lang = 'markup' if lang == 'html+django'
          lang = 'bash' if lang == 'bash'
          lang = 'python' if lang == 'default' || lang.start_with?('python') || lang.start_with?('ipython')
          pre['data-language'] = lang
          node.replace(pre)
        end

        # Support code blocks in jupyter notebook files
        css('.code_cell div.highlight').each do |node|
          pre = node.at_css('pre')
          pre['data-language'] = 'python'
          node.replace(pre)
        end

        unless context[:sphinx_keep_empty_ids]
          css('span[id]:empty').each do |node|
            (node.next_element || node.previous_element)['id'] ||= node['id'] if node.next_element || node.previous_element
            node.remove
          end
        end

        css('.section').each do |node|
          if node['id']
            if node.first_element_child['id']
              node.element_children[1]['id'] = node['id'] if node.element_children[1]
            else
              node.first_element_child['id'] = node['id']
            end
          end

          node.before(node.children).remove
        end

        css('h2 > a > code').each do |node|
          node.parent.before(node.content).remove
        end

        css('dt').each do |node|
          next if current_url.host == 'matplotlib.org'
          next if current_url.host == 'numpy.org'
          next if current_url.host == 'requests.readthedocs.io'
          next if current_url.host == 'scikit-learn.org'
          next unless node['id'] || node.at_css('code, .classifier')
          links = []
          links << node.children.last.remove while node.children.last.try(:name) == 'a'
          node.inner_html = "<code>#{CGI::escapeHTML(node.content.strip)}</code> "
          links.reverse_each { |link| node << link }
        end

        css('li > p:first-child:last-child').each do |node|
          node.before(node.children).remove
        end

        css('blockquote > div:first-child:last-child').each do |node|
          node.parent.before(node.parent.children).remove
          node.before(node.children).remove
        end

        css('.admonition-example').each do |node|
          title = node.at_css('.admonition-title')
          title.name = 'h4'
          title.remove_attribute 'class'
          node.before(node.children).remove
        end

        css('table[border]').each do |node|
          node.remove_attribute 'border'
          node.remove_attribute 'cellpadding'
          node.remove_attribute 'cellspacing'
        end

        css('code', 'tr').remove_attr('class')

        css('h1').each do |node|
          node.content = node.content
        end

        css('p.rubric').each do |node|
          node.name = 'h4'
        end

        doc
      end
    end
  end
end
