#!/usr/bin/env python

# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
import sys

if not sys.version_info >= (3, 10, 0):
    print(
        "Warning: Autorest for Python extension requires Python 3.10 at least. We will run your code with Pyodide since your Python version isn't adequate."
    )
    sys.exit(2)  # Exit code 2 for inadequate environment

try:
    from package_manager import detect_package_manager, PackageManagerNotFoundError

    detect_package_manager()  # Just check if we have a package manager
except (ImportError, ModuleNotFoundError, PackageManagerNotFoundError):
    print(
        "Warning: Your Python installation doesn't have a suitable package manager (pip or uv) available. We will run your code with Pyodide since your Python environment isn't adequate."
    )
    sys.exit(2)  # Exit code 2 for inadequate environment

try:
    import venv
except (ImportError, ModuleNotFoundError):
    print(
        "Warning: Your Python installation doesn't have venv available. We will run your code with Pyodide since your Python version isn't adequate."
    )
    sys.exit(2)  # Exit code 2 for inadequate environment


# Now we have a package manager (uv or pip) and Py >= 3.10, go to work
# At this point, both Python and package manager are confirmed to be available
# Any failures from here should fail the npm install, not fallback to Pyodide

from pathlib import Path

# eng/scripts/setup/install.py -> need to go up 4 levels to get to package root
_ROOT_DIR = Path(__file__).parent.parent.parent.parent


def main():
    venv_path = _ROOT_DIR / "venv"

    # Create virtual environment using package manager abstraction
    from package_manager import create_venv_with_package_manager, install_packages

    try:
        venv_context = create_venv_with_package_manager(venv_path)

        # Install required packages - install_packages handles package manager logic
        install_packages([f"{_ROOT_DIR}/generator"], venv_context, cwd=_ROOT_DIR)
    except Exception as e:
        # Since Python and package manager are available, any failure here should fail the npm install
        print(f"Error: Installation failed despite Python and package manager being available: {e}")
        sys.exit(1)  # Exit code 1 for installation failure


if __name__ == "__main__":
    main()