grimm-nixos-laptop/common/graphics/sway.nix

{ pkgs, config, lib, ... }:
let
  cfg = config.grimmShared;
  sway_conf = with lib; types.submodule ({ config, ... }: rec {
    options = {
      keybinds = mkOption {
        type = types.attrsOf types.str;
        default = { };
        description = "set of keybinds assigning key combo to action";
      };

      autolaunch = mkOption {
        type = types.listOf (types.either types.nonEmptyStr types.package);
        default = [ ];
        description = "set of commands to be run at sway startup";
      };

      execAlways = mkOption {
        type = types.listOf (types.either types.nonEmptyStr types.package);
        default = [ ];
        description = "set of commands to be run at sway reload";
      };

      extraConfig = mkOption {
        type = types.str;
        default = "";
        description = "additional sway config to be included";
      };

      definitions = mkOption {
        type = types.attrsOf types.str;
        default = { };
        description = "set of definitions assigning variable to value";
      };

      modes = mkOption {
        type = types.attrsOf sway_conf;
        default = { };
        description = "possible modes to switch to, e.g. resize";
      };
    };
  });

  build_screen_def = fps_func: with lib; let
    output_def = mapAttrsToList
      (name: value:
        "output ${value.id} mode ${value.mode}@${toString (fps_func value.fps)}Hz"
        + (optionalString (value.pos != null) " position ${value.pos}")
      )
      cfg.screens;
  in
  ''
    for pid in $(${pkgs.procps}/bin/pgrep sway -x)
    do
      uid=$(id -u $(${pkgs.procps}/bin/ps -o user= -p $pid))
      export SWAYSOCK="/run/user/$uid/sway-ipc.$uid.$pid.sock"
      if [[ -e "$SWAYSOCK" ]] ; then
        echo "sock is $SWAYSOCK"
        ${config.programs.sway.package}/bin/swaymsg '${concatMapStrings (s: s + " ; ") output_def}'
      fi
    done
  '';
  inherit (lib) getExe;

  fps_min = fps: with lib; if isInt fps then fps else (foldl' min 2147483647 fps);
  fps_max = fps: with lib; if isInt fps then fps else (foldl' max 0 fps);
  init_screens_min_fps = with lib; pkgs.writeShellScriptBin "init-screens-min"
    (build_screen_def fps_min);
  init_screens_max_fps = with lib; pkgs.writeShellScriptBin "init-screens-max"
    (build_screen_def fps_max);
  init_screens_auto = pkgs.writeShellScriptBin "init-screens-auto" "which run-on-ac && which run-on-bat && run-on-ac ${getExe init_screens_max_fps} && run-on-bat ${getExe init_screens_min_fps}  || ${getExe init_screens_max_fps}";
in
{
  config =
    let
      bar_conf_file = if (lib.isPath cfg.sway.bar.config) then cfg.sway.bar.config else pkgs.writers.writeJSON "config.json" cfg.sway.bar.config;
      waybar_full = pkgs.writeShellScriptBin "waybar-full" (
        (lib.getExe config.programs.waybar.package)
        + (lib.optionalString (!isNull cfg.sway.bar.config) " -c ${bar_conf_file}")
        + (lib.optionalString (!isNull cfg.sway.bar.style) " -s ${cfg.sway.bar.style}")
      );

      bar_config = ''
        bar { 
              swaybar_command ${getExe waybar_full}
        }
      '';

      dbus-sway-environment = pkgs.writeShellScriptBin "dbus-sway-environment" ''
        dbus-update-activation-environment --systemd WAYLAND_DISPLAY XDG_CURRENT_DESKTOP=sway
        systemctl --user stop xdg-desktop-portal xdg-desktop-portal-wlr
        systemctl --user start xdg-desktop-portal xdg-desktop-portal-wlr
      '';

      build_conf = with lib; sway_conf:
        let
          build_definition_lines = mapAttrsToList (name: value: "set \$${name} ${value}");
          build_keybind_lines = mapAttrsToList (key: value: "bindsym ${key} ${value}");
          build_exec_lines = e: map (item: "${e} " + (if isString item then item else (getExe item)));
          build_mode_lines = mapAttrsToList (name: value: ''
            mode "${name}" {
            ${concatLines (map (item: "    " + item ) (build_conf value))}}'');
        in
        ([ ]
          ++ (build_definition_lines sway_conf.definitions)
          ++ (build_keybind_lines sway_conf.keybinds)
          ++ (build_exec_lines "exec" sway_conf.autolaunch)
          ++ (build_exec_lines "exec_always" sway_conf.execAlways)
          ++ (build_mode_lines sway_conf.modes)
          ++ optional (sway_conf.extraConfig != "") sway_conf.extraConfig
        );

      sway_conf = lib.concatLines (
        (build_conf cfg.sway.config)
        ++ lib.optional cfg.sway.bar.enable bar_config
        ++ (lib.mapAttrsToList
          (name: value:
            "output ${value.id} mode ${value.mode}"
            + (lib.optionalString (value.pos != null) " position ${value.pos}")
          )
          cfg.screens)
      );

      conf_path = "sway.conf";
    in
    with cfg; lib.mkIf (enable && sway.enable) {
      environment.etc."${conf_path}".text = sway_conf;

      grimmShared.sway.config.execAlways = [
        dbus-sway-environment
        init_screens_auto
      ];

      environment.systemPackages = [
        waybar_full
        dbus-sway-environment
        init_screens_min_fps
        init_screens_max_fps
        init_screens_auto
      ] ++ (with pkgs; [
        procps
        slurp
        libnotify
      ]);

      systemd.services.reload-sway = {
        description = "Reload all running sway instances";
        wantedBy = [ "multi-user.target" ];

        serviceConfig.Type = "oneshot";
        script = ''
          for pid in $(${pkgs.procps}/bin/pgrep sway -x)
          do
            uid=$(id -u $(${pkgs.procps}/bin/ps -o user= -p $pid))
            export SWAYSOCK="/run/user/$uid/sway-ipc.$uid.$pid.sock"
            if [[ -e "$SWAYSOCK" ]] ; then
              echo "sock is $SWAYSOCK"
              ${config.programs.sway.package}/bin/swaymsg reload
            fi
          done

          rm -rf /home/*/.cache/rmenu
        '';
        reloadTriggers = [ config.environment.etc."${conf_path}".source ];
      };

      programs.waybar.enable = true;

      programs.dconf.enable = true;

      programs.sway = {
        enable = true;

        wrapperFeatures = {
          gtk = true;
          base = true;
        };

        extraPackages = with pkgs; [
          swaylock
          swayidle
          wl-clipboard
          wf-recorder
          dmenu
          wmenu
          waybar-mpris
        ];
        extraOptions = [
          "--config"
          "/etc/${conf_path}"
        ];
        extraSessionCommands = ''
          # source /etc/profile
          # test -f $HOME/.profile && source $HOME/.profile
          systemctl --user import-environment

          export XDG_CURRENT_DESKTOP=sway
          export SDL_VIDEODRIVER=wayland
          export QT_QPA_PLATFORM=wayland
          export QT_WAYLAND_DISABLE_WINDOWDECORATION="1"
          export _JAVA_AWT_WM_NONREPARENTING=1
          export MOZ_ENABLE_WAYLAND=1
          # export WLR_RENDERER=vulkan
          # export DRI_PRIME=1
          export NIXOS_OZONE_WL=1
        '';
      };
    };

  options.grimmShared.sway = with lib; {
    enable = mkEnableOption "grimm-sway";

    bar = {
      enable = mkEnableOption "grimm-sway-bar";

      style = mkOption {
        type = types.nullOr types.path;
        default = null;
        description = "waybar style sheet to use";
      };

      config = mkOption {
        type = types.nullOr (types.either types.path types.attrs);
        default = null;
        description = "waybar config to use";
      };
    };

    config = mkOption {
      type = sway_conf;
      description = "sway config to use";
    };
  };
}