1 files changed, 22 insertions, 17 deletions

diff --git a/overlays/postfix-mta-sts-resolver/default.nix b/overlays/postfix-mta-sts-resolver/default.nix
index c210cf79..919992ff 100644
--- a/overlays/postfix-mta-sts-resolver/default.nix
+++ b/overlays/postfix-mta-sts-resolver/default.nix
@@ -1,21 +1,26 @@
-{ prev, sources, ... }:
+{ prev, final, flake, flakeInputs, sources, ... }:
 {
   postfix-mta-sts-resolver =
-    with prev.poetry2nix;
-    mkPoetryApplication {
-      python = prev.python311;
-      inherit (sources.postfix-mta-sts-resolver) pname version;
-      overrides = overrides.withDefaults (self: super: {
-        pyyaml = super.pyyaml.override { preferWheel = true; };
-      });
-      projectDir = cleanPythonSources {
-        src = prev.runCommand "sources" {} ''
-          mkdir $out
-          cp -r --no-preserve=all ${sources.postfix-mta-sts-resolver.src}/. $out
-
-          cp ${./pyproject.toml} $out/pyproject.toml
-          cp ${./poetry.lock} $out/poetry.lock
-        '';
+    let
+      workspace = flakeInputs.uv2nix.lib.workspace.loadWorkspace { workspaceRoot = prev.runCommand "sources" {} ''
+        mkdir $out
+        cp -r --no-preserve=all \
+          ${sources.postfix-mta-sts-resolver.src}/. \
+          ${prev.lib.sources.sourceByRegex ./. ["^pyproject\.toml$" "^uv\.lock$"]}/. \
+          $out
+      '';
       };
-    };
+      pythonSet = flake.lib.pythonSet {
+        pkgs = final;
+        python = final.python312;
+        overlay = workspace.mkPyprojectOverlay {
+          sourcePreference = "wheel";
+        };
+      };
+      virtualEnv = pythonSet.mkVirtualEnv "postfix-mta-sts-resolver-env" workspace.deps.default;
+    in virtualEnv.overrideAttrs (oldAttrs: {
+      meta = (oldAttrs.meta or {}) // {
+        mainProgram = "mta-sts-daemon";
+      };
+    });
 }