nix-config/modules/hosts/vps/disko.nix

{inputs, ...}: {
  flake.modules.nixos.vps = {
    config,
    lib,
    ...
  }: let
    hasOptinPersistence = config.tnix.boot.impermanence.enable;
    isLegacy = config.tnix.boot.legacy.enable;
  in {
    imports = [
      inputs.disko.nixosModules.disko
    ];

    disko.devices.disk.primary = {
      device = "/dev/sda";
      type = "disk";
      content = {
        type = "gpt";
        partitions =
          {
            ESP = {
              size = "1G";
              type = "EF00";
              content = {
                type = "filesystem";
                format = "vfat";
                mountpoint = "/boot";
                mountOptions = [
                  "defaults"
                  "umask=0077"
                ];
              };
            };
            root = {
              size = "100%";
              type = "8300";
              content = {
                type = "btrfs";
                # Base subvolumes that always exist
                subvolumes =
                  {
                    "/root" = {
                      mountOptions = [
                        "compress=zstd"
                        "noatime"
                        "space_cache=v2"
                      ];
                      mountpoint = "/";
                    };
                    "/nix" = {
                      mountOptions = [
                        "compress=zstd"
                        "noatime"
                        "noacl"
                        "space_cache=v2"
                      ];
                      mountpoint = "/nix";
                    };
                  }
                  # Conditionally merge /persist only when impermanence is enabled
                  // lib.optionalAttrs hasOptinPersistence {
                    "/persist" = {
                      mountOptions = [
                        "compress=zstd"
                        "noatime"
                        "space_cache=v2"
                      ];
                      mountpoint = "/persist";
                    };
                  };
              };
            };
          }
          // lib.optionalAttrs isLegacy {
            boot = {
              name = "boot";
              size = "1M";
              type = "EF02";
            };
          };
      };
    };
  };
}