local wezterm = require('wezterm')
local act = wezterm.action
local config = wezterm.config_builder()
local mux = wezterm.mux

-- helper functions
local function get_current_working_dir(tab)
  local current_dir = tab.active_pane.current_working_dir
  local HOME_DIR = string.format('file://%s', os.getenv('HOME'))
  return current_dir == HOME_DIR and '~' or string.gsub(current_dir, '(.*[/\\])(.*)', '%2')
end

local function isVim(pane)
  return pane:get_foreground_process_name():find('n?vim') ~= nil
end

local function conditionalActivatePane(window, pane, pane_direction, vim_direction)
  if isVim(pane) then
    window:perform_action(act.SendKey({ key = vim_direction, mods = 'CTRL' }), pane)
  else
    window:perform_action(act.ActivatePaneDirection(pane_direction), pane)
  end
end

-- open new development session with splits and commands
-- this expects the default src / public-dist structure to work
local function newDevelopmentSession(path_and_title)
  local sites_dir = wezterm.home_dir .. '/Sites/'
  local dev_tab, dev_pane, dev_window = mux.spawn_window({
    cwd = sites_dir .. path_and_title,
    workspace = path_and_title,
  })
  local src_tab, src_pane, dev_window = dev_window:spawn_tab({
    cwd = sites_dir .. path_and_title,
  })
  local git_tab, git_pane, dev_window = dev_window:spawn_tab({
    cwd = sites_dir .. path_and_title,
  })
  dev_tab:set_title('zsh')
  dev_pane:send_text('v package.json\n')
  devcli_pane = dev_pane:split({
    workspace = path_and_title,
    direction = 'Right',
    cwd = sites_dir .. path_and_title,
  })
  devcli_pane:send_text('ggpl && gs\n')
  src_tab:set_title('src')
  src_pane:send_text('cd src && v\n:Telescope find_files\n')
  git_tab:set_title('git')
  git_pane:send_text('lg\n')
  dev_tab:activate()

  mux.set_active_workspace(path_and_title)
end

-- base config
config.color_scheme = 'GruvboxDark'
config.font = wezterm.font('MonaspiceAr Nerd Font')

config.font = wezterm.font({
  family = 'MonaspiceAr Nerd Font',
  weight = 'Regular',
  harfbuzz_features = { 'calt', 'liga', 'dlig', 'ss01', 'ss02', 'ss03', 'ss04', 'ss05', 'ss06', 'ss07', 'ss08' },
})

config.font_rules = {
  { -- Italic (comments)
    intensity = 'Normal',
    italic = true,
    font = wezterm.font({
      family = "MonaspiceRn Nerd Font",
      weight = "ExtraLight",
      stretch = "Normal",
      style = "Italic",
      harfbuzz_features = { 'calt', 'liga', 'dlig', 'ss01', 'ss02', 'ss03', 'ss04', 'ss05', 'ss06', 'ss07', 'ss08' },
    })
  },
}

config.default_prog = { '/opt/homebrew/bin/zsh' }
config.font_size = 14.0
config.line_height = 1.1
config.initial_cols = 120
config.initial_rows = 32
config.inactive_pane_hsb = { saturation = 0.5, brightness = 0.9 }
config.leader = { key = 'a', mods = 'CTRL', timeout_milliseconds = 1000 }
config.scrollback_lines = 5000
config.send_composed_key_when_left_alt_is_pressed = true
config.show_new_tab_button_in_tab_bar = false
config.tab_bar_at_bottom = true
config.use_dead_keys = false
config.use_fancy_tab_bar = false
config.window_decorations = 'RESIZE'

-- key mappings
config.keys = {
  { key = '\\', mods = 'LEADER', action = act.SplitHorizontal },
  { key = '-', mods = 'LEADER', action = act.SplitVertical },
  { key = ']', mods = 'LEADER', action = act.ActivateCopyMode },
  { key = 'g', mods = 'LEADER', action = act.ShowLauncherArgs { flags = 'FUZZY|WORKSPACES' } },
  -- Navigator.nvim keys
  { key = 'h', mods = 'CTRL', action = act.EmitEvent('ActivatePaneDirection-left') },
  { key = 'j', mods = 'CTRL', action = act.EmitEvent('ActivatePaneDirection-down') },
  { key = 'k', mods = 'CTRL', action = act.EmitEvent('ActivatePaneDirection-up') },
  { key = 'l', mods = 'CTRL', action = act.EmitEvent('ActivatePaneDirection-right') },
  -- leader-key mappings
  {
    key = ',',
    mods = 'LEADER',
    action = act.PromptInputLine {
      description = 'Enter new name for tab',
      action = wezterm.action_callback(function(window, pane, line)
        if line then
          window:active_tab():set_title(line)
        end
      end),
    },
  },
  {
    key = 'c',
    mods = 'LEADER',
    action = act.PromptInputLine {
      description = 'Enter name for new workspace',
      action = wezterm.action_callback(function(window, pane, line)
        if line then
          window:perform_action(
            act.SwitchToWorkspace {
              name = line,
            },
            pane
          )
        end
      end),
    },
  },
  {
    key = 'd',
    mods = 'LEADER',
    action = act.PromptInputLine {
      description = 'Enter name for new workspace',
      action = wezterm.action_callback(function(window, pane, line)
        if line then
          newDevelopmentSession(line:gsub('%s+', ''))
        end
      end),
    },
  },
  {
    key = '$',
    mods = 'LEADER',
    action = act.PromptInputLine {
      description = 'Rename current workspace',
      action = wezterm.action_callback(function(window, pane, line)
        if line then
          mux.rename_workspace(mux.get_active_workspace(), line)
        end
      end),
    },
  },
}

-- Navigator.nvim events
wezterm.on('ActivatePaneDirection-right', function(window, pane)
  conditionalActivatePane(window, pane, 'Right', 'l')
end)
wezterm.on('ActivatePaneDirection-left', function(window, pane)
  conditionalActivatePane(window, pane, 'Left', 'h')
end)
wezterm.on('ActivatePaneDirection-up', function(window, pane)
  conditionalActivatePane(window, pane, 'Up', 'k')
end)
wezterm.on('ActivatePaneDirection-down', function(window, pane)
  conditionalActivatePane(window, pane, 'Down', 'j')
end)

-- set tab-title to current dir if not set explicitly
wezterm.on('format-tab-title', function(tab)
  local cwd = get_current_working_dir(tab)
  local title = tab.tab_title
  local num = tab.tab_index + 1

  if title and #title > 0 then
    return string.format('  %s %s  ', num, title)
  end

  return string.format('  %s %s  ', num, cwd)
end)

-- Vim ZenMode fontsize-sync
wezterm.on('user-var-changed', function(window, pane, name, value)
  local overrides = window:get_config_overrides() or {}
  if name == "ZEN_MODE" then
    local incremental = value:find("+")
    local number_value = tonumber(value)
    if incremental ~= nil then
      while (number_value > 0) do
        window:perform_action(wezterm.action.IncreaseFontSize, pane)
        number_value = number_value - 1
      end
      overrides.enable_tab_bar = false
    elseif number_value < 0 then
      window:perform_action(wezterm.action.ResetFontSize, pane)
      overrides.font_size = nil
      overrides.enable_tab_bar = true
    else
      overrides.font_size = number_value
      overrides.enable_tab_bar = false
    end
  end
  window:set_config_overrides(overrides)
end)

-- show workspace on the left side of tabbar
wezterm.on('update-status', function(window)
  local workspace = mux.get_active_workspace()
  window:set_left_status(string.format('  %s  ', workspace))
end)

-- always maximise and start default session with mutt, notes and dotfiles
wezterm.on('gui-startup', function()
  local home_dir = wezterm.home_dir
  local tab, pane, window = mux.spawn_window({ cwd = home_dir })
  local todo_tab, todo_pane, window = window:spawn_tab({ cwd = home_dir .. '/Notes' })
  local dot_tab, dot_pane, window = window:spawn_tab({ cwd = home_dir .. '/Dotfiles' })
  local git_pane = dot_pane:split({ direction = 'Right', cwd = home_dir .. '/Dotfiles' })
  tab:set_title('mail')
  pane:send_text('O && m\n')
  dot_tab:set_title('dotfiles')
  dot_pane:send_text('v nvim/init.lua\n')
  todo_tab:set_title('todo')
  todo_pane:send_text('v index.md\n')
  git_pane:send_text('gs\n')
  tab:activate()

  -- initialize some sessions for MRU projects and folders
  newDevelopmentSession('ag/core')
  newDevelopmentSession('ag/web')
  newDevelopmentSession('ag/sp')
  newDevelopmentSession('pax/frontend')
  newDevelopmentSession('wwz/frontend')

  mux.set_active_workspace('default')
  window:gui_window():maximize()
end)

return config