97 lines
3.3 KiB
Nix
97 lines
3.3 KiB
Nix
{ cluster, config, lib, pkgs, ... }:
|
|
|
|
let
|
|
inherit (cluster.config.services.patroni) secrets;
|
|
|
|
patroni = cluster.config.links.patroni-pg-access;
|
|
|
|
cfg = cluster.config.patroni;
|
|
|
|
writeQueryFile = pkgs.writeText "patroni-query.sql";
|
|
|
|
psqlRunFile = file: ''
|
|
export PGPASSWORD="$(< ${secrets.PATRONI_SUPERUSER_PASSWORD.path})"
|
|
while ! ${config.services.patroni.postgresqlPackage}/bin/psql 'host=${patroni.ipv4} port=${patroni.portStr} dbname=postgres user=postgres' --tuples-only --csv --file="${file}"; do
|
|
sleep 3
|
|
done
|
|
'';
|
|
|
|
psql = query: psqlRunFile (writeQueryFile query);
|
|
|
|
psqlSecret = getSecret: queryTemplate: let
|
|
queryTemplateFile = writeQueryFile queryTemplate;
|
|
in ''
|
|
umask 77
|
|
secretFile="$(mktemp -ut patroniSecret.XXXXXXXXXXXXXXXX)"
|
|
queryFile="$(mktemp -ut patroniQuery.XXXXXXXXXXXXXXXX)"
|
|
trap "rm -f $secretFile $queryFile" EXIT
|
|
${getSecret} > "$secretFile"
|
|
cp --no-preserve=mode ${queryTemplateFile} "$queryFile"
|
|
${pkgs.replace-secret}/bin/replace-secret '@SECRET@' "$secretFile" "$queryFile"
|
|
${psqlRunFile "$queryFile"}
|
|
'';
|
|
|
|
genPassword = pkgs.writeShellScript "patroni-generate-user-password" ''
|
|
umask 77
|
|
base64 -w0 /dev/urandom | tr -d /+ | head -c256 | tee "/run/keys/locksmith-provider-patroni-$1"
|
|
'';
|
|
in
|
|
|
|
{
|
|
services.incandescence.providers.patroni = lib.mkIf config.services.haproxy.enable {
|
|
locksmith = true;
|
|
wantedBy = [ "patroni.service" "multi-user.target" ];
|
|
partOf = [ "patroni.service" ];
|
|
wants = [ "postgresql.service" ];
|
|
after = [ "postgresql.service" ];
|
|
|
|
formulae = {
|
|
user = {
|
|
destroyAfterDays = 0;
|
|
create = user: psqlSecret "${genPassword} ${user}" ''
|
|
SELECT 'CREATE USER ${user}'
|
|
WHERE NOT EXISTS (SELECT FROM pg_roles WHERE rolname = '${user}')\gexec
|
|
ALTER USER ${user} PASSWORD '@SECRET@';
|
|
'';
|
|
destroy = psqlSecret "printenv OBJECT" ''
|
|
DROP USER @SECRET@;
|
|
'';
|
|
};
|
|
database = {
|
|
destroyAfterDays = 30;
|
|
deps = [ "user" ];
|
|
create = db: psql ''
|
|
SELECT 'CREATE DATABASE ${db} OWNER ${cfg.databases.${db}.owner}'
|
|
WHERE NOT EXISTS (SELECT FROM pg_database WHERE datname = '${db}')\gexec
|
|
'';
|
|
change = db: psql ''
|
|
ALTER DATABASE ${db} OWNER TO ${cfg.databases.${db}.owner};
|
|
'';
|
|
destroy = psqlSecret "printenv OBJECT" ''
|
|
DROP DATABASE @SECRET@;
|
|
'';
|
|
};
|
|
};
|
|
};
|
|
|
|
services.locksmith.providers.patroni = lib.mkIf config.services.haproxy.enable {
|
|
secrets = lib.mapAttrs (user: userConfig: {
|
|
command = {
|
|
envFile = ''
|
|
echo "PGPASSWORD=$(cat /run/keys/locksmith-provider-patroni-${user})"
|
|
rm -f /run/keys/locksmith-provider-patroni-${user}
|
|
'';
|
|
pgpass = ''
|
|
echo "*:*:*:${user}:$(cat /run/keys/locksmith-provider-patroni-${user})"
|
|
rm -f /run/keys/locksmith-provider-patroni-${user}
|
|
'';
|
|
raw = ''
|
|
cat /run/keys/locksmith-provider-patroni-${user}
|
|
rm -f /run/keys/locksmith-provider-patroni-${user}
|
|
'';
|
|
}.${userConfig.locksmith.format};
|
|
checkUpdate = "test -e /run/keys/locksmith-provider-patroni-${user}";
|
|
inherit (userConfig.locksmith) nodes;
|
|
}) cfg.users;
|
|
};
|
|
}
|