__author__ = "D. Howe"
__version__ = "0.1.0"
__email__ = "d.howe@wrl.unsw.edu.au"

import os
import sys
import glob
import argparse
import subprocess


def pdf_unlock():
    # Get pdfs in same folder
    fnames = [
        os.path.join(name) for name in os.listdir('.') if name.endswith('.pdf')
    ]

    # Use pdf names from user (if provided)
    if len(sys.argv[1:]) > 0:
        fnames = sys.argv[1:]

    for fname in fnames:

        # Strip file extension
        base_name = os.path.splitext(fname)[0]

        # save as temporary file
        cmd_str = ('gs '
                   '-dNOPAUSE '
                   '-dQUIET '
                   '-dBATCH '
                   '-sDEVICE=pdfwrite '
                   '-sOutputFile="{output_pdf}" '
                   '"{input_pdf}"').format(output_pdf, input_pdf)

        print('Processing ' + base_name + '.pdf ...')

        with subprocess.Popen(
                cmd_str,
                stdout=subprocess.PIPE,
                bufsize=1,
                universal_newlines=True) as p:
            for line in p.stdout:
                print(line, end='')

        # Overwrite original file
        os.remove(base_name + '.pdf')
        os.rename(base_name + '_temp.pdf', base_name + '.pdf')


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('folder', help='name of input folder', default=None)
    parser.add_argument(
        '-r',
        '--recursive',
        help='search for files recursively',
        action='store_true')
    args = parser.parse_args()

    print(args.recursive)


if __name__ == '__main__':
    main()