blob: 1fdeb6dac18766d5a4f28f1bd9fcb7abbd41420f (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
{
lib,
config,
pkgs,
...
}:
let
cfg = config.services.redpanda-connect;
in
{
options.services.redpanda-connect = {
enable = lib.mkEnableOption "Enable Redpanda Connect";
package = lib.mkOption {
type = lib.types.package;
default = (pkgs.callPackage ./package.nix { });
description = "The Redpanda Connect package";
};
pipelines = lib.mkOption {
type = lib.types.attrsOf (
lib.types.submodule {
options = {
enable = lib.mkEnableOption "Enable the pipeline";
allowSudo = lib.mkEnableOption "Allow sudo";
config = lib.mkOption {
type = lib.types.attrs;
description = "The configuration for the pipeline";
};
};
}
);
};
};
config = lib.mkIf cfg.enable {
systemd.services = builtins.listToAttrs (
lib.mapAttrsToList (name: pipeline: {
name = "redpanda-connect-${name}";
value = {
wantedBy = [ "multi-user.target" ];
serviceConfig = {
Type = "simple";
ExecStart =
let
configFile = pkgs.writeText "redpanda-connect-${name}.json" (builtins.toJSON pipeline.config);
in
"${cfg.package}/bin/redpanda-connect run ${configFile}";
Restart = "always";
RestartSec = "5";
DynamicUser = true;
};
};
}) cfg.pipelines
);
# Allow sudo access to allowed services:
#security.polkit.extraConfig = ''
# polkit.addRule(function(action, subject) {
# if (action.id == "org.freedesktop.systemd1.manage-units" &&
# action.lookup("unit").startsWith("botcore.") &&
# subject.user == "botcore") {
# return polkit.Result.YES;
# }
# });
#'';
security.polkit.extraConfig =
let
allowedServices = builtins.filter (pipeline: pipeline.value.allowSudo) (lib.attrsToList cfg.pipelines);
in
builtins.concatStringsSep "\n" (
builtins.map (value: ''
polkit.addRule(function(action, subject) {
if (action.id == "org.freedesktop.systemd1.manage-units" &&
action.lookup("unit") == "redpanda-connect-${value.name}") {
return polkit.Result.YES;
}
});
'') allowedServices
);
};
}
|
