#!/usr/bin/env python

from os.path import join
from subprocess import check_call, check_output
from tempfile import TemporaryDirectory
import re


def parse_setup(lines, which):
    match = re.search(fr'{which}\w*=\w*[\(\[](?P<list>[^\)\]]*)', lines,
                      flags=re.DOTALL)
    packages = match.groups('list')[0]
    packages = re.sub(r"[\"'\s]+", '', packages, flags=re.MULTILINE)
    packages = [p for p in packages.split(',') if p]
    return packages


with open('setup.py') as fh:
    lines = fh.read()

install_requires = parse_setup(lines, 'install_requires')
tests_require = parse_setup(lines, 'tests_require')
dev_requires = [
    'build>=0.7.0',
    'pycodestyle>=2.6.0',
    'pyflakes>=2.2.0',
    'readme_renderer[md]>=26.0',
    'twine>=3.4.2',
]


def print_packages(packages, heading):
    print(f'{heading}:')
    print('  ', end='')
    print('\n  '.join(packages))


print_packages(install_requires, 'install_requires')
print_packages(tests_require, 'tests_require')
print_packages(dev_requires, 'dev_requires')

with TemporaryDirectory() as tmpdir:
    check_call(['python3', '-m', 'venv', tmpdir])

    check_call([join(tmpdir, 'bin', 'pip'), 'install', *install_requires])
    frozen = check_output([join(tmpdir, 'bin', 'pip'), 'freeze'])
    frozen = set(frozen.decode('utf-8').split())

    check_call([join(tmpdir, 'bin', 'pip'), 'install', *tests_require,
                *dev_requires])
    dev_frozen = check_output([join(tmpdir, 'bin', 'pip'), 'freeze'])
    dev_frozen = set(dev_frozen.decode('utf-8').split()) - frozen

print_packages(frozen, 'frozen')
print_packages(dev_frozen, 'dev_frozen')

with open('requirements.txt', 'w') as fh:
    fh.write('\n'.join(sorted(frozen)))
    fh.write('\n')

with open('requirements-dev.txt', 'w') as fh:
    fh.write('\n'.join(sorted(dev_frozen)))
    fh.write('\n')