{ config, lib, pkgs, ... }:

let 
  cfg = config.samfelag.modules.desktop.i3;
  mod = "Mod4";

  cmdModule = lib.types.submodule ({ ... }: {
    options = {
      enable = lib.mkOption {
        type = lib.types.bool;
        default = true;
      };
      command = lib.mkOption {
        type = lib.types.str;
        example = "${pkgs.rofi}/bin/rofi rofi -show run";
      };
    };
  });

in {

  options.samfelag.modules.desktop.i3 = {
    enable = lib.mkEnableOption "i3 window manager";
    launcher = lib.mkOption {
      type = cmdModule;
    };
    terminal = lib.mkOption {
      type = cmdModule;
    };
    browser = lib.mkOption {
      type = cmdModule;
    };
  };

  config = lib.mkIf cfg.enable {

    # - Dependencies -----------------------------

    home.packages = with pkgs; [
      (nerdfonts.override { fonts = [ "Iosevka" ]; })
    ];

    # - Configuration ----------------------------

    xsession.windowManager.i3 = {
      enable = true;
      config = {
        modifier = mod;

        fonts = {
          names =  ["DejaVu Sans Mono" "Iosevka Nerd Font"];
          style = "regular";
          size = 12.0;
        };

        focus.followMouse = false;

        keybindings = {
          "${mod}+q" = "kill";

          # Focus
          "${mod}+h" = "focus left";
          "${mod}+j" = "focus down";
          "${mod}+k" = "focus up";
          "${mod}+l" = "focus right";
          "${mod}+Left" = "focus left";
          "${mod}+Down" = "focus down";
          "${mod}+Up" = "focus up";
          "${mod}+Right" = "focus right";

          # Move
          "${mod}+Shift+h" = "move left";
          "${mod}+Shift+j" = "move down";
          "${mod}+Shift+k" = "move up";
          "${mod}+Shift+l" = "move right";
          "${mod}+Shift+Left" = "move left";
          "${mod}+Shift+Down" = "move down";
          "${mod}+Shift+Up" = "move up";
          "${mod}+Shift+Right" = "move right";

          # split orientation
          "${mod}+z" = "split h;exec notify-send 'tile horizontally'";
          "${mod}+v" = "split v;exec notify-send 'tile vertically'";
          
          # enter fullscreen mode for the focused container
          "${mod}+f" = "fullscreen toggle";
          
          # change container layout (stacked, tabbed, toggle split)
          "${mod}+s" = "layout stacking";
          "${mod}+w" = "layout tabbed";
          "${mod}+e" = "layout toggle split";
          
          # toggle tiling / floating
          "${mod}+Shift+space" = "floating toggle";
          
          # focus the parent container
          "${mod}+a" = "focus parent";

          # - Workspaces -------------------------
          "${mod}+1" = "workspace 1";
          "${mod}+2" = "workspace 2";
          "${mod}+3" = "workspace 3";
          "${mod}+4" = "workspace 4";
          "${mod}+5" = "workspace 5";
          "${mod}+6" = "workspace 6";
          "${mod}+7" = "workspace 7";
          "${mod}+8" = "workspace 8";
          "${mod}+9" = "workspace 9";
          "${mod}+0" = "workspace 0";

          "${mod}+Shift+1" = "move container to workspace 1; workspace 1";
          "${mod}+Shift+2" = "move container to workspace 2; workspace 2";
          "${mod}+Shift+3" = "move container to workspace 3; workspace 3";
          "${mod}+Shift+4" = "move container to workspace 4; workspace 4";
          "${mod}+Shift+5" = "move container to workspace 5; workspace 5";
          "${mod}+Shift+6" = "move container to workspace 6; workspace 6";
          "${mod}+Shift+7" = "move container to workspace 7; workspace 7";
          "${mod}+Shift+8" = "move container to workspace 8; workspace 8";
          "${mod}+Shift+9" = "move container to workspace 9; workspace 9";
          "${mod}+Shift+0" = "move container to workspace 0; workspace 0";

        }
        // (if cfg.terminal.enable then {
          "${mod}+Return" = "exec ${cfg.terminal.command}";
        } else {})
        // (if cfg.launcher.enable then {
          "${mod}+space" = "exec ${cfg.launcher.command}";
        } else {})
        // (if cfg.browser.enable then {
          "Control+Alt+3" = "exec ${cfg.browser.command}";
        } else {});
      };
    };
  };
}