python-reverse-lookup

#!/usr/bin/ruby

"""
Usage: $0 pypi_package [repository]
"""

require 'concurrent'

$pkg = ARGV[0].downcase
$repos = ARGV[1] || "all"

$repos = $repos == "all" ? ["core", "extra", "multilib", "core-testing", "extra-testing", "multilib-testing"] : $repos.split(",")

CACHEDIR = "#{Dir.home}/.cache/python-reverse-lookup"
Dir.mkdir(CACHEDIR) unless Dir.exist?(CACHEDIR)

POOL = Concurrent::FixedThreadPool.new(Concurrent.processor_count)
PACMAN_LOCK = Mutex.new

def parse_metadata(pattern)
    `pacman -Fx #{pattern}`.split(/\n/).each do |line|
        case line.split
        in [file, "is", "owned", "by", package, version]
            repo, package = package.split("/")
            next unless $repos.include?(repo)

            POOL.post do
                cachefile = "#{CACHEDIR}/" + "#{package}-#{version}-#{file}".gsub("/", "_")

                unless File.exist?(cachefile)
                    path = "/var/cache/pacman/pkg/#{package}-#{version}-*.pkg.tar.zst"
                    if ! File.exist? path
                        PACMAN_LOCK.synchronize do
                            out = `sudo pacman -Sw --noconfirm #{repo}/#{package} 2>&1`
                            puts out unless $?.success?
                        end
                    end
                    content = `bsdtar xOf #{path} #{file}`
                    File.write(cachefile, content) if $?.success?
                end

                File.readlines(cachefile).each do |line|
                    yield [repo, package].join("/"), line
                end if File.exist?(cachefile)
            end
        end
    end
end

parse_metadata("/usr/lib/python3\\.[0-9]+/site-packages/.*\\.dist-info/METADATA") do |package, line|
    if line.downcase.start_with? /requires-dist: #{$pkg}[^\w]/
        puts "#{package}: #{line}"
    end
end

$last_package = nil
parse_metadata("/usr/lib/python3\\.[0-9]+/site-packages/.*\\.egg-info/requires.txt") do |package, line|
    if package != $last_package
        $extra = ""
        $last_package = package
    end
    if line.match /\[.+\]/
        $extra = " #{line.chomp}"
    end
    if line.downcase.start_with? /#{$pkg}[^\w]/
        puts "#{package}: #{line.chomp}#{$extra}"
    end
end

POOL.shutdown
POOL.wait_for_termination