local a = require("./utils/ansi")
local strings = require("./utils/strings")
local debug = require("./utils/print_r")
local calculations = require("./utils/calculations")

local maze2d = require("./maze")
local maze3d = require("./maze3d")
local openscad = require("./openscad")

-- ANSI escape sequence helpers
function a.header(str) return a.fblue(a.hicol(str)) end
function a.item(str) return a.fyellow(str) end
function a.value(str) return a.fgreen(str) end

local settings = {
	mode = "help",
	extras = {},
	
	width = 11,
	height = 11,
	depth = 7,
	
	branching_factor = nil, -- Depends on the mode - nil will cause the engine to pick the default mode instead
	
	seed = os.time(),
	-- TODO: Add branching factor setting here & associated CLI argument
	filename = nil,
	
	scale = 5,
	inverted = true,
	
	colour = "#6495ED", -- Cornflower blue :D (if you get the reference, get in touch for a virtual cookie :D)
	
	path_length = 2,
	path_width = 1,
	path_depth = 1
}

local function help()
	--lua main.lua ]]..a.item("maze")..[[ [--width <int:11>] [--height <int:11>] [--seed <string>] [--path-length <int:2>] [--path-width <int:1>] [--branching-factor <int:6>]
	--lua main.lua ]]..a.item("maze3d")..[[ [--width <int:11>] [--height <int:11>] [--depth <int:7>] [--seed <string>] [--path-length <int:2>] [--path-width <int:1>] [--path-depth <int:1>] [--branching-factor <int:6>]
	print([[
]]..a.fgreen(a.hicol("multimaze: "))..a.fyellow(a.hicol("Multidimensional maze generator"))..[[

	]]..a.locol("By Starbeamrainbowlabs")..[[


]]..a.header("Usage:")..[[

	lua main.lua ]]..a.item("{mode} [{options}]")..[[


]]..a.header("Modes:")..[[

    ]]..a.item("help")..[[             Displays this message
    ]]..a.item("maze")..[[             Generates a 2d maze
    ]]..a.item("maze_svg")..[[         Generates a 2d maze as an SVG file
    ]]..a.item("maze3d")..[[           Generates a 3d maze
    ]]..a.item("maze3d_openscad")..[[  Generates a 3d maze as an openscad file

]]..a.header("Options:")..[[

        ]]..a.item("--help")..[[			Shows this message
    ]]..a.item("-w  --width")..a.value(" {int}")..[[		Sets the width of the maze
    ]]..a.item("-h  --height")..a.value(" {int}")..[[		Sets the height of the maze
    ]]..a.item("-d  --depth")..a.value(" {int}")..[[		Sets the depth of the maze ]]..a.locol("[3d only]")..[[

    ]]..a.item("-s  --seed")..a.value(" {string}")..[[		Sets the seed
    ]]..a.item("-f  --filename")..a.value(" {string}")..[[	Sets the output filename
        ]]..a.item("--toggle-invert")..[[		Toggle inversion of the output ]]..a.locol("[maze_svg, maze3d_openscad only]")..[[

    ]]..a.item("-m  --scale")..a.value(" {int}")..[[		Sets the scale of the output ]]..a.locol("[maze_svg, openscad output only]")..[[

    ]]..a.item("-l  --path-length")..a.value(" {int}")..[[	Sets the path length
    ]]..a.item("-y  --path-width")..a.value(" {int}")..[[	Sets the path width
    ]]..a.item("-z  --path-depth")..a.value(" {int}")..[[	Sets the path depth ]]..a.locol("[3d only]")..[[

    ]]..a.item("-b  --branching-factor")..a.value(" {int}")..[[Sets the path branching factor (higher = fewer branching paths, default: 6 for 2D, 3 for 3D)
        ]]..a.item("--colour")..a.value(" {string}")..[[	Sets the colour ]]..a.locol("[maze_svg only]")..[[

		
]]..a.header("Environment Variables:")..[[

	]]..a.item("NO_COLOR")..[[         Disable ANSI escape sequences (aka coloured output)

]])
end

for i=1,#arg do
	if not strings.starts_with(arg[i], "-") then
		table.insert(settings.extras, arg[i])
	elseif arg[i] == "--help" then
		help()
		os.exit()	
	elseif arg[i] == "--width" or arg[i] == "-w" then
		i = i + 1
		settings.width = tonumber(arg[i])
	elseif arg[i] == "--height" or arg[i] == "-h" then
		i = i + 1
		settings.height = tonumber(arg[i])
	elseif arg[i] == "--depth" or arg[i] == "-d" then
		i = i + 1
		settings.depth = tonumber(arg[i])
	elseif arg[i] == "--branching-factor" or arg[i] == "-b" then
		i = i + 1
		settings.branching_factor = tonumber(arg[i])
	elseif arg[i] == "--seed" or arg[i] == "-s" then
		i = i + 1
		settings.seed = strings.makeseed(arg[i])
	elseif arg[i] == "--filename" or arg[i] == "-f" then
		i = i + 1
		settings.filename = arg[i]
	elseif arg[i] == "--colour" then
		i = i + 1
		settings.colour = arg[i]
	elseif arg[i] == "--scale" or arg[i] == "-m" then
		i = i + 1
		settings.scale = tonumber(arg[i])
	elseif arg[i] == "--toggle-invert" then
		settings.inverted = false
	elseif arg[i] == "--path-length" or arg[i] == "-l" then
		i = i + 1
		settings.path_length = tonumber(arg[i])
	elseif arg[i] == "--path-width" or arg[i] == "-y" then
		i = i + 1
		settings.path_width = tonumber(arg[i])
	elseif arg[i] == "--path-depth" or arg[i] == "-z" then
		i = i + 1
		settings.path_depth = tonumber(arg[i])
	end
end

if #settings.extras > 0 then
	settings.mode = table.remove(settings.extras, 1)
end


-------------------------------------------------------------------------------


local function maze_text()
	local world, time = maze2d.generate_maze(
		settings.seed,
		settings.width,
		settings.height,
		settings.path_length,
		settings.path_width,
		settings.branching_factor
	)
	maze2d.printspace(world, settings.width, settings.height)
	print("Generation completed in " .. time .. "ms with seed "..settings.seed..".")
end

local function maze_svg()
	local world, time = maze2d.generate_maze(
		settings.seed,
		settings.width,
		settings.height,
		settings.path_length,
		settings.path_width,
		settings.branching_factor
	)
	
	local svg_settings = {
		scale = settings.scale,
		colour = settings.colour,
		inverted = settings.inverted
	}
	
	local svg = maze2d.make_svg(world, settings.width, settings.height, svg_settings)
	if not settings.filename then
		print(svg)
	else
		local out = io.open(settings.filename, "w")
		out:write(svg)
		out:close()
		print("Generation completed in "..time.."ms with seed "..settings.seed..".")
	end
end

local function maze3d_text()
	local world, time = maze3d.generate_maze(
		settings.seed,
		settings.width,
		settings.height,
		settings.depth,
		settings.path_length,
		settings.path_width,
		settings.path_depth,
		settings.branching_factor
	)
	
	maze3d.printspace(world, settings.width, settings.height, settings.depth)
	
	print("Generation completed in " .. time .. "ms with seed "..settings.seed..".")
end

local function maze3d_openscad()
	local world, time = maze3d.generate_maze(
		settings.seed,
		settings.width,
		settings.height,
		settings.depth,
		settings.path_length,
		settings.path_width,
		settings.path_depth,
		settings.branching_factor
	)
	
	if not settings.filename then
		print(a.fred(a.hicol("Error: No output filename specified.")))
		return false
	end
	
	local openscad_settings = {
		scale = settings.scale,
		inverted = settings.inverted
	}
	
	openscad.write3d(
		world,
		settings.width, settings.height, settings.depth,
		settings.filename,
		openscad_settings
	)
	print("Maze written to "..a.fgreen(a.hicol(settings.filename)))
end

local mode_functions = {
	["help"] = help,
	["maze"] = maze_text,
	["maze_svg"] = maze_svg,
	["maze3d"] = maze3d_text,
	["maze3d_openscad"] = maze3d_openscad
}

local selection = mode_functions[settings.mode]
if not selection then
	print(a.fred(a.hicol("Error: Unrecognised mode '"..a.FBLE..settings.mode..a.FRED.."'")))
	print()
	selection = mode_functions["help"]
end

selection()