#!/usr/bin/env python

import sys, os, json, argparse, subprocess, hashlib
from kumikolib import Kumiko
from lib.html import HTML
import cv2
import glob


parser = argparse.ArgumentParser(description='Kumiko CLI')

# Input/Output
parser.add_argument('-i', '--input', nargs='+', required=True, help='A file or folder name to parse')
parser.add_argument('-o', '--output', nargs=1, help='A file name to save json/html output to')
parser.add_argument('--rtl', action='store_true', help='Pass this option to number panels right-to-left')

# HTML reader page options
parser.add_argument('-b', '--browser', nargs=1, help='Opens given browser to read your comic page(s) panel-by-panel! (implies --html)', choices=['firefox','konqueror','chromium'])
parser.add_argument('--html', action='store_true', help='Generates an HTML file for reading')
parser.add_argument('--html-static-dir', nargs=1, help='When generating HTML, this will be the relative directory for javascript files: <script src="$static-dir/...">  (implies --html)')

# Configuration tweaks
parser.add_argument('--min-panel-size-ratio', nargs=1, type=float, help='Panels will be considered too small and exluded if they have a width < img.width * ratio or height < img/height * ratio (default is 1/15th)')

# Utilities
parser.add_argument('--debug', action='store_true', help="Generate an HTML debug file to show Kumiko's processing steps")
parser.add_argument('--progress', action='store_true', help='Prints progress information')



args = parser.parse_args()
k = Kumiko({
	'debug': args.debug,
	'progress': args.progress,
	'rtl': args.rtl,
	'min_panel_size_ratio': args.min_panel_size_ratio[0] if args.min_panel_size_ratio else False
})

file_or_folder = args.input[0]
folder = None
html_file = None

# Folder
if len(args.input) == 1 and os.path.isdir(args.input[0]):
	folder = args.input[0]
	if folder[-1] == '/':
		folder = folder[0:-1]
	info = k.parse_dir(folder)
	html_file = os.path.join('tests/results',os.path.basename(folder)+'.html')

# File
elif len(args.input) == 1 and os.path.isfile(args.input[0]):
	f = args.input[0]
	folder = os.path.dirname(f)
	if not folder:
		folder = './'
	info = k.parse_images([f])
	html_file = os.path.join('tests/results',os.path.basename(f)+'.html')

# URL list
else:
	folder = 'urls'
	info = k.parse_url_list(args.input)
	filehash = hashlib.sha1(';'.join(sorted(args.input)).encode()).hexdigest()
	html_file = os.path.join('tests/results',filehash+'.html')

if len(info) == 0:
	print("--input (-i) is not a file, or directory, or URL list: '"+str(args.input)+"'")
	sys.exit(1)

info = json.dumps(info)


# Generate HTML
if args.html or args.browser or args.html_static_dir or args.debug:
	images_dir = 'urls' if folder == 'urls' else os.path.relpath(folder,'tests/results')+'/'
	reldir = args.html_static_dir[0] if args.html_static_dir else '../../'
	
	html = ''
	if args.debug:
		html = k.dbg.html(folder,reldir)
	else:
		html += HTML.header(reldir = reldir)
		html += HTML.reader(info,images_dir)
		html += HTML.footer
	
	if args.output:
		html_file = args.output[0]
	
	if args.browser or args.output or args.debug:
		fh = open(html_file, 'w')
		fh.write(html)
		fh.close()
		print('Saved HTML file:',html_file)
	else:
		print(html)

# Or output Image
else:
	if args.output:
		print(info)
		info = json.loads(info)
		for index, input in enumerate(glob.glob(args.input[0]+"/*")):
			image = cv2.imread(input)
			panels = info[index]['panels']
			for panel in panels:
				cv2.rectangle(image, (panel[0], panel[1]), (panel[0]+panel[2], panel[1]+panel[3]), (0, 255, 0), 3)
			cv2.imwrite(info[index]['filename'], image)
	else:
		print(info)


# Open in browser
if args.browser:
	subprocess.run([args.browser[0],html_file])