#! /usr/bin/python3

from __future__ import print_function

import sys
import re
import os.path
import locale
from optparse import OptionParser
from gettext import gettext as _

import language_support_pkgs

if __name__ == "__main__":
try:
locale.setlocale(locale.LC_ALL, '')
except locale.Error:
# We might well be running from the installer before locales have
# been properly configured.
pass

parser = OptionParser()
parser.add_option("-l", "--language",
help=_("target language code"))
parser.add_option("-d", "--datadir",
default="/usr/share/language-selector/",
help=_("alternative datadir"))
parser.add_option("-p", "--package", default=None, help=_("check for the given package(s) only -- separate packagenames by comma"))
parser.add_option("-a", "--all", action='store_true', default=False,
help=_("output all available language support packages for all languages"))
parser.add_option("--show-installed",
action='store_true', default=False,
help=_("show installed packages as well as missing ones"))
(options, args) = parser.parse_args()

# sanity check for language code
if (options.language and
not re.match('^([a-z]{2,3}(_[A-Z]{2})?(@[a-z]+)?|(zh-han[st]))$', options.language)):
print("Error: Unsupported language code format.\n\
Supported formats: 'll' or 'lll'\n\
'll_CC' or 'lll_CC'\n\
'll@variant' or 'lll@variant'\n\
'll_CC@variant' or 'lll_CC@variant'\n\
Also supported are 'zh-hans' and 'zh-hant'.",
file=sys.stderr)
sys.exit(1)

if options.datadir:
pkg_depends = os.path.join(options.datadir, 'data', 'pkg_depends')
else:
pkg_depends = None

ls = language_support_pkgs.LanguageSupport(None, pkg_depends)

if options.all:
missing = set()
ls._countries_for_lang('de') # initialize cache
for lang, countries in ls.lang_country_map.items():
for country in countries:
missing.update(ls.by_locale('%s_%s' % (lang, country), options.show_installed))
elif options.package:
missing = set()
for package in options.package.split(','):
if options.language:
missing.update(ls.by_package_and_locale(package, options.language, options.show_installed))
else:
missing.update(ls.by_package(package, options.show_installed))
elif options.language:
missing = set(ls.by_locale(options.language, options.show_installed))
else:
missing = ls.missing(options.show_installed)

print(' '.join(sorted(missing)))