add nixops eval subcommand

Author: manveru

nix/eval-machine-info.nix

@@ -5,13 +5,15 @@
 , deploymentName
 , args
 , pluginNixExprs
+, evalFile ? null
 }:
 
 with import <nixpkgs> { inherit system; };
 with lib;
 
-
-rec {
+let
+  evaluator = if evalFile != null then (import evalFile) else id;
+in evaluator (rec {
 
   importedPluginNixExprs = map
     (expr: import expr)
@@ -200,4 +202,4 @@ rec {
   getNixOpsArgs = fs: lib.zipAttrs (lib.unique (lib.concatMap fileToArgs (getNixOpsExprs fs)));
 
   nixopsArguments = getNixOpsArgs networkExprs;
-}
+})

nixops/__main__.py

@@ -527,6 +527,18 @@
     help="include the physical specification in the evaluation",
 )
 
+subparser = add_subparser(
+    subparsers, "eval", help="eval the given file is nix code in the network expression"
+)
+subparser.set_defaults(op=op_eval)
+subparser.add_argument("code", metavar="CODE", help="code")
+subparser.add_argument(
+    "--json", action="store_true", help="print the option value in JSON format"
+)
+subparser.add_argument(
+    "--strict", action="store_true", help="enable strict evaluation"
+)
+
 subparser = add_subparser(
     subparsers,
     "list-generations",

nixops/deployment.py

@@ -380,22 +380,12 @@ def _eval_flags(self, exprs: List[str]) -> List[str]:
 
         flags.extend(
             [
-                "--arg",
-                "networkExprs",
-                py2nix(exprs_, inline=True),
-                "--arg",
-                "args",
-                py2nix(args, inline=True),
-                "--argstr",
-                "uuid",
-                self.uuid,
-                "--argstr",
-                "deploymentName",
-                self.name if self.name else "",
-                "--arg",
-                "pluginNixExprs",
-                py2nix(extraexprs),
-                "<nixops/eval-machine-info.nix>",
+                "--arg", "networkExprs", py2nix(exprs_, inline=True),
+                "--arg", "args", py2nix(args, inline=True),
+                "--argstr", "uuid", self.uuid,
+                "--argstr", "deploymentName", self.name if self.name else "",
+                "--arg", "pluginNixExprs", py2nix(extraexprs),
+                self.expr_path + "/eval-machine-info.nix",
             ]
         )
         return flags
@@ -503,6 +493,38 @@ def evaluate(self) -> None:
                 )
                 self.definitions[name] = defn
 
+    def evaluate_code(
+        self,
+        code: str,
+        json: bool = False,
+        strict: bool = False
+    ) -> str:
+        """Evaluate nix code in the deployment specification."""
+
+        exprs = self.nix_exprs
+        phys_expr = self.tempdir + "/physical.nix"
+        with open(phys_expr, "w") as f:
+            f.write(self.get_physical_spec())
+        exprs.append(phys_expr)
+
+        try:
+            return subprocess.check_output(
+                ["nix-instantiate"]
+                + self.extra_nix_eval_flags
+                + self._eval_flags(exprs)
+                + [
+                    "--eval-only",
+                    "--arg", "checkConfigurationOptions", "false",
+                    "--arg", "evalFile", code,
+                ]
+                + (["--strict"] if strict else [])
+                + (["--json"]   if json   else []),
+                stderr=self.logger.log_file,
+                text=True,
+            )
+        except subprocess.CalledProcessError:
+            raise NixEvalError
+
     def evaluate_option_value(
         self,
         machine_name: str,

nixops/script_defs.py

@@ -892,6 +892,12 @@ def op_show_option(args):
             )
         )
 
+def op_eval(args):
+    with deployment(args) as depl:
+        depl.evaluate()
+        sys.stdout.write(
+            depl.evaluate_code(args.code, json=args.json, strict=args.strict)
+        )
 
 @contextlib.contextmanager
 def deployment_with_rollback(args):