#!/usr/bin/env python


import os, subprocess, json, argparse, re, tempfile
from kumikolib import Kumiko,Panel,Debug
from lib.html import HTML



class Tester:
	
	files = []
	git_repo = 'https://framagit.org/nicooo/kumiko'
	git_versions = ['v1.4.1', 'current']
	
	def __init__(self, options):
		self.savedir = os.path.join('tests','results')
		if not os.path.isdir(self.savedir):
			raise Exception("'%s' is not a directory" % self.savedir)
		
		self.options = {
			'browser': options['browser'] if 'browser' in options and options['browser'] else False,
			'html': options['html'] if 'html' in options else False
		}
		
		# Test all files in tests/images/$folder/
		if len(self.files) == 0:
			for d in os.scandir('tests/images'):
				 if d.is_dir():
					 self.files.append(d)
			self.files.sort(key=lambda d: d.name)
	
	
	def run_all(self):
		for git_version in self.git_versions:
			self.run(git_version)
	
	
	def run(self, git_version):
		print('\n########## Finding panels with kumiko version',git_version,'##########')
		kumiko_bin = './kumiko'
		
		# Kumiko would not accept non-image files before v1.2, special case for .licence files
		accepts_license_files = git_version >= 'v1.2' or not re.match('^v\d+.\d+$', git_version)
		tmpfolder = None if accepts_license_files else tempfile.TemporaryDirectory(dir='./tests')
		
		# Get that Kumiko version
		if git_version != 'current':
			tempgit = '/tmp/kumikogit'
			subprocess.run(['rm','-rf',tempgit])
			subprocess.run(['git', 'clone', self.git_repo, tempgit], capture_output=True)
			subprocess.run(['git', 'checkout', git_version], cwd=tempgit, capture_output=True)
			kumiko_bin = os.path.join(tempgit,'kumiko')
		
		for f in self.files:
			# move .license files to tempdir (<v1.2 compatibility)
			if tmpfolder and os.path.isdir(f):
				for g in os.scandir(f):
					if re.search('\.license$',g.name):
						os.rename(g,os.path.join(tmpfolder.name,os.path.basename(g)))
			
			print("##### Kumiko-cutting {0} ({1}) #####".format(f if isinstance(f,str) else f.name, git_version))
			
			subprocess.run(['mkdir','-p',os.path.join(self.savedir,git_version)])
			jsonfile = os.path.join(self.savedir,git_version,os.path.basename(f)+'.json')
			subprocess.run(args=[kumiko_bin, '-i', f, '-o', jsonfile, '--progress'])
			
			if tmpfolder and os.path.isdir(f):
				for g in os.scandir(tmpfolder.name):
					if re.search('\.license$',g.name):
						os.rename(g,os.path.join(f,os.path.basename(g)))
	
	
	def compare_all(self):
		self.max_diffs = 20
		
		for i in range(len(self.git_versions)-1):
			v1 = self.git_versions[i]
			v2 = self.git_versions[i+1]
			print('\n########## Comparing kumiko results between versions',v1,'and',v2,'##########')
			
			files_diff = {}
			
			for file_or_dir in self.files:
				
				if len(files_diff) > 20:
					print('The maximum number of differences in files (20) has been reached, stopping')
					break
				
				with open(os.path.join(self.savedir,v1,os.path.basename(file_or_dir)+'.json')) as fh:
					json1 = json.load(fh)
				with open(os.path.join(self.savedir,v2,os.path.basename(file_or_dir)+'.json')) as fh:
					json2 = json.load(fh)
				
				files_diff.update(Debug.get_files_diff(file_or_dir,json1,json2))
			
			print('Found',len(files_diff),'differences')
			
			if not self.options['html']:
				return
			
			print('Generating HTML diff file')
			
			html_diff_file = os.path.join(self.savedir,'diff-'+v1+'-'+v2+'.html')
			diff_file = open(html_diff_file, 'w')
			diff_file.write(HTML.header('Comparing Kumiko results','../../'))
			
			diff_file.write(HTML.nbdiffs(files_diff))
			
			for img in files_diff:
				diff_file.write(HTML.side_by_side_panels(img,'',files_diff[img]['jsons'],v1,v2,images_dir=files_diff[img]['images_dir'], known_panels=files_diff[img]['known_panels']))
			
			diff_file.write(HTML.footer)
			diff_file.close()
			
			if self.options['browser']:
				subprocess.run([self.options['browser'],html_diff_file])
	


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

parser.add_argument('action', nargs='?', help="Just 'run' (compute information about files), 'compare' two versions of the code, or both", choices=['run','compare','run_compare'], default='run_compare')

parser.add_argument('-b', '--browser', nargs=1, help='Opens given browser to view the differences when ready (implies --html)', choices=['firefox','konqueror','chromium'])
parser.add_argument('--html', action='store_true', help='Generates an HTML file showing the differences between code versions')

args = parser.parse_args()

tester = Tester({
	'html': args.html or args.browser,
	'browser': args.browser[0] if args.browser != None else False,
})

if args.action in ['run','run_compare']:
	tester.run_all()
if args.action in ['compare','run_compare']:
	tester.compare_all()