republisher/shell.nix

{
  system ? "x86_64-linux",
  pkgs ? import <nixpkgs> { inherit system; },
  dev ? true,
}:

let
  pyCurrent = pkgs.python311;
  poetryExtras = if dev then [ "dev" ] else [ ];
  poetryInstallExtras = (
    if poetryExtras == [ ] then
      ""
    else
      pkgs.lib.concatStrings [
        " --with="
        (pkgs.lib.concatStringsSep "," poetryExtras)
      ]
  );
  packages = [
    pkgs.ffmpeg_5-headless
    #(pyCurrent (ps: with ps; [ ffmpeg-python ]))
    pkgs.zsh
    (pkgs.poetry.withPlugins (ps: with ps; [ poetry-plugin-up ]))
  ];

  LD_LIBRARY_PATH = pkgs.lib.makeLibraryPath [
    pkgs.stdenv.cc.cc
    # Add any missing library needed
    # You can use the nix-index package to locate them, e.g. nix-locate -w --top-level --at-root /lib/libudev.so.1
  ];

  # Put the venv on the repo, so direnv can access it
  POETRY_VIRTUALENVS_IN_PROJECT = "true";
  POETRY_VIRTUALENVS_PATH = "{project-dir}/.venv";

  # Use python from path, so you can use a different version to the one bundled with poetry
  POETRY_VIRTUALENVS_PREFER_ACTIVE_PYTHON = "true";
in
pkgs.mkShell {
  buildInputs = packages;
  shellHook = ''
    export SHELL=${pkgs.zsh}
    export LD_LIBRARY_PATH="${LD_LIBRARY_PATH}"
    export POETRY_VIRTUALENVS_IN_PROJECT="${POETRY_VIRTUALENVS_IN_PROJECT}"
    export POETRY_VIRTUALENVS_PATH="${POETRY_VIRTUALENVS_PATH}"
    export POETRY_VIRTUALENVS_PREFER_ACTIVE_PYTHON="${POETRY_VIRTUALENVS_PREFER_ACTIVE_PYTHON}"
    export PYTHON_KEYRING_BACKEND=keyring.backends.null.Keyring
    poetry env use "${pyCurrent}/bin/python"
    poetry install -vv --sync${poetryInstallExtras}
  '';
}