introduce formatters

Kiskae · Sep 9, 2023 · 5042673 · 5042673
1 parent e3b0146
commit 5042673
Show file tree

Hide file tree

Showing 6 changed files with 39 additions and 365 deletions.
diff --git a/.flake8 b/.flake8
diff --git a/dev/flake.lock b/dev/flake.lock
diff --git a/dev/flake.nix b/dev/flake.nix
@@ -1,19 +1,38 @@
 {
   description = "nvidia.nix dev flake";
-  
+
   inputs = {
     flake.url = "path:../";
-    
+
     nixpkgs.url = "github:nixos/nixpkgs/nixos-unstable-small";
-    
+
     treefmt-nix.url = "github:numtide/treefmt-nix";
     treefmt-nix.inputs.nixpkgs.follows = "nixpkgs";
   };
 
-  outputs = { self, nixpkgs, flake-utils, ... }: {};
+  outputs = {
+    self,
+    nixpkgs,
+    systems,
+    treefmt-nix,
+    ...
+  }: let
+    # Small tool to iterate over each systems
+    eachSystem = f: nixpkgs.lib.genAttrs (import systems) (system: f nixpkgs.legacyPackages.${system});
+
+    # Eval the treefmt modules from ./treefmt.nix
+    treefmtEval = eachSystem (pkgs: treefmt-nix.lib.evalModule pkgs ./treefmt.nix);
+  in {
+    # for `nix fmt`
+    formatter = eachSystem (pkgs: treefmtEval.${pkgs.system}.config.build.wrapper);
+    # for `nix flake check`
+    checks = eachSystem (pkgs: {
+      formatting = treefmtEval.${pkgs.sytem}.config.build.check self;
+    });
+  };
 
   /*
-  
+
   outputs = {
     self,
     nixpkgs,
@@ -57,5 +76,5 @@
     // {
       formatter.x86_64-linux = alejandra.packages.x86_64-linux.default;
     };
-    */
-}
+  */
+}
diff --git a/dev/treefmt.nix b/dev/treefmt.nix
@@ -0,0 +1,9 @@
+{pkgs, ...}: {
+  projectRootFile = ".git/config";
+  programs.alejandra.enable = true;
+  programs.black.enable = true;
+  programs.isort = {
+    enable = true;
+    profile = "black";
+  };
+}
diff --git a/flake.nix b/flake.nix
@@ -1,5 +1,5 @@
 {
   description = "nvidia.nix flake";
 
-  outputs = { self }: {};
+  outputs = {self}: {};
 }