From b1231eafb19c5151dc29609075430333f770edc1 Mon Sep 17 00:00:00 2001
From: Arthur Noel <arthur@ischium.net>
Date: Thu, 30 Nov 2023 16:45:20 +0000
Subject: [PATCH] shell: include treefmt wrapper

---
 flake.nix | 4 +++-
 shell.nix | 4 ++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/flake.nix b/flake.nix
index 10b84fa..6ff8480 100644
--- a/flake.nix
+++ b/flake.nix
@@ -28,7 +28,9 @@
               nixVersion = "unstable";
             };
           };
-          devShells.default = pkgs.callPackage ./shell.nix { };
+          devShells.default = pkgs.callPackage ./shell.nix {
+            packages = [ config.treefmt.build.wrapper ];
+          };
           apps.test-runner = {
             type = "app";
             program = "${config.packages.test-runner}";
diff --git a/shell.nix b/shell.nix
index 87e4645..d32920a 100644
--- a/shell.nix
+++ b/shell.nix
@@ -1,8 +1,8 @@
-{ pkgs ? import <nixpkgs> { } }:
+{ pkgs ? import <nixpkgs> { }, packages ? [ ] }:
 
 with pkgs;
 mkShell {
-  nativeBuildInputs = [
+  packages = packages ++ [
     python3.pkgs.pytest
     python3.pkgs.mypy
     ruff