rougail-cli/src/rougail/cli/__main__.py

"""
Silique (https://www.silique.fr)
Copyright (C) 2024-2025

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

Mtools is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with Mtools.  If not, see <http://www.gnu.org/licenses/>.
"""

from tiramisu_cmdline_parser import TiramisuCmdlineParser
from tiramisu.error import PropertiesOptionError
from tiramisu import Config
from pathlib import Path

from rougail import Rougail
from rougail.config import get_rougail_config
from rougail.utils import load_modules
try:
    from rougail.user_data_environment import RougailUserDataEnvironment
    from rougail.user_datas import UserDatas
except ImportError:
    RougailUserDataEnvironment = None

from .i18n import _


def _main():
    rougailconfig = get_rougail_config(
        backward_compatibility=False, add_extra_options=False
    )
    rougailconfig.generate_config()
    cmd_config = rougailconfig.config
    origin_prop = cmd_config.property.default('read_write', 'append')
    cmd_config.property.setdefault(frozenset(origin_prop | {"not_for_commandline"}), 'read_write', 'append')
    cmd_config.property.read_write()
    if RougailUserDataEnvironment:
        fake_rougail_config = {'step.user_data': 'environment',
                               'environment.default_environment_name': 'ROUGAILCLI',
                               }
        user_data = UserDatas(cmd_config).user_datas(RougailUserDataEnvironment(cmd_config, rougailconfig=fake_rougail_config).run())
        if user_data["errors"]:
            raise Exception(user_data["errors"][0])
        if user_data["warnings"]:
            raise Exception(user_data["warnings"][0])
    parser = TiramisuCmdlineParser(
        cmd_config,
        add_extra_options=False,
        short_name_max_len=2,
    )
    parser.parse_args()
    global print_traceback
    print_traceback = rougailconfig["cli.debug"]
    cmd_config.property.setdefault(origin_prop, 'read_write', 'append')
    cmd_config.property.remove("not_for_commandline")
    cmd_config.property.read_only()
    try:
        user_data_names = rougailconfig["step.user_data"]
    except PropertiesOptionError:
        user_data_names = []
    output_name = rougailconfig["step.output"]
    # structural
    rougail = Rougail(rougailconfig)
    config = rougail.run()
    # data user
    if not user_data_names:
        user_datas = None
    else:
        config.property.read_write()
        user_datas = []
        for user_data_name in user_data_names:
            path = (
                Path(__file__).parent.parent
                / ("user_data_" + user_data_name)
                / "__init__.py"
            )
            if not path.is_file():
                raise Exception(
                    _('cannot find "user_data" module "{0}"').format(user_data_name)
                )
            module = load_modules("rougail.user_data_" + user_data_name, str(path))
            if hasattr(module.RougailUserData, 'force_apply_user_data') and module.RougailUserData.force_apply_user_data is True:
                continue
            user_datas.extend(
                module.RougailUserData(
                    config,
                    rougailconfig=rougailconfig,
                ).run()
            )
    if user_datas:
        err_warn = rougail.user_datas(user_datas)
    else:
        err_warn = {"errors": [], "warnings": []}
    if user_data_names:
        config.property.read_write()
        user_datas = []
        for user_data_name in user_data_names:
            path = (
                Path(__file__).parent.parent
                / ("user_data_" + user_data_name)
                / "__init__.py"
            )
            module = load_modules("rougail.user_data_" + user_data_name, str(path))
            if hasattr(module.RougailUserData, 'force_apply_user_data') and module.RougailUserData.force_apply_user_data is True:
                ret = module.RougailUserData(
                    config,
                    rougailconfig=rougailconfig,
                ).run()
                err_warn["errors"].extend(ret["errors"])
                err_warn["warnings"].extend(ret["warnings"])
    # output
    config.property.read_only()
    path = Path(__file__).parent.parent / ("output_" + output_name) / "__init__.py"
    if not path.is_file():
        raise Exception(
            _('cannot find cli file for "output_name" module "{0}"').format(output_name)
        )
    module = load_modules("rougail.output_" + output_name, str(path))
    ret = module.RougailOutput(
        config=config,
        rougailconfig=rougailconfig,
        user_data_errors=err_warn["errors"],
        user_data_warnings=err_warn["warnings"],
    ).print()
    if ret is False:
        exit(1)


def main():
    global print_traceback
    print_traceback = False
    try:
        _main()
    except Exception as err:
        if print_traceback:
            import traceback
            traceback.print_exc()
        print(_("ERROR: {0}").format(err))
        exit(1)