#!/usr/bin/python3

import os
import argparse
import subprocess
import sys

_environ = os.environ.copy()

parser = argparse.ArgumentParser()
parser.add_argument("file", nargs=1, help="run PongChamp with the given file")
parser.add_argument("--lib", dest="lib", nargs=1, help="override the ld library path")

args = parser.parse_args()

cwd = os.getcwd()

# LD_LIBRARY_PATH=$(pwd)/engine-lib/ stack run -- examples/call-engine.js

if args.lib:
    lib = args.lib
else:
    lib = os.path.join(cwd, "engine-lib")

os.environ["LD_LIBRARY_PATH"] = lib

if sys.platform == 'win32':
    tmp_path = os.environ["PATH"]
    os.environ["PATH"] = tmp_path + ";" + lib + ";"

cmd = ['stack', 'run', '--', args.file[0]]
subprocess.run(' '.join(cmd), shell=True, env=os.environ)

if sys.platform == 'win32':
    os.environ["PATH"] = tmp_path