From e29d8f09d431538cd801e608f75ac26bdb318f86 Mon Sep 17 00:00:00 2001
From: Jinser Kafka <aimer@purejs.icu>
Date: Mon, 23 Feb 2026 16:04:44 +0800
Subject: [PATCH] feat: add nix devShell support

Signed-off-by: Jinser Kafka <aimer@purejs.icu>
---
 flake.nix | 107 +++++++++++++++++++++++++++++++++++++++---------------
 1 file changed, 78 insertions(+), 29 deletions(-)

diff --git a/flake.nix b/flake.nix
index 4fff6a919..f8e6b5ce7 100644
--- a/flake.nix
+++ b/flake.nix
@@ -47,44 +47,48 @@
       packages = forAllSystems (
         { pkgs, ... }:
         let
+          inherit (pkgs)
+            lib
+            python313
+            callPackage
+            ;
+          python = python313;
+          pyproject = lib.importTOML ./pyproject.toml;
+          workspace = uv2nix.lib.workspace.loadWorkspace { workspaceRoot = ./.; };
+
+          overlay = workspace.mkPyprojectOverlay {
+            sourcePreference = "wheel";
+          };
+          extraBuildOverlay = final: prev: {
+            # Add setuptools build dependency for ripgrepy
+            ripgrepy = prev.ripgrepy.overrideAttrs (old: {
+              nativeBuildInputs = (old.nativeBuildInputs or [ ]) ++ [ final.setuptools ];
+            });
+            # Replace README symlink with real file for Nix builds.
+            "kimi-code" = prev."kimi-code".overrideAttrs (old: {
+              postPatch = (old.postPatch or "") + ''
+                rm -f README.md
+                cp ${./README.md} README.md
+              '';
+            });
+          };
+          pythonSet = (callPackage pyproject-nix.build.packages { inherit python; }).overrideScope (
+            lib.composeManyExtensions [
+              pyproject-build-systems.overlays.wheel
+              overlay
+              extraBuildOverlay
+            ]
+          );
+
           kimi-cli =
             let
               inherit (pkgs)
                 lib
-                callPackage
-                python313
-                runCommand
                 ripgrep
                 stdenvNoCC
                 makeWrapper
                 versionCheckHook
                 ;
-              python = python313;
-              pyproject = lib.importTOML ./pyproject.toml;
-              workspace = uv2nix.lib.workspace.loadWorkspace { workspaceRoot = ./.; };
-              overlay = workspace.mkPyprojectOverlay {
-                sourcePreference = "wheel";
-              };
-              extraBuildOverlay = final: prev: {
-                # Add setuptools build dependency for ripgrepy
-                ripgrepy = prev.ripgrepy.overrideAttrs (old: {
-                  nativeBuildInputs = (old.nativeBuildInputs or [ ]) ++ [ final.setuptools ];
-                });
-                # Replace README symlink with real file for Nix builds.
-                "kimi-code" = prev."kimi-code".overrideAttrs (old: {
-                  postPatch = (old.postPatch or "") + ''
-                    rm -f README.md
-                    cp ${./README.md} README.md
-                  '';
-                });
-              };
-              pythonSet = (callPackage pyproject-nix.build.packages { inherit python; }).overrideScope (
-                lib.composeManyExtensions [
-                  pyproject-build-systems.overlays.wheel
-                  overlay
-                  extraBuildOverlay
-                ]
-              );
               kimiCliPackage = pythonSet.mkVirtualEnv "kimi-cli-virtual-env-${pyproject.project.version}" workspace.deps.default;
             in
             stdenvNoCC.mkDerivation ({
@@ -123,12 +127,57 @@
                 mainProgram = "kimi";
               };
             });
+
+          devEnv =
+            let
+              editableOverlay = workspace.mkEditablePyprojectOverlay {
+                root = "$REPO_ROOT";
+              };
+              editablePythonSet = pythonSet.overrideScope editableOverlay;
+            in
+            {
+              devVirtualEnv = editablePythonSet.mkVirtualEnv "kimi-cli-dev-virtual-env-${pyproject.project.version}" workspace.deps.all;
+              devPythonInterpreter = editablePythonSet.python.interpreter;
+            };
         in
         {
           inherit kimi-cli;
+          inherit (devEnv) devVirtualEnv devPythonInterpreter;
           default = kimi-cli;
         }
       );
       formatter = forAllSystems ({ pkgs, ... }: pkgs.nixfmt-tree);
+      devShells = forAllSystems (
+        { pkgs, system, ... }:
+        let
+          venv = self.packages.${system}.devVirtualEnv;
+          python = self.packages.${system}.devPythonInterpreter;
+        in
+        {
+          default = pkgs.mkShell {
+            packages = [
+              venv
+            ]
+            ++ (with pkgs; [
+              uv
+              prek
+              nodejs
+              biome
+            ]);
+
+            env = {
+              UV_NO_SYNC = "1";
+              UV_PYTHON = python;
+              UV_PROJECT_ENVIRONMENT = venv;
+              UV_PYTHON_DOWNLOADS = "never";
+            };
+
+            shellHook = ''
+              unset PYTHONPATH
+              export REPO_ROOT=$(git rev-parse --show-toplevel)
+            '';
+          };
+        }
+      );
     };
 }