"""Check that executable text files have a shebang."""
from __future__ import absolute_import
from __future__ import print_function
from __future__ import unicode_literals

import argparse
import pipes
import sys


def check_has_shebang(path):
    with open(path, 'rb') as f:
        first_bytes = f.read(2)

    if first_bytes != b'#!':
        print(
            '{path}: marked executable but has no (or invalid) shebang!\n'
            "  If it isn't supposed to be executable, try: chmod -x {quoted}\n"
            '  If it is supposed to be executable, double-check its shebang.'.format(
                path=path,
                quoted=pipes.quote(path),
            ),
            file=sys.stderr,
        )
        return 1
    else:
        return 0


def main(argv=None):
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument('filenames', nargs='*')
    args = parser.parse_args(argv)

    retv = 0

    for filename in args.filenames:
        retv |= check_has_shebang(filename)

    return retv