from optparse import OptionGroup
from hachoir.core.log import log
from hachoir.core.i18n import _, getTerminalCharset
from hachoir.core.tools import makePrintable
import hachoir.core.config as config


def getHachoirOptions(parser):
    """
    Create an option group (type optparse.OptionGroup) of Hachoir
    library options.
    """

    def setLogFilename(*args):
        log.setFilename(args[2])

    common = OptionGroup(parser, _("Hachoir library"),
                         "Configure Hachoir library")
    common.add_option("--verbose", help=_("Verbose mode"),
                      default=False, action="store_true")
    common.add_option("--log", help=_("Write log in a file"),
                      type="string", action="callback",
                      callback=setLogFilename)
    common.add_option("--quiet", help=_("Quiet mode (don't display warning)"),
                      default=False, action="store_true")
    common.add_option("--debug", help=_("Debug mode"),
                      default=False, action="store_true")
    return common


def configureHachoir(option):
    # Configure Hachoir using "option" (value from optparse)
    if option.quiet:
        config.quiet = True
    if option.verbose:
        config.verbose = True
    if option.debug:
        config.debug = True


def unicodeFilename(filename, charset=None):
    if not charset:
        charset = getTerminalCharset()
    try:
        return unicode(filename, charset)
    except UnicodeDecodeError:
        return makePrintable(filename, charset, to_unicode=True)