dotfiles/platforms/home-manager/modules/nmasur/presets/programs/wezterm.nix

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

let
  cfg = config.nmasur.presets.programs.wezterm;
in

{

  options.nmasur.presets.programs.wezterm = {
    enable = lib.mkEnableOption "WezTerm terminal";
    font = lib.mkOption {
      type = lib.types.str;
      description = "Name of the font for WezTerm";
    };
  };

  config = lib.mkIf cfg.enable {
    # Set the i3 terminal
    nmasur.presets.services.i3.terminal = pkgs.wezterm;

    # Display images in the terminal
    programs.fish.shellAliases = {
      icat = lib.mkForce "wezterm imgcat";
    };

    programs.wezterm = {
      enable = true;
      colorSchemes = {
        myTheme = {
          background = config.theme.colors.base00;
          foreground = config.theme.colors.base05;
          cursor_bg = config.theme.colors.base05;
          cursor_fg = config.theme.colors.base00;
          cursor_border = config.theme.colors.base05;
          selection_bg = config.theme.colors.base05;
          selection_fg = config.theme.colors.base00;
          scrollbar_thumb = config.theme.colors.base03;
          ansi = [
            config.theme.colors.base01 # black
            config.theme.colors.base0F # maroon
            config.theme.colors.base0B # green
            config.theme.colors.base0A # olive
            config.theme.colors.base0D # navy
            config.theme.colors.base0E # purple
            config.theme.colors.base0C # teal
            config.theme.colors.base06 # silver
          ];
          brights = [
            config.theme.colors.base03 # grey
            config.theme.colors.base08 # red
            config.theme.colors.base0B # lime
            config.theme.colors.base0A # yellow
            config.theme.colors.base0D # blue
            config.theme.colors.base0E # fuchsia
            config.theme.colors.base0C # aqua
            config.theme.colors.base07 # white
          ];
          compose_cursor = config.theme.colors.base09; # orange
          copy_mode_active_highlight_bg = {
            Color = config.theme.colors.base03;
          };
          copy_mode_active_highlight_fg = {
            Color = config.theme.colors.base07;
          };
          copy_mode_inactive_highlight_bg = {
            Color = config.theme.colors.base02;
          };
          copy_mode_inactive_highlight_fg = {
            Color = config.theme.colors.base06;
          };
          quick_select_label_bg = {
            Color = config.theme.colors.base02;
          };
          quick_select_label_fg = {
            Color = config.theme.colors.base06;
          };
          quick_select_match_bg = {
            Color = config.theme.colors.base03;
          };
          quick_select_match_fg = {
            Color = config.theme.colors.base07;
          };
        };
      };
      extraConfig = ''
        local wezterm = require("wezterm")
        local config = wezterm.config_builder()
        config.check_for_updates = false
        config.color_scheme = "myTheme"

        -- Scrollback
        config.scrollback_lines = 10000

        -- Window
        config.window_padding = {
          left = 10,
          right = 10,
          top = 10,
          bottom = 12,
        }

        config.font = wezterm.font('${cfg.font}', { weight = 'Bold'})
        config.font_size = ${if pkgs.stdenv.isLinux then "14.0" else "18.0"}

        -- Fix color blocks instead of text
        config.front_end = "WebGpu"

        -- Tab Bar
        config.hide_tab_bar_if_only_one_tab = true
        config.window_frame = {
          font = wezterm.font('${cfg.font}', { weight = 'Bold'}),
          font_size = ${if pkgs.stdenv.isLinux then "12.0" else "16.0"},
        }

        config.colors = {
          tab_bar = {
            active_tab = {
              bg_color = '${config.theme.colors.base00}',
              fg_color = '${config.theme.colors.base04}',
            },
          },
        }

        -- Disable audio
        config.audible_bell = "Disabled"

        config.initial_rows = 80
        config.initial_cols = 200

        config.unix_domains = {
          {
            name = 'unix',
          },
        }

        -- This causes `wezterm` to act as though it was started as
        -- `wezterm connect unix` by default, connecting to the unix
        -- domain on startup.
        -- If you prefer to connect manually, leave out this line.
        config.default_gui_startup_args = { 'connect', 'unix' }

        config.leader = {
          key = 'a',
          mods = 'CTRL',
          timeout_milliseconds = 2000,
        }

        config.keys = {
          -- Attach to muxer
          {
            key = 'a',
            mods = 'LEADER',
            action = wezterm.action.AttachDomain 'unix',
          },

          -- Detach from muxer
          {
            key = 'd',
            mods = 'LEADER',
            action = wezterm.action.DetachDomain { DomainName = 'unix' },
          },
          -- sends completion string for fish autosuggestions
          {
            key = 'Enter',
            mods = 'SHIFT',
            action = wezterm.action.SendString '\x1F'
          },
          -- ctrl-shift-h was "hide"
          {
            key = 'H',
            mods = 'SHIFT|CTRL',
            action = wezterm.action.DisableDefaultAssignment
          },
          -- alt-enter was "fullscreen"
          {
            key = 'Enter',
            mods = 'ALT',
            action = wezterm.action.DisableDefaultAssignment
          },
          -- make super-f "fullscreen"
          {
            key = 'f',
            mods = 'SUPER',
            action = wezterm.action.ToggleFullScreen
          },
          -- super-t open new tab in new dir
          {
            key = 't',
            mods = ${if pkgs.stdenv.isDarwin then "'SUPER'" else "'ALT'"},
            action = wezterm.action.SpawnCommandInNewTab {
              cwd = wezterm.home_dir,
            },
          },
          -- shift-super-t open new tab in same dir
          {
            key = 't',
            mods = 'SUPER|SHIFT',
            action = wezterm.action.SpawnTab 'CurrentPaneDomain'
          },
          -- project switcher
          {
             key = 'P',
             mods = 'SUPER',
             action = wezterm.action_callback(function(window, pane)
               local choices = {}

               wezterm.log_info "working?"

               function scandir(directory)
                   local i, t, popen = 0, {}, io.popen
                   local pfile = popen('${pkgs.fd}/bin/fd --search-path "'..directory..'" --type directory --exact-depth 2 | ${pkgs.proximity-sort}/bin/proximity-sort "'..os.getenv("HOME").."/dev/work"..'"')
                   for filename in pfile:lines() do
                       i = i + 1
                       t[i] = filename
                   end
                   pfile:close()
                   return t
               end

               for _, v in pairs(scandir(os.getenv("HOME").."/dev")) do
                 table.insert(choices, { label = v })
               end

               window:perform_action(
                 wezterm.action.InputSelector {
                   action = wezterm.action_callback(function(window, pane, id, label)
                     if not id and not label then
                       wezterm.log_info "cancelled"
                     else
                       window:perform_action(
                         wezterm.action.SpawnCommandInNewTab {
                           cwd = label,
                         },
                         pane
                       )
                     end
                   end),
                   fuzzy = true,
                   title = "Select Project",
                   choices = choices,
                 },
                 pane
               )
             end),
          },
        }

        -- print the workspace name at the upper right
        wezterm.on("update-right-status", function(window, pane)
          window:set_right_status(window:active_workspace())
        end)
        -- load plugin
        local workspace_switcher = wezterm.plugin.require("https://github.com/MLFlexer/smart_workspace_switcher.wezterm")
        -- set path to zoxide
        workspace_switcher.zoxide_path = "${pkgs.zoxide}/bin/zoxide"
        -- keymaps
        table.insert(config.keys, { key = "s", mods = "CTRL|SHIFT", action = workspace_switcher.switch_workspace() })
        -- table.insert(config.keys, { key = "t", mods = "CTRL|SHIFT", action = wezterm.action.ShowLauncherArgs({ flags = "FUZZY|WORKSPACES" }) })
        table.insert(config.keys, { key = "[", mods = "CTRL|SHIFT", action = wezterm.action.SwitchWorkspaceRelative(1) })
        table.insert(config.keys, { key = "]", mods = "CTRL|SHIFT", action = wezterm.action.SwitchWorkspaceRelative(-1) })

        return config
      '';
    };

  };
}