#!/usr/bin/env python3

import argparse
from pathlib import Path
from configparser import ConfigParser
from zipfile import ZipFile


parser = argparse.ArgumentParser(description='Install wheel scripts.')
parser.add_argument('wheel', type=Path, help='Wheel file')
parser.add_argument('--prefix', type=Path, default=Path('/usr/local/'), help='Installation prefix (/usr/, ~/.local, …)')


def get_console_scripts(whl):
	with ZipFile(whl) as archive:
		n = next(iter(n for n in archive.namelist() if n.endswith('.dist-info/entry_points.txt')), None)
		if n is None:
			return {}
		ini = archive.read(n).decode('utf-8')
	parser = ConfigParser()
	parser.optionxform = str  # case sensitive
	parser.read_string(ini)
	return dict(parser['console_scripts']) if 'console_scripts' in parser else {}


def main():
	args = parser.parse_args()
	
	scripts = get_console_scripts(args.wheel)
	for name, defn in scripts.items():
		defn, *_ = defn.split(';')  # extra condition
		mod, fun = defn.split(':')
		if mod.endswith('.__main__'):
			mod = mod[:-len('.__main__')]
			code = f'''\
#!/bin/sh
exec /usr/bin/python3 -m {mod} "$@"
'''
		else:
			code = f'''\
#!/usr/bin/python3
from {mod} import {fun}
{fun}()
'''
		bin_path = args.prefix / 'bin' / name
		bin_path.parent.mkdir(parents=True, exist_ok=True)
		bin_path.write_text(code)
		bin_path.chmod(0o755)


if __name__ == '__main__':
	main()