Start implementation of :CrazywallDry

Author: gitpushjoe

lua/crazywall/init.lua

@@ -3,9 +3,12 @@ local Context = require("core.context")
 local streams = require("core.streams")
 local fold = require("core.fold")
 local utils = require("core.utils")
+local validate = require("crazywall.validate")
+local default_config = require("core.defaults.config")
 local M = {}
 
 local configs = {}
+local current_config_name = "DEFAULT"
 
 local output_options = { "both", "planonly", "textonly" }
 local on_unsaved_options = { "warn", "write" }
@@ -15,7 +18,7 @@ local err_buf = nil
 
 ---@param path string?
 ---@return number?
-local function find_buffer(path)
+local find_buffer = function(path)
 	for _, buf_id in ipairs(vim.api.nvim_list_bufs()) do
 		local bufname = vim.api.nvim_buf_get_name(buf_id)
 		if bufname == path then
@@ -25,6 +28,90 @@ local function find_buffer(path)
 	return nil
 end
 
+---@param path string
+---@param text string
+---@return string? errmsg
+local write = function(path, text)
+	local dir = vim.fn.fnamemodify(path, ":p:h")
+	if vim.fn.isdirectory(dir) == 0 then
+		vim.fn.mkdir(dir, "p")
+	end
+
+	local file = io.open(path, "w")
+	if file then
+		local _, err = file:write(text)
+		file:close()
+		if err then
+			return err
+		end
+		return
+	end
+	return "Unable to write to " .. path
+end
+
+---@return string
+local get_plan_path = function()
+	return string.format(
+		"%s/crazywall/plan-%s.txt",
+		vim.fn.stdpath("data"),
+		os.time()
+	)
+end
+
+---@param lines string[]
+---@return number? buf_id
+---@return string? errmsg
+local display_floating_window = function(lines)
+	local plan_path_prefix =
+		string.format("%s/crazywall/plan-", vim.fn.stdpath("data"))
+	for _, buf_id in ipairs(vim.api.nvim_list_bufs()) do
+		local buf_name = vim.api.nvim_buf_get_name(buf_id)
+		if utils.str.starts_with(buf_name, plan_path_prefix) then
+			vim.api.nvim_buf_delete(buf_id, { force = true })
+		end
+	end
+	local buf_id = vim.api.nvim_create_buf(false, true)
+	if not buf_id then
+		return nil, "Failed to create buffer"
+	end
+	vim.api.nvim_buf_set_name(buf_id, get_plan_path())
+	vim.api.nvim_buf_set_option(buf_id, "buftype", "")
+	vim.api.nvim_buf_set_lines(buf_id, 0, -1, false, lines)
+	local editor_height = vim.api.nvim_get_option("lines")
+	local editor_width = vim.api.nvim_get_option("columns")
+	local win_height = math.min(math.floor(editor_height * 0.75), #lines)
+	local longest_line_length = (function()
+		local res = 0
+		for _, line in ipairs(lines) do
+			res = math.max(res, #line)
+		end
+		return res
+	end)()
+	local win_width =
+		math.min(math.floor(editor_width * 0.75), longest_line_length)
+	local row = math.floor((editor_height - win_height) / 2)
+	local col = math.floor((editor_width - win_width) / 2)
+
+	local win = vim.api.nvim_open_win(buf_id, true, {
+		relative = "editor",
+		width = win_width,
+		height = win_height,
+		row = row,
+		col = col,
+		border = "rounded",
+		style = "minimal",
+	})
+	if not win then
+		return nil, "Failed to open floating window."
+	end
+	vim.api.nvim_buf_set_option(buf_id, "modifiable", false)
+
+	local default_bg = vim.api.nvim_get_hl(0, { name= 'Normal'}).background
+	vim.api.nvim_set_hl(0, 'NormalFloat', { bg = default_bg })
+
+	return buf_id
+end
+
 ---@param err_text string?
 local display_err = function(err_text)
 	local err = err_text and utils.str.split_lines_to_list(err_text or "") or {}
@@ -33,6 +120,8 @@ local display_err = function(err_text)
 	end
 	local total_height = vim.api.nvim_get_option("lines") -- Total height of the Neovim window
 	local split_height = math.floor(total_height / 3)
+	split_height = math.min(#err, split_height)
+	split_height = math.max(split_height, 3)
 	vim.cmd("botright new")
 	local win_id = vim.api.nvim_get_current_win()
 	err_buf = vim.api.nvim_get_current_buf()
@@ -43,32 +132,91 @@ local display_err = function(err_text)
 	vim.api.nvim_buf_set_option(err_buf, "modifiable", false)
 end
 
+---@param buf_id integer?
+---@param ctx Context
+---@return Plan? plan
+---@return string? errmsg
+local do_fold = function(buf_id, ctx)
+	local root, err = fold.parse(ctx)
+	if not root then
+		return nil, err
+	end
+
+	_, err = fold.prepare(root, ctx)
+	if err then
+		return nil, err
+	end
+
+	local plan
+	plan, err = fold.execute(root, ctx, true)
+	if not plan then
+		return nil, err
+	end
+
+	if ctx.is_dry_run then
+		return plan
+	end
+
+	plan, err = fold.execute(root, ctx, false)
+	if err then
+		error(err)
+	end
+
+	if buf_id then
+		vim.api.nvim_buf_call(buf_id, function()
+			vim.cmd("edit")
+		end)
+	end
+
+	return plan
+end
+
+---@param buf_id integer?
+---@param write_on_unsaved boolean
+---@return boolean
+local handle_unsaved = function(buf_id, write_on_unsaved)
+	if not buf_id then
+		return true
+	end
+	if not vim.api.nvim_buf_get_option(buf_id, "modified") then
+		return true
+	end
+	if write_on_unsaved then
+		vim.api.nvim_buf_call(buf_id, function()
+			vim.cmd("write")
+		end)
+		return true
+	end
+	vim.api.nvim_err_writeln("crazywall: Buffer has unsaved changes.")
+	return false
+end
+
 vim.api.nvim_create_user_command("CrazywallQuick", function(opts)
-	local on_unsaved = opts.fargs[2] or "warn"
-	local src_path = opts.fargs[3] or vim.fn.expand("%")
-	local dest_path = opts.fargs[4] or vim.fn.expand("%")
+	local on_unsaved = opts.fargs[1] or "warn"
+	local err = validate.string_in_list(on_unsaved, on_unsaved_options)
+	if err then
+		vim.api.nvim_err_writeln(err)
+		return
+	end
 
-	local buf_id = find_buffer(src_path)
+	local src_path = opts.fargs[2] or vim.fn.expand("%")
+	local dest_path = opts.fargs[3] or vim.fn.expand("%")
 
-	if
-		buf_id
-		and on_unsaved == "warn"
-		and vim.api.nvim_buf_get_option(buf_id, "modified")
-	then
-		vim.api.nvim_err_writeln("crazywall: Buffer has unsaved changes.")
+	local buf_id = find_buffer(src_path)
+	if not handle_unsaved(buf_id, on_unsaved == "write") then
 		return
-	else
-		if buf_id and vim.api.nvim_buf_get_option(buf_id, "modified") then
-			vim.api.nvim_buf_call(buf_id, function()
-				vim.cmd("write")
-			end)
-		end
 	end
 
-	local config, err = Config:new({})
+	if configs[current_config_name] == nil then
+		return vim.api.nvim_err_writeln(
+			"crazywall: Could not find config " .. current_config_name .. "."
+		)
+	end
+
+	local config
+	config, err = Config:new(configs[current_config_name])
 	if not config then
-		display_err(err)
-		return
+		return display_err(err)
 	end
 
 	local ctx
@@ -87,52 +235,120 @@ vim.api.nvim_create_user_command("CrazywallQuick", function(opts)
 	)
 
 	if not ctx then
-		display_err(err)
-		return
+		return display_err(err)
 	end
 
-	local root
-	root, err = fold.parse(ctx)
-	if not root then
-		display_err(err)
-		return
+	local plan
+	plan, err = do_fold(buf_id, ctx)
+	if not plan then
+		return display_err(err)
 	end
 
-	_, err = fold.prepare(root, ctx)
+	local plan_path = get_plan_path()
+	err = write(plan_path, tostring(plan))
+	if err then
+		return display_err(err)
+	end
+
+	print("crazywall: Logs written to " .. plan_path)
+end, {
+	nargs = "*",
+	complete = function(_, line)
+		local args = vim.split(line, " ")
+		if #args == 2 then
+			return on_unsaved_options
+		end
+		if #args == 3 then
+			return { vim.fn.expand("%") }
+		end
+		if #args == 4 then
+			return { vim.fn.expand("%") }
+		end
+		return {}
+	end,
+	desc = "Folds a file with crazywall, skipping the confirmation window.",
+})
+
+vim.api.nvim_create_user_command("CrazywallDry", function(opts)
+	local output_style = opts.fargs[1] or "both"
+	local on_unsaved = opts.fargs[2] or "warn"
+	local err = validate.string_in_list(on_unsaved, on_unsaved_options)
+		or validate.string_in_list(output_style, output_options)
 	if err then
-		display_err(err)
+		vim.api.nvim_err_writeln(err)
 		return
 	end
 
-	local plan
-	plan, err = fold.execute(root, ctx, true)
-	if not plan then
-		display_err(err)
+	local src_path = opts.fargs[3] or vim.fn.expand("%")
+	local dest_path = opts.fargs[4] or vim.fn.expand("%")
+
+	local buf_id = find_buffer(src_path)
+	if not handle_unsaved(buf_id, on_unsaved == "write") then
 		return
 	end
 
-	plan, err = fold.execute(root, ctx, false)
+	if configs[current_config_name] == nil then
+		return vim.api.nvim_err_writeln(
+			"crazywall: Could not find config " .. current_config_name .. "."
+		)
+	end
+
+	local config
+	config, err = Config:new(configs[current_config_name])
+	if not config then
+		return display_err(err)
+	end
+
+	local ctx
+	ctx, err = Context:new(
+		config,
+		src_path,
+		dest_path,
+		vim.fn.readfile(src_path),
+		nil,
+		true,
+		false,
+		(output_style == "plan-only" or output_style == "both")
+				and streams.STDOUT
+			or streams.NONE,
+		(output_style == "text-only" or output_style == "both")
+				and streams.STDOUT
+			or streams.NONE,
+		true,
+		false
+	)
+
+	if not ctx then
+		return display_err(err)
+	end
+
+	local plan
+	plan, err = do_fold(buf_id, ctx)
 	if err then
-		error(err)
+		return display_err(err)
 	end
 
-	if buf_id then
-		vim.api.nvim_buf_call(buf_id, function()
-			vim.cmd("edit")
-		end)
+	_, err =
+		display_floating_window(utils.str.split_lines_to_list(tostring(plan)))
+	if err then
+		return nil, err
 	end
-	print("Done!")
+
+	vim.cmd('setlocal nowrap')
 end, {
-	nargs = "*", -- Expect exactly 2 arguments
+	nargs = "*",
 	complete = function(_, line)
 		local args = vim.split(line, " ")
 		if #args == 2 then
 			return output_options
-		elseif #args == 3 then
+		end
+		if #args == 3 then
 			return on_unsaved_options
-		elseif #args == 4 then
+		end
+		if #args == 4 then
 			return { vim.fn.expand("%") }
-		elseif #args == 5 then
+		end
+		if #args == 5 then
 			return { vim.fn.expand("%") }
 		end
 		return {}
@@ -142,7 +358,16 @@ end, {
 
 M.setup = function(opts)
 	opts = opts or {}
+	local keys = { "configs", "default_config_name" }
+	for key in pairs(opts) do
+		local err = validate.string_in_list(key, keys)
+		if err then
+			error(err)
+		end
+	end
 	configs = opts.configs or configs
+	configs.DEFAULT = configs.DEFAULT or default_config
+	current_config_name = opts.default_config_name or current_config_name
 end
 
 M.foo = 42