78 lines
1.6 KiB
Nix
78 lines
1.6 KiB
Nix
{
|
|
config,
|
|
lib,
|
|
pkgs,
|
|
...
|
|
}:
|
|
let
|
|
cfg = config.my_k3s;
|
|
inherit (lib)
|
|
mkEnableOption
|
|
mkIf
|
|
mkOption
|
|
mkMerge
|
|
;
|
|
types = lib.types;
|
|
in
|
|
{
|
|
options.my_k3s = {
|
|
enable = mkEnableOption "My k3s cluster";
|
|
tokenFile = mkOption {
|
|
type = types.nonEmptyStr;
|
|
};
|
|
role = mkOption {
|
|
type = types.nonEmptyStr;
|
|
default = "server";
|
|
};
|
|
clusterInit = mkOption {
|
|
type = types.bool;
|
|
default = false;
|
|
};
|
|
serverAddr = mkOption {
|
|
type = types.nonEmptyStr;
|
|
default = "server";
|
|
};
|
|
extraFlags = mkOption {
|
|
type = types.listOf types.str;
|
|
default = [ ];
|
|
};
|
|
nvidia = mkOption {
|
|
type = types.bool;
|
|
default = false;
|
|
};
|
|
|
|
};
|
|
|
|
config = mkIf cfg.enable (mkMerge [
|
|
{
|
|
services.k3s = {
|
|
enable = true;
|
|
role = cfg.role;
|
|
tokenFile = cfg.tokenFile;
|
|
clusterInit = cfg.clusterInit;
|
|
serverAddr = cfg.serverAddr;
|
|
extraFlags = cfg.extraFlags;
|
|
};
|
|
environment.systemPackages = [ pkgs.nfs-utils ];
|
|
services.openiscsi = {
|
|
enable = true;
|
|
name = "iqn.2016-04.com.open-iscsi:${config.networking.hostName}";
|
|
};
|
|
systemd.tmpfiles.rules = [
|
|
"L+ /usr/local/bin - - - - /run/current-system/sw/bin/"
|
|
];
|
|
}
|
|
(mkIf cfg.nvidia {
|
|
hardware.nvidia-container-toolkit.enable = true;
|
|
hardware.nvidia-container-toolkit.mount-nvidia-executables = true;
|
|
# virtualisation.docker = {
|
|
# enable = true;
|
|
# enableNvidia = true;
|
|
# };
|
|
environment.systemPackages = with pkgs; [
|
|
nvidia-container-toolkit
|
|
];
|
|
})
|
|
]);
|
|
}
|