{ config, lib, pkgs, ... }:

let
  inherit (lib) fetchFromGitHub mkDefault mkOption types versions;

  # Set the full kernel version and hashes
  version =
    if config.microsoft-surface.kernelVersion == "lts" then
      "6.12.19"
    else if config.microsoft-surface.kernelVersion == "latest" then
      "6.13.6"
    else
      abort "Invalid kernel version: ${config.microsoft-surface.kernelVersion}";
  
  shortVersion = versions.majorMinor version;
  packageHash =
    if shortVersion == "6.12" then
      "sha256-Pv7O8D8ma+MPLhYP3HSGQki+Yczp8b7d63qMb6l4+mY="
    else if shortVersion == "6.13" then
      "sha256-otD1ckNxNnvV8xipf9SZpbfg+bBq5EPwyieYtLIV4Ck="
    else
      abort "Invalid kernel version: ${shortVersion}";

  srcHash =
    if shortVersion == "6.12" then
      "sha256-1zvwV77ARDSxadG2FkGTb30Ml865I6KB8y413U3MZTE="
    else if shortVersion == "6.13" then
      "sha256-3gBTy0E9QI8g/R1XiCGZUbikQD5drBsdkDIJCTis0Zk="
    else
      abort "Invalid kernel version: ${shortVersion}";

  # Fetch the release from the linux-surface project
  rev = "arch-${version}-1";
  repos = pkgs.callPackage ({ fetchFromGitHub, rev, packageHash }: {
    linux-surface = fetchFromGitHub {
      owner = "linux-surface";
      repo = "linux-surface";
      rev = rev;
      hash = packageHash;
    };
  }) { inherit rev packageHash; };

  # Build the kernel package
  inherit (pkgs.callPackage ./kernel/linux-package.nix { inherit repos; }) linuxPackage surfacePatches;
  kernelPatches = surfacePatches {
    inherit version;
    patchFn = ./kernel/${shortVersion}/patches.nix;
  };
  kernelPackages = linuxPackage {
    inherit version kernelPatches;
    hash = srcHash;
    ignoreConfigErrors=true;
  };

in {
  options.microsoft-surface.kernelVersion = mkOption {
    description = "Kernel Version to use (patched for MS Surface)";
    type = types.enum [
      "lts"
      "latest"
    ];
    default = "lts";
  };

  config = {
    boot = {
      inherit kernelPackages;

      # Seems to be required to properly enable S0ix "Modern Standby":
      kernelParams = mkDefault [ "mem_sleep_default=deep" ];
    };

    # NOTE: Check the README before enabling TLP:
    services.tlp.enable = mkDefault false;

    # Needed for wifi firmware, see https://github.com/NixOS/nixos-hardware/issues/364
    hardware = {
      enableRedistributableFirmware = mkDefault true;
      sensor.iio.enable = mkDefault true;
    };
  };
}