1 files changed, 19 insertions, 4 deletions
diff --git a/krebs/3modules/htgen.nix b/krebs/3modules/htgen.nix
index 375e2697..334a83cb 100644
--- a/krebs/3modules/htgen.nix
+++ b/krebs/3modules/htgen.nix
@@ -1,7 +1,13 @@
 { config, lib, pkgs, ... }:
 
-with import <stockholm/lib>;
+with import ../../lib/pure.nix { inherit lib; };
 let
+  optionalAttr = name: value:
+    if name != null then
+      { ${name} = value; }
+    else
+      {};
+
   cfg = config.krebs.htgen;
 
   out = {
@@ -30,8 +36,15 @@ let
         };
 
         script = mkOption {
-          type = types.str;
+          type = types.nullOr types.str;
+          default = null;
+        };
+
+        scriptFile = mkOption {
+          type = types.nullOr (types.either types.package types.pathname);
+          default = null;
         };
+
         user = mkOption {
           type = types.user;
           default = {
@@ -54,8 +67,10 @@ let
         after = [ "network.target" ];
         environment = {
           HTGEN_PORT = toString htgen.port;
-          HTGEN_SCRIPT = htgen.script;
-        };
+        }
+        // optionalAttr "HTGEN_SCRIPT" htgen.script
+        // optionalAttr "HTGEN_SCRIPT_FILE" htgen.scriptFile
+        ;
         serviceConfig = {
           SyslogIdentifier = "htgen";
           User = htgen.user.name;