args @ { config, pkgs, lib, ... }:
let
  HOME = config.home-manager.users.krizej.home.homeDirectory;
  monitors = import ../utils/monitors.nix args;
in {
  services.xserver.windowManager.i3.enable = true;
  home-manager.users.krizej = {
    xsession.windowManager.i3 = {
      enable = true;
      config = rec {
        assigns = {
          "2" = [
            { class = "discord"; }
            { class = "SchildiChat"; }
            { class = "element"; }
          ];
        };

        window = {
          border = 1;
          titlebar = true;
        };

        floating = {
          border = 1;
          titlebar = true;
          criteria = [
            { class = "zenity"; }
            { class = "fceux"; }
            { title = "Friends List"; }
          ];
        };

        gaps = {
          inner = 0;
          outer = 0;
        };

        fonts = {
          names = [ "JetBrains Mono NL" ];
          size = 10.0;
        };

        bars = [
          {
            inherit fonts;
            trayOutput = "primary";
            statusCommand = "${pkgs.i3status}/bin/i3status";
          }
        ];

        defaultWorkspace = "workspace number 1";
        workspaceOutputAssign = [
          { workspace = "1"; output = monitors.left; }
          { workspace = "2"; output = monitors.right; }
        ];

        focus = {
          followMouse = true;
          mouseWarping = true;
          newWindow = "urgent";
        };

        startup = [
          { command = "autotiling &"; notification = false; }

          # { command = "mpd &"; notification = false; }
          { command = "dunst &"; notification = false; }
          { command = "flameshot &"; notification = false; }

          { command = "i3-msg \"workspace 2; append_layout ~/.config/i3/workspace2.json\""; notification = false; }
          { command = "discord &"; notification = false; }
          { command = "element-desktop &"; notification = false; }
        ];

        modes = let
          cmd_and_center = cmd: "exec --no-startup-id i3-msg '${cmd}' && i3-center-mouse-on-focus";
        in {
          resize = {
            "Left"   = cmd_and_center "resize shrink width  10 px or 10 ppt";
            "Right"  = cmd_and_center "resize grow   width  10 px or 10 ppt";
            "Up"     = cmd_and_center "resize shrink height 10 px or 10 ppt";
            "Down"   = cmd_and_center "resize grow   height 10 px or 10 ppt";
            "Escape" = "mode default";
          };
        };

        modifier = "Mod4";
        floating.modifier = "Mod4";

        keybindings = let
          mod = modifier;
          execns = "exec --no-startup-id";
          cmd_and_center = cmd: "${execns} i3-msg '${cmd}' && i3custom-center-mouse-on-focus";
        in {
          # Program keybinds
          "${mod}+Return"      = "${execns} alacritty";
          "${mod}+Shift+s"     = "${execns} flameshot gui";
          "${mod}+b"           = "${execns} firefox";
          "${mod}+d"           = "${execns} dmenu_run";
          "--release ${mod}+k" = "${execns} xkill";

          # Volume control
          "Ctrl+F9"  = "${execns} pamixer --toggle-mute";
          "Ctrl+F10" = "${execns} pamixer --decrease 5";
          "Ctrl+F11" = "${execns} pamixer --increase 5";

          # MPD controls
          # "Ctrl+F5" = "exec ~/music/select_music.sh";
          # "Ctrl+F6" = "exec mpc prev";
          # "Ctrl+F7" = "exec mpc toggle";
          # "Ctrl+F8" = "exec mpc next";
          # "Ctrl+Shift+F6" = "exec mpc seek -3";
          # "Ctrl+Shift+F7" = "exec mpc clear";
          # "Ctrl+Shift+F8" = "exec mpc seek +3";
          # "Ctrl+Shift+F10" = ''exec mpc volume -5 && dunstify "mpc $(mpc volume)" -r 420'';
          # "Ctrl+Shift+F11" = ''exec mpc volume +5 && dunstify "mpc $(mpc volume)" -r 420'';

          # General i3 keybindings
          "${mod}+f"       = cmd_and_center "fullscreen toggle";
          "${mod}+r"       = "mode resize";
          "${mod}+Shift+q" = cmd_and_center "kill";
          "${mod}+Shift+r" = "restart";
          "${mod}+Shift+e" = "${execns} i3-nagbar -t warning -m 'exit?' -b 'yes' 'i3-msg exit'";
          # Mouse bindings
          "--border button3" = "${execns} i3custom-window-actions";

          "${mod}+Left"        = cmd_and_center "focus left";
          "${mod}+Right"       = cmd_and_center "focus right";
          "${mod}+Up"          = cmd_and_center "focus up";
          "${mod}+Down"        = cmd_and_center "focus down";
          "${mod}+Shift+Left"  = cmd_and_center "move left";
          "${mod}+Shift+Right" = cmd_and_center "move right";
          "${mod}+Shift+Up"    = cmd_and_center "move up";
          "${mod}+Shift+Down"  = cmd_and_center "move down";

          "${mod}+q" = cmd_and_center "layout stacking";
          "${mod}+w" = cmd_and_center "layout tabbed";
          "${mod}+e" = cmd_and_center "layout toggle split";

          "${mod}+Shift+space" = "floating toggle; ${execns} i3custom-center-mouse-on-focus";
          "${mod}+space" = cmd_and_center "focus mode_toggle";

          "${mod}+1" = cmd_and_center "workspace number 1";
          "${mod}+2" = cmd_and_center "workspace number 2";
          "${mod}+3" = cmd_and_center "workspace number 3";
          "${mod}+4" = cmd_and_center "workspace number 4";
          "${mod}+5" = cmd_and_center "workspace number 5";
          "${mod}+6" = cmd_and_center "workspace number 6";
          "${mod}+7" = cmd_and_center "workspace number 7";
          "${mod}+8" = cmd_and_center "workspace number 8";
          "${mod}+9" = cmd_and_center "workspace number 9";
          "${mod}+0" = cmd_and_center "workspace number 10";

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

    programs.i3status = {
      enable = true;
      enableDefault = false;

      general = {
        interval = 1;
        colors = true;
        color_good = "#91d956";
        color_bad = "#f43841";
        color_degraded = "#ffdd33";
        output_format = "i3bar";
      };

      modules = {
        "cpu_usage" = {
          position = 0;
          settings = {
            format = "cpu: %usage";
          };
        };

        "memory" = {
          position = 1;
          settings = {
            format = "mem: %used";
            threshold_degraded = "1G";
            format_degraded = "MEMORY < %available";
          };
        };

        "volume master" = {
          position = 2;
          settings = {
            color_degraded = "#333333";
            format = "vol: %volume";
            format_muted = "vol: ---";
          };
        };

        "tztime local" = {
          position = 3;
          settings = {
            format = "%Y-%m-%d %H:%M:%S (%A)";
          };
        };
      };
    };

    home.packages = with pkgs; [
      autotiling

      (pkgs.writeShellScriptBin "i3custom-center-mouse-on-focus" ''
        eval $(xdotool getwindowfocus getwindowgeometry --shell)
        MX=$(($X + $WIDTH / 2))
        MY=$(($Y + $HEIGHT / 2))
        xdotool mousemove $MX $MY
      '')

      (pkgs.writeShellScriptBin "i3custom-window-actions" ''
        yad --splash --mouse --title="i3 window actions" \
            --gtkrc="${HOME}/.config/i3/i3gtk.css"       \
            --text="i3 window actions"                   \
            --button float:0 --button kill:1

        case $? in
          0) i3-msg "floating toggle";;
          1) i3-msg "kill";;
         -1) ;;
        esac
      '')

      (pkgs.writeShellScriptBin "i3custom-floating-toggle" ''
        i3-msg "mark --add _; [con_mark=f] floating disable, border pixel 1, mark --add --toggle _; [con_mark=_] floating enable, border normal; mark --add --toggle f;"
      '')
    ];

    home.file = {
      ".config/i3/workspace2.json".source = ../files/workspace2.json;
      ".config/i3/i3gtk.css".source = ../files/i3gtk.css;
    };
  };
}