link-bin-outputs.py

#!/usr/bin/env python3
import itertools
import argparse
import os.path
import json


argparser = argparse.ArgumentParser(
    description='Link bin outputs based on package.json contents')
argparser.add_argument(
    'bin_out', type=str,
    help='bin output path')
argparser.add_argument(
    'lib_out', type=str,
    help='lib output path')


def get_bin_attr_files(package_json):
    try:
        bins = package_json['bin']
    except KeyError:
        return tuple()
    else:
        if isinstance(bins, str):
            return ((os.path.basename(package_json['name']), bins),)
        else:
            return bins.items()


def get_directories_bin_attr_files(package_json, lib_out):
    try:
        bins = package_json['directories']['bin']
    except Exception:
        pass
    else:
        for f in os.path.listdir(os.path.join(lib_out, bins)):
            yield os.path.join(lib_out, f)


def resolve_bin_outputs(bin_out, lib_out, entries):
    for e in entries:
        yield (
            os.path.join(bin_out, e[0]),
            os.path.join(lib_out, e[1])
        )


if __name__ == '__main__':
    args = argparser.parse_args()

    with open(os.path.join(args.lib_out, 'package.json')) as f:
        package_json = json.load(f)

    for fout, fin in resolve_bin_outputs(
            args.bin_out, args.lib_out, itertools.chain(
                get_bin_attr_files(package_json),
                get_directories_bin_attr_files(args.lib_out, package_json))):

        os.makedirs(os.path.dirname(fout), exist_ok=True)  # Allows for things like @mapbox/node-pre-gyp
        os.symlink(fin, fout)
        os.chmod(fout, 0o755)

        # Print input file to stdout so we can pipe it to patchShebangs
        print(fin)