From 3e8a08d63850df045d91b6d416b3de3a8a98e39c Mon Sep 17 00:00:00 2001
From: Mikhail Klementev <blame@dumpstack.io>
Date: Thu, 2 Jan 2020 17:44:46 +0000
Subject: [PATCH] Shutdown vm at app close, initial support for args

---
 generate.go | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/generate.go b/generate.go
index 5d73deb..23a32d6 100644
--- a/generate.go
+++ b/generate.go
@@ -14,14 +14,23 @@ import (
 
 var template = `
 {pkgs, ...}:
-{
+let
+  application = "${pkgs.%s}/bin/%s";
+  appRunner = pkgs.writeShellScriptBin "app" ''
+    ARGS_FILE=/home/user/.args
+    ARGS=$(cat $ARGS_FILE)
+    rm $ARGS_FILE
+
+    ${application} $ARGS
+    systemctl poweroff
+  '';
+in {
   imports = [
     <nixpkgs/nixos/modules/virtualisation/qemu-vm.nix>
     <nix/base.nix>
   ];
 
-  services.xserver.displayManager.sessionCommands =
-    "while [ 1 ]; do ${pkgs.%s}/bin/%s; done &";
+  services.xserver.displayManager.sessionCommands = "${appRunner}/bin/app &";
 }
 `