neoshell.lua

#!/usr/bin/env lua

package.path = package.path .. ";/home/jadu/.luarocks/share/lua/5.4/?.lua"
package.cpath = package.cpath .. ";/home/jadu/.local/share/luarocks/lib/lua/5.4/?.so"

local lfs_available, lfs = pcall(require, "lfs")
local linenoise_available, linenoise = pcall(require, "linenoise")

local function get_home()
	return os.getenv("HOME") or "/"
end

local function set_env(name, value)
	if value then
		os.execute("export " .. name .. "='" .. value:gsub("'", "'\\''") .. "' >/dev/null 2>&1")
	else
		os.execute("unset " .. name .. " >/dev/null 2>&1")
	end
end

local config = {
	prompt = "%u@%h %w $ ",
	rprompt = "", -- Right prompt support
	aliases = {},
	global_aliases = {},
	colors = true,
	syntax_highlighting = true, -- Syntax highlighting while typing
	auto_suggestions = true, -- Grey suggestions from history
	fuzzy_completion = true, -- Fuzzy tab completion
	auto_cd = true, -- Auto-cd when typing directory
	theme = {
		base = "#b4befe",
	},
	cdpath = { ".", "..", "~" },
	path = {},
	history_file = get_home() .. "/.neoshell_history",
	history_size = 500,
	syntax_colors = {
		command = "green",
		argument = "text",
		string = "yellow",
		variable = "pink",
		comment = "overlay0",
		error = "red",
		suggestion = "overlay0",
	},
}

local current_dir = os.getenv("PWD") or get_home()
local dir_stack = { current_dir }
local aliases = {}
local global_aliases = {}
local shell_vars = {}
local jobs = {}
local job_id_counter = 1

local history = {}
local history_index = 0

-- External commands cache for syntax highlighting
local external_commands = {}
local function refresh_external_commands()
	external_commands = {}
	local handle = io.popen("compgen -c 2>/dev/null | sort -u")
	if handle then
		for line in handle:lines() do
			if line and line ~= "" then
				external_commands[line] = true
			end
		end
		handle:close()
	end
	-- Also add builtins
	for cmd, _ in pairs(builtin_cmds or {}) do
		external_commands[cmd] = true
	end
end

-- Git info cache
local git_info = { branch = "", dirty = false, ahead = 0, behind = 0 }
local function get_git_info()
	local handle = io.popen(
		"git symbolic-ref --short HEAD 2>/dev/null || git describe --tags --exact-match 2>/dev/null || git rev-parse --short HEAD 2>/dev/null"
	)
	if handle then
		git_info.branch = handle:read("*l") or ""
		handle:close()

		if git_info.branch ~= "" then
			local status_handle = io.popen("git status --porcelain 2>/dev/null | wc -l")
			if status_handle then
				local dirty_count = tonumber(status_handle:read("*l")) or 0
				git_info.dirty = dirty_count > 0
				status_handle:close()
			end

			local ahead_handle = io.popen("git rev-list --count HEAD@{upstream}..HEAD 2>/dev/null || echo 0")
			if ahead_handle then
				git_info.ahead = tonumber(ahead_handle:read("*l")) or 0
				ahead_handle:close()
			end

			local behind_handle = io.popen("git rev-list --count HEAD..HEAD@{upstream} 2>/dev/null || echo 0")
			if behind_handle then
				git_info.behind = tonumber(behind_handle:read("*l")) or 0
				behind_handle:close()
			end
		end
	end
end

-- Levenshtein distance for typo suggestions
local function levenshtein(a, b)
	local len_a, len_b = #a, #b
	if len_a == 0 then
		return len_b
	end
	if len_b == 0 then
		return len_a
	end

	local matrix = {}
	for i = 0, len_a do
		matrix[i] = {}
		matrix[i][0] = i
	end
	for j = 0, len_b do
		matrix[0][j] = j
	end

	for i = 1, len_a do
		for j = 1, len_b do
			local cost = (a:sub(i, i) == b:sub(j, j)) and 0 or 1
			matrix[i][j] = math.min(matrix[i - 1][j] + 1, matrix[i][j - 1] + 1, matrix[i - 1][j - 1] + cost)
		end
	end
	return matrix[len_a][len_b]
end

local function find_closest_command(cmd)
	local best_match = nil
	local best_distance = math.huge

	for known_cmd, _ in pairs(external_commands) do
		local dist = levenshtein(cmd, known_cmd)
		if dist < best_distance and dist <= 3 then
			best_distance = dist
			best_match = known_cmd
		end
	end

	return best_match
end

local function get_theme_color(color_name)
	local base = config.theme and config.theme.base or "#b4befe"
	local r = tonumber(base:sub(2, 3), 16)
	local g = tonumber(base:sub(4, 5), 16)
	local b = tonumber(base:sub(6, 7), 16)

	local adjustments = {
		rosewater = { 0, -20, -20 },
		flamingo = { 0, -15, -15 },
		pink = { 10, -10, 20 },
		mauve = { 20, 0, 30 },
		red = { 20, -30, -30 },
		maroon = { 20, -20, -20 },
		peach = { 20, -10, -10 },
		yellow = { 30, 10, -20 },
		green = { -20, 30, -20 },
		teal = { -20, 30, 30 },
		sky = { -20, 10, 30 },
		sapphire = { -10, 5, 30 },
		blue = { 0, 0, 0 },
		lavender = { 20, 10, 30 },
		text = { 0, -10, -10 },
		subtext1 = { -10, -5, -10 },
		subtext0 = { -15, -10, -10 },
		overlay2 = { -15, -10, -5 },
		overlay1 = { -20, -15, -10 },
		overlay0 = { -25, -20, -15 },
		surface2 = { -30, -25, -20 },
		surface1 = { -35, -30, -25 },
		surface0 = { -40, -35, -30 },
		base = { -50, -45, -40 },
		mantle = { -55, -50, -45 },
		crust = { -60, -55, -50 },
		white = { 30, 30, 30 },
		black = { -30, -30, -30 },
		reset = { 0, 0, 0 },
		bright_red = { 40, -20, -20 },
		bright_green = { -20, 40, -20 },
		bright_yellow = { 40, 40, -10 },
		bright_blue = { -10, -10, 40 },
		bright_magenta = { 40, -10, 40 },
		bright_cyan = { -10, 40, 40 },
		bright_white = { 40, 40, 40 },
	}

	local adj = adjustments[color_name] or { 0, 0, 0 }
	local nr = math.min(255, math.max(0, r + adj[1]))
	local ng = math.min(255, math.max(0, g + adj[2]))
	local nb = math.min(255, math.max(0, b + adj[3]))

	return string.format("38;2;%d;%d;%d", nr, ng, nb)
end

local colors = {}

local function init_colors()
	colors = {
		rosewater = get_theme_color("rosewater"),
		flamingo = get_theme_color("flamingo"),
		pink = get_theme_color("pink"),
		mauve = get_theme_color("mauve"),
		red = get_theme_color("red"),
		maroon = get_theme_color("maroon"),
		peach = get_theme_color("peach"),
		yellow = get_theme_color("yellow"),
		green = get_theme_color("green"),
		teal = get_theme_color("teal"),
		sky = get_theme_color("sky"),
		sapphire = get_theme_color("sapphire"),
		blue = get_theme_color("blue"),
		lavender = get_theme_color("lavender"),
		text = get_theme_color("text"),
		subtext1 = get_theme_color("subtext1"),
		subtext0 = get_theme_color("subtext0"),
		overlay2 = get_theme_color("overlay2"),
		overlay1 = get_theme_color("overlay1"),
		overlay0 = get_theme_color("overlay0"),
		surface2 = get_theme_color("surface2"),
		surface1 = get_theme_color("surface1"),
		surface0 = get_theme_color("surface0"),
		base = get_theme_color("base"),
		mantle = get_theme_color("mantle"),
		crust = get_theme_color("crust"),
		white = get_theme_color("white"),
		black = get_theme_color("black"),
		reset = "0",
		bright_red = get_theme_color("bright_red"),
		bright_green = get_theme_color("bright_green"),
		bright_yellow = get_theme_color("bright_yellow"),
		bright_blue = get_theme_color("bright_blue"),
		bright_magenta = get_theme_color("bright_magenta"),
		bright_cyan = get_theme_color("bright_cyan"),
		bright_white = get_theme_color("bright_white"),
	}
end

local function c(color)
	if not config.colors then
		return ""
	end
	return "\27[" .. colors[color] .. "m"
end

local function expand_tilde(path)
	if path:sub(1, 1) == "~" then
		if path == "~" then
			return get_home()
		end
		return get_home() .. path:sub(2)
	end
	return path
end

local function resolve_path(path, silent)
	if path == "" then
		return get_home()
	elseif path == "-" then
		return os.getenv("OLDPWD") or get_home()
	end

	local target = expand_tilde(path)

	if lfs_available then
		local attr = lfs.attributes(target)
		if attr and attr.mode == "directory" then
			return target
		end
	else
		local handle = io.popen("test -d '" .. target .. "' 2>/dev/null && echo 'yes'")
		if handle then
			local result = handle:read("*a"):match("^%s*(.-)%s*$")
			handle:close()
			if result == "yes" then
				return target
			end
		end
	end

	if not silent then
		for _, p in ipairs(config.cdpath) do
			local base = expand_tilde(p)
			local full_path = base .. "/" .. path
			if lfs_available then
				local attr = lfs.attributes(full_path)
				if attr and attr.mode == "directory" then
					return full_path
				end
			else
				local handle = io.popen("test -d '" .. full_path .. "' 2>/dev/null && echo 'yes'")
				if handle then
					local result = handle:read("*a"):match("^%s*(.-)%s*$")
					handle:close()
					if result == "yes" then
						return full_path
					end
				end
			end
		end
	end

	return target
end

local function load_history()
	local file = io.open(config.history_file, "r")
	if not file then
		return
	end

	for line in file:lines() do
		if line ~= "" then
			table.insert(history, line)
		end
	end
	file:close()

	if #history > config.history_size then
		local new_history = {}
		for i = #history - config.history_size + 1, #history do
			table.insert(new_history, history[i])
		end
		history = new_history
	end

	history_index = #history + 1
end

local function save_history()
	local file = io.open(config.history_file, "w")
	if not file then
		return
	end

	for _, line in ipairs(history) do
		file:write(line .. "\n")
	end
	file:close()
end

local function add_to_history(line)
	line = line:gsub("^%s+", ""):gsub("%s+$", "")
	if line == "" then
		return
	end

	if history[#history] == line then
		return
	end

	table.insert(history, line)
	history_index = #history + 1

	if #history > config.history_size then
		table.remove(history, 1)
		history_index = #history + 1
	end
end

local function history_search(search_str)
	for i = #history, 1, -1 do
		if history[i]:match(search_str) then
			return history[i], i
		end
	end
	return nil
end

-- Config parsers for different formats
local function parse_toml(content)
	local result = {}
	local current_section = result
	local in_array = false
	local array_name = nil
	local array_values = {}
	
	for line in content:gmatch("[^\r\n]+") do
		line = line:gsub("%s*#.*$", "")  -- Remove comments
		line = line:gsub("^%s+", ""):gsub("%s+$", "")  -- Trim
		
		if line == "" then goto continue end
		
		-- Section headers
		local section = line:match("^%[([^%]]+)%]$")
		if section then
			if array_name then
				current_section[array_name] = array_values
				array_name = nil
				array_values = {}
			end
			
			result[section] = result[section] or {}
			current_section = result[section]
			goto continue
		end
		
		-- Arrays like aliases = ["ll = ls -la", "la = ls -a"]
		local array_start = line:match("^(%w+)%s*=%s*%[")
		if array_start then
			array_name = array_start
			in_array = true
			array_values = {}
			local values = line:match("%[(.*)%]")
			if values then
				for val in values:gmatch('"([^"]+)"') do
					table.insert(array_values, val)
				end
				current_section[array_name] = array_values
				array_name = nil
				in_array = false
			end
			goto continue
		end
		
		-- Inline tables like theme = { base = "#b4befe" }
		local key, inline_table = line:match("^(%w+)%s*=%s*{(.+)}")
		if key and inline_table then
			local tbl = {}
			for k, v in inline_table:gmatch("(%w+)%s*=%s*\"?([^\"},]+)\"?") do
				tbl[k] = v:gsub("^%s+", ""):gsub("%s+$", "")
			end
			current_section[key] = tbl
			goto continue
		end
		
		-- Key-value pairs
		local key, value = line:match("^([^=]+)%s*=%s*(.+)$")
		if key then
			key = key:gsub("^%s+", ""):gsub("%s+$", "")
			value = value:gsub("^%s+", ""):gsub("%s+$", "")
			
			-- Remove quotes
			if value:match("^%'") and value:match("%'.$") then
				value = value:sub(2, -2)
			elseif value:match('^"') and value:match('"$') then
				value = value:sub(2, -2)
			elseif value == "true" then
				value = true
			elseif value == "false" then
				value = false
			elseif tonumber(value) then
				value = tonumber(value)
			end
			
			-- Handle nested keys like syntax_colors.command
			local parent, child = key:match("^([^.]+)%.(.+)$")
			if parent then
				result[parent] = result[parent] or {}
				result[parent][child] = value
			else
				result[key] = value
			end
		end
		
		::continue::
	end
	
	return result
end

local function parse_yaml(content)
	-- Simple YAML parser (subset)
	local result = {}
	local current_section = nil
	local indent_stack = {}
	
	for line in content:gmatch("[^\r\n]+") do
		-- Skip comments and empty lines
		if line:match("^%s*#") or line:match("^%s*$") then goto continue end
		
		local indent = #line:match("^( *)")
		line = line:gsub("^%s+", ""):gsub("%s+$", "")
		
		-- Remove inline comments
		line = line:gsub("%s*#.*$", "")
		
		-- Section
		if line:match("^%w+:$") then
			local section = line:gsub(":", "")
			result[section] = {}
			current_section = result[section]
			goto continue
		end
		
		-- Array item under current section
		if current_section and line:match("^%- ") then
			local value = line:gsub("^%- ", "")
			value = value:gsub("^%s*\"?", ""):gsub("\"?%s*$", "")
			if not current_section.array then
				current_section.array = {}
			end
			table.insert(current_section.array, value)
			goto continue
		end
		
		-- Key-value pairs
		local key, value = line:match("^([^:]+):%s*(.+)$")
		if key then
			key = key:gsub("^%s+", ""):gsub("%s+$", "")
			value = value:gsub("^%s+", ""):gsub("%s+$", "")
			
			-- Handle "key: [" or "key: {" starts
			if value == "[" then
				if current_section then
					current_section[key] = {}
				else
					result[key] = {}
				end
				goto continue
			end
			
			-- Remove quotes
			if value:match("^%'") and value:match("%'.$") then
				value = value:sub(2, -2)
			elseif value:match('^"') and value:match('"$') then
				value = value:sub(2, -2)
			elseif value == "true" then
				value = true
			elseif value == "false" then
				value = false
			elseif tonumber(value) then
				value = tonumber(value)
			end
			
			if current_section then
				-- Parse alias definitions
				if key == "aliases" or key == "global_aliases" then
					local alias_key, alias_val = value:match("^(%S+)%s*=%s*(.+)$")
					if alias_key then
						if type(current_section) == "table" then
							if not current_section[key] then
								current_section[key] = {}
							end
							current_section[key][alias_key] = alias_val
						end
					else
						current_section[key] = value
					end
				else
					current_section[key] = value
				end
			else
				result[key] = value
			end
		end
		
		::continue::
	end
	
	return result
end

local function parse_json(content)
	-- Very basic JSON parser for simple config
	local result = {}
	
	-- Remove whitespace
	content = content:gsub("%s+", " ")
	
	-- Parse top-level object
	for key, value in content:gmatch('"([^"]+)":%s*([^,{]+)') do
		value = value:gsub("^%s+", ""):gsub("%s+$", "")
		
		if value:match('^"') then
			value = value:gsub('"', "")
		elseif value == "true" then
			value = true
		elseif value == "false" then
			value = false
		elseif tonumber(value) then
			value = tonumber(value)
		end
		
		result[key] = value
	end
	
	-- Parse arrays
	for key, arr_content in content:gmatch('"([^"]+)":%s*(%[[^%]]+%])') do
		local arr = {}
		for val in arr_content:gmatch('"([^"]+)"') do
			table.insert(arr, val)
		end
		result[key] = arr
	end
	
	-- Parse nested objects (simple)
	for key, obj_content in content:gmatch('"([^"]+)":%s*({[^}]+})') do
		local obj = {}
		for k, v in obj_content:gmatch('"([^"]+)":%s*"?([^",}]+)"?') do
			obj[k] = v
		end
		result[key] = obj
	end
	
	return result
end

local function parse_ini(content)
	local result = {}
	local current_section = result
	
	for line in content:gmatch("[^\r\n]+") do
		line = line:gsub("%s*;.*$", ""):gsub("^%s+", ""):gsub("%s+$", "")
		if line == "" then goto continue end
		
		-- Section
		local section = line:match("^%[([^%]]+)%]$")
		if section then
			if section == "aliases" then
				result.aliases = {}
				current_section = result.aliases
			elseif section == "global_aliases" then
				result.global_aliases = {}
				current_section = result.global_aliases
			elseif section == "theme" then
				result.theme = {}
				current_section = result.theme
			elseif section == "syntax_colors" then
				result.syntax_colors = {}
				current_section = result.syntax_colors
			else
				result[section] = result[section] or {}
				current_section = result[section]
			end
			goto continue
		end
		
		-- Key-value pairs
		local key, value = line:match("^([^=]+)%s*=%s*(.+)$")
		if key then
			key = key:gsub("^%s+", ""):gsub("%s+$", "")
			value = value:gsub("^%s+", ""):gsub("%s+$", "")
			
			-- Handle quotes
			if value:match('^"') and value:match('"$') then
				value = value:sub(2, -2)
			elseif value:match("^%'") and value:match("%'.$") then
				value = value:sub(2, -2)
			elseif value == "true" then
				value = true
			elseif value == "false" then
				value = false
			elseif tonumber(value) then
				value = tonumber(value)
			end
			
			current_section[key] = value
		end
		
		::continue::
	end
	
	return result
end

local function load_config()
	local home = get_home()
	local config_files = {
		{ path = home .. "/.shell.toml", parser = parse_toml },
		{ path = home .. "/.shell.yaml", parser = parse_yaml },
		{ path = home .. "/.shell.yml", parser = parse_yaml },
		{ path = home .. "/.shell.json", parser = parse_json },
		{ path = home .. "/.shell.conf", parser = parse_ini },
		{ path = home .. "/.shellrc", parser = parse_ini },
		{ path = home .. "/.neoshell", parser = nil },  -- Lua format, backward compatible
	}
	
	local loaded_config = nil
	local loaded_path = nil
	
	for _, cfg in ipairs(config_files) do
		local file = io.open(cfg.path, "r")
		if file then
			local content = file:read("*all")
			file:close()
			
			if cfg.parser then
				local success, result = pcall(cfg.parser, content)
				if success then
					loaded_config = result
					loaded_path = cfg.path
					break
				else
					print("Error parsing " .. cfg.path .. ": " .. tostring(result))
				end
			else
				-- Lua format
				local env = {}
				setmetatable(env, { __index = _G })
				local chunk, err = load(content, cfg.path, "t", env)
				if chunk then
					local ok, result = pcall(chunk)
					if ok and type(result) == "table" then
						loaded_config = result
						loaded_path = cfg.path
						break
					end
				end
			end
		end
	end
	
	if not loaded_config then
		return
	end
	
	-- Apply loaded config
	if loaded_config.prompt then
		config.prompt = loaded_config.prompt
	end
	if loaded_config.aliases then
		config.aliases = loaded_config.aliases
	end
	if loaded_config.global_aliases then
		config.global_aliases = loaded_config.global_aliases
	end
	if loaded_config.colors ~= nil then
		config.colors = loaded_config.colors
	end
	if loaded_config.theme then
		config.theme = loaded_config.theme
	end
	if loaded_config.cdpath then
		config.cdpath = loaded_config.cdpath
	end
	if loaded_config.history_file then
		config.history_file = loaded_config.history_file
	end
	if loaded_config.history_size then
		config.history_size = loaded_config.history_size
	end
	if loaded_config.rprompt then
		config.rprompt = loaded_config.rprompt
	end
	if loaded_config.syntax_highlighting ~= nil then
		config.syntax_highlighting = loaded_config.syntax_highlighting
	end
	if loaded_config.auto_suggestions ~= nil then
		config.auto_suggestions = loaded_config.auto_suggestions
	end
	if loaded_config.fuzzy_completion ~= nil then
		config.fuzzy_completion = loaded_config.fuzzy_completion
	end
	if loaded_config.auto_cd ~= nil then
		config.auto_cd = loaded_config.auto_cd
	end
	if loaded_config.syntax_colors then
		for k, v in pairs(loaded_config.syntax_colors) do
			config.syntax_colors[k] = v
		end
	end

	-- Source additional config files
	if loaded_config.source and loaded_config.source.files then
		for _, file in ipairs(loaded_config.source.files) do
			local expanded_path = file:gsub("^~", home)
			local f = io.open(expanded_path, "r")
			if f then
				local content = f:read("*all")
				f:close()
				local env = {}
				setmetatable(env, { __index = _G })
				local chunk, err = load(content, expanded_path, "t", env)
				if chunk then
					pcall(chunk)
				end
			end
		end
	end

	aliases = config.aliases
	global_aliases = config.global_aliases
end

local function format_prompt()
	local prompt = config.prompt
	local username = os.getenv("USER") or os.getenv("USERNAME") or "user"
	local hostname = os.getenv("HOSTNAME") or "localhost"
	local git_str = ""
	
	-- Update git info
	get_git_info()
	if git_info.branch ~= "" then
		git_str = c("mauve") .. "(" .. git_info.branch .. ")" .. c("reset")
		if git_info.dirty then
			git_str = git_str .. c("yellow") .. "*" .. c("reset")
		end
		if git_info.ahead > 0 then
			git_str = git_str .. c("green") .. "↑" .. git_info.ahead .. c("reset")
		end
		if git_info.behind > 0 then
			git_str = git_str .. c("red") .. "↓" .. git_info.behind .. c("reset")
		end
		git_str = git_str .. " "
	end

	-- Get short directory name
	local short_dir = current_dir
	if short_dir == get_home() then
		short_dir = "~"
	else
		-- Replace home with ~ in path
		short_dir = short_dir:gsub("^" .. get_home(), "~")
	end
	
	prompt = prompt:gsub("%%u", username)
	prompt = prompt:gsub("%%h", hostname)
	prompt = prompt:gsub("%%w", short_dir)
	prompt = prompt:gsub("%%%%", "%%")  -- Replace %% with % (escaped properly)
	
	-- Add git branch placeholder
	prompt = prompt:gsub("%%g", git_info.branch ~= "" and git_info.branch or "")

	if config.colors then
		-- Build colored prompt
		local prompt_parts = {}
		local i = 1
		while i <= #prompt do
			if prompt:sub(i, i) == "%" then
				local code = prompt:sub(i+1, i+1)
				if code == "n" then
					table.insert(prompt_parts, c("blue") .. username .. c("reset"))
					i = i + 2
				elseif code == "m" then
					table.insert(prompt_parts, c("mauve") .. hostname .. c("reset"))
					i = i + 2
				elseif code == "~" then
					table.insert(prompt_parts, c("green") .. short_dir .. c("reset"))
					i = i + 2
				elseif code == "g" then
					if git_info.branch ~= "" then
						table.insert(prompt_parts, c("mauve") .. "(" .. git_info.branch .. ")" .. c("reset"))
						if git_info.dirty then
							table.insert(prompt_parts, c("yellow") .. "*" .. c("reset"))
						end
					end
					i = i + 2
				elseif code == "#" then
				table.insert(prompt_parts, c("red") .. "#" .. c("reset"))
					i = i + 2
				elseif code == "$" then
					table.insert(prompt_parts, c("yellow") .. "$" .. c("reset"))
					i = i + 2
				elseif code == "%" then
					table.insert(prompt_parts, "%")
					i = i + 2
				else
					table.insert(prompt_parts, prompt:sub(i, i))
					i = i + 1
				end
			else
				table.insert(prompt_parts, prompt:sub(i, i))
				i = i + 1
			end
		end
		prompt = table.concat(prompt_parts)
	end

	return prompt
end

local function format_rprompt()
	local rprompt = config.rprompt
	if rprompt == "" then
		return ""
	end

	local username = os.getenv("USER") or os.getenv("USERNAME") or "user"
	local hostname = os.getenv("HOSTNAME") or "localhost"
	local time_str = os.date("%H:%M")

	rprompt = rprompt:gsub("%%u", username)
	rprompt = rprompt:gsub("%%h", hostname)
	rprompt = rprompt:gsub("%%w", current_dir)
	rprompt = rprompt:gsub("%%t", time_str)
	rprompt = rprompt:gsub("%%%", "%%")

	if config.colors then
		rprompt = c("overlay0") .. rprompt .. c("reset")
	end

	return rprompt
end

local builtin_cmds = {
	cd = true,
	pwd = true,
	exit = true,
	help = true,
	echo = true,
	printf = true,
	test = true,
	export = true,
	alias = true,
	unalias = true,
	source = true,
	set = true,
	unset = true,
	pushd = true,
	popd = true,
	dirs = true,
	jobs = true,
	fg = true,
	bg = true,
	type = true,
	["true"] = true,
	["false"] = true,
	return_builtin = true,
}

local function fuzzy_match(str, pattern)
	if not config.fuzzy_completion then
		return str:sub(1, #pattern) == pattern
	end

	local pattern_idx = 1
	local str_idx = 1

	while pattern_idx <= #pattern and str_idx <= #str do
		if pattern:sub(pattern_idx, pattern_idx):lower() == str:sub(str_idx, str_idx):lower() then
			pattern_idx = pattern_idx + 1
		end
		str_idx = str_idx + 1
	end

	return pattern_idx > #pattern
end

local function get_command_completions(prefix)
	local matches = {}
	for cmd, _ in pairs(builtin_cmds) do
		if cmd ~= "return_builtin" and fuzzy_match(cmd, prefix) then
			table.insert(matches, cmd)
		end
	end
	for cmd, _ in pairs(aliases) do
		if fuzzy_match(cmd, prefix) then
			table.insert(matches, cmd)
		end
	end

	local handle = io.popen("compgen -c 2>/dev/null || echo ''")
	if handle then
		for line in handle:lines() do
			if line and fuzzy_match(line, prefix) then
				local found = false
				for _, m in ipairs(matches) do
					if m == line then
						found = true
						break
					end
				end
				if not found then
					table.insert(matches, line)
				end
			end
		end
		handle:close()
	end

	return matches
end

local function get_path_completions(prefix)
	local matches = {}
	local dir = current_dir
	local search_prefix = prefix

	if prefix:match("^~/") then
		dir = get_home()
		search_prefix = prefix:sub(3)
	elseif prefix:sub(1, 1) == "~" then
		dir = get_home()
		search_prefix = prefix:sub(2)
	elseif prefix:match("^%./") then
		dir = current_dir
		search_prefix = prefix:sub(3)
	elseif prefix:match("^%/") then
		dir = "/"
		search_prefix = prefix:sub(2)
	else
		local last_slash = prefix:match(".*/")
		if last_slash then
			dir = current_dir .. "/" .. last_slash
			search_prefix = prefix:match("([^/]+)$") or ""
		end
	end

	local handle = io.popen("ls -la '" .. dir .. "' 2>/dev/null")
	if not handle then
		return matches
	end

	for line in handle:lines() do
		local parts = {}
		for w in line:gmatch("%S+") do
			table.insert(parts, w)
		end
		if #parts >= 9 then
			local fname = parts[9]
			local ftype = parts[1]
			if fname and fuzzy_match(fname, search_prefix) then
				local is_dir = ftype:sub(1, 1) == "d"
				local match = prefix
				if not prefix:match("/$") and prefix:match("/") then
					match = prefix:match(".*/") .. fname
				elseif prefix == "" or prefix == "." then
					match = "./" .. fname
				elseif prefix:sub(1, 2) == "~/" then
					match = "~/" .. fname
				else
					match = fname
				end
				if is_dir then
					match = match .. "/"
				end
				table.insert(matches, match)
			end
		end
	end
	handle:close()

	return matches
end

local function do_complete(line)
	local completions = {}

	if line:match("^%s*$") or line:match("%s$") then
		return nil
	end

	local tokens = {}
	for word in line:gmatch("%S+") do
		table.insert(tokens, word)
	end

	if #tokens == 0 or (tokens[#tokens] == line:match("%S+$")) then
		local last_token = tokens[#tokens] or ""
		local cmd_completions = get_command_completions(last_token)
		for _, c in ipairs(cmd_completions) do
			table.insert(completions, c)
		end

		local search_prefix = last_token
		if line:sub(-#last_token) == last_token then
			local path_completions = get_path_completions(search_prefix)
			for _, c in ipairs(path_completions) do
				table.insert(completions, c)
			end
		end
	end

	if #completions == 0 then
		return nil
	elseif #completions == 1 then
		if #tokens > 0 then
			local prefix_len = #tokens[#tokens]
			return completions[1]:sub(prefix_len + 1)
		else
			return completions[1]
		end
	else
		local prefix = tokens[#tokens] or ""
		local common = completions[1]
		for i = 2, #completions do
			local comp = completions[i]
			local new_common = ""
			for j = 1, math.min(#common, #comp) do
				if common:sub(j, j) == comp:sub(j, j) then
					new_common = new_common .. common:sub(j, j)
				else
					break
				end
			end
			common = new_common
		end

		if config.colors then
			io.write("\n")
			for _, comp in ipairs(completions) do
				if comp:sub(-1) == "/" then
					io.write(c("blue") .. comp .. c("reset") .. "  ")
				else
					io.write(c("text") .. comp .. c("reset") .. "  ")
				end
			end
			io.write("\n")
			io.write(format_prompt() .. line)
			io.flush()
		end

		return common:sub(#prefix + 1)
	end
end

if linenoise_available then
	linenoise.setcompletion(do_complete)
end

local function expand_env_vars(str)
	local function expand_var(var)
		local name = var:match("^%{(.-)%}$") or var
		if name:sub(1, 1) == "$" then
			name = name:sub(2)
		end

		if shell_vars[name] then
			return shell_vars[name]
		end

		local val = os.getenv(name)
		if val then
			return val
		end

		return ""
	end

	str = str:gsub("%$%b{}", expand_var)
	str = str:gsub("%$%a[%w_]*", expand_var)

	return str
end

local function expand_command_substitution(str)
	local function expand_cmd(cmd)
		local handle = io.popen(cmd .. " 2>/dev/null")
		if handle then
			local result = handle:read("*a"):gsub("%s+$", "")
			handle:close()
			return result
		end
		return ""
	end

	local function match_cmd_sub(s)
		local start = s:find("%$%(")
		if not start then
			return s
		end

		if s:sub(start + 2, start + 2) == "(" then
			return s
		end

		local remaining = s:sub(start + 2)
		local depth = 1
		local finish = nil

		for i = 1, #remaining do
			local c = remaining:sub(i, i)
			if c == "(" then
				depth = depth + 1
			elseif c == ")" then
				depth = depth - 1
				if depth == 0 then
					finish = start + i
					break
				end
			end
		end

		if finish then
			local cmd = s:sub(start + 2, finish - 1)
			local before = s:sub(1, start - 1)
			local after = s:sub(finish + 1)
			return before .. expand_cmd(cmd) .. match_cmd_sub(after)
		end

		return s
	end

	local function expand_backtick(cmd)
		local handle = io.popen(cmd .. " 2>/dev/null")
		if handle then
			local result = handle:read("*a"):gsub("%s+$", "")
			handle:close()
			return result
		end
		return ""
	end

	str = str:gsub("`([^`]+)`", expand_backtick)

	return match_cmd_sub(str)
end

local function expand_arithmetic(str)
	local function eval_arith(expr)
		local safe_expr = expr:gsub("[^%d%+%-%*%/%^%(%s%)]", "")

		local function pow(a, b)
			return a ^ b
		end

		local env = { math = math, pow = pow }
		local func, err = load("return " .. safe_expr, "arith", "t", env)
		if func then
			local ok, result = pcall(func)
			if ok then
				return tostring(math.floor(result * 100 + 0.5) / 100)
			end
		end
		return "$(( " .. expr .. " ))"
	end

	local function match_arith(s)
		local start = s:find("$%(%(")
		if not start then
			return s
		end

		local close_pos = s:find("))", start + 3)

		if close_pos then
			local expr = s:sub(start + 3, close_pos - 1)
			local before = s:sub(1, start - 1)
			local after = s:sub(close_pos + 1)
			return before .. eval_arith(expr) .. match_arith(after)
		end

		return s
	end

	return match_arith(str)
end

local function expand_zsh_globs(pattern)
	-- Handle ** (recursive) patterns
	if pattern:match("%*%*") then
		local parts = {}
		local base_path = ""
		local glob_pattern = pattern

		-- Separate base path from pattern
		if pattern:match("^/") then
			base_path = "/"
			glob_pattern = pattern:sub(2)
		elseif pattern:match("^%./") then
			base_path = current_dir .. "/"
			glob_pattern = pattern:sub(3)
		elseif pattern:match("^~/") then
			base_path = get_home() .. "/"
			glob_pattern = pattern:sub(3)
		else
			base_path = current_dir .. "/"
		end

		-- Convert ** to find command
		local find_cmd = "find "
			.. base_path
			.. ' -type f -name "'
			.. glob_pattern:gsub("%*%*", "*"):gsub("%*", "*")
			.. '" 2>/dev/null'

		if glob_pattern:match("%*%*/") then
			-- Directory search
			find_cmd = "find "
				.. base_path
				.. ' -type d -name "'
				.. glob_pattern:gsub("%*%*.*", "*"):gsub(".*/", "")
				.. '" 2>/dev/null'
		end

		local handle = io.popen(find_cmd)
		if handle then
			for line in handle:lines() do
				if line and line ~= "" then
					table.insert(parts, line)
				end
			end
			handle:close()
		end

		if #parts > 0 then
			return table.concat(parts, " ")
		end
	end

	return nil
end

local function expand_globs(str)
	local has_glob = str:match("%*") or str:match("%?") or str:match("%[")
	if not has_glob then
		return str
	end

	-- Try zsh-style ** glob first
	local zsh_result = expand_zsh_globs(str)
	if zsh_result then
		return zsh_result
	end

	local handle = io.popen("ls -1d " .. str .. " 2>/dev/null || echo ''")
	if handle then
		local result = handle:read("*a"):gsub("%s+$", "")
		handle:close()
		if result ~= "" and result ~= str then
			local parts = {}
			for part in result:gmatch("%S+") do
				table.insert(parts, part)
			end
			if #parts > 0 then
				return table.concat(parts, " ")
			end
		end
	end

	return str
end

local function apply_modifiers(word, mod)
	if mod == "h" then
		return word:match("^(.+)/[^/]+$") or word
	elseif mod == "t" then
		return word:match("[^/]+$") or word
	elseif mod == "r" then
		return word:match("^(.+)%.[^.]+$") or word
	elseif mod == "e" then
		return word:match("%.([^.]+)$") or ""
	elseif mod == "p" then
		return word:match("^.*/") or ""
	elseif mod == "q" then
		return "'" .. word:gsub("'", "'\\''") .. "'"
	elseif mod == "u" then
		return word:sub(1, 1):upper() .. word:sub(2)
	elseif mod == "l" then
		return word:sub(1, 1):lower() .. word:sub(2)
	elseif mod == "U" then
		return word:upper()
	elseif mod == "L" then
		return word:lower()
	end
	return word
end

local function expand_history_ref(str)
	local ref = str:match("^!(.+)$") or str:match("^%!(.+)")
	if ref then
		if ref == "!" then
			return history[#history] or ""
		elseif ref == "#" then
			return ""
		elseif ref:match("^%d+$") then
			local idx = tonumber(ref)
			if idx > 0 and idx <= #history then
				return history[idx]
			end
		else
			local matched = history_search(ref)
			if matched then
				return matched
			end
		end
		return str
	end

	local last_cmd = str:match("!(%S+)")
	if last_cmd then
		local matched = history_search(last_cmd)
		if matched then
			return str:gsub("!" .. last_cmd, matched)
		end
	end

	return str
end

local function expand_line(line)
	line = expand_history_ref(line)
	line = expand_env_vars(line)
	line = expand_command_substitution(line)
	line = expand_arithmetic(line)

	return line
end

local function builtin_cd(args)
	local path = args[1] or get_home()
	local target = resolve_path(path, false)

	local old_dir = current_dir

	local success, err
	if lfs_available then
		success, err = lfs.chdir(target)
	else
		success = os.execute("cd " .. target .. " 2>/dev/null")
	end

	if success then
		if lfs_available then
			current_dir = lfs.currentdir()
		else
			local handle = io.popen("cd " .. target .. " && pwd")
			if handle then
				current_dir = handle:read("*a"):gsub("%s+$", "")
				handle:close()
			end
		end
		set_env("PWD", current_dir)
		set_env("OLDPWD", old_dir)

		if args._raw ~= true then
			table.insert(dir_stack, 1, current_dir)
			if #dir_stack > 100 then
				table.remove(dir_stack)
			end
		end
	else
		if config.colors then
			print(c("red") .. "cd: " .. path .. ": No such file or directory" .. c("reset"))
		else
			print("cd: " .. path .. ": No such file or directory")
		end
	end
end

local function builtin_pwd(args)
	if config.colors then
		print(c("green") .. current_dir .. c("reset"))
	else
		print(current_dir)
	end
end

local function builtin_exit(args)
	local code = tonumber(args[1]) or 0
	save_history()
	os.exit(code)
end

local function builtin_echo(args)
	local result = table.concat(args, " ")
	result = result:gsub("\\n", "\n")
	result = result:gsub("\\t", "\t")
	result = expand_env_vars(result)
	print(result)
end

local function builtin_printf(args)
	local format = args[1] or ""
	local params = {}
	for i = 2, #args do
		table.insert(params, args[i])
	end

	local format_expanded = expand_env_vars(format)

	local ok, result = pcall(string.format, format_expanded, unpack(params))
	if ok then
		io.write(result)
	else
		if config.colors then
			print(c("red") .. "printf: " .. result .. c("reset"))
		else
			print("printf: " .. result)
		end
	end
end

local function builtin_test(args)
	if #args == 0 then
		return false
	end

	local expr = table.concat(args, " ")

	if expr:match("^%-%w+%s+$") then
		return true
	end

	if expr:match("^%-%w+%s+%S+$") then
		local op, path = expr:match("^%-%w+%s+(.+)$")
		local flag = expr:match("^(%-%w+)")

		if flag == "-e" or flag == "-f" or flag == "-d" or flag == "-r" or flag == "-w" or flag == "-x" then
			local test_path = expand_env_vars(path)
			local handle = io.popen("test " .. flag .. " '" .. test_path .. "' 2>/dev/null && echo 'yes'")
			if handle then
				local result = handle:read("*a"):match("^%s*(.-)%s*$")
				handle:close()
				os.exit(result == "yes" and 0 or 1)
			end
		end

		if flag == "-z" then
			os.exit(#path == 0 and 0 or 1)
		elseif flag == "-n" then
			os.exit(#path > 0 and 0 or 1)
		elseif flag == "-n" and op:match("^%S+$") then
			os.exit(#op > 0 and 0 or 1)
		end
	end

	local handle = io.popen("test " .. expr .. " 2>/dev/null && echo 'yes'")
	if handle then
		local result = handle:read("*a"):match("^%s*(.-)%s*$")
		handle:close()
		os.exit(result == "yes" and 0 or 1)
	else
		os.exit(1)
	end
end

local function builtin_export(args)
	if #args == 0 then
		-- Sort and print all environment variables like zsh
		local env_vars = {}
		for name, value in pairs(shell_vars) do
			table.insert(env_vars, {name=name, value=value})
		end
		-- Also get system env vars
		local handle = io.popen("env 2>/dev/null")
		if handle then
			for line in handle:lines() do
				local name, value = line:match("^([^=]+)=(.*)$")
				if name and not shell_vars[name] then
					table.insert(env_vars, {name=name, value=value})
				end
			end
			handle:close()
		end
		-- Sort by name
		table.sort(env_vars, function(a, b) return a.name < b.name end)
		for _, v in ipairs(env_vars) do
			if config.colors then
				print(c("green") .. "export " .. c("yellow") .. v.name .. c("reset") .. "=" .. c("pink") .. "\"" .. v.value .. "\"" .. c("reset"))
			else
				print("export " .. v.name .. "=\"" .. v.value .. "\"")
			end
		end
		return
	end

	for _, arg in ipairs(args) do
		local name, value = arg:match("^([^=]+)=(.*)$")
		if name then
			-- Remove quotes if present
			if value:match("^\"") and value:match("\"$") then
				value = value:sub(2, -2)
			elseif value:match("^'") and value:match("'$") then
				value = value:sub(2, -2)
			end
			
			-- Expand variables in value
			value = expand_env_vars(value)
			
			shell_vars[name] = value
			set_env(name, value)
			
			-- Special handling for PATH
			if name == "PATH" then
				refresh_external_commands()
			end
		else
			local val = shell_vars[arg] or os.getenv(arg)
			if val then
				if config.colors then
					print(c("green") .. "export " .. c("yellow") .. arg .. c("reset") .. "=" .. c("pink") .. "\"" .. val .. "\"" .. c("reset"))
				else
					print("export " .. arg .. "=\"" .. val .. "\"")
				end
			end
		end
	end
end

local function builtin_alias(args)
	if #args == 0 then
		for name, value in pairs(aliases) do
			if config.colors then
				print(c("green") .. "alias " .. c("yellow") .. name .. c("reset") .. "='" .. value .. "'")
			else
				print("alias " .. name .. "='" .. value .. "'")
			end
		end
		for name, value in pairs(global_aliases) do
			if config.colors then
				print(c("green") .. "alias " .. c("yellow") .. name .. c("reset") .. "='" .. value .. "'")
			else
				print("alias " .. name .. "='" .. value .. "'")
			end
		end
		return
	end

	for _, arg in ipairs(args) do
		local name, value = arg:match("^([^=]+)%s*=%s*(.+)$")
		if name then
			aliases[name] = value
		end
	end
end

local function builtin_unalias(args)
	if #args == 0 then
		return
	end

	for _, name in ipairs(args) do
		aliases[name] = nil
		global_aliases[name] = nil
	end
end

local function builtin_path(args)
	if #args == 0 then
		-- Print PATH in zsh-like format
		local path = os.getenv("PATH") or ""
		local paths = {}
		for dir in path:gmatch("[^:]+") do
			table.insert(paths, dir)
		end
		for i, dir in ipairs(paths) do
			if config.colors then
				print(c("yellow") .. i .. c("reset") .. " " .. c("blue") .. dir .. c("reset"))
			else
				print(i .. " " .. dir)
			end
		end
		return
	end
	
	local cmd = args[1]
	if cmd == "add" then
		local new_path = args[2]
		if new_path then
			new_path = expand_tilde(new_path)
			local path = os.getenv("PATH") or ""
			if path:find(new_path, 1, true) == nil then
				os.execute("export PATH=\"" .. new_path .. ":$PATH\" 2>/dev/null")
				set_env("PATH", new_path .. ":" .. path)
			end
		end
	elseif cmd == "remove" then
		local rm_path = args[2]
		if rm_path then
			rm_path = expand_tilde(rm_path)
			local path = os.getenv("PATH") or ""
			local new_path = path:gsub(rm_path .. ":?", "", 1)
			new_path = new_path:gsub(":+", ":"):gsub(":$", "")
			os.execute("export PATH=\"" .. new_path .. "\" 2>/dev/null")
			set_env("PATH", new_path)
		end
	end
	
	refresh_external_commands()
end

local function builtin_source(args)
	local path = args[1]
	if not path then
		if config.colors then
			print(c("red") .. "source: missing file operand" .. c("reset"))
		else
			print("source: missing file operand")
		end
		return
	end

	local file = io.open(path, "r")
	if not file then
		if config.colors then
			print(c("red") .. "source: " .. path .. ": No such file" .. c("reset"))
		else
			print("source: " .. path .. ": No such file")
		end
		return
	end
	local content = file:read("*all")
	file:close()

	local env = {}
	setmetatable(env, { __index = _G })
	local chunk, err = load(content, "source", "t", env)
	if not chunk then
		if config.colors then
			print(c("red") .. "source: " .. err .. c("reset"))
		else
			print("source: " .. err)
		end
		return
	end

	pcall(chunk)
end

local function builtin_set(args)
	if #args == 0 then
		for name, value in pairs(shell_vars) do
			print(name .. "=" .. value)
		end
		return
	end

	for _, arg in ipairs(args) do
		if arg == "-e" or arg == "-x" or arg == "-v" then
			-- ignore set options for now
		else
			local name, value = arg:match("^([^=]+)=(.*)$")
			if name then
				shell_vars[name] = value
				set_env(name, value)
			end
		end
	end
end

local function builtin_unset(args)
	for _, name in ipairs(args) do
		shell_vars[name] = nil
		set_env(name, nil)
	end
end

local function builtin_pushd(args)
	local path = args[1]

	if not path then
		if #dir_stack > 1 then
			local new_dir = dir_stack[2]
			table.remove(dir_stack, 1)
			builtin_cd({ _raw = true, new_dir })
			table.insert(dir_stack, 1, current_dir)
		end
		return
	end

	table.insert(dir_stack, 1, current_dir)
	builtin_cd(args)
end

local function builtin_popd(args)
	if #dir_stack <= 1 then
		if config.colors then
			print(c("red") .. "popd: directory stack empty" .. c("reset"))
		else
			print("popd: directory stack empty")
		end
		return
	end

	local prev_dir = table.remove(dir_stack, 1)
	builtin_cd({ _raw = true, prev_dir })
end

local function builtin_dirs(args)
	for i, dir in ipairs(dir_stack) do
		if config.colors then
			if i == 1 then
				print(c("green") .. dir .. c("reset") .. c("white") .. " (top)" .. c("reset"))
			else
				print(dir)
			end
		else
			if i == 1 then
				print(dir .. " (top)")
			else
				print(dir)
			end
		end
	end
end

local function builtin_jobs(args)
	local has_jobs = false
	for id, job in pairs(jobs) do
		has_jobs = true
		local status = job.completed and "Done" or "Running"
		local mark = job.completed and "+" or "-"
		if config.colors then
			print(c("white") .. "[" .. id .. "]" .. c("reset") .. "  " .. status .. "  " .. mark .. "  " .. job.cmd)
		else
			print("[" .. id .. "]  " .. status .. "  " .. mark .. "  " .. job.cmd)
		end
	end

	if not has_jobs then
		if config.colors then
			print(c("yellow") .. "jobs: no jobs" .. c("reset"))
		else
			print("jobs: no jobs")
		end
	end
end

local function builtin_fg(args)
	local job_id = tonumber(args[1]) or 1

	local job = jobs[job_id]
	if not job then
		if config.colors then
			print(c("red") .. "fg: " .. job_id .. ": no such job" .. c("reset"))
		else
			print("fg: " .. job_id .. ": no such job")
		end
		return
	end

	if job.completed then
		if config.colors then
			print(c("yellow") .. job.cmd .. c("reset"))
		else
			print(job.cmd)
		end
		jobs[job_id] = nil
	else
		print(job.cmd)
	end
end

local function builtin_bg(args)
	local job_id = tonumber(args[1]) or 1

	local job = jobs[job_id]
	if not job then
		if config.colors then
			print(c("red") .. "bg: " .. job_id .. ": no such job" .. c("reset"))
		else
			print("bg: " .. job_id .. ": no such job")
		end
		return
	end

	print("[" .. job_id .. "] " .. job.cmd .. " &")
end

local function builtin_type(args)
	if #args == 0 then
		return
	end

	local cmd = args[1]

	if builtin_cmds[cmd] then
		if config.colors then
			print(cmd .. c("white") .. " is a shell builtin" .. c("reset"))
		else
			print(cmd .. " is a shell builtin")
		end
		return
	end

	if aliases[cmd] then
		if config.colors then
			print(cmd .. c("white") .. " is an alias for " .. c("yellow") .. aliases[cmd] .. c("reset"))
		else
			print(cmd .. " is an alias for " .. aliases[cmd])
		end
		return
	end

	local path = os.getenv("PATH")
	if path then
		for dir in path:gmatch("[^:]+") do
			local full_path = dir .. "/" .. cmd
			local handle =
				io.popen("test -f '" .. full_path .. "' -a -x '" .. full_path .. "' 2>/dev/null && echo 'yes'")
			if handle then
				local result = handle:read("*a"):match("^%s*(.-)%s*$")
				handle:close()
				if result == "yes" then
					if config.colors then
						print(cmd .. c("white") .. " is " .. c("green") .. full_path .. c("reset"))
					else
						print(cmd .. " is " .. full_path)
					end
					return
				end
			end
		end
	end

	if config.colors then
		print(c("red") .. "type: " .. cmd .. ": not found" .. c("reset"))
	else
		print("type: " .. cmd .. ": not found")
	end
end

local builtins = {
	cd = builtin_cd,
	pwd = builtin_pwd,
	exit = builtin_exit,
	help = builtin_help,
	echo = builtin_echo,
	printf = builtin_printf,
	test = builtin_test,
	export = builtin_export,
	alias = builtin_alias,
	unalias = builtin_unalias,
	path = builtin_path,
	source = builtin_source,
	set = builtin_set,
	unset = builtin_unset,
	pushd = builtin_pushd,
	popd = builtin_popd,
	dirs = builtin_dirs,
	jobs = builtin_jobs,
	fg = builtin_fg,
	bg = builtin_bg,
	type = builtin_type,
	["true"] = function()
		os.exit(0)
	end,
	["false"] = function()
		os.exit(1)
	end,
	["return"] = function() end,
}

local function builtin_help()
	if config.colors then
		print(c("lavender") .. "neoshell - An Enhanced Lua Shell" .. c("reset"))
		print()
		print(c("yellow") .. "Built-in commands:" .. c("reset"))
		print("  " .. c("green") .. "cd [dir]" .. c("reset") .. "       Change directory (supports ~, -, cdpath)")
		print("  " .. c("green") .. "pwd" .. c("reset") .. "          Print working directory")
		print("  " .. c("green") .. "pushd [dir]" .. c("reset") .. "   Push directory to stack and cd")
		print("  " .. c("green") .. "popd" .. c("reset") .. "         Pop directory from stack")
		print("  " .. c("green") .. "dirs" .. c("reset") .. "         Display directory stack")
		print("  " .. c("green") .. "exit [n]" .. c("reset") .. "      Exit shell with optional exit code")
		print("  " .. c("green") .. "echo [args]" .. c("reset") .. "   Print arguments")
		print("  " .. c("green") .. "printf [fmt]" .. c("reset") .. " Printf-like output")
		print("  " .. c("green") .. "test [expr]" .. c("reset") .. "  Test expression")
		print("  " .. c("green") .. "export [var[=val]]" .. c("reset") .. "  Set environment variables")
		print("  " .. c("green") .. "alias [name=cmd]" .. c("reset") .. " Define/list aliases")
		print("  " .. c("green") .. "unalias [name]" .. c("reset") .. " Remove alias")
		print("  " .. c("green") .. "source [file]" .. c("reset") .. " Execute commands from file")
		print("  " .. c("green") .. "set [var=val]" .. c("reset") .. "  Set/list shell variables")
		print("  " .. c("green") .. "unset [var]" .. c("reset") .. "  Unset shell variable")
		print("  " .. c("green") .. "type [cmd]" .. c("reset") .. "  Show command type")
		print("  " .. c("green") .. "jobs" .. c("reset") .. "         List background jobs")
		print("  " .. c("green") .. "fg [job]" .. c("reset") .. "     Bring job to foreground")
		print("  " .. c("green") .. "bg [job]" .. c("reset") .. "     Resume job in background")
		print("  " .. c("green") .. "help" .. c("reset") .. "         Show this help message")
		print()
		print(c("yellow") .. "Features:" .. c("reset"))
		print("  " .. c("peach") .. "$VAR, ${VAR}" .. c("reset") .. "      Environment variable expansion")
		print("  " .. c("peach") .. "$(cmd), `cmd`" .. c("reset") .. "   Command substitution")
		print("  " .. c("peach") .. "$((expr))" .. c("reset") .. "       Arithmetic expansion")
		print("  " .. c("peach") .. "!, !!, !pattern" .. c("reset") .. "  History expansion")
		print("  " .. c("peach") .. "*.txt, **/*.lua" .. c("reset") .. "   Glob expansion")
		print("  " .. c("peach") .. "%, %:h, %:t, %:r" .. c("reset") .. "  Word modifiers")
		print()
		print(c("yellow") .. "Configuration:" .. c("reset"))
		print("  Place a .neoshell file in your home directory with:")
		print("    return {")
		print('        prompt = "%u@%h %w $ ",')
		print('        aliases = { ll = "ls -la" },')
		print('        global_aliases = { G = "| grep" },')
		print('        cdpath = {".", "..", "~/", "~/projects"},')
		print("        colors = true,")
		print('        theme = { base = "#b4befe" },')
		print('        history_file = "~/.neoshell_history",')
		print("        history_size = 500")
		print("    }")
	else
		print([[neoshell - An Enhanced Lua Shell

Built-in commands:
  cd [dir]       Change directory (supports ~, -, cdpath)
  pwd            Print working directory
  pushd [dir]    Push directory to stack and cd
  popd           Pop directory from stack
  dirs           Display directory stack
  exit [n]       Exit shell with optional exit code
  echo [args]    Print arguments
  printf [fmt]   Printf-like output
  test [expr]    Test expression
  export [var]   Set environment variables
  alias [name]   Define/list aliases
  unalias [name] Remove alias
  source [file]  Execute commands from file
  set [var]      Set/list shell variables
  unset [var]    Unset shell variable
  type [cmd]     Show command type
  jobs           List background jobs
  fg [job]       Bring job to foreground
  bg [job]       Resume job in background
  help           Show this help message

Features:
  $VAR, ${VAR}      Environment variable expansion
  $(cmd), `cmd`     Command substitution
  $((expr))         Arithmetic expansion
  !, !!, !pattern   History expansion
  *.txt, **/*.lua   Glob expansion
  %:h, %:t, %:r     Word modifiers

Configuration:
  Place a .neoshell file in your home directory with:
    return {
        prompt = "%u@%h %w $ ",
        aliases = { ll = "ls -la" },
        global_aliases = { G = " | grep" },
        cdpath = {".", "..", "~", "~/projects"},
        colors = true,
        theme = { base = "#b4befe" },
        history_file = "~/.neoshell_history",
        history_size = 500
    }]])
	end
end

builtins.help = builtin_help

local function expand_alias(cmd)
	if aliases[cmd] then
		return aliases[cmd]
	end

	for pattern, replacement in pairs(global_aliases) do
		if cmd:match(pattern) then
			return cmd:gsub(pattern, replacement)
		end
	end

	return cmd
end

local function parse_command(line)
	local tokens = {}
	local current = ""
	local in_single_quote = false
	local in_double_quote = false

	for i = 1, #line do
		local char = line:sub(i, i)

		if char == "'" and not in_double_quote then
			in_single_quote = not in_single_quote
		elseif char == '"' and not in_single_quote then
			in_double_quote = not in_double_quote
		elseif char == " " and not in_single_quote and not in_double_quote then
			if current ~= "" then
				table.insert(tokens, current)
				current = ""
			end
		else
			current = current .. char
		end
	end

	if current ~= "" then
		table.insert(tokens, current)
	end

	return tokens
end

local function is_background(tokens)
	for i = #tokens, 1, -1 do
		if tokens[i] == "&" then
			table.remove(tokens, i)
			return true
		end
	end
	return false
end

local function run_background(cmd)
	local job_id = job_id_counter
	job_id_counter = job_id_counter + 1

	jobs[job_id] = {
		cmd = cmd,
		completed = false,
	}

	if config.colors then
		print("[" .. job_id .. "] " .. c("blue") .. cmd .. c("reset") .. " &")
	else
		print("[" .. job_id .. "] " .. cmd .. " &")
	end

	local handle = io.popen(cmd .. " 2>&1 &")
	if handle then
		handle:close()
	end

	jobs[job_id].completed = true
end

local function execute_command(line)
	line = line:gsub("^%s+", ""):gsub("%s+$", "")

	if line == "" then
		return
	end

	line = expand_line(line)

	line = expand_alias(line)

	local tokens = parse_command(line)
	local cmd = tokens[1]

	if builtins[cmd] then
		table.remove(tokens, 1)
		builtins[cmd](tokens)
		return
	end

	local is_bg = is_background(tokens)
	if is_bg then
		local full_cmd = table.concat(tokens, " ")
		run_background(full_cmd)
		return
	end

	-- Auto-cd: if cmd is a directory and auto_cd is enabled
	if config.auto_cd and not builtins[cmd] then
		local expanded = expand_tilde(cmd)
		local attr = lfs_available and lfs.attributes(expanded)
		if attr and attr.mode == "directory" then
			builtin_cd({ cmd })
			return
		end
		-- Also check if it's a relative path
		if lfs_available then
			local full = current_dir .. "/" .. cmd
			attr = lfs.attributes(full)
			if attr and attr.mode == "directory" then
				builtin_cd({ cmd })
				return
			end
		end
	end

	local full_cmd = cmd .. " " .. table.concat(tokens, " ", 2)

	-- Check if this is a TUI (Terminal UI) application that needs direct terminal access
	local tui_apps = {
		vim = true,
		nvim = true,
		vi = true,
		nano = true,
		emacs = true,
		less = true,
		more = true,
		man = true,
		htop = true,
		top = true,
		btm = true,
		btop = true,
		yazi = true,
		ranger = true,
		nnn = true,
		lf = true,
		fzf = true,
		gh = true,
		lazygit = true,
		tig = true,
		ncdu = true,
		mc = true,
		lynx = true,
		w3m = true,
	}

	local is_tui = tui_apps[cmd]

	if is_tui then
		-- TUI apps need direct terminal access - use os.execute instead of io.popen
		-- Restore terminal settings for TUI apps
		if linenoise_available then
			-- Restore raw mode before running TUI app
			os.execute("stty sane 2>/dev/null")
		end

		-- Execute directly without capturing output
		local exit_code = os.execute(full_cmd .. " 2>&1")

		-- Re-initialize linenoise if available
		if linenoise_available then
			linenoise.historysetmaxlen(config.history_size)
		end

		return
	end

	-- For non-TUI apps, use io.popen to capture output
	local handle = io.popen(full_cmd .. " 2>&1")
	if handle then
		local output = handle:read("*a")
		handle:close()

		if output ~= "" then
			io.write(output)
		end
	else
		-- Better error message with typo suggestions
		if config.colors then
			io.write(c("red") .. "sh: " .. cmd .. ": command not found" .. c("reset"))
		else
			io.write("sh: " .. cmd .. ": command not found")
		end

		local suggestion = find_closest_command(cmd)
		if suggestion then
			if config.colors then
				print(c("yellow") .. "  Did you mean: " .. suggestion .. "?" .. c("reset"))
			else
				print("  Did you mean: " .. suggestion .. "?")
			end
		else
			print()
		end
	end
end

local function repl()
	load_history()

	local function add_and_execute(line)
		add_to_history(line)
		local success, err = pcall(execute_command, line)
		if not success then
			if config.colors then
				print(c("red") .. "Error: " .. tostring(err) .. c("reset"))
			else
				print("Error: " .. tostring(err))
			end
		end
	end

	if linenoise_available then
		linenoise.historysetmaxlen(config.history_size)
		linenoise.historyadd("")

		while true do
			local prompt = format_prompt()
			local line = linenoise.line(prompt)

			if not line then
				print()
				break
			end

			if line ~= "" then
				add_and_execute(line)
			end
		end
	else
		while true do
			io.write(format_prompt())
			io.flush()

			local line = io.read()
			if not line then
				print()
				break
			end

			add_and_execute(line)
		end
	end

	save_history()
end

load_config()
init_colors()
repl()