local function nextcol(state)
	local from, to = state.str:find(state.pattern, state.last)
	if from then
		state.last = to+1
		return from, to
	end
end

local function unquote(args)
	local result = {}
	local term
	if args[1]:find([=[^["']]=]) then
		term = args[1]:sub(1, 1) .. "$"
	elseif args[1]:find("^%[=*%[") then
		local num = args[1]:find("[", 2, true) - 2
		term = "]" .. string.rep("=", num) .. "]$"
	else
		return args
	end
	for _, arg in ipairs(args) do
		table.insert(result, arg)
		if term and arg:find(term) then
			result = {table.concat(result, " "):sub(#term, -#term)}
			term = nil
		end
	end
	return result
end

local function colmatch(str, pattern)
	return nextcol, {str=str,pattern=pattern,last=0}
end

local function findLocations(args)
	local files = vim.fs.find(function(file)
		if #args>1 then
			for _, pattern in ipairs(args), args, 1 do
				if vim.fs.basename(file):find(pattern) then
					return true
				end
			end
		else
			return true
		end
		return false
	end, {
		limit=math.huge;
		type="file";
	})

	local locations = {}

	for _, file in ipairs(files) do
		local lnum = 0
		for line in io.lines(file) do
			lnum = lnum + 1
			for from, to in colmatch(line, args[1]) do
				table.insert(locations, {
					filename = file;
					lnum = lnum;
					col = from;
					end_col = to;
					text = line:sub(from, to)
				})
			end
		end
	end

	return locations
end

vim.api.nvim_create_user_command("Matcha", function(params)
	vim.fn.setqflist(findLocations(unquote(params.fargs)))
	vim.cmd("copen")
	vim.cmd("crewind")
end, {nargs="+"})

vim.api.nvim_create_user_command("MatchaLocal", function(params)
	vim.fn.setloclist(0, findLocations(unquote(params.fargs)))
	vim.cmd("lopen")
	vim.cmd("lrewind")
end, {nargs="+"})