infra/nix/os/modules/opinionatedDisk.nix

{ lib, config, ... }:
with lib;

let
  cfg = config.hardware.opinionatedDisk;
  ownLib = import ../lib/default.nix { };
in {
  options.hardware.opinionatedDisk = {
    enable = mkEnableOption "Enable opinionated filesystem layout";
    diskId = mkOption { type = types.str; };
    encrypted = mkOption {
      default = true;
      type = types.bool;
    };
  };

  config = lib.mkIf cfg.enable {
    fileSystems."/boot" = {
      device = (ownLib.disk.bootFsDevice cfg.diskId);
      fsType = "vfat";
    };

    fileSystems."/" = {
      device = (ownLib.disk.rootFsDevice cfg.diskId);
      fsType = "btrfs";
      options = [ "subvol=nixos" ];
    };

    fileSystems."/home" = {
      device = (ownLib.disk.rootFsDevice cfg.diskId);
      fsType = "btrfs";
      options = [ "subvol=home" ];
    };

    swapDevices = [{ device = (ownLib.disk.swapFsDevice cfg.diskId); }];

    boot.loader.grub = {
      device = (ownLib.disk.bootGrubDevice cfg.diskId);
      enableCryptodisk = cfg.encrypted;
    };

    boot.initrd.luks.devices = lib.optionalAttrs cfg.encrypted
      (builtins.listToAttrs [{
        name = let
          splitstring =
            builtins.split "/" (ownLib.disk.bootLuksDevice cfg.diskId);
          lastelem = (builtins.length splitstring) - 1;
        in builtins.elemAt splitstring lastelem;
        value = {
          device = (ownLib.disk.bootLuksDevice cfg.diskId);
          preLVM = true;
          allowDiscards = true;
        };
      }]);
  };
}