rust: Add treefmt-nix

akirak · Oct 8, 2023 · 4196d9c · 4196d9c
1 parent f9a131b
commit 4196d9c
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/rust/flake.nix b/rust/flake.nix
@@ -2,11 +2,17 @@
   inputs = {
     rust-overlay.url = "github:oxalica/rust-overlay";
     systems.url = "github:nix-systems/default";
+    treefmt-nix = {
+      url = "github:numtide/treefmt-nix";
+      inputs.nixpkgs.url = "nixpkgs";
+    };
   };
 
   outputs = {
+    self,
     systems,
     nixpkgs,
+    treefmt-nix,
     ...
   } @ inputs: let
     eachSystem = f:
@@ -17,6 +23,8 @@
             overlays = [inputs.rust-overlay.overlays.default];
           })
       );
+
+    treefmtEval = eachSystem (pkgs: treefmt-nix.lib.evalModule pkgs ./treefmt.nix);
   in {
     devShells = eachSystem (pkgs: {
       default = pkgs.mkShell {
@@ -28,5 +36,11 @@
         ];
       };
     });
+
+    formatter = eachSystem (pkgs: treefmtEval.${pkgs.system}.config.build.wrapper);
+
+    checks = eachSystem (pkgs: {
+      formatting = treefmtEval.${pkgs.system}.config.build.check self;
+    });
   };
 }
diff --git a/rust/treefmt.nix b/rust/treefmt.nix
@@ -0,0 +1,6 @@
+{
+  projectRootFile = "treefmt.nix";
+
+  programs.alejandra.enable = true;
+  programs.rustfmt.enable = true;
+}