import ./make-test-python.nix (
  { pkgs, lib, ... }:

  let
    cagebreakConfigfile = pkgs.writeText "config" ''
      workspaces 1
      escape C-t
      bind t exec env DISPLAY=:0 ${pkgs.xterm}/bin/xterm -cm -pc
    '';
  in
  {
    name = "cagebreak";
    meta = with pkgs.lib.maintainers; {
      maintainers = [ berbiche ];
    };

    nodes.machine =
      { config, ... }:
      {
        # Automatically login on tty1 as a normal user:
        imports = [ ./common/user-account.nix ];
        services.getty.autologinUser = "alice";
        programs.bash.loginShellInit = ''
          if [ "$(tty)" = "/dev/tty1" ]; then
            set -e

            mkdir -p ~/.config/cagebreak
            cp -f ${cagebreakConfigfile} ~/.config/cagebreak/config

            cagebreak
          fi
        '';

        hardware.graphics.enable = true;
        programs.xwayland.enable = true;
        security.polkit.enable = true;
        environment.systemPackages = [
          pkgs.cagebreak
          pkgs.wayland-utils
        ];

        # Need to switch to a different GPU driver than the default one (-vga std) so that Cagebreak can launch:
        virtualisation.qemu.options = [ "-vga none -device virtio-gpu-pci" ];
      };

    enableOCR = true;

    testScript =
      { nodes, ... }:
      let
        user = nodes.machine.config.users.users.alice;
        XDG_RUNTIME_DIR = "/run/user/${toString user.uid}";
      in
      ''
        start_all()
        machine.wait_for_unit("multi-user.target")
        machine.wait_for_file("${XDG_RUNTIME_DIR}/wayland-0")

        with subtest("ensure wayland works with wayinfo from wallutils"):
            print(machine.succeed("env XDG_RUNTIME_DIR=${XDG_RUNTIME_DIR} wayland-info"))

        # TODO: Fix the XWayland test (log the cagebreak output to debug):
        # with subtest("ensure xwayland works with xterm"):
        #     machine.send_key("ctrl-t")
        #     machine.send_key("t")
        #     machine.wait_until_succeeds("pgrep xterm")
        #     machine.wait_for_text("${user.name}@machine")
        #     machine.screenshot("screen")
        #     machine.send_key("ctrl-d")
      '';
  }
)
