default.nix

{ python, pypkgs }:
{ pkgs-unstable
, lib
, ...
}:
let
  pythonVersion = lib.versions.majorMinor python.version;

  pkgs = pkgs-unstable;

  pylibs-dir = ".pythonlibs";

  userbase = "$REPL_HOME/${pylibs-dir}";

  pythonUtils = import ../../python-utils {
    inherit pkgs python pypkgs;
  };
  pythonWrapper = pythonUtils.pythonWrapper;
  python-ld-library-path = pythonUtils.python-ld-library-path;

  pip = import ./pip.nix (
    pkgs
    // {
      inherit python pypkgs;
    }
  );
  pip-wrapper = pythonWrapper {
    bin = "${pip.pip}/bin/pip";
    name = "pip";
  };

  poetry = pkgs.callPackage (../../poetry/poetry-py + "${pythonVersion}.nix") {
    inherit python;
    inherit pypkgs;
  };

  poetry-config = pkgs.writeTextFile {
    name = "poetry-config";
    text = "";
    destination = "/config.toml";
  };

  python3-wrapper = pythonWrapper {
    bin = "${python}/bin/python3";
    name = "python3";
    aliases = [
      "python"
      "python${pythonVersion}"
    ];
  };

  poetry-wrapper = pythonWrapper {
    bin = "${poetry}/bin/poetry";
    name = "poetry";
  };

  binary-wrapped-python = pkgs.callPackage ../../python-wrapped {
    inherit pkgs python python-ld-library-path;
  };

  sitecustomize = pkgs.callPackage ./sitecustomize.nix { };

  uv = pkgs.callPackage ./uv {
    rustPlatform = (
      let
        toolchain = pkgs.fenix.latest.toolchain;
      in
      pkgs.makeRustPlatform {
        cargo = toolchain;
        rustc = toolchain;
      }
    );
  };

in
{
  id = "python-${pythonVersion}";
  name = "Python Tools";
  displayVersion = pythonVersion;
  imports = [ (import ../ty) ];
  description = ''
    Development tools for Python. Includes Python interpreter, Pip, Poetry, the ty language server, and Ruff formatting.
  '';

  replit.packages = [
    binary-wrapped-python
    pip-wrapper
    poetry-wrapper
    uv
  ];

  replit.runners.python = {
    name = "Python ${pythonVersion}";
    displayVersion = python.version;
    fileParam = true;
    language = "python3";
    start = "${python3-wrapper}/bin/python3 $file";
    defaultEntrypoints = [
      "main.py"
      "app.py"
      "run.py"
    ];
  };

  replit.dev.packagers.upmPython = {
    name = "Python packager";
    language = "python3";
    ignoredPackages = [ "unit_tests" ];
    ignoredPaths = [ pylibs-dir ];
    features = {
      packageSearch = true;
      guessImports = true;
      enabledForHosting = false;
    };
  };

  replit.env = {
    POETRY_CONFIG_DIR = poetry-config.outPath;
    POETRY_CACHE_DIR = "$REPL_HOME/.cache/pypoetry";
    POETRY_VIRTUALENVS_CREATE = "0";
    POETRY_INSTALLER_MODERN_INSTALLATION = "1";
    POETRY_DOWNLOAD_WITH_CURL = "1";
    POETRY_PIP_USE_PIP_CACHE = "1";
    POETRY_PIP_NO_ISOLATE = "1";
    POETRY_PIP_NO_PREFIX = "1";
    POETRY_PIP_FROM_PATH = "1";
    POETRY_USE_USER_SITE = "1";
    PIP_CONFIG_FILE = pip.config.outPath;
    PYTHONUSERBASE = userbase;
    PYTHONPATH = "${sitecustomize}:${pip.pip}/${python.sitePackages}";
    REPLIT_PYTHONPATH = "${userbase}/${python.sitePackages}:${pypkgs.setuptools}/${python.sitePackages}";
    UV_PROJECT_ENVIRONMENT = "$REPL_HOME/.pythonlibs";
    UV_PYTHON_DOWNLOADS = "never";
    UV_PYTHON_PREFERENCE = "only-system";
    # Even though it is set-default in the wrapper, add it to the
    # environment too, so that when someone wants to override it,
    # they can keep the defaults if they want to.
    REPLIT_PYTHON_LD_LIBRARY_PATH = python-ld-library-path;
    PATH = "${userbase}/bin";
  };
}