#!/bin/env python3
"""
Calls the `ld` linker using a `yaml` manifest
"""

# Import
import argparse
from pathlib import Path
import subprocess
import yaml
import util


def main(args):
	"""
	Main function
	"""
	config = yaml.safe_load(open(args.input_yaml, encoding="utf-8"))
	input_dir = Path(args.input_yaml).parent

	entry = config.get("entry")
	start_addr = config["start_addr"]
	objs = config["objs"]
	objs = list(map(lambda obj_path: str(util.process_path(obj_path, input_dir)), objs))

	sections = config["sections"]
	sections = map(lambda section: f"KEEP(*({section}));", sections)
	sections = '\n\t\t'.join(sections)

	link_with = config.get("link_with") or []
	link_with = list(map(lambda file: f"--just-symbols={file}", link_with))

	# Create the linker script
	with open(args.linker_script_output, "w", encoding="utf-8") as linker_script_file:

		linker_script_file.write(f"""\
SECTIONS {{
	_.text = {hex(start_addr)};
	.text _.text : {{
		{sections}
	}}
}}

SECTIONS {{
	/DISCARD/ : {{ *(*) }}
}}""")

		if entry is not None:
			linker_script_file.write(f"""\


ENTRY({entry})""")

	args = [
	    args.ld_bin,  #
	    "-o",
	    args.output,
	    "-EL",
	    "--omagic",
	    "--script",
	    args.linker_script_output,
	    "--warn-section-align",
	    "--no-warn-mismatch",  # TODO: Might be worth considering some mismatches?
	    "--warn-common",
	] + link_with + objs
	subprocess.run(args, check=True)


if __name__ == "__main__":
	parser = argparse.ArgumentParser(description="`ld` wrapper")
	parser.add_argument("input_yaml", type=str)
	parser.add_argument("-o", dest="output", type=str, required=True)
	parser.add_argument("--linker-script-output", dest="linker_script_output", type=str, required=True)
	parser.add_argument("--ld-bin", dest="ld_bin", type=str, required=True)

	args = parser.parse_args()
	main(args)