-- This file is executed once on we start up. The state perseveres
-- through callbacks
--
-- wehack.runprocess: Wait for exit code, report errors (grimext)
-- wehack.runprocess2: Wait for exit code, don't report errors (jasshelper)
-- wehack.execprocess: Don't wait for exit code (War3)
--
grimregpath = "Software\\Grimoire\\"
--warcraftdir = grim.getregpair(grimregpath,"War3InstallPath")
--if warcraftdir == 0 then
--    wehack.messagebox("Error, could not find warcraft install path in wehack.lua")
--end

isstartup = true
grimdir = grim.getcwd()
dofile("wehacklib.lua")
dofile("findpath.lua")
if path==0 or path=="" then
path = "."
end
mapvalid = true
cmdargs = "" -- used to execute external tools on save

confregpath = "HKEY_CURRENT_USER\\Software\\Grimoire\\"

haveext = grim.exists("grimext\\grimex.dll")
if haveext then
utils = wehack.addmenu("Extensions")
end
haveumswe = haveext and grim.exists("umswe\\umswecore.lua")
if haveumswe then
ums = wehack.addmenu("UMSWE")
end

whmenu = wehack.addmenu("Grimoire")
wh_window = TogMenuEntry:New(whmenu,"Start War3 with -window",nil,true)
wh_opengl = TogMenuEntry:New(whmenu,"Start War3 with -opengl",nil,false)

wehack.addmenuseparator(whmenu)
wh_tesh = TogMenuEntry:New(whmenu,"Enable TESH",nil,true)
if grim.isdotnetinstalled() then
wh_colorizer = TogMenuEntry:New(whmenu,"Enable Colorizer",nil,true)
end
wh_nolimits = TogMenuEntry:New(whmenu,"Enable no limits",
function(self) grim.nolimits(self.checked) end,false)
wh_oehack = TogMenuEntry:New(whmenu,"Enable object editor hack",
function(self) grim.objecteditorhack(self.checked) end,true)
wh_syndisable = TogMenuEntry:New(whmenu,"Disable WE syntax checker",
function(self) grim.syndisable(self.checked) end,true)
wh_descpopup = TogMenuEntry:New(whmenu,"Disable default description nag",
function(self) grim.descpopup(self.checked) end,true)
wh_autodisable = TogMenuEntry:New(whmenu,"Don't let WE disable triggers",
function(self) grim.autodisable(self.checked) end,true)
wh_alwaysenable = TogMenuEntry:New(whmenu,"Always allow trigger enable",
function(self) grim.alwaysenable(self.checked) end,true)
wh_disablesound = TogMenuEntry:New(whmenu,"Mute editor sounds",nil,true)
wh_firstsavenag = TogMenuEntry:New(whmenu,"Disable first save warning",nil,true)

-- when warmachine is turned on, war3err must not be on
--function setwar3err()
--    if wh_machine.checked then
--  grim.setregstring(confregpath,"Enable war3err","off")
-- wehack.checkmenuentry(wh_enablewar3err.menu,wh_enablewar3err.id,0)
-- end
--end

-- when warmachine is turned on, war3err must not be on
--function setwarmachine()
--    if wh_enablewar3err.checked then
--  grim.setregstring(confregpath,"Enable warmachine","off")
-- wehack.checkmenuentry(wh_machine.menu,wh_machine.id,0)
-- end
--end

--if not grim.isnewcompiler(path.."\\war3.exe") then
--wh_enablewar3err.cb = setwarmachine
--wh_machine.cb = setwar3err
--end

wehack.addmenuseparator(whmenu)

usetestmapconf = (grim.getregpair(confregpath,"Use custom test map settings") == "on")
function testmapconfig()
usetestmapconf = wehack.testmapconfig(path,usetestmapconf)
if usetestmapconf then
grim.setregstring(confregpath,"Use custom test map settings","on")
else
grim.setregstring(confregpath,"Use custom test map settings","off")
end
end
wh_configtest = MenuEntry:New(whmenu,"Customize test map settings",testmapconfig);

function aboutpopup()
wehack.showaboutdialog("Grimoire 1.5")
end
wh_about = MenuEntry:New(whmenu,"About Grimoire ...",aboutpopup)

havejh = grim.exists("jasshelper\\jasshelper.exe")
if havejh then
jhmenu = wehack.addmenu("vJass")
jh_enable = TogMenuEntry:New(jhmenu,"Enable JassHelper",nil,true)
jh_debug = TogMenuEntry:New(jhmenu,"Debug Mode",nil,false)
jh_disable = TogMenuEntry:New(jhmenu,"Disable vJass syntax",nil,false)
jh_disableopt = TogMenuEntry:New(jhmenu,"Disable script optimization",nil,false)
wehack.addmenuseparator(jhmenu)

function jhshowerr()
--  if jh_fast ~= nil and jh_fast.checked then
--   wehack.showpreviouserrors();
--  else
wehack.execprocess("jasshelper\\jasshelper.exe --showerrors")
--  end
end

function jhabout()
--  if jh_fast ~= nil and jh_fast.checked then
--   wehack.jasshelperabout();
--  else
wehack.execprocess("jasshelper\\jasshelper.exe --about")
--  end
end

jhshowerrm = MenuEntry:New(jhmenu,"Show previous errors",jhshowerr)
jhaboutm = MenuEntry:New(jhmenu,"About JassHelper ...",jhabout)
end
-- cJass#1
have_ah = grim.exists("adichelper\\adichelper.exe")
if have_ah then
ah_menu = wehack.addmenu("cJass")
ah_enable = TogMenuEntry:New(ah_menu,"Enable AdicParser",nil,true)
ah_enableopt = TogMenuEntry:New(ah_menu,"Enable AdicOptimizer",nil,true)

wehack.addmenuseparator(ah_menu)

if (grim.getregpair(confregpath,"First launch passed") ~= "yes") then
ah_firstlaunch = true
end

if ah_firstlaunch then
if (wehack.runprocess2("AdicHelper\\AHupdate.exe /ask") == 6) then
ah_enableupdate = true
end

grim.setregstring(confregpath,"First launch passed","yes")
if ah_enableupdate then
grim.setregstring(confregpath,"Enable AutoUpdate","on")
else
grim.setregstring(confregpath,"Enable AutoUpdate","off")
end
end

ah_enableupdate = TogMenuEntry:New(ah_menu,"Enable AutoUpdate",nil,false)

if ah_enableupdate.checked then
wehack.execprocess("adichelper\\AHupdate.exe /silent")
end

function ah_execupdate()
wehack.execprocess("adichelper\\AHupdate.exe")
end
ah_update = MenuEntry:New(ah_menu,"Check for updates now",ah_execupdate)

wehack.addmenuseparator(ah_menu)

function ah_about()
wehack.execprocess("adichelper\\adichelper.exe")
end
ah_aboutm = MenuEntry:New(ah_menu,"About AdicHelper ...",ah_about)

end
-- /cJass#1

-- Reinventing the Craft
havertc = grim.exists("rtc\\rtcexehack.exe") and grim.exists("rtc\\rcncore.dll")
if havertc then
rtcmenu = wehack.addmenu("RtC")
rtc_enabled = TogMenuEntry:New(rtcmenu, "Enable Reinventing the Craft", nil, true)

wehack.addmenuseparator(rtcmenu)
function rtcabout()
wehack.execprocess("rtc\\rtcexehack.exe --about")
end
rtc_about = MenuEntry:New(rtcmenu, "About Reinventing the Craft ...", rtcabout)
else
grim.log("Reinventing the Craft")
if(not grim.exists("rtc\\rtcexehack.exe")) then
grim.log("rtc\\rtcexehack.exe missing!")
end
if(not grim.exists("rtc\\rcncore.dll")) then
grim.log("rtc\\rcncore.dll missing!")
end
end

function initshellext()
local first, last = string.find(grim.getregpair("HKEY_CLASSES_ROOT\\WorldEdit.Scenario\\shell\\open\\command\\", ""),"NewGen",1)
if first then
wehack.checkmenuentry(shellext.menu,shellext.id,1)
else
local second, third = string.find(grim.getregpair("HKEY_CLASSES_ROOT\\WorldEdit.Scenario\\shell\\open\\command\\", ""),".bat",1)
if second then
wehack.checkmenuentry(shellext.menu,shellext.id,1)
else
wehack.checkmenuentry(shellext.menu,shellext.id,0)
end
end
end

function fixopencommand(disable,warpath,grimpath,filetype)
--local curval = grim.getregpair("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\open\\command\\","")
--if curval ~= 0 then
-- if disable then
-- grim.setregstring("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\open\\command\\","",string.gsub(curval, "%%L", "%%1"))
-- else
-- grim.setregstring("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\open\\command\\","",string.gsub(curval, "%%1", "%%L"))
-- end
--end

local wepath = "\""..grimpath.."\\NewGen WE.exe\""
if not grim.exists(grimpath.."\\NewGen WE.exe") then
wepath = "\""..grimpath.."\\we.bat\""
end
if disable then
grim.setregstring("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\open\\command\\","","\""..warpath.."\\World Editor.exe\" -loadfile \"%L\"")
else
grim.setregstring("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\open\\command\\","",wepath.." -loadfile \"%L\"")
end
end

function registerextension(disable,warpath,grimpath,filetype,istft)
if disable then
grim.deleteregkey("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\fullscreen\\command\\");
grim.deleteregkey("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\fullscreen\\");
grim.deleteregkey("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\windowed\\command\\");
grim.deleteregkey("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\windowed\\");
grim.deleteregkey("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\opengl\\command\\");
grim.deleteregkey("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\opengl\\");
else
--if istft then
-- gamepath = "\""..warpath.."\\Frozen Throne.exe\""
--else
-- gamepath = "\""..warpath.."\\Warcraft III.exe\""
--end
--grim.setregstring("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\fullscreen\\","","Play Fullscreen")
--grim.setregstring("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\fullscreen\\command\\","",gamepath.." -loadfile \"%L\"")
--grim.setregstring("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\windowed\\","","Play Windowed")
--grim.setregstring("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\windowed\\command\\","",gamepath.." -window -loadfile \"%L\"")

local gamepath = "\""..grimpath.."\\NewGen Warcraft.exe\""
if not grim.exists(grimpath.."\\NewGen Warcraft.exe") then
gamepath = "\""..grimpath.."\\startwar3.bat\""
end
grim.setregstring("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\fullscreen\\","","Play Fullscreen")
grim.setregstring("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\fullscreen\\command\\","",gamepath.." -loadfile \"%L\"")
grim.setregstring("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\windowed\\","","Play Windowed")
grim.setregstring("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\windowed\\command\\","",gamepath.." -window -loadfile \"%L\"")
grim.setregstring("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\opengl\\","","Play With OpenGL")
grim.setregstring("HKEY_CLASSES_ROOT\\WorldEdit."..filetype.."\\shell\\opengl\\command\\","",gamepath.." -window -opengl -loadfile \"%L\"")
end
end

function toggleshellext()
local istft = (grim.getregpair("HKEY_CURRENT_USER\\Software\\Blizzard Entertainment\\Warcraft III\\", "InstallPathX") ~= 0)
local first, last = string.find(grim.getregpair("HKEY_CLASSES_ROOT\\WorldEdit.Scenario\\shell\\open\\command\\", ""),"NewGen",1)
local found = false
if first then
found = true
else
local second, third = string.find(grim.getregpair("HKEY_CLASSES_ROOT\\WorldEdit.Scenario\\shell\\open\\command\\", ""),".bat",1)
if second then
found = true
end
end

if path ~= 0 and grimdir ~= 0 then
fixopencommand(found,path,grimdir,"Scenario")
registerextension(found,path,grimdir,"Scenario",istft)
fixopencommand(found,path,grimdir,"ScenarioEx")
registerextension(found,path,grimdir,"ScenarioEx",istft)
fixopencommand(found,path,grimdir,"Campaign")
registerextension(found,path,grimdir,"Campaign",istft)
fixopencommand(found,path,grimdir,"AIData")
if found then
wehack.checkmenuentry(shellext.menu,shellext.id,0)
else
wehack.checkmenuentry(shellext.menu,shellext.id,1)
end
end
end

function initlocalfiles()
if grim.getregpair("HKEY_CURRENT_USER\\Software\\Blizzard Entertainment\\Warcraft III\\", "Allow Local Files") == 0 then
wehack.checkmenuentry(localfiles.menu,localfiles.id,0)
else
wehack.checkmenuentry(localfiles.menu,localfiles.id,1)
end
end

function togglelocalfiles()
if grim.getregpair("HKEY_CURRENT_USER\\Software\\Blizzard Entertainment\\Warcraft III\\", "Allow Local Files") == 0 then
grim.setregdword("HKEY_CURRENT_USER\\Software\\Blizzard Entertainment\\Warcraft III\\", "Allow Local Files", 1)
wehack.checkmenuentry(localfiles.menu,localfiles.id,1)
else
grim.setregdword("HKEY_CURRENT_USER\\Software\\Blizzard Entertainment\\Warcraft III\\", "Allow Local Files", 0)
wehack.checkmenuentry(localfiles.menu,localfiles.id,0)
end
end

function runobjectmerger(mode)
curmap = wehack.findmappath()
if curmap ~= "" then
source = wehack.openfiledialog("Unit files (*.w3u)|*.w3u|Item files (*.w3t)|*w3t|Doodad files (*.w3d)|*.w3d|Destructable files (*.w3b)|*.w3b|Ability files (*.w3a)|*.w3a|Buff files (*.w3h)|*.w3h|Upgrade files (*.w3q)|*.w3q|", "w3a", "Select files to import ...", true)
grim.log("got in lua: " .. source)
if source ~= "" then
list = strsplit("|", source);
-- cmdargs = "ObjectMerger \""..curmap.."\" "..wehack.getlookupfolders().." "..mode..fileargsjoin(list)
cmdargs = "grimext\\ObjectMerger.exe \""..curmap.."\" "..wehack.getlookupfolders().." "..mode..fileargsjoin(list)
grim.log("assembled cmdline: " .. cmdargs)
-- wehack.messagebox(cmdargs,"Grimoire",false)
wehack.savemap()
grim.log("called saved map")
end
else
showfirstsavewarning()
end
end

function runconstantmerger()
curmap = wehack.findmappath()
if curmap ~= "" then
source = wehack.openfiledialog("Text files (*.txt)|*.txt|", "txt", "Select files to import ...", true)
if source ~= "" then
list = strsplit("|", source);
-- cmdargs = "ConstantMerger \""..curmap.."\" "..wehack.getlookupfolders()..fileargsjoin(list)
cmdargs = "grimext\\ConstantMerger.exe \""..curmap.."\" "..wehack.getlookupfolders()..fileargsjoin(list)
-- wehack.messagebox(cmdargs,"Grimoire",false)
wehack.savemap()
end
else
showfirstsavewarning()
end
end

function runtriggermerger()
curmap = wehack.findmappath()
if curmap ~= "" then
source = wehack.openfiledialog("GUI Trigger files (*.wtg)|*.wtg|Custom Text Trigger files (*.wct)|*wct|", "wtg", "Select trigger data to import ...", true)
if source ~= "" then
list = strsplit("|", source);
-- cmdargs = "TriggerMerger \""..curmap.."\" "..wehack.getlookupfolders()..fileargsjoin(list)
cmdargs = "grimext\\TriggerMerger.exe \""..curmap.."\" "..wehack.getlookupfolders()..fileargsjoin(list)
-- wehack.messagebox(cmdargs,"Grimoire",false)
wehack.savemap()
end
else
showfirstsavewarning()
end
end

function runfileimporterfiles()
curmap = wehack.findmappath()
if curmap ~= "" then
source = wehack.openfiledialog("All files (*.*)|*.*|", "*", "Select files to import ...", true)
if source ~= "" then
list = strsplit("|", source);
inmpqpath = wehack.inputbox("Specify the target path ...","FileImporter","Units\\")
-- cmdargs = "FileImporter \""..curmap.."\" "..wehack.getlookupfolders()..argsjoin(inmpqpath,list)
cmdargs = "grimext\\FileImporter.exe \""..curmap.."\" "..wehack.getlookupfolders()..argsjoin(inmpqpath,list)
-- wehack.messagebox(cmdargs,"Grimoire",false)
wehack.savemap()
end
else
showfirstsavewarning()
end
end

function runfileimporterdir()
curmap = wehack.findmappath()
if curmap ~= "" then
source = wehack.browseforfolder("Select the source directory ...")
if source ~= "" then
-- cmdargs = "FileImporter \""..curmap.."\" "..wehack.getlookupfolders().." \""..source.."\""
cmdargs = "grimext\\FileImporter.exe \""..curmap.."\" "..wehack.getlookupfolders().." \""..source.."\""
-- wehack.messagebox(cmdargs,"Grimoire",false)
wehack.savemap()
end
else
showfirstsavewarning()
end
end

function runfileexporter()
curmap = wehack.findmappath()
if curmap ~= "" then
target = wehack.browseforfolder("Select the target directory ...")
if target ~= "" then
--   wehack.rungrimextool("FileExporter", curmap, removequotes(wehack.getlookupfolders()), target)
wehack.runprocess("grimext\\FileExporter.exe \""..curmap.."\" "..wehack.getlookupfolders().." \""..target.."\"")
end
else
showfirstsavewarning()
end
end

function runtilesetter()
curmap = wehack.findmappath()
if curmap ~= "" then
map = wehack.openarchive(curmap,15)
oldtiles = wehack.getcurrenttiles()
wehack.closearchive(map)
if oldtiles ~= "" then
newtiles = wehack.tilesetconfig(string.sub(oldtiles,1,1), string.sub(oldtiles,2))
if newtiles ~= "" then
tileset = string.sub(newtiles,1,1)
tiles = string.sub(newtiles,2)
if tileset ~= "" and tiles ~= "" then
--        cmdargs = "TileSetter \""..curmap.."\" "..wehack.getlookupfolders().." "..tileset.." "..tiles
cmdargs = "grimext\\TileSetter.exe \""..curmap.."\" "..wehack.getlookupfolders().." "..tileset.." "..tiles
wehack.savemap()
end
end

-- tileset = wehack.inputbox("Specify the tileset ...","TileSetter",string.sub(oldtiles,1,1))
-- if tileset ~= "" then
-- tiles = wehack.inputbox("Specify the tile list ...","TileSetter",string.sub(oldtiles,2))
-- if tiles ~= "" then
-- cmdargs = "grimext\\TileSetter.exe \""..curmap.."\" "..wehack.getlookupfolders().." "..tileset.." "..tiles
-- wehack.savemap()
-- end
-- end
end
else
showfirstsavewarning()
end
end

function showfirstsavewarning()
if wh_firstsavenag.checked then
return
else
wehack.messagebox("Could not find path to map, please try saving again","Grimoire",false)
end
end

function testmap(cmdline)
--if havejh and jh_enable.checked and not mapvalid then
--    return
--end
--wehack.messagebox(cmdline)
--mappath = strsplit(" ",cmdline)[2]
--compilemap_path(mappath)

if wh_opengl.checked then
cmdline = cmdline .. " -opengl"
end
if wh_window.checked then
cmdline = cmdline .. " -window"
end
if rtc_enabled.checked then
--  local args = string.gsub(cmdline,"\"([^\"]*)\" ","")
--  wehack.messagebox(args,"Grimoire",false)
--  cmdline = "startwar3.bat " .. args

local testmaparglist = argsplit(cmdline)
local len = table.getn(testmaparglist)
if len > 2 then
cmdline = "startwar3.bat"
--if not grim.exists(cmdline) then
-- cmdline = "NewGen Warcraft.exe" -- NewGen version doesn't use batch files
--end
for i = 2, len do
if (i < len) and (usetestmapconf) then
local arglen = string.len(testmaparglist[i+1])
if testmaparglist[i] == "-loadfile" and arglen > 3 then
local ext = string.lower(string.sub(testmaparglist[i+1],arglen-4+1))
if ext == ".w3m" or ext == ".w3x" then
local substitute = wehack.setupwgcfile(testmaparglist[i+1]);
if (substitute ~= testmaparglist[i+1]) then
testmaparglist[i+1] = "\"" .. substitute .. "\""
end
end
end
end
cmdline = cmdline .. " " .. testmaparglist[i]
end
end
end
wehack.execprocess(cmdline)
end

function compilemap_path(mappath)
if mappath == "" then
showfirstsavewarning()
return
end
map = wehack.openarchive(mappath,15)
wehack.extractfile("jasshelper\\common.j","scripts\\common.j")
wehack.extractfile("jasshelper\\Blizzard.j","scripts\\Blizzard.j")
wehack.extractfile("war3map.j","war3map.j")
wehack.closearchive(map)
if cmdargs ~= "" then
local cmdtable = argsplit(cmdargs)
--  local len = table.getn(cmdtable)
--  for i = 1, len do
--   cmdtable[i] = removequotes(cmdtable[i])
--  end
--  wehack.rungrimextool(cmdtable)
grim.log("running tool on save: "..cmdargs)
wehack.runprocess(cmdargs)
cmdargs = ""
end

-- fly-data
fly_preprocess(mappath)
-- /fly-data

-- cJass#2
if have_ah and ah_enable.checked then
cmdline = "AdicHelper\\AdicHelper.exe"
if jh_debug.checked then
cmdline = cmdline .. " /dbg"
end
cmdline = cmdline .. " /mappars=\"" .. mappath.."\""
adicresult = wehack.runprocess2(cmdline)
if adicresult == 1 then
mapvalid = false
return
end
end
-- /cJass#2
if havejh and jh_enable.checked then
cmdline = "jasshelper\\jasshelper.exe"
if jh_debug.checked then
cmdline = cmdline .. " --debug"
end
if jh_disable.checked then
cmdline = cmdline .. " --nopreprocessor"
end
if jh_disableopt.checked then
cmdline = cmdline .. " --nooptimize"
end
cmdline = cmdline .. " jasshelper\\common.j jasshelper\\blizzard.j \"" .. mappath .."\""
toolresult = 0
--  if jh_fast ~= nil and jh_fast.checked then
--   toolresult = wehack.runjasshelper(jh_debug.checked, jh_disable.checked, "jasshelper\\common.j", "jasshelper\\blizzard.j", mappath, "")
--  else
toolresult = wehack.runprocess2(cmdline)
--  end
if toolresult == 0 then
-- cJass#3
if have_ah and ah_enableopt.checked then
wehack.runprocess2("AdicHelper\\AdicHelper.exe /mapoptz=\"" .. mappath.."\"")
end
-- /cJass#3
mapvalid = true
else
mapvalid = false
end
end
end

dofile("ScExp\\ScExp.lua")

dofile("fly\\fly.lua")

function compilemap()
mappath = wehack.findmappath()
if mappath == "" then
scexpBuildCampaign()
else
compilemap_path(mappath)
end
end

if haveext then
localfiles = MenuEntry:New(utils,"Enable Local Files",togglelocalfiles)
shellext = MenuEntry:New(utils,"Register Shell Extensions",toggleshellext)
initlocalfiles()
initshellext()
wehack.addmenuseparator(utils)
end
if haveext and grim.exists("grimext\\tilesetter.exe") then
tilesetter = MenuEntry:New(utils,"Edit Tileset",runtilesetter)
end
if haveext and grim.exists("grimext\\fileexporter.exe") then
fileexporter = MenuEntry:New(utils,"Export Files",runfileexporter)
end
if haveext and grim.exists("grimext\\fileimporter.exe") then
fileimporterdir = MenuEntry:New(utils,"Import Directory",runfileimporterdir)
fileimporterfiles = MenuEntry:New(utils,"Import Files",runfileimporterfiles)
end
if haveext and grim.exists("grimext\\objectmerger.exe") then
objectmerger = MenuEntry:New(utils,"Merge Object Editor Data",function(self) runobjectmerger("m") end)
objectreplacer = MenuEntry:New(utils,"Replace Object Editor Data",function(self) runobjectmerger("r") end)
objectimporter = MenuEntry:New(utils,"Import Object Editor Data",function(self) runobjectmerger("i") end)
end
if haveext and grim.exists("grimext\\constantmerger.exe") then
constantmerger = MenuEntry:New(utils,"Merge Constants Data",runconstantmerger)
end
if haveext and grim.exists("grimext\\triggermerger.exe") then
triggermerger = MenuEntry:New(utils,"Merge Trigger Data",runtriggermerger)
end

function extabout()
grim.openlink("http://www.wc3campaigns.net")
end
if haveext then
wehack.addmenuseparator(utils)
aboutextensions = MenuEntry:New(utils,"About Grimex ...",extabout)
end

if haveumswe then
ums_enabled = TogMenuEntry:New(ums,"Enable UMSWE",nil,false)
ums_cat = TogMenuEntry:New(ums,"Custom Editor Categories",nil,false)
ums_til = TogMenuEntry:New(ums,"Non Tileset Specific Objects",nil,false)
ums_pat = TogMenuEntry:New(ums,"Custom Tile Pathability",nil,false)

function unloadumswe()
local umswehandle = wehack.getarchivehandle("umswe\\umswe.mpq")
if umswehandle ~= 0 then
wehack.closearchive(umswehandle)
wehack.setarchivehandle("umswe\\umswe.mpq", 0)
end
end

function getumsweargs()
local umsargs = "";
if (ums_enabled.checked) then
umsargs = umsargs .. " umscore=1"
else
umsargs = umsargs .. " umscore=0"
end
if (ums_cat.checked) then
umsargs = umsargs .. " umscategories=1"
else
umsargs = umsargs .. " umscategories=0"
end
if (ums_til.checked) then
umsargs = umsargs .. " umstiles=1"
else
umsargs = umsargs .. " umstiles=0"
end
if (ums_pat.checked) then
umsargs = umsargs .. " umspathing=1"
else
umsargs = umsargs .. " umspathing=0"
end
return umsargs
end

function toggleumswe()
if not isstartup then
unloadumswe()
wehack.setwaitcursor(true)
wehack.runprocess("grimext\\PatchGenerator.exe umswe\\umswecore.lua "..wehack.getlookupfolders().." umswe"..getumsweargs())
wehack.setwaitcursor(false)
end
end

function toggleumswecat()
if ums_enabled.checked and not isstartup then
unloadumswe()
wehack.setwaitcursor(true)
wehack.runprocess("grimext\\PatchGenerator.exe umswe\\umswecategories.lua "..wehack.getlookupfolders().." umswe"..getumsweargs())
wehack.setwaitcursor(false)
end
end

function toggleumswetil()
if ums_enabled.checked and not isstartup then
unloadumswe()
wehack.setwaitcursor(true)
wehack.runprocess("grimext\\PatchGenerator.exe umswe\\umswetilesets.lua "..wehack.getlookupfolders().." umswe"..getumsweargs())
wehack.setwaitcursor(false)
end
end

function toggleumswepat(newstate)
if ums_enabled.checked and not isstartup then
unloadumswe()
wehack.setwaitcursor(true)
wehack.runprocess("grimext\\PatchGenerator.exe umswe\\umswepathing.lua "..wehack.getlookupfolders().." umswe"..getumsweargs())
wehack.setwaitcursor(false)
end
end

ums_enabled.cb = toggleumswe
ums_cat.cb = toggleumswecat
ums_til.cb = toggleumswetil
ums_pat.cb = toggleumswepat

function categoryconfig()
if wehack.showcategorydialog("umswe\\umswecategories.conf.lua") and ums_enabled.checked then
if ums_cat.checked then
unloadumswe()
wehack.setwaitcursor(true)
wehack.runprocess("grimext\\PatchGenerator.exe umswe\\umswecategories.lua "..wehack.getlookupfolders().." umswe"..getumsweargs())
wehack.setwaitcursor(false)
end
end
end

function pathabilityconfig()
unloadumswe()
if wehack.showpathdialog("umswe\\umswepathing.conf.lua","umswe\\umswe.mpq") and ums_enabled.checked then
if ums_pat.checked then
wehack.setwaitcursor(true)
wehack.runprocess("grimext\\PatchGenerator.exe umswe\\umswepathing.lua "..wehack.getlookupfolders().." umswe"..getumsweargs())
wehack.setwaitcursor(false)
end
end
end

function umsweabout()
wehack.showumsweabout("UMSWE 5.0")
end

wehack.addmenuseparator(ums)
ums_catconf = MenuEntry:New(ums,"Customize Editor Categories",categoryconfig)
ums_pathconf = MenuEntry:New(ums,"Customize Tile Pathability",pathabilityconfig)
ums_about = MenuEntry:New(ums,"About UMSWE ...",umsweabout)

end

isstartup = false