localcheck.lua

-- If you want to hardcode toc or slug in and not enter it, do so here
local ADDON_SLUG
local TOC_FILE
-- Automatically identify the slug based on the .git or .svn configuration data, change the path if it's not in root, or set to false to disable
local AUTO_IDENTIFY_SLUG = "./"
-- Automatically find the TOC in the given path, set to false to disable
local AUTO_FIND_TOC = "./"
-- Only necessary if you aren't auto identify the slug, curseforge or wowace.
local SITE_LOCATION = nil
-- Personally I keep the api key in another file and just have this reference that to get it
-- If you want to do this, create the file with CURSE_API_KEY = "<key>" in it and set the path here
-- set this to nil and it will ask you for your API key
local API_KEY_FILE = "../TestCode/api-key.lua"
-- Patterns that should not be scrapped, case-insensitive
-- Anything between the no-lib-strip is automatically ignored
local FILE_BLACKLIST = {"^localization", "^lib"}
-- 1 = English
DEFAULT_LANGUAGE = "1"
-- Removes phrases that are not found through this
DELETE_UNIMPORTED = true

-- No more modifying!
local OS_TYPE = os.getenv("HOME") and "linux" or "windows"

-- Mak sure we have LuaSockets
local _, http = pcall(require, "socket.http")
local _, ltn = pcall(require, "ltn12")
if( not http ) then
	print("Failed to find socket.http, did you install LuaSockets?")
elseif( not ltn ) then
	print("Failed to find ltn12, did you install LuaSockets?")
end


-- Figure out the API key
if( API_KEY_FILE ) then
	local file = io.open(API_KEY_FILE)
	if( not file ) then
		print(string.format("It appears the API key file %s does not exist.", API_KEY_FILE))
	else
		file:close()
		dofile(API_KEY_FILE)
		
		if( not CURSE_API_KEY ) then
			print("You did not define CURSE_API_KEY in your key file, make sure it does not have local next to it.")
		end
	end
end


if( not CURSE_API_KEY ) then
	while( not CURSE_API_KEY ) do
		io.stdout:write("Enter API key: ")
		CURSE_API_KEY = io.stdin:read("*line")
		CURSE_API_KEY = CURSE_API_KEY ~= "" and CURSE_API_KEY or nil
	end
end

-- Attempt to automatically identify the addon slug
if( AUTO_IDENTIFY_SLUG ) then
	local git = io.open(AUTO_IDENTIFY_SLUG .. ".git/config")
	local svn = io.open(AUTO_IDENTIFY_SLUG .. ".svn/entries")

	if( git ) then
		local contents = git:read("*all")
		git:close()

		SITE_LOCATION, ADDON_SLUG = string.match(contents, "git%.(.-)%.com:wow/(.-)/mainline%.git")
	elseif( svn ) then
		local contents = svn:read("*all")
		svn:close()
	
		SITE_LOCATION, ADDON_SLUG = string.match(contents, "svn%.(.-)%.com/wow/(.-)/mainline")
	end
	
	if( not ADDON_SLUG ) then
		print("Failed to identify addon slug.")
	elseif( not SITE_LOCATION ) then
		print("Failed to identify site location.")
	end
end

if( not SITE_LOCATION ) then
	while( not SITE_LOCATION ) do
		io.stdout:write("Site location [wowace/curseforge]: ")
		SITE_LOCATION = io.stdin:read("*line")
		SITE_LOCATION = ( SITE_LOCATION == "wowace" or SITE_LOCATION == "curseforge" ) and SITE_LOCATION or nil
	end
end

-- Manually ask for it
if( not ADDON_SLUG ) then
	while( not ADDON_SLUG ) do
		io.stdout:write("Enter slug: ")
		ADDON_SLUG = io.stdin:read("*line")
		ADDON_SLUG = ADDON_SLUG ~= "" and ADDON_SLUG or nil
	end
end

print(string.format("Using addon slug: %s", ADDON_SLUG))

-- Find the TOC now
if( AUTO_FIND_TOC ) then
	local pipe = OS_TYPE == "windows" and io.popen(string.format("dir /B \"%s\"", AUTO_FIND_TOC)) or io.popen(string.format("ls -1 \"%s\"", AUTO_FIND_TOC))
	if( type(pipe) == "userdata" ) then
		for file in pipe:lines() do
			if( string.match(file, "(.+)%.toc") ) then
				TOC_FILE = file
				break
			end
		end

		pipe:close()
		if( not TOC_FILE ) then print("Failed to auto detect toc file.") end
	else
		print("Failed to auto find toc, cannot run dir /B or ls -1")
	end
end

if( not TOC_FILE ) then
	while( not TOC_FILE ) do
		io.stdout:write("TOC path: ")
		TOC_FILE = io.stdin:read("*line")
		TOC_FILE = TOC_FILE ~= "" and TOC_FILE or nil
		if( TOC_FILE ) then
			local file = io.open(TOC_FILE)
			if( file ) then
				file:close()
				break
			else
				print(string.format("%s does not exist.", TOC_FILE))
			end
		end
	end
end

print(string.format("Using TOC file %s", TOC_FILE))
print("")

-- Parse through the TOC file so we know what to scan
local ignore
local localizedKeys = {}
for line in io.lines(TOC_FILE) do
	line = string.gsub(line, "\r", "")
	
	if( string.match(line, "#@no%-lib%-strip@") ) then
		ignore = true
	elseif( string.match(line, "#@end%-no%-lib%-strip@") ) then
		ignore = nil
	end
		
	if( not ignore and string.match(line, "%.lua") ) then
		-- Make sure it's a valid file
		local blacklist
		for _, check in pairs(FILE_BLACKLIST) do
			if( string.match(string.lower(line), check) ) then
				blacklist = true
				break
			end
		end
	
		-- File checks out, scrap everything
		if( not blacklist ) then
			-- Fix slashes
			if( OS_TYPE == "linux" ) then
				line = string.gsub(line, "\\", "/")
			end
			
			local keys = 0
			local contents = io.open(line):read("*all")
		
			for match in string.gmatch(contents, "L%[\"(.-)%\"]") do
				if( not localizedKeys[match] ) then keys = keys + 1 end
				localizedKeys[match] = true
			end
			
			print(string.format("%s (%d keys)", line, keys))
		end
	end
end

-- Compile all of the localization we found into string form
local totalLocalizedKeys = 0
local localization = ""
for key in pairs(localizedKeys) do
	localization = string.format("%s\nL[\"%s\"] = true", localization, key, key)
	totalLocalizedKeys = totalLocalizedKeys + 1
end

if( totalLocalizedKeys == 0 ) then
	print("Warning, failed to find any localizations, perhaps you messed up a configuration variable?")
	return
end

print(string.format("Found %d keys total", totalLocalizedKeys))

local addonData = {
	["format"] = "lua_additive_table",
	["language"] = DEFAULT_LANGUAGE,
	["delete_unimported"] = DELETE_UNIMPORTED and "y" or "n",
	["text"] = localization,
}

-- Send it off
local boundary = string.format("-------%s", os.time())
local source = {}
local body = ""

for key, data in pairs(addonData) do
	body = string.format("%s--%s\r\n", body, boundary)
	body = string.format("%sContent-Disposition: form-data; name=\"%s\"\r\n\r\n", body, key)
	body = string.format("%s%s\r\n", body, data)
end

body = string.format("%s--%s\r\n", body, boundary)

http.request({
	method = "POST",
	url = string.format("http://www.%s.com/addons/%s/localization/import/?api-key=%s", SITE_LOCATION, ADDON_SLUG, CURSE_API_KEY),
	sink = ltn12.sink.table(source),
	source = ltn12.source.string(body),
	headers = {
		["Content-Type"] = string.format("multipart/form-data; boundary=\"%s\"", boundary),
		["Content-Length"] = string.len(body),
	},
})

local contents = table.concat(source, "\n")
local errors = {}
for line in string.gmatch(contents, "(.-)\n") do
	local msg = string.match(line, "<span class=\"error\">(.-)</p>")
	if( msg ) then
		table.insert(errors, msg)
	end
end

print("")

if( #(errors) == 0 ) then
	print(string.format("Updated localization for %s on %s!", ADDON_SLUG, SITE_LOCATION))
else
	print("Localization failed:")
	for _, line in pairs(errors) do
		print(line)
	end
end