gokill/flake.nix

{
  description = "A very basic flake";

  #nixpkgs for testing framework
  inputs.nixpkgs.url = "github:nixos/nixpkgs/nixpkgs-unstable";

  outputs = { self, nixpkgs, ... }: 
  let
    forAllSystems = nixpkgs.lib.genAttrs [ "x86_64-linux" ];
    pkgs = nixpkgs.legacyPackages."x86_64-linux";
  in
  {
    devShell."x86_64-linux" = pkgs.mkShell {
      packages = with pkgs; [
        go
        gotools
        mdbook
      ];
    };

    packages.x86_64-linux.gokill = nixpkgs.legacyPackages.x86_64-linux.buildGoModule rec {
      pname = "gokill";
      version = "1.0";
      vendorHash = "sha256-aKEOMeW9QVSLsSuHV4b1khqM0rRrMjJ6Eu5RjY+6V8k=";
      src = ./.;

      postInstall = ''
        '';
    };

    packages.x86_64-linux.gokill-docbuilder = nixpkgs.legacyPackages.x86_64-linux.buildGoModule rec {
      pname = "docbuilder";
      version = "1.0";
      vendorHash = null;
      buildFLags = "-o . $dest/cmd/gokill/docbuilder";
      src = ./.;

      postInstall = ''
        '';
    };


    packages.x86_64-linux.docs = pkgs.callPackage (import ./docs/default.nix) { self = self; };

    packages.x86_64-linux.default = self.packages.x86_64-linux.gokill;

    nixosModules.gokill = { config, lib, pkgs, ... }: 
    let
      cfg = config.services.gokill;
      configFile = pkgs.writeText "config.json" ''${cfg.extraConfig}''; 
      gokill-pkg = self.packages.x86_64-linux.gokill;
    in
    {
      options = {
        services.gokill = {
          enable = lib.mkOption {
            default = false;
            type = lib.types.bool;
            description = lib.mdDoc ''
              Enables gokill daemon
              '';
          };

          extraConfig = lib.mkOption {
            type = lib.types.str;
            description = lib.mdDoc ''
              gokill config.json
              '';
          };
        };
      };

      config = lib.mkIf cfg.enable {
        systemd.services.gokill = {
          description = "gokill daemon";
          serviceConfig = {
            Type = "simple";
            ExecStart = "${gokill-pkg}/bin/gokill -c ${configFile}";
            Restart = "on-failure";
          };

          wantedBy = [ "default.target" ];
        };
      };
    };

    packages.x86_64-linux.testVm = 
    let
      nixos = nixpkgs.lib.nixosSystem {
        system = "x86_64-linux";
        modules = [
          self.nixosModules.gokill
          {
            services.gokill.enable = true;
            services.gokill.extraConfig = ''
              [
                  {
                      "type": "Timeout", 
                      "name": "custom timeout",
                      "options": {
                        "duration": 30
                      },
                      "actions": [
                          {
                              "type": "Print",
                              "options": {
                                "message": "Stage 1 triggered. Waiting 25 seconds"
                              },
                              "stage": 1
                          },
                          {
                              "type": "Timeout",
                              "options": {
                                "duration": 20
                              },
                              "stage": 1
                          },
                          {
                              "type": "Timeout",
                              "options": {
                                "duration": 5
                              },
                              "stage": 2
                          },
                          {
                              "type": "Print",
                              "options": {
                                "message": "Shutdown in 5 seconds..."
                              },
                              "stage": 2
                          },
                          {
                              "type": "Shutdown",
                              "options": {
                              },
                              "stage": 3
                          }
                      ]
                  }
              ]
              '';
            users.users.root.password = "root";
            virtualisation.vmVariant.virtualisation.graphics = false;
          }
        ];

      };
    in
    nixos.config.system.build.vm;

    apps.x86_64-linux.testVm = {
      type = "app";
      program = builtins.toString (nixpkgs.legacyPackages."x86_64-linux".writeScript "vm" ''
        ${self.packages."x86_64-linux".testVm}/bin/run-nixos-vm
      '');
    };

    apps.x86_64-linux.docs = {
      type = "app";
      program = builtins.toString (nixpkgs.legacyPackages."x86_64-linux".writeScript "docs" ''
        ${pkgs.python3}/bin/python3 -m http.server --directory ${self.packages."x86_64-linux".docs}/share/doc'');
    };
  };
}