import os, argparse, sys, re
from mwclient import Site
from functions import print_colormsg

p = argparse.ArgumentParser(description='Upload files from a directory, with metadata values to the wiki.\n'
                                        'Note that any value containing spaces should be between quotation marks',
                            formatter_class=argparse.ArgumentDefaultsHelpFormatter)
# TODO: Add example of command to description
p.add_argument('--host', default='hub.xpub.nl/sandbox', help='wiki host')
p.add_argument('--path', default='/itchwiki/', help='Wiki path. Should end with /')
p.add_argument('--scheme', default='https', help='http or https')
p.add_argument('--dry', '-d', action='store_true',
               help='dry-run: will only print the metadata of each file that will be upload, but does NOT upload')

p.add_argument('--title', required=True, help='Metadata **Title** value of publication. Must not exist yet in the wiki.')
p.add_argument('--date', required=True, help='Metadata **Date** value of publication. Format yyyy/mm/dd '
                                             'For dates without day or date use 01 ie. 1986 --date "1986/01/01"  March 1985: --date "1984/05/01"')
p.add_argument('--creator', required=True, action='append', help='Metadata **Creator** value(s) of publication.'
                                                                       'Multiple values should be SEPARATED BY COMMA')
p.add_argument('--dir', required=True, help='Full path of the image directory, that you wish to upload')
args = p.parse_args()


site = Site(host=args.host, path=args.path, scheme=args.scheme)

wd =os.path.dirname(os.path.abspath(__file__))  # parent working directory
with open(os.path.join(wd, 'login.txt'), 'r') as login:  # read login user & pwd
    loginlines = login.read()
    user, pwd = loginlines.split('\n')
    site.login(username=user, password=pwd)  # login to wiki

site = Site(host=args.host, path=args.path)
# login
with open(os.path.join(wd, 'login.txt'), 'r') as login:  # read login user & pwd
    loginlines = login.read()
    user, pwd = loginlines.split('\n')
    site.login(username=user, password=pwd)  # login to wiki


if os.path.isdir(args.dir) is False:
    print_colormsg(f'Error: --dir {args.dir} absolute path cannot be found', level='fail')
    sys.exit()
elif not re.match(r'\d{4}\/\d{2}\/\d{2}', args.date):
    print_colormsg(f'Error:  --date {args.date} format should be --date "yyyy/mm/dd"', level='fail')
    sys.exit()
elif len(list(site.ask(f'[[Title::{args.title}]]'))) > 0:
    print_colormsg(f'Error: --title "{args.title}" already exists in wiki. Provide a different one', level='fail')
    sys.exit()

# TODO os.listdir(args.dir) ORDER!
# TODO PartOf from os.listdir(args.dir) that fit on to condition
# TODO Part base on os.lisdir order!
for _file in os.listdir(args.dir):
    if (os.path.splitext(_file)[-1]).lower() not in ['.jpg', '.jpeg', '.png']:
        print_colormsg(f'File: {_file} is not a .jpg .jpeg or .png hence will Not be upload', level='warning')
    else:
        print_colormsg(_file, level='ok')

if args.dry == True:
    print(args)
    sys.exit()


smw_prop_val = '''{{ImageMetadata
|Title={title}
|Date={date}
|Part={part}
|Partof={partof}
|Creator={creator}
}}
[[Template:ImageMetadata]]
'''

img_smw_prop_val = smw_prop_val.format(title='cat from hell', date='2020/02/13', part='1', partof='1', creator='test')
print(img_smw_prop_val)

filename = 'cat2.jpg'
page = site.images[filename]
img_url = f'https://{args.host}{args.path}/index.php/File:{filename}'

if page.exists is True:
    print(f'{filename} is already upload it, you can see it at {img_url}' )
    print('The upload process wont proceed. Please upload all images in folder by hand')
else:
    with open(filename, 'rb') as _file:
        site.upload(file=_file, filename=filename, description='img_smw_prop_val', ignore=True)