nixpkgs/nixos/modules/hardware/usb-storage.nix

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

{
  options.hardware.usbStorage.manageShutdown = lib.mkOption {
    type = lib.types.bool;
    default = false;
    description = ''
      Enable this option to gracefully spin-down external storage during shutdown.
      If you suspect improper head parking after poweroff, install `smartmontools` and check
      for the `Power-Off_Retract_Count` field for an increment.
    '';
  };

  config = lib.mkIf config.hardware.usbStorage.manageShutdown {
    services.udev.extraRules = ''
      ACTION=="add|change", SUBSYSTEM=="scsi_disk", DRIVERS=="usb-storage|uas", ATTR{manage_shutdown}="1"
    '';
  };

  imports = [
    (lib.mkRenamedOptionModule
      [ "hardware" "usbStorage" "manageStartStop" ]
      [ "hardware" "usbStorage" "manageShutdown" ]
    )
  ];
}