core/pkgs/by-name/py/python/manylinux/default.nix

{ lib, pkgs }:

let
  # Create a derivation that links all desired manylinux libraries
  createManyLinuxPackage =
    name: libs:
    let
      drvs = lib.unique (lib.attrValues libs);
      names = lib.attrNames libs;
    in
    pkgs.runCommand name { buildInputs = drvs; } ''
      mkdir -p $out/lib
      num_found=0

      IFS=:
      export DESIRED_LIBRARIES=${lib.concatStringsSep ":" names}
      export LIBRARY_PATH=${lib.makeLibraryPath drvs}
      for desired in $DESIRED_LIBRARIES; do
        for path in $LIBRARY_PATH; do
          if [ -e $path/$desired ]; then
            echo "FOUND $path/$desired"
            ln -s $path/$desired $out/lib/$desired
            num_found=$((num_found+1))
            break
          fi
        done
      done

      num_desired=${toString (lib.length names)}
      echo "Found $num_found of $num_desired libraries"
      if [ "$num_found" -ne "$num_desired" ]; then
        echo "Error: not all desired libraries were found"
        exit 1
      fi
    '';

  getLibOutputs = lib.mapAttrs (k: v: lib.getLib v);

  # https://www.python.org/dev/peps/pep-0599/
  manylinux2014Libs = getLibOutputs (
    with pkgs;
    {
      "libgcc_s.so.1" = glibc;
      "libstdc++.so.6" = stdenv.cc.cc;
      "libm.so.6" = glibc;
      "libdl.so.2" = glibc;
      "librt.so.1" = glibc;
      "libc.so.6" = glibc;
      "libnsl.so.1" = glibc;
      "libutil.so.1" = glibc;
      "libpthread.so.0" = glibc;
      "libresolv.so.2" = glibc;
      "libX11.so.6" = xorg.libX11;
      "libXext.so.6" = xorg.libXext;
      "libXrender.so.1" = xorg.libXrender;
      "libICE.so.6" = xorg.libICE;
      "libSM.so.6" = xorg.libSM;
      "libGL.so.1" = libGL;
      "libgobject-2.0.so.0" = glib;
      "libgthread-2.0.so.0" = glib;
      "libglib-2.0.so.0" = glib;
      "libz.so.1" = zlib;
      "libexpat.so.1" = expat;
    }
  );

  # https://www.python.org/dev/peps/pep-0571/
  manylinux2010Libs = manylinux2014Libs;

  # https://www.python.org/dev/peps/pep-0513/
  manylinux1Libs = getLibOutputs (
    manylinux2010Libs
    // (with pkgs; {
      "libpanelw.so.5" = ncurses5;
      "libncursesw.so.5" = ncurses5;
      "libcrypt.so.1" = libxcrypt;
    })
  );

in
{
  # List of libraries that are needed for manylinux compatibility.
  # When using a wheel that is manylinux1 compatible, just extend
  # the `buildInputs` with one of these `manylinux` lists.
  # Additionally, add `autoPatchelfHook` to `nativeBuildInputs`.
  manylinux1 = lib.unique (lib.attrValues manylinux1Libs);
  manylinux2010 = lib.unique (lib.attrValues manylinux2010Libs);
  manylinux2014 = lib.unique (lib.attrValues manylinux2014Libs);

  # These are symlink trees to the relevant libs and are typically not needed
  # These exist so as to quickly test whether all required libraries are provided
  # by the mapped packages.
  manylinux1Package = createManyLinuxPackage "manylinux1" manylinux1Libs;
  manylinux2010Package = createManyLinuxPackage "manylinux2010" manylinux2010Libs;
  manylinux2014Package = createManyLinuxPackage "manylinux2014" manylinux2014Libs;
}