module Docs
  class Python
    class EntriesV2Filter < Docs::EntriesFilter
      REPLACE_TYPES = {
        'compiler package'                        => 'Compiler',
        'Cryptographic'                           => 'Cryptography',
        'Custom Interpreters'                     => 'Interpreters',
        'Data Compression & Archiving'            => 'Data Compression',
        'Generic Operating System'                => 'Operating System',
        'Graphical User Interfaces with Tk'       => 'Tk',
        'Internet Data Handling'                  => 'Internet Data',
        'Internet Protocols & Support'            => 'Internet',
        'Interprocess Communication & Networking' => 'Networking',
        'MacOSA'                                  => 'Mac OS',
        'Program Frameworks'                      => 'Frameworks',
        'Structured Markup Processing Tools'      => 'Structured Markup' }

      def get_name
        name = at_css('h1').content
        name.remove! %r{\A[\d\.]+ } # remove list number
        name.remove! "\u{00B6}" # remove pilcrow sign
        name.remove! %r{ [\u{2013}\u{2014}].+\z} # remove text after em/en dash
        name.remove! 'Built-in'
        name.strip!
        name
      end

      def get_type
        return 'Logging' if slug.start_with? 'library/logging'

        type = at_css('.related a[accesskey="U"]').content

        if type == 'The Python Standard Library'
          type = at_css('h1').content
        elsif type.include?('I/O') || %w(select selectors).include?(name)
          type = 'Input/ouput'
        elsif type.start_with? '18'
          type = 'Internet Data Handling'
        elsif type.include? 'Mac'
          type = 'Mac OS'
        end

        type.remove! %r{\A\d+\.\s+} # remove list number
        type.remove! "\u{00b6}" # remove paragraph character
        type.sub! ' and ', ' & '
        [' Services', ' Modules', ' Specific', 'Python '].each { |str| type.remove!(str) }

        REPLACE_TYPES[type] || type
      end

      def include_default_entry?
        !at_css('.body > .section:only-child > .toctree-wrapper:last-child') && !type.in?(%w(Superseded SunOS))
      end

      def additional_entries
        return [] if root_page? || !include_default_entry? || name == 'errno'
        clean_id_attributes
        entries = []

        css('.class > dt[id]', '.exception > dt[id]', '.attribute > dt[id]', '.data > dt[id]').each do |node|
          entries << [node['id'], node['id']]
        end

        css('.function > dt[id]', '.method > dt[id]', '.staticmethod > dt[id]', '.classmethod > dt[id]').each do |node|
          entries << [node['id'] + '()', node['id']]
        end

        entries
      end

      def clean_id_attributes
        css('.section > .target[id]').each do |node|
          if dt = node.at_css('+ dl > dt')
            dt['id'] ||= node['id'].remove(/\w+\-/)
          end
          node.remove
        end
      end
    end
  end
end
