#!/usr/bin/env python
# This script generates a table of dependencies in HTML format on stdout.
# It expects to be run in the tahoe-lafs-dep-eggs directory.

from __future__ import print_function

import re, os, sys
import pkg_resources

extensions = ('.egg', '.tar.bz2', '.tar.gz', '.exe')
platform_aliases = [('i686','x86'), ('i386','x86'), ('i86pc','x86'), ('win32','windows-x86'),
                    ('win-amd64','windows-x86_64'), ('amd64','x86_64')]
min_supported_python = {'windows-x86': '2.7', 'windows-x86_64': '2.7'}
pkg_name_continuations = ('modules')

FILENAME_RE  = re.compile(r'([a-zA-Z_0-9\.]*)-([0-9\.a-vx-z_]*)(-py[0-9\.]*)?(-.*)?')
FILENAME_RE2 = re.compile(r'([a-zA-Z_0-9\.]*)-([0-9\.a-vx-z_]*)(win32|win-amd64)?(-py[0-9\.]*)?')

matrix = {}
pkgs = set()
platform_dependent_pkgs = set()
python_versions = set()

depdirs = ['.', '../tahoe-dep-sdists']
if len(sys.argv) > 1:
    depdirs = sys.argv[1 :]

filenames = set()
for depdir in depdirs:
    filenames = filenames.union(os.listdir(depdir))

def add(d, k, v):
    if k in d:
        d[k] += [v]
    else:
        d[k] = [v]

for fname in filenames:
    for ext in extensions:
        if fname.endswith(ext):
            m = FILENAME_RE.match(fname[:-len(ext)])
            try:
                pkg       = m.group(1)
                pkg2      = m.group(2)
                if pkg2 in pkg_name_continuations:
                    pkg += '-' + pkg2
                else:
                    pythonver = (m.group(3) or '-py')[3:]
                    platform  = (m.group(4) or '-')[1:]
            except (IndexError, AttributeError, TypeError):
                continue

            if not pkg2 in pkg_name_continuations and not pythonver:
                m = FILENAME_RE2.match(fname[:-len(ext)])
                if m.group(3):
                    try:
                        platform  = m.group(3)
                        pythonver = (m.group(4) or '-py')[3:]
                    except (IndexError, AttributeError, TypeError):
                        continue

            for (alias, replacement) in platform_aliases:
                if platform.endswith(alias):
                    platform = platform[:-len(alias)] + replacement
                    break

            pkgs.add(pkg)
            if platform:
                platform_dependent_pkgs.add(pkg)
            if pythonver not in matrix:
                python_versions.add(pythonver)
                matrix[pythonver] = {}
            add(matrix[pythonver], platform, (pkg, fname))
            break

platform_independent_pkgs = pkgs - platform_dependent_pkgs

width = 100 / (len(platform_dependent_pkgs) + 1)

def file_list(all_files, pkg):
    files = sorted([(pkg_resources.parse_version(n), n) for (p, n) in all_files if pkg == p])
    return '<br>&nbsp;'.join(['<a href="%s">%s</a>' % (f, f) for (v, f) in files])

greybgstyle = '; background-color: #E0E0E0'
nobgstyle = ''
unsupportedstyle = '; color: #C00000'

print('<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd">')
print('<html>')
print('<head>')
print('  <meta http-equiv="Content-Type" content="text/html;charset=us-ascii">')
print('  <title>Software packages that Tahoe-LAFS depends on</title>')
print('</head>')
print('<body>')
print('<h2>What is this?</h2>')
print('<p>See <a href="https://tahoe-lafs.org/trac/tahoe-lafs/browser/docs/quickstart.rst">quickstart.rst</a>, <a href="https://tahoe-lafs.org/trac/tahoe-lafs/wiki/Installation">wiki:Installation</a>, and <a href="https://tahoe-lafs.org/trac/tahoe-lafs/wiki/CompileError">wiki:CompileError</a>.')
print('<h2>Software packages that Tahoe-LAFS depends on</h2>')
print()
for pyver in reversed(sorted(python_versions)):
    greybackground = False
    if pyver:
        print('<p>Packages for Python %s that have compiled C/C++ code:</p>' % (pyver,))
        print('<table border="1">')
        print('  <tr>')
        print('    <th style="background-color: #FFFFD0" width="%d%%">&nbsp;Platform&nbsp;</th>' % (width,))
        for pkg in sorted(platform_dependent_pkgs):
            print('    <th style="background-color: #FFE8FF;" width="%d%%">&nbsp;%s&nbsp;</th>' % (width, pkg))
        print('  </tr>')

        first = True
        for platform in sorted(matrix[pyver]):
            unsupported_python = (platform in min_supported_python and
                                  pyver.split('.') < min_supported_python[platform].split('.'))

            if greybackground:
                bgstyle = greybgstyle
            else:
                bgstyle = nobgstyle
            greybackground = not greybackground
            row_files = sorted(matrix[pyver][platform])
            style1 = first and 'border-top: 2px solid #000000' or ''
            style1 += bgstyle
            style1 += unsupported_python and unsupportedstyle or ''
            style2 = first and 'border-top: 2px solid #000000' or ''
            style2 += bgstyle
            annotated_platform = platform.replace('-', '&#x2011;') + (unsupported_python and '&nbsp;(unsupported)' or '')
            print('  <tr>')
            print('    <td style="%s">&nbsp;%s&nbsp;</td>' % (style1, annotated_platform))
            for pkg in sorted(platform_dependent_pkgs):
                if pkg == 'pywin32' and not platform.startswith('windows'):
                    print('    <td style="border: 0; text-align: center; %s"> n/a </td>' % (style2,))
                else:
                    print('    <td style="%s">&nbsp;%s</td>' % (style2, file_list(row_files, pkg)))
            print('  </tr>')
            first = False

    print('</table>')
    print()

print('<p>Packages that are platform-independent or source-only:</p>')
print('<table border="1">')
print('  <tr>')
print('    <th style="background-color:#FFFFD0;">&nbsp;Package&nbsp;</th>')
print('    <th style="background-color:#FFE8FF;">&nbsp;All Python versions&nbsp;</th>')
print('  </tr>')

style1 = 'border-top: 2px solid #000000; background-color:#FFFFF0;'
style2 = 'border-top: 2px solid #000000;'
m = matrix['']['']
for pkg in sorted(platform_independent_pkgs):
    print('  <tr>')
    print('    <th style="%s">&nbsp;%s&nbsp;</th>' % (style1, pkg))
    print('    <td style="%s">&nbsp;%s</td>' % (style2, file_list(m, pkg)))
    print('  </tr>')

print('</table>')

# The document does validate, but not when it is included at the bottom of a directory listing.
#print('<hr>')
#print('<a href="http://validator.w3.org/check?uri=referer" target="_blank"><img border="0" src="http://www.w3.org/Icons/valid-html401-blue" alt="Valid HTML 4.01 Transitional" height="31" width="88"></a>')
print('</body></html>')