Add pre-commit hook for nix users that does auto formatting + check with ruff and other improvements.

.envrc

@@ -1 +1,3 @@
+watch_dir nix
+
 use flake

.gitignore

@@ -3,6 +3,7 @@
 .flatpak-builder/
 .ignore/
 .direnv/
+/.pre-commit-config.yaml
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

flake.nix

@@ -3,6 +3,7 @@
 
   inputs = {
     nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
+    git-hooks.url = "github:cachix/git-hooks.nix";
   };
 
   outputs =
@@ -12,6 +13,7 @@
       ...
     }:
     let
+      inherit (self) inputs;
       systems = [
         "x86_64-linux"
         "aarch64-linux"
@@ -28,7 +30,7 @@
         in
         {
           default = pkgs.callPackage ./nix/package.nix { };
-          anifetch = self.packages.default;
+          anifetch = self.packages.${system}.default;
         }
       );
 
@@ -39,17 +41,97 @@
         anifetch = self.overlays.default;
       };
 
-      formatter = forAllSystems (system: nixpkgs.legacyPackages.${system}.nixfmt-tree);
+      formatter = forAllSystems (
+        system:
+        let
+          pkgs = nixpkgs.legacyPackages.${system};
+          config = self.checks.${system}.pre-commit-check.config;
+          inherit (config) package configFile;
+          script = ''
+            ${pkgs.lib.getExe package} run --all-files --config ${configFile}
+          '';
+        in
+        pkgs.writeShellScriptBin "pre-commit-run" script
+      );
+
+      checks = forAllSystems (
+        system:
+        let
+          pkgs = import nixpkgs { inherit system; };
+        in
+        {
+          pre-commit-check = inputs.git-hooks.lib.${system}.run {
+            src = ./.;
+            hooks = {
+              nixfmt.enable = true;
+
+              ruff-check = {
+                enable = true;
+                entry = "${pkgs.lib.getExe pkgs.ruff}";
+                args = [
+                  "check"
+                  "--fix"
+                ];
+                types = [
+                  "file"
+                  "python"
+                ];
+              };
+              ruff-format = {
+                enable = true;
+                entry = "${pkgs.lib.getExe pkgs.ruff}";
+                args = [
+                  "format"
+                ];
+                types = [
+                  "file"
+                  "python"
+                ];
+              };
+            };
+
+            package = pkgs.prek;
+          };
+        }
+      );
 
-      devShell = forAllSystems (
+      devShells = forAllSystems (
         system:
         let
           pkgs = import nixpkgs { inherit system; };
+          myPython = pkgs.python3;
+          pythonWithPkgs = myPython.withPackages (ps: [
+            ps.pip
+            ps.setuptools
+          ]);
+
+          venv = "venv";
+          inherit (self.checks.${system}.pre-commit-check) shellHook enabledPackages;
         in
-        pkgs.mkShell {
-          packages = [
-            self.packages.${pkgs.stdenv.hostPlatform.system}.default
-          ];
+        {
+          default = pkgs.mkShell {
+            nativeBuildInputs = [
+              pythonWithPkgs
+              pkgs.bc
+              pkgs.chafa
+              pkgs.ffmpeg
+            ]
+            ++ enabledPackages;
+            shellHook = shellHook + ''
+              export "CPATH=${pkgs.linuxHeaders}/include:$CPATH"
+              if [ ! -d "${venv}" ]; then
+                echo "Creating Python venv..."
+                python3 -m venv ${venv}
+              fi
+              echo "Activating venv..."
+              source ${venv}/bin/activate
+              if ! pip show anifetch &>/dev/null; then
+                echo "Aniftech not installed: Install anifetch..."
+                pip install -e .
+              fi
+              echo "Venv activated."
+            '';
+          };
         }
       );
     };

nix/package.nix

@@ -8,10 +8,11 @@
 let
   fs = lib.fileset;
   sourceFiles = ../.;
+  anifetchPyprojectToml = fromTOML (builtins.readFile ../pyproject.toml);
 in
 fs.trace sourceFiles python3Packages.buildPythonApplication {
-  name = "anifetch-wrapped";
-  version = "git";
+  pname = "anifetch";
+  version = "${anifetchPyprojectToml.project.version}-git";
   pyproject = true;
   src = fs.toSource {
     root = ../.;

ruff.sh

@@ -1,4 +1,4 @@
-#!/bin/bash
+#!/usr/bin/env bash
 
 ruff check . --fix
-ruff format .
+ruff format .