hiro-flake-utils/flake.nix

{
  inputs = {
    nixpkgs.url = "nixpkgs/nixos-unstable";
    flake-utils.url = "github:numtide/flake-utils";
    poetry2nix.url = "github:nix-community/poetry2nix";
  };

  outputs = { self, nixpkgs, flake-utils, poetry2nix }: {
    lib = rec {
      currentDefaultPackage = flake: system: flake.defaultPackage.${system};
      makeDefaultPackageOverrides =
        (flakes: system:
          let
            body = (nixpkgs.lib.attrsets.mapAttrs'
              (name: flake:
                {
                  name = nixpkgs.lib.strings.toLower name;
                  value = (currentDefaultPackage flake system);
                })
              flakes);
          in
          (self: super:
            body
          ));

      makeAddCythonOverrides =
        (flakes:
          (builtins.map
            (name:
              (self: super: {
                ${name} = super.${name}.overridePythonAttrs (
                  old: {
                    buildInputs = (old.buildInputs or [ ]) ++ [
                      self.cython
                    ];
                  }
                );
              })
            )
            flakes
          ));

      poetry2nixWrapper = nixpkgs: pythonInputs:
        { name
        , poetryArgs ? { }
        , buildInputs ? _: [ ]
        , nixpkgsConfig ? { }
        , addCythonTo ? [ ]
        }:
        (flake-utils.lib.eachDefaultSystem (system:
          let
            overlay = nixpkgs.lib.composeManyExtensions [
              poetry2nix.overlay

              (final: prev:
                let overrides = (nixpkgs.lib.composeManyExtensions [
                  (prev.poetry2nix.overrides.withDefaults
                    (makeDefaultPackageOverrides pythonInputs system))
                    ] ++ (makeAddCythonOverrides addCythonTo));
                in
                {
                  ${name} = (prev.poetry2nix.mkPoetryApplication ({
                    preferWheels = true;
                    overrides = overrides;
                  } // poetryArgs));

                  "${name}Shell" = (prev.poetry2nix.mkPoetryEnv ({
                    overrides = overrides;
                    preferWheels = true;
                    editablePackageSources = {
                      ${name} = poetryArgs.projectDir + "/${name}";
                    };
                  } // poetryArgs));
                })

            ];
            pkgs = import nixpkgs {
              inherit system;
              overlays = [ overlay ];
              config = nixpkgsConfig;
            };
          in
          rec {
            packages = {
              ${name} = pkgs.${name};
            };

            defaultPackage = packages.${name};
            devShell = pkgs."${name}Shell".env.overrideAttrs (oldAttrs: {
              buildInputs = (buildInputs pkgs) ++ [ pkgs.poetry ];
            });
          }));
    };
  };
}