import re
from pathlib import Path


def read_version_from_pyproject(file_path):
    with open(file_path, "r") as file:
        for line in file:
            match = re.search(r'version = "(.*)"', line)
            if match:
                return match.group(1)
    return None


# def get_version_from_pypi(package_name):
#     import requests

#     response = requests.get(f"https://pypi.org/pypi/{package_name}/json")
#     if response.ok:
#         return response.json()["info"]["version"]
#     return None


def get_version_from_pypi(package_name):
    # Use default python lib to make the GET for this because it runs in github actions
    import json
    import urllib.request

    response = urllib.request.urlopen(f"https://pypi.org/pypi/{package_name}/json")
    if response.getcode() == 200:
        return json.loads(response.read())["info"]["version"]
    return None


def is_development_release(version):
    """
    Determines if the version is a development version based on PEP 440.

    We consider a development version (.devN) as our nightly versions
    """
    return "dev" in version


def update_pyproject_dependency(pyproject_path, version, is_nightly):
    # Strip "v" prefix from version if present
    if version.startswith("v"):
        version = version[1:]

    pattern = re.compile(r'langflow-base = \{ path = "\./src/backend/base", develop = true \}')
    if is_nightly:
        # NOTE: This process can be simplified; see the note in update_lf_base_dependency.py
        replacement = f'langflow-base-nightly = "{version}"'
    else:
        replacement = f'langflow-base = "^{version}"'
    with open(pyproject_path, "r") as file:
        content = file.read()
    content = pattern.sub(replacement, content)
    with open(pyproject_path, "w") as file:
        file.write(content)


if __name__ == "__main__":
    # Backing up files
    pyproject_path = Path(__file__).resolve().parent / "../pyproject.toml"
    pyproject_path = pyproject_path.resolve()
    with open(pyproject_path, "r") as original, open(pyproject_path.with_name("pyproject.toml.bak"), "w") as backup:
        backup.write(original.read())
    # Now backup poetry.lock
    with open(pyproject_path.with_name("poetry.lock"), "r") as original, open(
        pyproject_path.with_name("poetry.lock.bak"), "w"
    ) as backup:
        backup.write(original.read())

    # Reading version and updating pyproject.toml
    langflow_base_path = Path(__file__).resolve().parent / "../src/backend/base/pyproject.toml"
    version = read_version_from_pyproject(langflow_base_path)
    if version:
        # Nightly versions contain "dev"
        # WARNING: This will cause issues if we release `.dev` versions to `langflow` or `langflow-base`
        is_nightly = is_development_release(version)
        update_pyproject_dependency(pyproject_path, version, is_nightly)
    else:
        print("Error: Version not found.")