feat: pass through rocket options in nixos-module

tweag · Apr 12, 2024 · 53de4c5 · 53de4c5
1 parent 8af8853
commit 53de4c5
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 8 deletions.
diff --git a/nix/configuration.nix b/nix/configuration.nix
@@ -1,5 +1,8 @@
 { ... }:
 
+let
+  port = 9000;
+in
 {
   virtualisation.vmVariant = {
     virtualisation = {
@@ -8,8 +11,8 @@
       forwardPorts = [
         {
           from = "host";
-          guest.port = 8000;
-          host.port = 8000;
+          guest.port = port;
+          host.port = port;
         }
       ];
     };
@@ -21,6 +24,14 @@
     password = "genealogos";
   };
 
-  services.genealogos.enable = true;
+  services.genealogos = {
+    enable = true;
+    rocketConfig = {
+      release = {
+        port = port;
+        address = "0.0.0.0";
+      };
+    };
+  };
 }
 
diff --git a/nix/genealogos-module.nix b/nix/genealogos-module.nix
@@ -5,6 +5,7 @@ with lib;
 
 let
   cfg = config.services.genealogos;
+  rocketConfigFormat = pkgs.formats.toml { };
 in
 {
   options = {
@@ -19,16 +20,43 @@ in
           The genealogos-api package to use.
         '';
       };
+
+      rocketConfig = lib.mkOption {
+        type = rocketConfigFormat.type;
+        default = { };
+        example = lib.literalExpression ''
+          {
+            release = {
+              address = "0.0.0.0";
+              port = "8000";
+              limits = {
+                form = "64 kB";
+                json = "1 MiB";
+              };
+            };
+          }
+        '';
+
+        description = lib.mdDoc ''
+          Configuration file for Genealogos.
+
+          Genealogos-api uses rocket as its http implementation.
+          For all configuration options, see https://rocket.rs/guide/v0.5/configuration/#configuration-parameters
+        '';
+      };
     };
   };
 
   config = mkIf (cfg.enable) {
-    systemd.services.genealogos =
-      {
-        description = "Genealogos sbom generator";
-        wantedBy = [ "multi-user.target" ];
+    systemd.services.genealogos = {
+      description = "Genealogos sbom generator";
+      wantedBy = [ "multi-user.target" ];
 
-        serviceConfig.ExecStart = "${cfg.package}/bin/genealogos-api";
+      serviceConfig.ExecStart = "${cfg.package}/bin/genealogos-api";
+
+      environment = {
+        ROCKET_CONFIG = rocketConfigFormat.generate "Rocket.toml" cfg.rocketConfig;
       };
+    };
   };
 }