from waflib import Logs, Utils, TaskGen, Task
from waflib.Tools import ccroot, c, cxx

MAIN_BINARY = 'xash'

def options(opt):
	opt.add_option('--static-linking', action='store', dest='STATIC_LINKING', default=None)

def configure(conf):
	if conf.options.STATIC_LINKING:
		conf.find_program('ld')
		conf.find_program('objcopy')
		conf.env.STATIC_LINKING = conf.options.STATIC_LINKING

def build(bld):
	if bld.env.STATIC_LINKING:
		apply_static(MAIN_BINARY,*bld.env.STATIC_LINKING.split(','))


class objcopy(Task.Task):
	no_errcheck_out = True
	run_str = '${OBJCOPY} -G lib_${NAME}_exports ${SRC[0].abspath()} ${TGT[0].abspath()}'
	def keyword(self):
		return 'ObjCopy'

class xshlib(ccroot.link_task):
	no_errcheck_out = True
	run_str = '${LD} -r -o ${TGT[0].abspath()} -melf_i386 ${CCLNK_SRC_F}${SRC}'
	def add_target(self, target):
		base = self.generator.path
		target_unstripped = base.find_or_declare('%s.unstripped.o'% target)
		target_stripped = base.find_or_declare('%s.o'% target)

		self.set_outputs(target_unstripped)
		self.generator.objcopy_task= self.generator.create_task('objcopy', target_unstripped, target_stripped)
		self.generator.objcopy_task.env['NAME'] = target
		print self.generator.name
#	def run(self):
#		xshlib.run(self)
#		objcopy.run(self)


class cprogram_static(c.cprogram):
	run_str = '${LINK_CC} -static ${LINKFLAGS} ${CCLNK_SRC_F}${SRC} ${CCLNK_TGT_F}${TGT[0].abspath()} ${RPATH_ST:RPATH} ${FRAMEWORKPATH_ST:FRAMEWORKPATH} ${FRAMEWORK_ST:FRAMEWORK} ${ARCH_ST:ARCH} ${STLIB_MARKER} ${STLIBPATH_ST:STLIBPATH} ${STLIB_ST:STLIB} ${STLIB_MARKER} ${LIBPATH_ST:LIBPATH} ${LIB_ST:LIB} ${LDFLAGS}'
class cxxprogram_static(cxx.cxxprogram):
	run_str = '${LINK_CXX} -static ${LINKFLAGS} ${CXXLNK_SRC_F}${SRC} ${CXXLNK_TGT_F}${TGT[0].abspath()} ${RPATH_ST:RPATH} ${FRAMEWORKPATH_ST:FRAMEWORKPATH} ${FRAMEWORK_ST:FRAMEWORK} ${ARCH_ST:ARCH} ${STLIB_MARKER} ${STLIBPATH_ST:STLIBPATH} ${STLIB_ST:STLIB} ${STLIB_MARKER} ${LIBPATH_ST:LIBPATH} ${LIB_ST:LIB} ${LDFLAGS}'

ccroot.USELIB_VARS['cprogram_static'] = ccroot.USELIB_VARS['cxxprogram_static'] = ccroot.USELIB_VARS['cxxprogram']


@TaskGen.feature('xshlib')
@TaskGen.after('apply_link')
def add_objcopy_task(self):
	if getattr(self, 'link_task', None):
		exe_node = self.link_task.outputs[0]
		self.create_task('objcopy', self.link_task.outputs[0], self.link_task.outputs[1])



def apply_static(main, *reloc):
	def write_libraries_list(out_node):
		libraries = reloc
		externs = '\n'.join(['extern table_t lib_%s_exports[];' % e for e in libraries])
		table = '\n'.join(['{ "%s", &lib_%s_exports },' % (e, e) for e in libraries])
		out_node.write('%s\nstruct {const char *name;void *func;} libs[] = {\n%s\n{0,0}\n};\n' % (externs, table ))


	def write_export_list(name,in_node, out_node):
		exports = in_node.read().split('\n')
		externs = '\n'.join(['extern void %s;' % e for e in exports])
		table = '\n'.join(['{ "%s", &%s },' % (e, e) for e in exports])
		out_node.write('%s\nstruct {const char *name;void *func;} lib_%s_exports[] = {\n%s\n{0,0}\n};\n' % (externs, name, table ))

	@TaskGen.feature('cshlib', 'cxxshlib')
	@TaskGen.before('process_source')
	def add_link_helper(self):
		if self.name in reloc:
			for k in ('cshlib', 'cxxshlib'):
				if k in self.features:
					self.features.remove(k)
			self.features.append('xshlib')
			in_node = self.path.get_src().make_node('exports.txt')
			bldnode = self.path.get_bld()
			bldnode.mkdir()
			out_node = bldnode.make_node('link_helper.c')
			write_export_list(self.name,in_node, out_node)
			self.source = Utils.to_list(self.source) + [out_node]

	@TaskGen.feature('cshlib', 'cxxshlib', 'cprogram', 'cxxprogram', 'cprogram_static', 'cxxprogram_static')
	@TaskGen.before('process_source')
	def add_deps(self):
		if self.name == main:
			write_libraries_list(self.path.get_bld().make_node('generated_library_tables.h'))

			for t in reloc:
				self.source += [self.bld.get_tgen_by_name(t).objcopy_task.outputs[0]]