""" 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 . """ from warnings import warn from pathlib import Path from sys import exit from tiramisu_cmdline_parser import TiramisuCmdlineParser from tiramisu.error import PropertiesOptionError from tiramisu import Config from rougail import Rougail from rougail.config import get_rougail_config from rougail.utils import load_modules from rougail.user_datas import UserDatas try: from rougail.user_data_yaml import RougailUserDataYaml except ImportError: RougailUserDataYaml = None try: from rougail.user_data_environment import RougailUserDataEnvironment except ImportError: RougailUserDataEnvironment = None from .i18n import _ def _main(arguments, do_not_print): 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() fake_rougail_config = { "step.user_data": [], } fake_user_datas = [] if RougailUserDataYaml: config_file = cmd_config.forcepermissive.option("cli.config_file").value.get() if Path(config_file).is_file(): fake_rougail_config = { "step.user_data": "yaml", "yaml.filename": [config_file], "yaml.file_with_secrets": "all", } fake_user_datas.extend( RougailUserDataYaml(cmd_config, rougailconfig=fake_rougail_config).run() ) if RougailUserDataEnvironment: fake_rougail_config = { "step.user_data": "environment", "environment.default_environment_name": "ROUGAILCLI", } fake_user_datas.extend( RougailUserDataEnvironment( cmd_config, rougailconfig=fake_rougail_config ).run() ) if fake_user_datas: user_datas = UserDatas(cmd_config).user_datas(fake_user_datas, return_values_not_error=True) else: user_datas = {} parser = TiramisuCmdlineParser( cmd_config, # add_extra_options=False, short_name_max_len=2, ) parser.parse_args(arguments) if user_datas: user_data = UserDatas(cmd_config).user_datas(user_datas, user_datas_type="values") if user_data["warnings"]: for warning in user_data["warnings"]: warn(warning) if user_data["errors"]: raise Exception(user_data["errors"][0]) 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() if rougailconfig["cli.versions"]: versions = [] from tiramisu import __version__ versions.append(f"tiramisu: {__version__}") from tiramisu_cmdline_parser import __version__ versions.append(f"tiramisu-cmdline-parser: {__version__}") from rougail import __version__ versions.append(f"rougail: {__version__}") from . import __version__ versions.append(f"rougail-cli: {__version__}") for step in ["structural", "user_data", "output"]: display_step = step.replace("_", "-") for step_name in sorted(cmd_config.unrestraint.option(f"step.{step}").value.list()): path = ( Path(__file__).parent.parent / (step + "_" + step_name) / "__init__.py" ) if path.is_file(): try: module = load_modules( "rougail." + step + "_" + step_name, str(path) ) versions.append( f"rougail-{display_step}-{step_name}: {module.__version__}" ) except Exception as err: pass if do_not_print: return versions for version in versions: print(version) return try: user_data_names = rougailconfig["step.user_data"] except PropertiesOptionError: user_data_names = [] # structural if rougailconfig["cli.load_config"]: rougail = Rougail(rougailconfig) config = rougail.run() else: config = None # 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() 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 if config: config.property.read_only() output_name = rougailconfig["step.output"] 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)) output = module.RougailOutput( config=config, rougailconfig=rougailconfig, user_data_errors=err_warn["errors"], user_data_warnings=err_warn["warnings"], ) if do_not_print: return output.run() ret = output.print() if ret is False: exit(1) def main(arguments=None, do_not_print=False): global print_traceback print_traceback = True try: return _main(arguments, do_not_print) except Exception as err: if print_traceback: import traceback traceback.print_exc() exit(_("ERROR: {0}").format(err))