about summary refs log tree commit diff
path: root/modules/vlock/default.nix
blob: 11e9a475c6ae283a0f9d1037c1817d92ca0cc2a7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
{ pkgs, config, lib, ... }:

let
  messageFile = pkgs.runCommand "message.cat" {} ''
    echo -en '\e[H\e[2J\e[?25l' > "$out"
    "${pkgs.aacolorize}/bin/aacolorize" \
      "${./message.cat}" "${./message.colmap}" \
      >> "$out"
  '';

  esc = "\\\\033";
  unlockCSI = "${esc}[16;39H${esc}[?25h${esc}[K";

  vlock = lib.overrideDerivation pkgs.vlock (o: {
    postPatch = (o.postPatch or "") + ''
      echo -n '"' > src/message.h
      sed -e ':nl;N;$!bnl;s/[\\"]/\\&/g;s/\n/\\n/g' "${messageFile}" \
        >> src/message.h
      sed -i -e '$s/$/"/' src/message.h
      sed -i -e 's!getenv("VLOCK_MESSAGE")!\n#include "message.h"\n!' \
        src/vlock-main.c
      sed -i -re 's/(fprintf[^"]*")(.*user)/\1${unlockCSI}\2/' \
        src/auth-pam.c
    '';
  });
in {
  options.vuizvui.vlock.enable = lib.mkEnableOption "console lock";

  config.systemd.sockets.vlock = {
    description = "Console Lock Socket";
    wantedBy = [ "sockets.target" ];
    socketConfig.ListenStream = "/run/console-lock.sock";
    socketConfig.Accept = true;
  };

  config.systemd.services."vlock@" = lib.mkIf config.vuizvui.vlock.enable {
    description = "Lock All Consoles";
    serviceConfig.Type = "oneshot";

    #environment.USER = "%i"; XXX
    environment.USER = "aszlig";

    script = ''
      retval=0
      oldvt="$("${pkgs.kbd}/bin/fgconsole")"
      "${vlock}/bin/vlock" -asn || retval=$?
      if [ $retval -ne 0 ]; then "${pkgs.kbd}/bin/chvt" "$oldvt"; fi
      exit $retval
    '';
  };
}