Refactor to allow overriding for customizing kernel

README.md

@@ -109,7 +109,7 @@ If you want to construct your own `linuxPackages` attrset with `linuxKernel.pack
         { pkgs, ... }:
         {
           nixpkgs.overlays = [ self.overlay ];
-          boot.kernelPackages = pkgs.cachyosKernels.linuxPackages-cachyos-latest;
+          boot.kernelPackages = pkgs.linuxKernel.packagesFor pkgs.cachyosKernels.linux-cachyos-latest;
 
           # ZFS config
           boot.supportedFilesystems.zfs = true;
@@ -124,3 +124,31 @@ If you want to construct your own `linuxPackages` attrset with `linuxKernel.pack
   };
 }
 ```
+
+## How to apply CachyOS patches on your own kernel
+
+The kernels provided in this flake can be overridden to use your own kernel source. This is helpful if you want to use a kernel version not available in Nixpkgs.
+
+```nix
+{
+  kernel = pkgs.cachyosKernels.linux-cachyos-latest.override {
+    pname = "linux-cachyos-with-custom-source";
+    version = "6.12.34";
+    src = pkgs.fetchurl {
+      # ...
+    };
+    # Additional args are available. See kernel-cachyos/mkCachyKernel.nix
+  };
+
+  # For non-LTO kernels
+  kernelPackages = pkgs.linuxKernel.packagesFor kernel;
+
+  # helpers.nix provides a few utilities for building kernel with LTO.
+  # I haven't figured out a clean way to expose it in flakes.
+  helpers = pkgs.callPackage "${inputs.nix-cachyos-kernel.outPath}/helpers.nix" {};
+
+  # For LTO kernels, helpers.kernelModuleLLVMOverride fixes compilation for some
+  # out-of-tree modules in nixpkgs.
+  kernelPackagesWithLTOFix = helpers.kernelModuleLLVMOverride (pkgs.linuxKernel.packagesFor kernel);
+}
+```

flake.nix

@@ -28,17 +28,21 @@
         loadPackages =
           pkgs:
           let
-            kernels =
+            load =
+              path:
               lib.removeAttrs
-                (pkgs.callPackage ./kernel-cachyos {
+                (pkgs.callPackage path {
                   inherit inputs;
                 })
                 [
                   "override"
                   "overrideDerivation"
                 ];
+            kernels = load ./kernel-cachyos;
+            packages = load ./kernel-cachyos/packages.nix;
           in
           kernels
+          // packages
           // {
             zfs-cachyos = pkgs.callPackage ./zfs-cachyos {
               inherit inputs;

helpers.nix

@@ -77,5 +77,4 @@ rec {
           v
       ) prev
     );
-
 }

kernel-cachyos/default.nix

@@ -8,39 +8,32 @@
 }:
 let
   mkCachyKernel = callPackage ./mkCachyKernel.nix { inherit inputs; };
-
+in
-  batch =
+builtins.listToAttrs (
-    {
+  builtins.map (v: lib.nameValuePair v.pname v) [
-      pnameSuffix,
-      version,
-      src,
-      configVariant,
-      ...
-    }:
-    [
     (mkCachyKernel {
-        pnameSuffix = "${pnameSuffix}";
+      pname = "linux-cachyos-latest";
-        inherit version src configVariant;
+      inherit (linux_latest) version src;
+      configVariant = "linux-cachyos";
       lto = false;
     })
     (mkCachyKernel {
-        pnameSuffix = "${pnameSuffix}-lto";
+      pname = "linux-cachyos-latest-lto";
-        inherit version src configVariant;
-        lto = true;
-      })
-    ];
-
-  batches = [
-    (batch {
-      pnameSuffix = "latest";
       inherit (linux_latest) version src;
       configVariant = "linux-cachyos";
+      lto = true;
     })
-    (batch {
+    (mkCachyKernel {
-      pnameSuffix = "lts";
+      pname = "linux-cachyos-lts";
       inherit (linux) version src;
       configVariant = "linux-cachyos-lts";
+      lto = false;
     })
-  ];
+    (mkCachyKernel {
-in
+      pname = "linux-cachyos-lts-lto";
-builtins.listToAttrs (lib.flatten batches)
+      inherit (linux) version src;
+      configVariant = "linux-cachyos-lts";
+      lto = true;
+    })
+  ]
+)

kernel-cachyos/mkCachyKernel.nix

@@ -5,40 +5,57 @@
   buildLinux,
   stdenv,
   kernelPatches,
-  linuxKernel,
   ...
 }:
+lib.makeOverridable (
   {
-  pnameSuffix,
+    pname,
     version,
     src,
+
+    # Kernel config variant to be used as defconfig, e.g. "linux-cachyos-lts".
+    # See https://github.com/CachyOS/linux-cachyos for available values.
     configVariant,
+
+    # Set to true to enable Clang+ThinLTO.
     lto,
-}:
+
+    # Patches to be applied in patchedSrc phase. This is different from buildLinux's kernelPatches.
+    prePatch ? "",
+    patches ? [ ],
+    postPatch ? "",
+
+    # See nixpkgs/pkgs/os-specific/linux/kernel/generic.nix for additional options.
+    # Additional args are passed to buildLinux.
+    ...
+  }@args:
   let
     helpers = callPackage ../helpers.nix { };
-  inherit (helpers) stdenvLLVM ltoMakeflags kernelModuleLLVMOverride;
+    inherit (helpers) stdenvLLVM ltoMakeflags;
 
     splitted = lib.splitString "-" version;
     ver0 = builtins.elemAt splitted 0;
     major = lib.versions.pad 2 ver0;
 
     cachyosConfigFile = "${inputs.cachyos-kernel.outPath}/${configVariant}/config";
+    cachyosPatch = "${inputs.cachyos-kernel-patches.outPath}/${major}/all/0001-cachyos-base-all.patch";
 
     # buildLinux doesn't accept postPatch, so adding config file early here
     patchedSrc = stdenv.mkDerivation {
-    pname = "linux-cachyos-${pnameSuffix}-src";
+      pname = "${pname}-src";
-    inherit version src;
+      inherit version src prePatch;
       patches = [
         kernelPatches.bridge_stp_helper.patch
         kernelPatches.request_key_helper.patch
-      "${inputs.cachyos-kernel-patches.outPath}/${major}/all/0001-cachyos-base-all.patch"
+        cachyosPatch
-    ];
+      ]
+      ++ patches;
       postPatch = ''
         for DIR in arch/*/configs; do
           install -Dm644 ${cachyosConfigFile} $DIR/cachyos_defconfig
         done
-    '';
+      ''
+      + postPatch;
       dontConfigure = true;
       dontBuild = true;
       dontFixup = true;
@@ -47,19 +64,29 @@ let
         cp -r * $out/
       '';
     };
-
+  in
-  kernelPackage = buildLinux {
+  buildLinux (
-    pname = "linux-cachyos-${pnameSuffix}";
+    (lib.removeAttrs args [
-    inherit version;
+      "pname"
+      "version"
+      "src"
+      "configVariant"
+      "lto"
+      "prePatch"
+      "patches"
+      "postPatch"
+    ])
+    // {
+      inherit pname version;
       src = patchedSrc;
-    stdenv = if lto then stdenvLLVM else stdenv;
+      stdenv = args.stdenv or (if lto then stdenvLLVM else stdenv);
 
-    extraMakeFlags = lib.optionals lto ltoMakeflags;
+      extraMakeFlags = (lib.optionals lto ltoMakeflags) ++ (args.extraMakeFlags or [ ]);
 
-    defconfig = "cachyos_defconfig";
+      defconfig = args.defconfig or "cachyos_defconfig";
 
       # Clang has some incompatibilities with NixOS's default kernel config
-    ignoreConfigErrors = lto;
+      ignoreConfigErrors = args.ignoreConfigErrors or lto;
 
       structuredExtraConfig =
         with lib.kernel;
@@ -76,31 +103,22 @@ let
             OVERLAY_FS_METACOPY = no;
             OVERLAY_FS_DEBUG = no;
           }
-        // lib.optionalAttrs lto {
+          // (lib.optionalAttrs lto {
             LTO_NONE = no;
             LTO_CLANG_THIN = yes;
-        }
+          })
+          // (args.structuredExtraConfig or { })
         );
 
       extraMeta = {
         description = "Linux CachyOS Kernel" + lib.optionalString lto " with Clang+ThinLTO";
-    };
+      }
-  };
+      // (args.extraMeta or { });
 
-  zfsPackage = callPackage ../zfs-cachyos {
+      extraPassthru = {
-    inherit inputs;
+        inherit cachyosConfigFile cachyosPatch;
-    kernel = kernelPackage;
+      }
-  };
+      // (args.extraPassthru or { });
-in
-[
-  (lib.nameValuePair "linux-cachyos-${pnameSuffix}" kernelPackage)
-  (lib.nameValuePair "linuxPackages-cachyos-${pnameSuffix}" (
-    kernelModuleLLVMOverride (
-      (linuxKernel.packagesFor kernelPackage).extend (
-        final: prev: {
-          zfs_cachyos = zfsPackage;
     }
   )
 )
-  ))
-]

kernel-cachyos/packages.nix

@@ -1,14 +1,31 @@
 {
-  mode ? null,
+  inputs,
   callPackage,
   lib,
   linuxKernel,
-  sources,
   ...
 }:
 let
-  kernels = callPackage ./default.nix { inherit mode sources; };
+  helpers = callPackage ../helpers.nix { };
+  inherit (helpers) kernelModuleLLVMOverride;
+
+  kernels = lib.filterAttrs (_: lib.isDerivation) (callPackage ./. { inherit inputs; });
 in
-lib.mapAttrs (n: v: linuxKernel.packagesFor v) (
+lib.mapAttrs' (
-  lib.filterAttrs (n: nv: !lib.hasSuffix "configfile" n) kernels
+  n: v:
+  let
+    zfsPackage = callPackage ../zfs-cachyos {
+      inherit inputs;
+      kernel = v;
+    };
+
+    packages = kernelModuleLLVMOverride (
+      (linuxKernel.packagesFor v).extend (
+        final: prev: {
+          zfs_cachyos = zfsPackage;
+        }
       )
+    );
+  in
+  lib.nameValuePair "linuxPackages-${lib.removePrefix "linux-" n}" packages
+) kernels