mas-cli 5.1.4__py3-none-any.whl

mas/cli/displayMixins.py

@@ -0,0 +1,132 @@
+# *****************************************************************************
+# Copyright (c) 2024 IBM Corporation and other Contributors.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Eclipse Public License v1.0
+# which accompanies this distribution, and is available at
+# http://www.eclipse.org/legal/epl-v10.html
+#
+# *****************************************************************************
+
+from os import getenv
+from prompt_toolkit import prompt, print_formatted_text, HTML, PromptSession
+from prompt_toolkit.completion import WordCompleter
+from prompt_toolkit.validation import Validator
+
+from .validators import YesNoValidator, FileExistsValidator, DirectoryExistsValidator
+
+import logging
+logger = logging.getLogger(__name__)
+
+H1COLOR = "SkyBlue"
+H2COLOR = "SkyBlue"
+DESCRIPTIONCOLOR = "LightSlateGrey"
+SUMMARYCOLOR = "SkyBlue"
+UNDEFINEDPARAMCOLOR = "LightSlateGrey"
+PROMPTCOLOR = "Yellow"
+
+
+class PrintMixin():
+    def printTitle(self, message):
+        print_formatted_text(HTML(f"<b><u>{message.replace(' & ', ' &amp; ')}</u></b>"))
+
+    def printH1(self, message):
+        self.h1count += 1
+        self.h2count = 0
+        print()
+        print_formatted_text(HTML(f"<u><{H1COLOR}>{self.h1count}) {message.replace(' & ', ' &amp; ')}</{H1COLOR}></u>"))
+
+    def printH2(self, message):
+        self.h2count += 1
+        print()
+        print_formatted_text(HTML(f"<u><{H2COLOR}>{self.h1count}.{self.h2count}) {message.replace(' & ', ' &amp; ')}</{H2COLOR}></u>"))
+
+    def printDescription(self, content: list) -> None:
+        content[0] = f"<{DESCRIPTIONCOLOR}>{content[0]}"
+        content[len(content) - 1] = f"{content[len(content) - 1]}</{DESCRIPTIONCOLOR}>"
+        print_formatted_text(HTML("\n".join(content)))
+
+    def printHighlight(self, message: str) -> None:
+        if isinstance(message, list):
+            message = "\n".join(message)
+
+        print_formatted_text(HTML(f"<MediumTurquoise>{message.replace(' & ', ' &amp; ')}</MediumTurquoise>"))
+
+    def printWarning(self, message):
+        print_formatted_text(HTML(f"<Red>Warning: {message.replace(' & ', ' &amp; ')}</Red>"))
+
+    def printSummary(self, title: str, value: str) -> None:
+        titleLength = len(title)
+        message = f"{title} {'.' * (40 - titleLength)} {value}"
+        print_formatted_text(HTML(f"  <{SUMMARYCOLOR}>{message.replace(' & ', ' &amp; ')}</{SUMMARYCOLOR}>"))
+
+    def printParamSummary(self, message: str, param: str) -> None:
+        if self.getParam(param) is None:
+            self.printSummary(message, f"<{UNDEFINEDPARAMCOLOR}>Undefined</{UNDEFINEDPARAMCOLOR}>")
+        elif self.getParam(param) == "":
+            self.printSummary(message, f"<{UNDEFINEDPARAMCOLOR}>Default</{UNDEFINEDPARAMCOLOR}>")
+        else:
+            self.printSummary(message, self.getParam(param))
+
+
+def masPromptYesOrNo(message):
+    return HTML(f"<{PROMPTCOLOR}>{message.replace(' & ', ' &amp; ')}? [y/n]</{PROMPTCOLOR}> ")
+
+
+def masPromptValue(message):
+    return HTML(f"<{PROMPTCOLOR}>{message.replace(' & ', ' &amp; ')}</{PROMPTCOLOR}> ")
+
+
+class PromptMixin():
+    def yesOrNo(self, message: str, param: str = None) -> bool:
+        response = prompt(masPromptYesOrNo(message), validator=YesNoValidator(), validate_while_typing=False)
+        responseAsBool = response.lower() in ["y", "yes"]
+
+        if param is not None:
+            self.params[param] = "true" if responseAsBool else "false"
+        return responseAsBool
+
+    def promptForString(self, message: str, param: str = None, default: str = "", isPassword: bool = False, validator: Validator = None, completer: WordCompleter = None) -> str:
+        if param is not None and default == "":
+            default = getenv(param.upper(), default="")
+
+        if completer is not None:
+            promptSession = PromptSession()
+            response = promptSession.prompt(masPromptValue(message), is_password=isPassword, default=default, completer=completer, validator=validator, validate_while_typing=False, pre_run=promptSession.default_buffer.start_completion)
+        else:
+            response = prompt(masPromptValue(message), is_password=isPassword, default=default, completer=completer, validator=validator, validate_while_typing=False)
+
+        if param is not None:
+            self.params[param] = response
+        return response
+
+    def promptForInt(self, message: str, param: str = None, default: int = None) -> int:
+        if param is not None and default is None:
+            default = getenv(param.upper(), default=None)
+
+        if default is None:
+            response = int(prompt(masPromptValue(message)))
+        else:
+            response = int(prompt(masPromptValue(message), default=str(default)))
+        if param is not None:
+            self.params[param] = str(response)
+        return response
+
+    def promptForListSelect(self, message: str, options: list, param: str = None, default: int = None) -> str:
+        selection = self.promptForInt(message=message, default=default)
+        # List indices are 0 origin, so we need to subtract 1 from the selection made to arrive at the correct value
+        self.setParam(param, options[selection - 1])
+
+    def promptForFile(self, message: str, mustExist: bool = True, default: str = "", envVar: str = "") -> None:
+        if default == "" and envVar != "":
+            default = getenv(envVar, "")
+        if mustExist:
+            return prompt(masPromptValue(message), validator=FileExistsValidator(), validate_while_typing=False, default=default)
+        else:
+            return prompt(masPromptValue(message), default=default)
+
+    def promptForDir(self, message: str, mustExist: bool = True, default: str = "") -> None:
+        if mustExist:
+            return prompt(masPromptValue(message), validator=DirectoryExistsValidator(), validate_while_typing=False, default=default)
+        else:
+            return prompt(masPromptValue(message), default=default)

mas/cli/gencfg.py

@@ -0,0 +1,113 @@
+# *****************************************************************************
+# Copyright (c) 2024 IBM Corporation and other Contributors.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Eclipse Public License v1.0
+# which accompanies this distribution, and is available at
+# http://www.eclipse.org/legal/epl-v10.html
+#
+# *****************************************************************************
+
+from os import path
+from jinja2 import Template
+from base64 import b64encode
+from json import loads
+
+
+class ConfigGeneratorMixin():
+    def generateJDBCCfg(self, instanceId: str, scope: str, destination: str, appId: str = "", workspaceId: str = "") -> None:
+        templateFile = path.join(self.templatesDir, "jdbccfg.yml.j2")
+        with open(templateFile) as tFile:
+            template = Template(tFile.read())
+
+        if scope == "workspace-application":
+            assert appId != ""
+            assert workspaceId != ""
+
+        name = self.promptForString("Configuration Display Name")
+        url = self.promptForString("JDBC Connection String")
+
+        username = self.promptForString("JDBC Username")
+        password = self.promptForString("JDBC Password", isPassword=True)
+
+        sslEnabled = self.yesOrNo("Enable SSL Connection")
+
+        if sslEnabled:
+            sslCertFile = self.promptForFile("Path to certificate file")
+            with open(sslCertFile) as cFile:
+                certLocalFileContent = cFile.read()
+        else:
+            certLocalFileContent = ""
+
+        cfg = template.render(
+            scope=scope,
+
+            mas_instance_id=instanceId,
+            mas_workspace_id=workspaceId,
+            mas_application_id=appId,
+
+            cfg_display_name=name,
+
+            jdbc_url=url,
+            jdbc_username=username,
+            jdbc_password=password,
+
+            jdbc_ssl_enabled=sslEnabled,
+            jdbc_cert_local_file_content=certLocalFileContent
+        )
+
+        with open(destination, 'w') as f:
+            f.write(cfg)
+            f.write('\n')
+
+    def generateMongoCfg(self, instanceId: str, destination: str) -> None:
+        templateFile = path.join(self.templatesDir, "suite_mongocfg.yml.j2")
+
+        with open(templateFile) as tFile:
+            template = Template(tFile.read())
+
+        name = self.promptForString("Configuration Display Name")
+        hosts = self.promptForString("MongoDb Hosts (comma-separated list)")
+
+        username = self.promptForString("MongoDb Username")
+        password = self.promptForString("MongoDb Password", isPassword=True)
+        encoded_username = b64encode(username.encode('ascii')).decode("ascii")
+        encoded_password = b64encode(password.encode('ascii')).decode("ascii")
+        sslCertFile = self.promptForFile("Path to certificate file")
+        with open(sslCertFile) as cFile:
+            certLocalFileContent = cFile.read()
+
+        cfg = template.render(
+            mas_instance_id=instanceId,
+            cfg_display_name=name,
+            mongodb_hosts=hosts,
+            mongodb_admin_username=encoded_username,
+            mongodb_admin_password=encoded_password,
+            mongodb_ca_pem_local_file=certLocalFileContent
+        )
+
+        with open(destination, 'w') as f:
+            f.write(cfg)
+            f.write('\n')
+
+    def generateFacilitiesCfg(self, destination: str) -> None:
+        templateFile = path.join(self.templatesDir, "facilities-configs.yml.j2")
+
+        with open(templateFile) as tFile:
+            template = Template(tFile.read())
+
+        dwfagents = self.getParam("mas_ws_facilities_dwfagents")
+        maxconnpoolsize = self.getParam("mas_ws_facilities_db_maxconnpoolsize")
+        userfiles_size = self.getParam("mas_ws_facilities_storage_userfiles_size")
+        log_size = self.getParam("mas_ws_facilities_storage_log_size")
+        cfg = template.render(
+            mas_instance_id=self.getParam("mas_instance_id"),
+            mas_ws_facilities_storage_log_size=log_size if log_size != "" else 30,
+            mas_ws_facilities_storage_userfiles_size=userfiles_size if userfiles_size != "" else 50,
+            mas_ws_facilities_db_maxconnpoolsize=maxconnpoolsize if maxconnpoolsize != "" else 200,
+            mas_ws_facilities_dwfagents=loads(dwfagents) if dwfagents != '' else ''
+        )
+
+        with open(destination, 'w') as f:
+            f.write(cfg)
+            f.write('\n')

mas/cli/install/__init__.py

@@ -0,0 +1,11 @@
+# *****************************************************************************
+# Copyright (c) 2024 IBM Corporation and other Contributors.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Eclipse Public License v1.0
+# which accompanies this distribution, and is available at
+# http://www.eclipse.org/legal/epl-v10.html
+#
+# *****************************************************************************
+
+from ..cli import BaseApp  # noqa: F401