~ion/restricted-manager/ion

#!/usr/bin/ruby

# nvidia_supported – create a modules.alias compatible listing of devices

# supported by nvidia drivers

#

# Copyright © 2007  Canonical Ltd.

#

# Author: Johan Kiviniemi <debian@johan.kiviniemi.name>

#

# This program is free software; you can redistribute it and/or modify it under

# the terms of the GNU General Public License as published by the Free Software

# Foundation; either version 2 of the License, or (at your option) any later

# version.

#

# This program is distributed in the hope that it will be useful, but WITHOUT

# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS

# FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more

# details.

#

# You should have received a copy of the GNU General Public License along with

# this program; if not, write to the Free Software Foundation, Inc., 51

# Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

begin

  require 'rubygems'

rescue LoadError

  # It's ok if mechanize has been installed via something else than rubygems.

end

require 'mechanize'

require 'logger'

def main

  usage if ARGV.empty?

  versions = []

  ARGV.each do |arg|

    driver, version = arg.split /:/, 2

    usage unless driver and version

    versions << [driver, version]

  end

  scraper = NvidiaSupported.new

  scraper.log = Logger.new $stderr

  scraper.log.level = Logger::INFO

  puts "# Listing generated by nvidia_supported. Do not edit manually."

  vendor_id = 0x10DE

  versions.each do |driver, version|

    puts "# #{driver}: #{version}"

    scraper.process(version).each do |dev_id|

      puts 'alias pci:v%08Xd%08Xsv*sd*bc*sc*i* %s' % [vendor_id,

        dev_id, driver]

    end

  end

  puts

end

def usage

  $stderr.puts "USAGE: #$0 nvidia:1.0-9876 nvidia_legacy:1.0-1234 ..."

  exit 1

end

class NvidiaSupported < WWW::Mechanize

  def initialize

    super

    get 'http://www.nvidia.com/object/linux_display_archive.html'

  end

  def process version

    transact do

      log.info "Clicking the driver link, version #{version}"

      link_re = Regexp.new '_v?' + Regexp.quote(version) + '\.html$'

      click page.links.with.href(link_re)

      log.info "Clicking the Supported Products List link"

      click page.links.with.text(/Supported Products List/)

      # A workaround for older hpricot (< 0.5) without support for expressions

      # like td[text() = ...]. The original form would be faster.

      #(page / 'td[text() ^= "0x"]').map {|td| td.inner_html.strip.hex }

      (page / 'td').

        map {|td| td.inner_html.strip }.

        find_all {|text| text =~ /^0x/ }.

        map {|text| text.hex }

    end

  end

end

main if $0 == __FILE__