98 lines
		
	
	
	
		
			3.1 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
			
		
		
	
	
			98 lines
		
	
	
	
		
			3.1 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
| {
 | |
|   config,
 | |
|   lib,
 | |
|   pkgs,
 | |
|   ...
 | |
| }: let
 | |
|   adminUsers = lib.lists.filter (user: user.isAdmin) (lib.attrsets.mapAttrsToList (_: user: user) config.host.postgres.extraUsers);
 | |
|   clientUsers = lib.lists.filter (user: user.isClient) (lib.attrsets.mapAttrsToList (_: user: user) config.host.postgres.extraUsers);
 | |
|   createUsers = lib.lists.filter (user: user.createUser) (lib.attrsets.mapAttrsToList (_: user: user) config.host.postgres.extraUsers);
 | |
|   createDatabases = lib.attrsets.mapAttrsToList (_: user: user) config.host.postgres.extraDatabases;
 | |
| in {
 | |
|   options = {
 | |
|     host.postgres = {
 | |
|       enable = lib.mkEnableOption "enable postgres";
 | |
|       extraUsers = lib.mkOption {
 | |
|         type = lib.types.attrsOf (lib.types.submodule ({name, ...}: {
 | |
|           options = {
 | |
|             name = lib.mkOption {
 | |
|               type = lib.types.str;
 | |
|               default = name;
 | |
|             };
 | |
|             isAdmin = lib.mkOption {
 | |
|               type = lib.types.bool;
 | |
|               default = false;
 | |
|             };
 | |
|             isClient = lib.mkOption {
 | |
|               type = lib.types.bool;
 | |
|               default = false;
 | |
|             };
 | |
|             createUser = lib.mkOption {
 | |
|               type = lib.types.bool;
 | |
|               default = false;
 | |
|             };
 | |
|           };
 | |
|         }));
 | |
|         default = {};
 | |
|       };
 | |
|       extraDatabases = lib.mkOption {
 | |
|         type = lib.types.attrsOf (lib.types.submodule ({name, ...}: {
 | |
|           options = {
 | |
|             name = lib.mkOption {
 | |
|               type = lib.types.str;
 | |
|               default = name;
 | |
|             };
 | |
|           };
 | |
|         }));
 | |
|         default = {};
 | |
|       };
 | |
|     };
 | |
|   };
 | |
| 
 | |
|   config = lib.mkIf config.host.postgres.enable {
 | |
|     services = {
 | |
|       postgresql = {
 | |
|         enable = true;
 | |
|         package = pkgs.postgresql_16;
 | |
|         ensureUsers =
 | |
|           [
 | |
|             {
 | |
|               name = "postgres";
 | |
|             }
 | |
|           ]
 | |
|           ++ (
 | |
|             builtins.map (user: {
 | |
|               name = user.name;
 | |
|               ensureDBOwnership = true;
 | |
|             })
 | |
|             createUsers
 | |
|           );
 | |
|         ensureDatabases = builtins.map (database: database.name) createDatabases;
 | |
|         identMap =
 | |
|           ''
 | |
|             # ArbitraryMapName systemUser DBUser
 | |
| 
 | |
|             # Administration Users
 | |
|             superuser_map      root      postgres
 | |
|             superuser_map      postgres  postgres
 | |
|           ''
 | |
|           + (
 | |
|             lib.strings.concatLines (builtins.map (user: "superuser_map      ${user.name}   postgres") adminUsers)
 | |
|           )
 | |
|           + ''
 | |
| 
 | |
|             # Client Users
 | |
|           ''
 | |
|           + (
 | |
|             lib.strings.concatLines (builtins.map (user: "user_map      ${user.name}   ${user.name}") clientUsers)
 | |
|           );
 | |
|         # configuration here lets users access the db that matches their name and lets user postgres access everything
 | |
|         authentication = pkgs.lib.mkOverride 10 ''
 | |
|           # type database DBuser    origin-address auth-method   optional_ident_map
 | |
|           local  all      postgres                 peer          map=superuser_map
 | |
|           local  sameuser all                      peer          map=user_map
 | |
|         '';
 | |
|       };
 | |
|     };
 | |
|   };
 | |
| }
 |