Localize.py

import sublime, sublime_plugin
import os, re, json, codecs, zipfile

p = sublime.platform()
version = sublime.version()
v = version[0]

ps = os.path.abspath(__file__).split(os.sep)
pName = ps[-2]
if '.' in pName:
	pName = os.path.splitext(pName)[0]
pkgs = sublime.packages_path()
if not pkgs:
	pkgs = os.sep.join(ps[:-2])
dDir = pkgs + os.sep + 'Default'
pDir = pkgs + os.sep + pName
mDir = pDir + os.sep + 'menu'
lDir = pDir + os.sep + 'locale'

mExt = '.sublime-menu.json'
mMenu = 'Main.sublime-menu'
sFile = pName + '.sublime-settings'
cFile = pName + '.sublime-commands'

class LocalizeRunCommand(sublime_plugin.WindowCommand):
	def run(self, action):
		if action == 'reset':
			restoreMenu()
			saveSetting('locale', '')
			return
		elif action == 'detect':
			makeMenu('', True)
			makeCommand('', True)
			return
		elif action == 'add':
			self.window.run_command("open_dir", {"dir": pDir})
		elif action == 'submit':
			self.window.run_command("open_url", {"url": "https://github.com/zam1024t/" + pName + "#submit-a-language"})

class LocalizeCommand(sublime_plugin.ApplicationCommand):
	def run(self, locale):
		saveSetting('locale', locale)
	def is_checked(self, locale):
		return locale == getSetting('locale')

def setLocale(locale, force = False):
	if not locale:
		return

	if (not force) and locale == getSetting('locale'):
		return

	link = getLink(locale)
	if link:
		locale = link

	m = {}
	d = os.path.join(mDir, v, locale)
	m.update(findMenu(d))

	sDir = ''
	d = os.path.join(mDir, version)
	ld = os.path.join(d, locale)
	if (not os.path.isdir(ld)) and getSetting('findSimilarVer', True):
		sDir = findSimilarVer()
		if not sDir:
			return
		d = os.path.join(mDir, sDir)

	d = os.path.join(d, locale)
	if not os.path.isdir(d):
		return
	m.update(findMenu(d))

	d = os.path.join(d, p)
	m.update(findMenu(d))

	d = os.path.join(d, sublime.arch())
	m.update(findMenu(d))

	updateMenu(m)
	if getSetting('updateTopMenu', True):
		updateTopMenu()
	sublime.status_message('Locale ' + locale + ' has loaded.')

def getLink(locale):
	f = os.path.join(lDir, locale, locale + '.json')
	if not os.path.isfile(f):
		return locale
	conf = getJson(f)
	if isset(conf, 'link') and conf['link']:
		locale = conf['link']
	return locale

def findSimilarVer():
	sVer = ''
	for item in os.listdir(mDir):
		subDir = os.path.join(mDir, item)
		if os.path.isdir(subDir) and item.isdigit() and item < version:
			sVer = item
	return sVer

def isset(dict, key):
	try:
		locale = dict[key]
	except KeyError:
		return False
	return True

def updateHotkey(menu):
	for idx, value in enumerate(menu):
		if isset(value, 'mnemonic') and isset(value, 'caption'):
			if value['mnemonic'] not in value['caption']:
				altKey = '(' + value['mnemonic'].upper() + ')'
				if altKey in value['caption']:
					value['mnemonic'] = value['mnemonic'].upper()
		if isset(value, 'children'):
			value['children'] = updateHotkey(value['children'])
		menu[idx] = value
	return menu

def makeCommand(locale, force = False):
	if not locale and not force:
		return

	f = os.path.join(pDir, '.command.json')
	if not os.path.isfile(f):
		return
	command = getJson(f)

	for item in sorted(os.listdir(lDir)):
		f = os.path.join(lDir, item, item + '.json')
		if os.path.isfile(f):
			conf = getJson(f)
			if isset(conf, 'hidden') and conf['hidden']:
				continue
			caption = item
			if isset(conf, 'caption') and conf['caption']:
				caption = conf['caption']
			command.append({
				"command": "localize",
				"args": {
					"locale": item
				},
				"caption": pName + ': ' + caption + ' (' + item + ')'
			})
	saveJson(os.path.join(pDir, cFile), command)

def findMenu(d):
	m = {}
	if not os.path.isdir(d):
		return m
	for item in os.listdir(d):
		filename = os.path.join(d, item)
		if os.path.isfile(filename) and item.endswith(mExt):
			m[item] = filename
	return m

def makeMenu(locale, force = False):
	if not locale and not force:
		return
	f = os.path.join(lDir, locale, 'menu.json')
	if not os.path.isfile(f):
		link = getLink(locale)
		if link:
			f = os.path.join(lDir, link, 'menu.json')
	if not os.path.isfile(f):
		f = os.path.join(pDir, '.menu.json')
	menu = getJson(f)

	for item in sorted(os.listdir(lDir)):
		f = os.path.join(lDir, item, item + '.json')
		if os.path.isfile(f):
			conf = getJson(f)
			if isset(conf, 'hidden') and conf['hidden']:
				continue
			caption = item
			if isset(conf, 'caption') and conf['caption']:
				caption = conf['caption']
			menu[0]['children'][2]['children'].append({
				"command": "localize",
				"checkbox": True,
				"args": {
					"locale": item
				},
				"caption": caption + ' (' + item + ')'
			})
	mFile = os.path.join(pDir, mMenu)
	saveJson(mFile, menu)

def updateMenu(m):
	if not os.path.isdir(dDir):
		os.makedirs(dDir)
	for k in m:
		target = os.path.join(dDir, k[:-5])
		menu = getJson(m[k])
		menu = updateHotkey(menu)
		if p == 'osx' and target[-17:] == mMenu:
			menu[8]['caption'] = 'Preferences'
			menu[8]['mnemonic'] = 'n'
		saveJson(target, menu)

def updateTopMenu():
	tDir = pkgs + os.sep + 'ZZZZZZZZ-' + pName
	if not os.path.isdir(tDir):
		os.makedirs(tDir)
	topMenu = []
	menu = getJson(dDir + os.sep + mMenu)
	for idx, subMenu in enumerate(menu):
		if isset(subMenu, 'children'):
			del(subMenu['children'])
		topMenu.append(subMenu)
	target = os.path.join(tDir, mMenu)
	saveJson(target, topMenu)

def backupMenu():
	if not os.path.isdir(dDir):
		return
	if not getSetting('auto_backup', True):
		return
	bDir = os.path.join(mDir, version, 'backup')
	if not os.path.isdir(bDir):
		os.makedirs(bDir)
	for item in os.listdir(dDir):
		filename = os.path.join(dDir, item)
		if os.path.isfile(filename) and item.endswith(mExt[:-5]):
			target = os.path.join(bDir, item + mExt[-5:])
			open(target, 'wb').write(open(filename, 'rb').read())

def restoreMenu():
	if not os.path.isdir(dDir):
		os.makedirs(dDir)
	for item in os.listdir(dDir):
		filename = os.path.join(dDir, item)
		if os.path.isfile(filename) and item.endswith(mExt[:-5]):
			os.remove(filename)
	bDir = os.path.join(mDir, version, 'backup')
	if not os.path.isdir(bDir):
		return
	for item in os.listdir(bDir):
		filename = os.path.join(bDir, item)
		if os.path.isfile(filename) and item.endswith(mExt):
			target = os.path.join(dDir, item[:-5])
			open(target, 'wb').write(open(filename, 'rb').read())

def unpackMenu(dFile, eDir):
	z = zipfile.ZipFile(dFile, 'r')
	files = [i.filename for i in z.infolist()]
	for item in files:
		if item.endswith(mExt[:-5]):
			z.extract(item, eDir)
			file = os.path.join(eDir, item)
			target = os.path.join(eDir, item + '.json');
			os.rename(file, target);
	z.close()

def unpackSelf(pkgFile, pkgDir):
	z = zipfile.ZipFile(pkgFile, 'r')
	files = [i.filename for i in z.infolist()]
	for item in files:
		z.extract(item, pkgDir)
	z.close()

def getSetting(key, value = None):
	conf = sublime.load_settings(sFile)
	return conf.get(key, value)

def setSetting(setting, key, value = None):
	setting.set(key, value)
	return setting

def saveSetting(key, value = None):
	conf = sublime.load_settings(sFile)
	conf.set(key, value)
	return sublime.save_settings(sFile)

def getJson(filename):
	if not os.path.isfile(filename):
		return
	data = ''
	with codecs.open(filename, 'rb', 'utf-8') as file:
		for line in file:
			if line.lstrip().startswith('//'):
				continue
			data = data + "\n" + line
	data = re.sub(",[ \t\r\n]+}", '}', data)
	data = re.sub(",[ \t\r\n]+\]", ']', data)
	arr = []
	try:
		arr = json.loads(data)
	except ValueError:
		pass
	return arr

def saveJson(path, data):
	data = json.dumps(data, ensure_ascii = False, indent = 4).encode('utf-8')
	open(path, 'wb').write(data)

def plugin_loaded():
	sublime.set_timeout(init, 200)

def init():
	if v == '3' and (sublime.installed_packages_path() in pDir):
		pkgDir = os.path.join(sublime.packages_path(), pName);
		if not os.path.isdir(pkgDir):
			pkgFile = os.path.dirname(os.path.abspath(__file__))
			unpackSelf(pkgFile, pkgDir)
			return
	locale = ''
	firstRun = False
	fFile = os.path.join(pDir, '.firstRun')
	if not os.path.isfile(fFile):
		firstRun = True
		backupMenu()
		open(fFile, 'wt').write('')
		locale = getSetting('locale', '')
	eDir = os.path.join(mDir, version, 'en');
	if v == '3' and not os.path.isdir(eDir):
		eFile = sublime.executable_path();
		dFile = os.path.join(os.path.dirname(eFile), 'Packages', 'Default.sublime-package');
		unpackMenu(dFile, eDir);
	makeMenu(locale, firstRun)
	makeCommand(locale, firstRun)
	setLocale(locale, firstRun)

	s = sublime.load_settings(sFile)
	s.add_on_change('locale', updateLocale)

def updateLocale():
	locale = getSetting('locale')
	if not locale:
		return
	lFile = os.path.join(pDir, '.lastRun')
	if os.path.isfile(lFile) and open(lFile, 'r').read() == locale:
		return
	makeMenu(locale, True)
	makeCommand(locale, True)
	setLocale(locale, True)
	open(lFile, 'wt').write(locale)

if (v == '2'):
	init()