local wezterm = require("wezterm")

local function basename(s)
	return string.gsub(s or "", "(.*[/\\])(.*)", "%2")
end

local SEPARATOR = " | "
local function setup()
	-- STATUSBAR
	-- show currently active key table in lower right status bar
	-- mimicing Vim modes
	wezterm.on("update-status", function(window, pane)
		local displayed = { left = {}, right = {} }
		local keytable = window:active_key_table()
		if keytable then
			displayed.left[#displayed.left + 1] = "MODE: " .. keytable
		end

		local workspace = window:active_workspace()
		if workspace and workspace ~= "default" then
			displayed.left[#displayed.left + 1] = "WORKSPACE: " .. workspace
		end

		local bat = ""
		for _, b in ipairs(wezterm.battery_info()) do
			bat = "🔋 " .. string.format("%.0f%%", b.state_of_charge * 100) .. " " .. b.state
		end
		displayed.right[#displayed.right + 1] = bat

		local currentprogram = pane:get_foreground_process_name()
		displayed.right[#displayed.right + 1] = basename(currentprogram)

		local statusleft = ""
		for _, v in ipairs(displayed.left) do
			statusleft = statusleft .. v .. SEPARATOR
		end
		local statusright = ""
		for _, v in ipairs(displayed.right) do
			statusright = statusright .. v .. SEPARATOR
		end

		window:set_left_status(statusleft or "")
		window:set_right_status(statusright or "")
	end)
end

return { setup = setup }