#!/usr/bin/env python3
import argparse
import os
import subprocess
import sys
from pathlib import Path

# Define the default values
TCL_SCRIPT_NAME = "xsct-init.tcl"
SCRIPTS_DIR = "scripts"
DEFAULT_IP_ADDRESS_HW_SERVER = "localhost"


def main():
    parser = argparse.ArgumentParser(
        description="Script to launch xsct, initialize and prepare the processing system for debugging "
        "and load an a bare-metal application to the board"
    )
    parser.add_argument(
        "-t",
        "--tools",
        # Required only if env var is not set
        required=not bool(os.getenv("VIVADO_TOOLS")),
        # Use env var if set
        default=os.getenv("VIVADO_TOOLS"),
        help="The path to the tool to use. Must point to a valid Vivado tools installation which"
        "also provides xsct, for example a Vitis installation. The directory where the path "
        " points to should contain a shell script named settings64.sh",
    )
    parser.add_argument(
        "--itcl",
        dest="init_tcl",
        required=True,
        help="Path to the ps7 initialization TCL file to prepare the processing system.",
    )
    parser.add_argument("-a", "--app", dest="app", help="Path to the app to program")
    parser.add_argument(
        "-i",
        "--ip",
        default=DEFAULT_IP_ADDRESS_HW_SERVER,
        help="The IP address of the hardware server (default: localhost)",
    )
    parser.add_argument(
        "-b",
        "--bit",
        dest="bit",
        help="Optional path to the bitstream which will also be programed to the device.",
    )

    args = parser.parse_args()

    settings_script = os.path.join(args.tools, "settings64.sh")

    if not os.path.isfile(settings_script):
        print(f"Invalid tool path {args.tools}, did not find settings file.")
        sys.exit(1)

    # Source the settings script and check for xsdb availability
    command = f"source {settings_script} && command -v xsct"
    result = subprocess.run(
        command,
        shell=True,
        capture_output=True,
        executable="/bin/bash",
    )

    if result.returncode != 0:
        print("Error: 'xsct' could not be found after sourcing settings64.sh.")
        sys.exit(1)

    if args.app and not os.path.isfile(args.app):
        print(f"The app '{args.app}' does not exist")
        sys.exit(1)

    # Export environment variables
    if args.app:
        os.environ["APP"] = args.app
    os.environ["IP_ADDRESS_HW_SERVER"] = args.ip
    if args.bit:
        os.environ["BITSTREAM"] = args.bit
    os.environ["PS_INIT_SCRIPT"] = args.init_tcl

    xsct_script = Path(TCL_SCRIPT_NAME)

    if not xsct_script.exists():
        xsct_script = Path(os.path.join(SCRIPTS_DIR, TCL_SCRIPT_NAME))
        if not xsct_script.exists():
            print(f"Could not find the xsct initialization script {TCL_SCRIPT_NAME}")
            sys.exit(1)

    # Launch xsct with the initialization script

    command = f"bash -c 'source {settings_script} && xsct {xsct_script} | tee xsct-output.log'"
    subprocess.run(
        command,
        shell=True,
        check=True,
    )


if __name__ == "__main__":
    main()