Move programming related stuff to a module

modules/nixos/environment/default.nix

@@ -5,14 +5,13 @@
 }: {
   imports = [
     ./gui
+    ./programming
 
     ./packages.nix
     ./cachix.nix
     ./fonts.nix
     ./latex.nix
     ./ld.nix
-    ./programming.nix
-    ./python.nix
 
     ./amd
     ./intel.nix
@@ -21,6 +20,7 @@
   cachix.enable = lib.mkDefault false;
   gui.enable = lib.mkDefault true;
   latex.enable = lib.mkDefault false;
+  programming.enable = lib.mkDefault true;
 
   amd.enable = lib.mkDefault false;
   intel.enable = lib.mkDefault false;

modules/nixos/environment/programming/default.nix

@@ -0,0 +1,35 @@
+{
+  pkgs,
+  lib,
+  config,
+  ...
+}: {
+  imports = [
+    ./python.nix
+  ];
+
+  options.programming.enable = lib.mkEnableOption "Setup programming tools";
+
+  config = lib.mkIf config.programming.enable {
+    python.enable = lib.mkDefault true;
+
+    environment = {
+      systemPackages = with pkgs; [
+        gfortran
+        gcc
+        gcc-unwrapped
+        gnumake
+        julia-bin
+        #unstable.ruff
+        #ruff
+
+        # rust
+        cargo
+        rustc
+
+        # misc
+        hyperfine
+      ];
+    };
+  };
+}

modules/nixos/environment/programming/python.nix

@@ -0,0 +1,49 @@
+{
+  pkgs,
+  lib,
+  config,
+  ...
+}: let
+  packages = ps:
+    with ps; [
+      # Jupyter
+      jupyter
+      ipython
+      python-lsp-server
+      # Must
+      jax
+      matplotlib
+      numba
+      numba-scipy
+      numpy
+      pandas
+      plotly
+      streamlit
+      # Misc
+      manim
+      psutil
+      # Optional
+      # opencv4
+      # pillow
+      # requests
+      # scikit-image
+    ];
+in {
+  options.python.enable = lib.mkEnableOption "Setup a python environment";
+
+  config = lib.mkIf config.python.enable {
+    environment = {
+      systemPackages = with pkgs; [
+        (python3.withPackages packages)
+        manim
+        manim-slides
+        mkdocs
+        poetry
+      ];
+
+      sessionVariables = {
+        PYTHON_KEYRING_BACKEND = "keyring.backends.null.Keyring";
+      };
+    };
+  };
+}