init.lua

-- CyberEngineTWeaks Mod Manager is a mod manager for Cyber Engine Tweaks based mods

-- Copyright (C) 2021 Mingming Cui
--
-- This program is free software: you can redistribute it and/or modify
-- it under the terms of the GNU General Public License as published by
-- the Free Software Foundation, either version 3 of the License, or
-- (at your option) any later version.
--
-- This program is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-- GNU General Public License for more details.
--
-- You should have received a copy of the GNU General Public License
-- along with this program.  If not, see <https://www.gnu.org/licenses/>.

i18n = require("i18n")
languages = require("lang/lang")

function check_module()
  CETMM_fs = json.CETMM_fs
	if type(CETMM_fs) == "table" then
		return true
	else
		return false
	end
end

function scan_dofiles()
	local dofile_names = {}
	local dofile_list = dir("./dofiles")
	for i in pairs(dofile_list) do
		if dofile_list[i].type == "file" and dofile_list[i].name:match("(.+)%.lua$") then
			table.insert(dofile_names, dofile_list[i].name)
		end
	end
	return dofile_names
end

function draw_mod_list(mods_data)
	for i in pairs(mods_data) do
		ImGui.Spacing()
		ImGui.PushID(i)
		if mods_data[i].name == "cet_mod_manager" then
			pushstylecolor(ImGuiCol.Text, theme.TextDisabled)
			pushstylecolor(ImGuiCol.FrameBg, theme.FrameBgDisabled)
			pushstylecolor(ImGuiCol.FrameBgHovered, theme.FrameBgHoveredDisabled)
			pushstylecolor(ImGuiCol.FrameBgActive, theme.FrameBgActiveDisabled)
			pushstylecolor(ImGuiCol.CheckMark, theme.CheckMarkTrueDisabled)
			ImGui.Checkbox(i18n("window_title"), true)
			ImGui.PopStyleColor(5)
		elseif mods_data[i].state == true then
			mods_data[i].state, mods_data[i].pressed = ImGui.Checkbox(modNameConvert(mods_data[i].name), mods_data[i].state)
		elseif mods_data[i].state == false then
			mods_data[i].state, mods_data[i].pressed = ImGui.Checkbox(modNameConvert(mods_data[i].name), mods_data[i].state)
		else
			pushstylecolor(ImGuiCol.Text, theme.TextDisabled)
			pushstylecolor(ImGuiCol.FrameBg, theme.FrameBgDisabled)
			pushstylecolor(ImGuiCol.FrameBgHovered, theme.FrameBgHoveredDisabled)
			pushstylecolor(ImGuiCol.FrameBgActive, theme.FrameBgActiveDisabled)
			pushstylecolor(ImGuiCol.CheckMark, theme.CheckMarkFalseDisabled)
			ImGui.Checkbox(modNameConvert(mods_data[i].name), false)
			ImGui.PopStyleColor(5)
		end
		ImGui.PopID()
	end
end

function draw_dofile_list(dofile_names)
	local btnRun = {}
	for i in pairs(dofile_names) do
		ImGui.Spacing()
		ImGui.PushID(i)
		pushstylecolor(ImGuiCol.Text, theme.CustomToggleOnText)
		pushstylecolor(ImGuiCol.Button, theme.CustomToggleOn)
		pushstylecolor(ImGuiCol.ButtonHovered, theme.CustomToggleOnHovered)
		pushstylecolor(ImGuiCol.ButtonActive, theme.CustomToggleOn)
		btnRun[i] = ImGui.Button(i18n("button_dofile_run"), 40*fontscale, 20*fontscale)
		ImGui.PopStyleColor(4)
		ImGui.PopID()
		ImGui.SameLine()
		ImGui.PushID(i)
		pushstylecolor(ImGuiCol.Text, theme.Separator)
		ImGui.Text(modNameConvert(dofile_names[i]:match("(.+)%.lua$")))
		ImGui.PopStyleColor(1)
		ImGui.PopID()
	end
	return btnRun
end

function pushstylecolor(style, color)
	ImGui.PushStyleColor(style, color[1], color[2], color[3], color[4])
end

function setThemeBegin()
	pushstylecolor(ImGuiCol.TitleBg,				theme.TitleBg)
	pushstylecolor(ImGuiCol.TitleBgCollapsed,		theme.TitleBgCollapsed)
	pushstylecolor(ImGuiCol.TitleBgActive,			theme.TitleBgActive)
	pushstylecolor(ImGuiCol.Border,					theme.Border)
	pushstylecolor(ImGuiCol.WindowBg,				theme.WindowBg)
	pushstylecolor(ImGuiCol.ScrollbarBg,			theme.ScrollbarBg)
	pushstylecolor(ImGuiCol.ScrollbarGrab,			theme.ScrollbarGrab)
	pushstylecolor(ImGuiCol.ScrollbarGrabHovered,	theme.ScrollbarGrabHovered)
	pushstylecolor(ImGuiCol.ScrollbarGrabActive,	theme.ScrollbarGrabActive)
	pushstylecolor(ImGuiCol.ResizeGrip, 			theme.ResizeGrip)
	pushstylecolor(ImGuiCol.ResizeGripHovered, 		theme.ResizeGripHovered)
	pushstylecolor(ImGuiCol.ResizeGripActive,		theme.ResizeGripActive)
	pushstylecolor(ImGuiCol.Text,					theme.Text)
	pushstylecolor(ImGuiCol.Header,					theme.Header)
	pushstylecolor(ImGuiCol.HeaderHovered,			theme.HeaderHovered)
	pushstylecolor(ImGuiCol.HeaderActive,			theme.HeaderActive)
	pushstylecolor(ImGuiCol.CheckMark,				theme.CheckMark)
	pushstylecolor(ImGuiCol.FrameBg,				theme.FrameBg)
	pushstylecolor(ImGuiCol.FrameBgHovered,			theme.FrameBgHovered)
	pushstylecolor(ImGuiCol.FrameBgActive,			theme.FrameBgActive)
	pushstylecolor(ImGuiCol.Button,					theme.Button)
	pushstylecolor(ImGuiCol.ButtonHovered,			theme.ButtonHovered)
	pushstylecolor(ImGuiCol.ButtonActive,			theme.ButtonActive)
	pushstylecolor(ImGuiCol.Separator,				theme.Border)
	ImGui.PushStyleVar(ImGuiStyleVar.WindowPadding, 6*fontscale, 6*fontscale)
	ImGui.PushStyleVar(ImGuiStyleVar.WindowRounding, 0)
end

function setThemeEnd()
	ImGui.PopStyleVar(2)
	ImGui.PopStyleColor(24)
end

function modNameConvert(name)
	if string.find(name, "_") then
		return string.gsub(" "..string.gsub(name, "_" , " "), "%W%l", string.upper):sub(2)
	elseif string.find(name, "-") then
		return string.gsub(" "..string.gsub(name, "-" , " "), "%W%l", string.upper):sub(2)
	else
		return string.gsub(" "..name, "%W%l", string.upper):sub(2)
	end
end

function btnToggle(onname, offname, state, width, height)
	if onname ~= nil and offname ~= nil and state ~= nil then
		if type(width) == "number" and type(height) == "number" then
			if state then
				return ImGui.Button(onname, width, height)
			else
				return ImGui.Button(offname, width, height)
			end
		else
			if state then
				return ImGui.Button(onname)
			else
				return ImGui.Button(offname)
			end
		end
	end
end

function btnToggleStyleBegin(state)
	if state then
		pushstylecolor(ImGuiCol.Button, theme.CustomToggleOn)
		pushstylecolor(ImGuiCol.ButtonHovered, theme.CustomToggleOnHovered)
		pushstylecolor(ImGuiCol.Text, theme.CustomToggleOnText)
	else
		pushstylecolor(ImGuiCol.Button, theme.Button)
		pushstylecolor(ImGuiCol.ButtonHovered, theme.ButtonHovered)
		pushstylecolor(ImGuiCol.Text, theme.Text)
	end
end

function btnToggleStyleEnd()
	ImGui.PopStyleColor(3)
end

function file_exists(name)
   local f=io.open(name,"r")
   if (f~=nil) then io.close(f) return true else return false end
end

function saveConfig(name, config)
	local file = io.open(name, "w")
	local jconfig = json.encode(config)
	file:write(jconfig)
	file:close()
end

function loadConfig(name)
	if file_exists(name) then
		local file = io.open(name, "r")
		local config = json.decode(file:read("*a"))
		file:close()
			if type(config.autoscan) ~= "boolean" then
				config.autoscan = false
			end
			if config.lang == nil then
				config.lang = "en_us"
			end
			if type(config.autoapear) ~= "boolean" then
				config.autoapear = true
			end
		saveConfig(name, config)
		return config
	else
		config = { autoscan = false, lang = "en_us", autoapear = true }
		saveConfig(name, config)
		return config
	end
end

function applyConfig(config)
	if config.autoscan then
		mods_data = CETMM_fs.GetModsData()
		dofile_names = scan_dofiles()
		scanned = true
	else
		scanned = false
	end
	if config.lang ~= "en_us" then
		i18n.loadFile("lang/en_us.lua")
	end
	i18n.loadFile("lang/"..config.lang..".lua")
	i18n.setLocale(config.lang)
end

function setLang(lang)
	i18n.loadFile("lang/"..lang..".lua")
	i18n.setLocale(lang)
	config.lang = lang
	saveConfig("config.json", config)
end

module = check_module()
config = loadConfig("config.json")
applyConfig(config)

-- Export mod lists

function loadModlist()
	if file_exists("mod_list.json") then
		local file = io.open("mod_list.json", "r")
		return json.decode(file:read("*a"))
	end
end

local mod_list = loadModlist()
local export = {}
function export.GetModList()
	return mod_list
end

function export.PrintModList()
	for k, mods in pairs(mod_list) do
		if mods then
			print(k..": ")
			for _, entry in ipairs(mods) do
				print("\t"..entry)
			end
		end
	end
end

registerForEvent("onInit", function()
	draw = false
	showHelp = false
	showDofileMods = false
	wWidth, wHeight = GetDisplayResolution()
	theme = require("theme")
	fontscale = ImGui.GetFontSize()/13
	ry2 = 0
	print(i18n("console_msg_loaded1"))
	print(i18n("console_msg_loaded2"))
	print(i18n("console_msg_loaded3"))
	print(i18n("console_msg_loaded4"))
	print(i18n("console_msg_loaded5"))
end)

registerHotkey("mod_manager_interface", i18n("hotkey_manager"), function()
	draw = not draw
	showDofileMods = false
end)

registerHotkey("dofile_interface", i18n("hotkey_dofiles"), function()
	draw = not draw
	showDofileMods = true
end)

registerForEvent("onUpdate", function()
	if btnScan then
		mods_data = CETMM_fs.GetModsData()
		dofile_names = scan_dofiles()
		scanned = true
		print(i18n("console_msg_scan"))
	end
	if btnOpenMods then
		CETMM_fs.OpenFolder("mods")
	end
	if btnOpenDofiles then
		CETMM_fs.OpenFolder("dofiles")
	end
	if btnAutoScan then
		config.autoscan = not config.autoscan
		saveConfig("config.json", config)
		if config.autoscan then print(i18n("console_msg_autoscan_on")) else print(i18n("console_msg_autoscan_off")) end
	end
	if btnAutoApear then
		config.autoapear = not config.autoapear
		saveConfig("config.json", config)
		if config.autoscan then print(i18n("console_msg_autoapear_on")) else print(i18n("console_msg_autoapear_off")) end
	end
	if btnDofiles then
		showDofileMods = not showDofileMods
	end
	if btnHelp then
		showHelp = not showHelp
	end
	if btnRun ~= nil then
		for i in pairs(btnRun) do
			if btnRun[i] then
				print(i18n("console_msg_dofile_run", { dofilename = modNameConvert(dofile_names[i]:match("(.+)%.lua$")) }))
				dofile("./dofiles/"..dofile_names[i])
				print(i18n("console_msg_dofile_done"))
			end
		end
	end
	if scanned then
		for i in pairs(mods_data) do
			if mods_data[i].pressed then
				local result = CETMM_fs.ToggleMod(mods_data[i])
				if result == 1 then print(i18n("console_msg_mod_enable", { modname = modNameConvert(mods_data[i].name) }))
				elseif result == 2 then print(i18n("console_msg_mod_enable_error", { modname = modNameConvert(mods_data[i].name) }))
				elseif result == 3 then print(i18n("console_msg_mod_disable", { modname = modNameConvert(mods_data[i].name) }))
				elseif result == 4 then print(i18n("console_msg_mod_enable_error", { modname = modNameConvert(mods_data[i].name) })) end
			end
		end
	end
	for l in pairs(languages) do
		if languages[l].selLang then
			setLang(languages[l].id)
			languages[l].selLang = false
		end
	end
end)

registerForEvent("onDraw", function()
		if not module then
			setThemeBegin()
			ImGui.SetNextWindowSize(520*fontscale, 0, ImGuiCond.FirstUseEver)
			ImGui.SetNextWindowPos(wWidth*0.5-260*fontscale, wHeight*0.5-150, ImGuiCond.FirstUseEver)
			ImGui.PushStyleVar(ImGuiStyleVar.WindowPadding, 10, 10)
			if ImGui.Begin("CETMM Error", bit32.bor(ImGuiWindowFlags.NoResize, ImGuiWindowFlags.NoSavedSettings, ImGuiWindowFlags.NoTitleBar, ImGuiWindowFlags.NoNav, ImGuiWindowFlags.NoMove)) then
				local win_width = ImGui.GetWindowSize()
				pushstylecolor(ImGuiCol.Button, theme.Hidden)
				pushstylecolor(ImGuiCol.ButtonHovered, theme.Hidden)
				pushstylecolor(ImGuiCol.ButtonActive, theme.Hidden)
				ImGui.PushStyleVar(ImGuiStyleVar.FramePadding, 0, 0)
				ImGui.Button(i18n("window_title"), 0, 25*fontscale)
				ImGui.PopStyleVar(1)
				ImGui.PopStyleColor(3)
				ImGui.SameLine()
				ImGui.SetCursorPosX(win_width-25*fontscale-10)
				if ImGui.Button("!", 25*fontscale, 25*fontscale) then
					ImGui.OpenPopup("Settings")
				end

				if ImGui.BeginPopup("Settings", ImGuiWindowFlags.NoMove) then
					ImGui.Spacing()
		      ImGui.Text(i18n("text_select_lang"))
					ImGui.Spacing()
					ImGui.Separator()
					ImGui.Spacing()
					ImGui.PushStyleVar(ImGuiStyleVar.SelectableTextAlign, 0, 0.5)
					for l in pairs(languages) do
						languages[l].selLang = ImGui.Selectable(languages[l].name, false, ImGuiSelectableFlags.None, 0, 22*fontscale)
					end
					ImGui.PopStyleVar(1)
		      ImGui.EndPopup()
		    end
				pushstylecolor(ImGuiCol.FrameBg, {0.27, 0.05, 0.07, 1})
				ImGui.Dummy(0,8)
				ImGui.TextWrapped(i18n("text_error_window_1"))
				ImGui.Dummy(0,6)
				ImGui.TextWrapped(i18n("text_error_window_2"))
				ImGui.Dummy(0,6)
				ImGui.TextWrapped(i18n("text_error_window_3"))
				ImGui.Dummy(0,6)
				ImGui.TextWrapped(i18n("text_error_window_4"))
				ImGui.Dummy(0,6)
				ImGui.TextWrapped(i18n("text_error_window_5"))
				ImGui.Dummy(0,8)
				ImGui.PopStyleColor(1)
			end
			ImGui.End()
			ImGui.PopStyleVar(1)
			setThemeEnd()
		end

    if draw and module then
		setThemeBegin()
		draw = ImGui.Begin(i18n("window_title"), true , ImGuiWindowFlags.NoResize)
		ImGui.SetWindowPos(wWidth/2-210*fontscale, wHeight/2-320*fontscale, ImGuiCond.FirstUseEver)
		if 600*fontscale > wHeight*0.8 then
			ImGui.SetWindowSize(420*fontscale, wHeight*0.8, ImGuiCond.Always)
		else
			ImGui.SetWindowSize(420*fontscale, 640*fontscale, ImGuiCond.Always)
		end
		ImGui.BeginGroup()
		ImGui.Spacing()
		local cursorbtnX, cursorbtnY = ImGui.GetCursorPos()
		local cabtnx, cabtny = ImGui.GetContentRegionAvail()
		btnToggleStyleBegin(showDofileMods)
		btnDofiles = ImGui.Button(i18n("button_dofiles"), ImGui.CalcTextSize(i18n("button_dofiles"))+20*fontscale, 25*fontscale)
		btnToggleStyleEnd()
		ImGui.SetCursorPos(cursorbtnX + cabtnx - 60*fontscale - (ImGui.CalcTextSize(i18n("button_scan"))+20*fontscale), cursorbtnY)
		btnScan = ImGui.Button(i18n("button_scan"), ImGui.CalcTextSize(i18n("button_scan"))+20*fontscale, 25*fontscale)
		ImGui.SetCursorPos(cursorbtnX + cabtnx - 55*fontscale, cursorbtnY)
		if ImGui.Button("!", 25*fontscale, 25*fontscale) then
			ImGui.OpenPopup("Settings")
		end
		if ImGui.IsItemHovered() then
			ImGui.SetTooltip(i18n("tooltip_btn_settings"))
		end
		ImGui.SetCursorPos(cursorbtnX + cabtnx - 25*fontscale, cursorbtnY)
		btnToggleStyleBegin(showHelp)
		btnHelp = ImGui.Button("?", 25*fontscale, 25*fontscale)
		btnToggleStyleEnd()
		if ImGui.IsItemHovered() then
			ImGui.SetTooltip(i18n("tooltip_btn_help"))
		end
		ImGui.Spacing()
		ImGui.EndGroup()
		if ImGui.BeginPopup("Settings", ImGuiWindowFlags.NoMove) then
			ImGui.Spacing()
			ImGui.Text(i18n("text_select_settings"))
			ImGui.Spacing()
			btnToggleStyleBegin(config.autoscan)
			btnAutoScan = btnToggle(i18n("button_autoscan_on"), i18n("button_autoscan_off"), config.autoscan, ImGui.CalcTextSize(i18n("button_autoscan_off"))+20*fontscale, 25*fontscale)
			btnToggleStyleEnd()
			ImGui.SameLine()
			btnToggleStyleBegin(config.autoapear)
			btnAutoApear = btnToggle(i18n("button_autoappear_on"), i18n("button_autoappear_off"), config.autoapear, ImGui.CalcTextSize(i18n("button_autoappear_off"))+20*fontscale, 25*fontscale)
			btnToggleStyleEnd()
			ImGui.Spacing()
			ImGui.Separator()
			ImGui.Spacing()
      ImGui.Text(i18n("text_select_lang"))
			ImGui.Spacing()
			ImGui.PushStyleVar(ImGuiStyleVar.SelectableTextAlign, 0, 0.5)
			for l in pairs(languages) do
				languages[l].selLang = ImGui.Selectable(languages[l].name, false, ImGuiSelectableFlags.None, 0, 22*fontscale)
			end
			ImGui.PopStyleVar(1)
      ImGui.EndPopup()
    end

		cax, cay = ImGui.GetContentRegionAvail()
		ImGui.BeginChild("Mod List", cax+5, cay-ry2-5)

		if showHelp then
			if not showDofileMods then
				pushstylecolor(ImGuiCol.Text, theme.Separator)
				ImGui.TextWrapped(i18n("text_help_manager_1"))
				ImGui.Spacing()
				ImGui.TextWrapped(i18n("text_help_manager_2"))
				ImGui.Spacing()
				ImGui.TextWrapped(i18n("text_help_manager_3"))
				ImGui.Spacing()
				ImGui.TextWrapped(i18n("text_help_manager_4"))
				ImGui.Spacing()
				ImGui.TextWrapped(i18n("text_help_manager_5"))
				ImGui.PopStyleColor(1)
			else
				ImGui.TextWrapped(i18n("text_help_dofiles_1"))
				ImGui.Spacing()
				ImGui.TextWrapped(i18n("text_help_dofiles_2"))
				ImGui.Spacing()
				ImGui.TextWrapped(i18n("text_help_dofiles_3"))
				ImGui.Spacing()
				ImGui.TextWrapped(i18n("text_help_dofiles_4"))
				ImGui.Spacing()
				ImGui.TextWrapped(i18n("text_help_dofiles_5"))
			end
			ImGui.Spacing()
		end

		if scanned then
			if showDofileMods then
				if dofile_names[1] ~= nil then
					btnRun = draw_dofile_list(dofile_names)
				else
					ImGui.Spacing()
					ImGui.Text(i18n("text_no_dofiles"))
				end
			else
				draw_mod_list(mods_data)
			end
		else
			ImGui.Spacing()
			if showDofileMods then
				pushstylecolor(ImGuiCol.Text, theme.Separator)
				ImGui.Text(i18n("text_please_scan"))
				ImGui.PopStyleColor(1)
			else
				ImGui.Text(i18n("text_please_scan"))
			end
		end

		ImGui.EndChild()
		ImGui.BeginGroup()
		ImGui.Spacing()
		btnOpenMods = ImGui.Button(i18n("button_mods_folder"), ImGui.CalcTextSize(i18n("button_mods_folder"))+20*fontscale, 25*fontscale)
		ImGui.SameLine()
		btnOpenDofiles = ImGui.Button(i18n("button_dofile_folder"), ImGui.CalcTextSize(i18n("button_dofile_folder"))+20*fontscale, 25*fontscale)
		ImGui.EndGroup()
		rx2, ry2 = ImGui.GetItemRectSize()
		ImGui.End()
		setThemeEnd()
    end
end)

registerForEvent("onOverlayOpen", function()
	if config.autoapear then
		draw = true
	end
end)

registerForEvent("onOverlayClose", function()
	if config.autoapear then
		draw = false
	end
end)

return export