import sys
from pathlib import Path
import stat

import toml

from . import get_response
from .static import Situation
from .config import CONFIG_FILES, CONFIG_DIRECTORY, generate_current_configuration


def development():
    s = "positive"
    if len(sys.argv) > 1:
        s = sys.argv[1]

    print(get_response(Situation(s)))
    

def write_current_config():
    f = "python-mommy.toml"
    if len(sys.argv) > 1:
        f = sys.argv[1]

    config_file = CONFIG_DIRECTORY / f
    print(f"writing to: {config_file}")

    data = toml.dumps(generate_current_configuration())
    print(data)
    with config_file.open("w") as f:
        f.write(data)


WRAPPER_TEMPLATE = """#!{inner_bin}
# -*- coding: utf-8 -*-

import sys, subprocess
from python_mommy_venv import get_response, Situation


INTERPRETER = "{inner_bin}"
result = subprocess.run([INTERPRETER] + sys.argv[1:])
code = result.returncode

print()
print(get_response(Situation.POSITIVE if code == 0 else Situation.NEGATIVE))
exit(code=code)
"""


PIP_HOOK = """# GENERATED BY MOMMY
    code = main()

    from pathlib import Path

    bin_path = Path(".venv", "bin")

    python_interpreter_wrappers = []
    for path in bin_path.iterdir():
        if path.is_symlink() and path.name.startswith("inner_"):
            python_interpreter_wrappers.append(path.name.replace("inner_", "", 1))

    for path in bin_path.iterdir():
        if path.is_symlink():
            continue
        if path.name in python_interpreter_wrappers:
            continue

        text: str
        with path.open("r") as f:
            text = f.read()

        first_line = text.split("\\n")[0]
        if not ("inner_" in first_line and first_line.startswith("#!")):
            continue 
        
        print(f"mommifying " + str(path))

        text = text.replace("inner_", "", 1)
        with path.open("w") as f:
            f.write(text)

    sys.exit(code)"""


def mommify_pip(path: Path):
    text: str
    with path.open("r") as f:
        text = f.read()

    if "# GENERATED BY MOMMY" in text:
        print(f"pip hook already installed in {path}")
        return
    
    print(f"installing pip hook in {path}")
    text = text.replace("sys.exit(main())", PIP_HOOK, 1)
    with path.open("w") as f:
        f.write(text)

def mommify_venv():
    v = ".venv"
    if len(sys.argv) > 1:
        v = sys.argv[1]

    bin_path = Path(v, "bin")
    bin_path = bin_path.resolve()
    print(bin_path)

    for path in bin_path.iterdir():
        if not path.is_symlink():
            if path.name.startswith("pip"):
                mommify_pip(path)
            continue
        
        name = path.name
        if name.startswith("inner_"):
            continue
        target = path.resolve()

        print("")
        print(f"modifying {name} ({target})")

        # creating inner symlink
        inner_bin = Path(bin_path, "inner_" + name)
        if inner_bin.exists():
            print(f"inner symlink does already exist {inner_bin}")
            print("skipping")
            continue

        print(f"creating symlink: {inner_bin} -> {target}")
        Path(bin_path, "inner_" + name).symlink_to(target)

        # remove original symlink
        print(f"removing original symlink: {path}")
        path.unlink()

        # creating the wrapper string
        print("writing wrapper script")
        with path.open("w") as f:
            f.write(WRAPPER_TEMPLATE.format(inner_bin=str(inner_bin)))
        print("making wrapper script executable")
        path.chmod(path.stat().st_mode | stat.S_IEXEC)