import os
import subprocess
import sys
from . import tools

global original_path

try :
	original_path = os.getcwd()
	testpath = os.path.dirname(os.path.abspath(os.path.join(os.getcwd(), sys.argv[0])))
	sys.path.append(testpath)
	import config

	SRCDIR = os.path.abspath(config.SRCDIR)
	BUILDDIR = os.path.abspath(config.BUILDDIR)
	distribute = config.DISTRIBUTE
	os.chdir(testpath)

except:
	print('ERROR: missing config.py, re-run configure script.', file=sys.stderr)
	sys.exit(1)

class Architecture:
	KnownArchitectures = {
		'x64'		: 'x64',
		'x86-64'	: 'x64',
		'x86_64'	: 'x64',
		'x86'		: 'x86',
		'aarch64'	: 'arm',
		'i386'		: 'x86',
		'i486'		: 'x86',
		'i686'		: 'x86',
		'Intel 80386'	: 'x86',
		'arm'		: 'arm',
		'ARM'		: 'arm',
	}

	CrossCompileFlags = {
		'x64' : 'ARCH_FLAGS=-m64',
		'x86' : 'ARCH_FLAGS=-m32',
	}

	def __init__(self, arch):
		try:
			canonical_host = Architecture.make_canonical( config.HOSTARCH )
		except KeyError:
			print("Unknown host architecture %s" % config.HOSTARCH, file=sys.stderr)
			sys.exit(1)

		if arch:
			try:
				arch = Architecture.make_canonical( arch )
			except KeyError:
				print("Unknown architecture %s" % arch, file=sys.stderr)
				sys.exit(1)

		if arch and arch != canonical_host:
			self.target = arch
			self.cross_compile = True
		else:
			self.target = canonical_host
			self.cross_compile = False


		try :
			self.flags = Architecture.CrossCompileFlags[self.target]
		except KeyError:
			print("Cross compilation not available for architecture %s" % self.target, file=sys.stderr)
			sys.exit(1)

		self.string = self.target

	def update(self):
		if not self.cross_compile:
			self.target = machine_default()
			self.string = self.target
			print("updated to %s" % self.target)

	def match(self, arch):
		return True if not arch else self.target == arch

	@classmethod
	def make_canonical(_, arch):
		return Architecture.KnownArchitectures[arch]


class Debug:
	def __init__(self, value):
		self.string = "debug" if value else "no debug"
		self.flags  = """DEBUG_FLAGS=%s""" % ("-debug -O0" if value else "-nodebug -O2")
		self.path   = "debug" if value else "nodebug"

class Install:
	def __init__(self, value):
		if value:
			distribute = False

		self.string = "installed" if value else "in-tree"
		self.flags  = """INSTALL_FLAGS=%s""" % ("" if value else "-in-tree")

class Timeouts:
	def __init__(self, ts, tg):
		self.single = Timeouts.check(ts)
		self.total  = Timeouts.check(tg)

	@classmethod
	def check(_, value):
		if value < 1:
			print("Timeouts must be at least 1 second", file=sys.stderr)
			sys.exit(1)

		return value

def init( options ):
	global arch
	global archive
	global debug
	global distcc
	global dry_run
	global generating
	global install
	global make
	global output_width
	global timeout

	arch         = Architecture(options.arch)
	archive      = os.path.abspath(os.path.join(original_path, options.archive_errors)) if options.archive_errors else None
	debug        = Debug(options.debug)
	distcc       = "DISTCC_CFA_PATH=~/.cfadistcc/%s/cfa" % tools.config_hash()
	dry_run      = options.dry_run
	generating   = options.regenerate_expected
	install      = Install(options.install)
	make         = ['make']
	output_width = 24
	timeout      = Timeouts(options.timeout, options.global_timeout)

	# if we distribute, distcc errors will fail tests, use log file for distcc
	# don't use "'DISTCC_LOG' not in os.environ" because it can be set to ''
	if distribute and not os.environ.get('DISTCC_LOG'):
		os.putenv('DISTCC_LOG', os.path.join(BUILDDIR, 'distcc_error.log'))

def update_make_cmd(force, jobs):
	global make

	make = ['make'] if not force else ['make', "-j%i" % jobs]

def validate():
	errf = os.path.join(BUILDDIR, ".validate.err")
	make_ret, out = tools.make( ".validate", error_file = errf, output_file=subprocess.DEVNULL, error=subprocess.DEVNULL )
	if make_ret != 0:
		with open (errf, "r") as myfile:
			error=myfile.read()
		print("ERROR: Invalid configuration %s:%s" % (arch.string, debug.string), file=sys.stderr)
		print("       verify returned : \n%s" % error, file=sys.stderr)
		tools.rm(errf)
		sys.exit(1)

	tools.rm(errf)

def prep_output(tests):
	global output_width
	output_width = max(map(lambda t: len(t.target()), tests))