xsoar-cli 1.0.3__tar.gz → 1.0.5__tar.gz

{xsoar_cli-1.0.3 → xsoar_cli-1.0.5}/.github/workflows/pull-request-open.yml

@@ -5,14 +5,18 @@ on:
 jobs:
   test:
     runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.10", "3.11", "3.12"]
     steps:
       - name: Checkout code
         uses: actions/checkout@v4
 
-      - name: Setup Python 3.12
+      - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v5
         with:
-          python-version: "3.12"
+          python-version: ${{ matrix.python-version }}
+          cache: "pip"
 
       - name: Create and activate virtual environment
         run: |
@@ -21,7 +25,6 @@ jobs:
 
       - name: Install requirements
         run: |
-          pip install -r requirements.txt
           pip install -r requirements_dev.txt
 
       - name: Install xsoar-cli

{xsoar_cli-1.0.3 → xsoar_cli-1.0.5}/.github/workflows/release.yml

@@ -1,12 +1,8 @@
 name: release
 
 on:
-  push:
-    tags:
-      - "[0-9]+.[0-9]+.[0-9]+"
-      - "[0-9]+.[0-9]+.[0-9]+a[0-9]+"
-      - "[0-9]+.[0-9]+.[0-9]+b[0-9]+"
-      - "[0-9]+.[0-9]+.[0-9]+rc[0-9]+"
+  release:
+    types: [published]
 
 env:
   PACKAGE_NAME: "xsoar-cli"
@@ -117,28 +113,3 @@ jobs:
 
       - name: Publish distribution to PyPI
         uses: pypa/gh-action-pypi-publish@release/v1
-
-  github_release:
-    name: Create GitHub Release
-    needs: [setup_and_build, details]
-    runs-on: ubuntu-latest
-    permissions:
-      contents: write
-    steps:
-      - name: Checkout Code
-        uses: actions/checkout@v3
-        with:
-          fetch-depth: 0
-
-      - name: Download artifacts
-        uses: actions/download-artifact@v4
-        with:
-          name: dist
-          path: dist/
-
-      - name: Create GitHub Release
-        id: create_release
-        env:
-          GH_TOKEN: ${{ github.token }}
-        run: |
-          gh release create ${{ needs.details.outputs.tag_name }} dist/* --title ${{ needs.details.outputs.tag_name }} --generate-notes

{xsoar_cli-1.0.3 → xsoar_cli-1.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xsoar-cli
-Version: 1.0.3
+Version: 1.0.5
 Project-URL: Documentation, https://github.com/tlium/xsoar-cli#readme
 Project-URL: Issues, https://github.com/tlium/xsoar-cli/issues
 Project-URL: Source, https://github.com/tlium/xsoar-cli
@@ -9,10 +9,12 @@ License-Expression: MIT
 License-File: LICENSE.txt
 Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
-Requires-Python: >=3.12
+Requires-Python: <3.13,>=3.10
 Requires-Dist: click==8.1.8
 Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: xsoar-client>=1.0.0

{xsoar_cli-1.0.3 → xsoar_cli-1.0.5}/pyproject.toml

@@ -7,13 +7,15 @@ name = "xsoar-cli"
 dynamic = ["version"]
 description = ''
 readme = "README.md"
-requires-python = ">=3.12"
+requires-python = ">=3.10, <3.13"
 license = "MIT"
 keywords = []
 authors = [{ name = "Torbjørn Lium", email = "torben@lium.org" }]
 classifiers = [
     "Development Status :: 4 - Beta",
     "Programming Language :: Python",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
     "Programming Language :: Python :: Implementation :: CPython",
     "Programming Language :: Python :: Implementation :: PyPy",

{xsoar_cli-1.0.3 → xsoar_cli-1.0.5}/src/xsoar_cli/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2025-present Torbjørn Lium <torben@lium.org>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "1.0.3"
+__version__ = "1.0.5"

{xsoar_cli-1.0.3 → xsoar_cli-1.0.5}/src/xsoar_cli/case/commands.py

@@ -3,7 +3,7 @@ from typing import TYPE_CHECKING
 
 import click
 
-from xsoar_cli.utilities import load_config, validate_environments
+from xsoar_cli.utilities import load_config, parse_string_to_dict, validate_environments
 
 if TYPE_CHECKING:
     from xsoar_client.xsoar_client import Client
@@ -15,12 +15,14 @@ def case() -> None:
 
 
 @click.argument("casenumber", type=int)
-@click.option("--environment", default="dev", show_default=True, help="Environment as defined in config file")
+@click.option("--environment", default=None, help="Default environment set in config file.")
 @click.command(help="Get basic information about a single case in XSOAR")
 @click.pass_context
 @load_config
-def get(ctx: click.Context, casenumber: int, environment: str) -> None:
-    xsoar_client: Client = ctx.obj["server_envs"][environment]
+def get(ctx: click.Context, casenumber: int, environment: str | None) -> None:
+    if not environment:
+        environment = ctx.obj["default_environment"]
+    xsoar_client: Client = ctx.obj["server_envs"][environment]["xsoar_client"]
     response = xsoar_client.get_case(casenumber)
     if response["total"] == 0 and not response["data"]:
         click.echo(f"Cannot find case ID {casenumber}")
@@ -40,7 +42,7 @@ def clone(ctx: click.Context, casenumber: int, source: str, dest: str) -> None:
     if not valid_envs:
         click.echo(f"Error: cannot find environments {source} and/or {dest} in config")
         ctx.exit(1)
-    xsoar_source_client: Client = ctx.obj["server_envs"][source]
+    xsoar_source_client: Client = ctx.obj["server_envs"][source]["xsoar_client"]
     results = xsoar_source_client.get_case(casenumber)
     data = results["data"][0]
     # Dbot mirror info is irrelevant. This will be added again if applicable by XSOAR after ticket creation in dev.
@@ -57,21 +59,37 @@ def clone(ctx: click.Context, casenumber: int, source: str, dest: str) -> None:
     # Ensure that playbooks run immediately when the case is created
     data["createInvestigation"] = True
 
-    xsoar_dest_client: Client = ctx.obj["server_envs"][dest]
+    xsoar_dest_client: Client = ctx.obj["server_envs"][dest]["xsoar_client"]
     case_data = xsoar_dest_client.create_case(data=data)
     click.echo(json.dumps(case_data, indent=4))
 
 
-@click.option("--environment", default="dev", show_default=True, help="Environment as defined in config file")
+@click.option("--environment", default=None, help="Default environment set in config file.")
 @click.option("--casetype", default="", show_default=True, help="Create case of specified type. Default type set in config file.")
+@click.option(
+    "--custom-fields",
+    default=None,
+    help='Additional fields on the form "myfield=my_value,anotherfield=another value". Use machine name for field names, e.g mycustomfieldname.',
+)
+@click.option("--custom-fields-delimiter", default=",", help='Delimiter when specifying additional fields. Default is ","')
 @click.argument("details", type=str, default="Placeholder case details")
 @click.argument("name", type=str, default="Test case created from xsoar-cli")
 @click.command()
 @click.pass_context
 @load_config
-def create(ctx: click.Context, environment: str, casetype: str, name: str, details: str) -> None:
+def create(  # noqa: PLR0913
+    ctx: click.Context,
+    environment: str | None,
+    casetype: str,
+    name: str,
+    custom_fields: str | None,
+    custom_fields_delimiter: str | None,
+    details: str,
+) -> None:
     """Creates a new case in XSOAR. If invalid case type is specified as a command option, XSOAR will default to using Unclassified."""
-    xsoar_client: Client = ctx.obj["server_envs"][environment]
+    if not environment:
+        environment = ctx.obj["default_environment"]
+    xsoar_client: Client = ctx.obj["server_envs"][environment]["xsoar_client"]
     if not casetype:
         casetype = ctx.obj["default_new_case_type"]
     data = {
@@ -79,6 +97,7 @@ def create(ctx: click.Context, environment: str, casetype: str, name: str, detai
         "name": name,
         "type": casetype,
         "details": details,
+        "CustomFields": parse_string_to_dict(custom_fields, custom_fields_delimiter),
     }
     case_data = xsoar_client.create_case(data=data)
     case_id = case_data["id"]

{xsoar_cli-1.0.3 → xsoar_cli-1.0.5}/src/xsoar_cli/config/commands.py

@@ -8,7 +8,13 @@ if TYPE_CHECKING:
 
 import contextlib
 
-from xsoar_cli.utilities import get_config_file_contents, get_config_file_path, get_config_file_template_contents, load_config
+from xsoar_cli.utilities import (
+    fail_if_no_artifacts_provider,
+    get_config_file_contents,
+    get_config_file_path,
+    get_config_file_template_contents,
+    load_config,
+)
 
 
 @click.group(help="Create/validate etc")
@@ -35,6 +41,7 @@ def show(ctx: click.Context, masked: bool) -> None:
 @click.option("--only-test-environment", default=None, show_default=True, help="Environment as defined in config file")
 @click.pass_context
 @load_config
+@fail_if_no_artifacts_provider
 def validate(ctx: click.Context, only_test_environment: str) -> None:
     """Validates that the configuration file is JSON and tests connectivity for each XSOAR Client environment defined."""
     return_code = 0
@@ -44,7 +51,7 @@ def validate(ctx: click.Context, only_test_environment: str) -> None:
             # what the user specified in option
             continue
         click.echo(f'Testing "{server_env}" environment...', nl=False)
-        xsoar_client: Client = ctx.obj["server_envs"][server_env]
+        xsoar_client: Client = ctx.obj["server_envs"][server_env]["xsoar_client"]
         try:
             xsoar_client.test_connectivity()
         except ConnectionError as ex:
@@ -53,6 +60,9 @@ def validate(ctx: click.Context, only_test_environment: str) -> None:
             return_code = 1
             continue
         click.echo("OK")
+    if ctx.obj["default_environment"] not in ctx.obj["server_envs"]:
+        click.echo(f'Error: default environment "{ctx.obj["default_environment"]}" not found in server config.')
+        return_code = 1
     ctx.exit(return_code)
 
 

{xsoar_cli-1.0.3 → xsoar_cli-1.0.5}/src/xsoar_cli/manifest/commands.py

@@ -39,14 +39,43 @@ def manifest() -> None:
     """Various commands to interact/update/deploy content packs defined in the xsoar_config.json manifest."""
 
 
-@click.option("--environment", default="dev", show_default=True, help="Environment as defined in config file")
+@click.option("--environment", default=None, help="Default environment set in config file.")
+@click.argument("manifest_path", type=str)
+@click.command()
+@click.pass_context
+@load_config
+def generate(ctx: click.Context, environment: str | None, manifest_path: str) -> None:
+    """Generate a new xsoar_config.json manifest from installed content packs.
+
+    This command assumes that you do not have any custom content packs uploaded to XSOAR.
+    All packs will be added as "marketplace_packs" in the manifest.
+    """
+    if not environment:
+        environment = ctx.obj["default_environment"]
+    xsoar_client: Client = ctx.obj["server_envs"][environment]["xsoar_client"]
+    installed_packs = xsoar_client.get_installed_packs()
+    manifest_data = {
+        "marketplace_packs": [],
+    }
+    for item in installed_packs:
+        tmpobj = {
+            "id": item["id"],
+            "version": item["currentVersion"],
+        }
+        manifest_data["marketplace_packs"].append(tmpobj)
+    write_manifest(manifest_path, manifest_data)
+
+
+@click.option("--environment", default=None, help="Default environment set in config file.")
 @click.argument("manifest", type=str)
 @click.command()
 @click.pass_context
 @load_config
-def update(ctx: click.Context, environment: str, manifest: str) -> None:
+def update(ctx: click.Context, environment: str | None, manifest: str) -> None:
     """Update manifest on disk with latest available content pack versions."""
-    xsoar_client: Client = ctx.obj["server_envs"][environment]
+    if not environment:
+        environment = ctx.obj["default_environment"]
+    xsoar_client: Client = ctx.obj["server_envs"][environment]["xsoar_client"]
     manifest_data = load_manifest(manifest)
     click.echo("Fetching outdated packs from XSOAR server. This may take a minute...", nl=False)
     results = xsoar_client.get_outdated_packs()
@@ -81,15 +110,17 @@ def update(ctx: click.Context, environment: str, manifest: str) -> None:
     write_manifest(manifest, manifest_data)
 
 
-@click.option("--environment", default="dev", show_default=True, help="Environment as defined in config file")
+@click.option("--environment", default=None, help="Default environment set in config file.")
 @click.argument("manifest", type=str)
 @click.command()
 @click.pass_context
 @load_config
-def validate(ctx: click.Context, environment: str, manifest: str) -> None:
+def validate(ctx: click.Context, environment: str | None, manifest: str) -> None:
     """Validate manifest JSON and all pack availability. Validates upstream pack availability by doing HTTP CONNECT.
     Custom pack availability is implementation dependant."""
-    xsoar_client: Client = ctx.obj["server_envs"][environment]
+    if not environment:
+        environment = ctx.obj["default_environment"]
+    xsoar_client: Client = ctx.obj["server_envs"][environment]["xsoar_client"]
     manifest_data = load_manifest(manifest)
     click.echo("Manifest is valid JSON")
     keys = ["custom_packs", "marketplace_packs"]
@@ -121,15 +152,17 @@ def validate(ctx: click.Context, environment: str, manifest: str) -> None:
     click.echo("Manifest is valid JSON and all packs are reachable.")
 
 
-@click.option("--environment", default="dev", show_default=True, help="Environment as defined in config file")
+@click.option("--environment", default=None, help="Default environment set in config file.")
 @click.argument("manifest", type=str)
 @click.command()
 @click.pass_context
 @load_config
-def diff(ctx: click.Context, manifest: str, environment: str) -> None:
+def diff(ctx: click.Context, manifest: str, environment: str | None) -> None:
     """Prints out the differences (if any) between what is defined in the xsoar_config.json manifest and what is actually
     installed on the XSOAR server."""
-    xsoar_client: Client = ctx.obj["server_envs"][environment]
+    if not environment:
+        environment = ctx.obj["default_environment"]
+    xsoar_client: Client = ctx.obj["server_envs"][environment]["xsoar_client"]
     manifest_data = load_manifest(manifest)
     installed_packs = xsoar_client.get_installed_packs()
     all_good = True
@@ -147,14 +180,14 @@ def diff(ctx: click.Context, manifest: str, environment: str) -> None:
         click.echo("All packs up to date.")
 
 
-@click.option("--environment", default="dev", show_default=True, help="Environment as defined in config file")
+@click.option("--environment", default=None, help="Default environment set in config file.")
 @click.option("--verbose", is_flag=True, default=False)
 @click.option("--yes", is_flag=True, default=False)
 @click.command()
 @click.argument("manifest", type=str)
 @click.pass_context
 @load_config
-def deploy(ctx: click.Context, environment: str, manifest: str, verbose: bool, yes: bool) -> None:  # noqa: FBT001
+def deploy(ctx: click.Context, environment: str | None, manifest: str, verbose: bool, yes: bool) -> None:  # noqa: FBT001
     """
     Deploys content packs to the XSOAR server as defined in the xsoar_config.json manifest.
     The PATH argument expects the full or relative path to xsoar_config.json
@@ -169,8 +202,9 @@ def deploy(ctx: click.Context, environment: str, manifest: str, verbose: bool, y
         )
     if not should_continue:
         ctx.exit()
-
-    xsoar_client: Client = ctx.obj["server_envs"][environment]
+    if not environment:
+        environment = ctx.obj["default_environment"]
+    xsoar_client: Client = ctx.obj["server_envs"][environment]["xsoar_client"]
     manifest_data = load_manifest(manifest)
     click.echo("Fetching installed packs...", err=True)
     installed_packs = xsoar_client.get_installed_packs()
@@ -191,10 +225,11 @@ def deploy(ctx: click.Context, environment: str, manifest: str, verbose: bool, y
                 # Print message that install is skipped
 
     if none_installed:
-        click.echo("No packs to install. XSOAR server is up to date with manifest.")
+        click.echo("No packs to install. All packs and versions in manifest is already installed on XSOAR server.")
 
 
 manifest.add_command(deploy)
 manifest.add_command(diff)
 manifest.add_command(update)
 manifest.add_command(validate)
+manifest.add_command(generate)

{xsoar_cli-1.0.3 → xsoar_cli-1.0.5}/src/xsoar_cli/pack/commands.py

@@ -15,14 +15,16 @@ def pack(ctx: click.Context) -> None:
     """Various content pack related commands."""
 
 
-@click.option("--environment", default="dev", show_default=True, help="Environment as defined in config file")
+@click.option("--environment", default=None, help="Default environment set in config file.")
 @click.command()
 @click.argument("pack_id", type=str)
 @click.pass_context
 @load_config
-def delete(ctx: click.Context, environment: str, pack_id: str) -> None:
+def delete(ctx: click.Context, environment: str | None, pack_id: str) -> None:
     """Deletes a content pack from the XSOAR server."""
-    xsoar_client: Client = ctx.obj["server_envs"][environment]
+    if not environment:
+        environment = ctx.obj["default_environment"]
+    xsoar_client: Client = ctx.obj["server_envs"][environment]["xsoar_client"]
     if not xsoar_client.is_installed(pack_id=pack_id):
         click.echo(f"Pack ID {pack_id} is not installed. Cannot delete.")
         sys.exit(1)
@@ -30,13 +32,15 @@ def delete(ctx: click.Context, environment: str, pack_id: str) -> None:
     click.echo(f"Deleted pack {pack_id} from XSOAR {environment}")
 
 
-@click.option("--environment", default="dev", show_default=True, help="Environment as defined in config file")
+@click.option("--environment", default=None, help="Default environment set in config file.")
 @click.command()
 @click.pass_context
 @load_config
-def get_outdated(ctx: click.Context, environment: str) -> None:
+def get_outdated(ctx: click.Context, environment: str | None) -> None:
     """Prints out a list of outdated content packs."""
-    xsoar_client: Client = ctx.obj["server_envs"][environment]
+    if not environment:
+        environment = ctx.obj["default_environment"]
+    xsoar_client: Client = ctx.obj["server_envs"][environment]["xsoar_client"]
     click.echo("Fetching outdated packs. This may take a little while...", err=True)
     outdated_packs = xsoar_client.get_outdated_packs()
     if not outdated_packs:

{xsoar_cli-1.0.3 → xsoar_cli-1.0.5}/src/xsoar_cli/playbook/commands.py

@@ -19,12 +19,12 @@ def playbook(ctx: click.Context) -> None:
     """Download/attach/detach playbooks"""
 
 
-@click.option("--environment", default="dev", show_default=True, help="Environment as defined in config file")
+@click.option("--environment", default=None, help="Default environment set in config file.")
 @click.command()
 @click.argument("name", type=str)
 @click.pass_context
 @load_config
-def download(ctx: click.Context, environment: str, name: str) -> None:
+def download(ctx: click.Context, environment: str | None, name: str) -> None:
     """Download and reattach playbook.
 
     We try to detect output path to $(cwd)/Packs/<Pack ID>/Playbooks/<name>.yml
@@ -32,7 +32,8 @@ def download(ctx: click.Context, environment: str, name: str) -> None:
     then demisto-sdk format --assume-yes --no-validate --no-graph is done on the downloaded playbook before the item
     is re-attached in XSOAR.
     """
-
+    if not environment:
+        environment = ctx.obj["default_environment"]
     xsoar_client: Client = ctx.obj["server_envs"][environment]
     # Maybe we should search for the playbook before attempting download in
     # case user specifies a cutsom playbook and not a system playbook

{xsoar_cli-1.0.3 → xsoar_cli-1.0.5}/src/xsoar_cli/plugins/README.md

@@ -40,20 +40,20 @@ class MyPlugin(XSOARPlugin):
     @property
     def name(self) -> str:
         return "myplugin"
-    
-    @property 
+
+    @property
     def version(self) -> str:
         return "1.0.0"
-    
+
     @property
     def description(self) -> str:
         return "My custom plugin"
-    
+
     def get_command(self) -> click.Command:
         @click.command(help="My custom command")
         def mycommand():
             click.echo("Hello from my plugin!")
-        
+
         return mycommand
 ```
 
@@ -75,23 +75,23 @@ class MyCustomPlugin(XSOARPlugin):
     @property
     def name(self) -> str:
         return "mycustom"
-    
+
     @property
     def version(self) -> str:
         return "1.0.0"
-    
+
     @property
     def description(self) -> str:
         return "A custom plugin with multiple commands"
-    
+
     def get_command(self) -> click.Command:
         """Return the main command group for this plugin."""
-        
+
         @click.group(help="My custom commands")
         def mycustom():
             """Main command group for my custom plugin."""
             pass
-        
+
         @click.command(help="Greet someone")
         @click.option("--name", default="World", help="Name to greet")
         @click.option("--times", default=1, help="Number of times to greet")
@@ -99,7 +99,7 @@ class MyCustomPlugin(XSOARPlugin):
             """Greet someone multiple times."""
             for i in range(times):
                 click.echo(f"Hello, {name}!")
-        
+
         @click.command(help="Show current status")
         @click.option("--verbose", "-v", is_flag=True, help="Verbose output")
         def status(verbose: bool):
@@ -108,7 +108,7 @@ class MyCustomPlugin(XSOARPlugin):
             if verbose:
                 click.echo(f"Description: {self.description}")
                 click.echo("Status: Active")
-        
+
         @click.command(help="Process a file")
         @click.argument("filename", type=click.Path(exists=True))
         @click.option("--output", "-o", help="Output file")
@@ -118,18 +118,18 @@ class MyCustomPlugin(XSOARPlugin):
             if output:
                 click.echo(f"Output will be saved to: {output}")
             # Your processing logic here
-        
+
         # Add commands to the group
         mycustom.add_command(greet)
         mycustom.add_command(status)
         mycustom.add_command(process)
-        
+
         return mycustom
-    
+
     def initialize(self):
         """Initialize the plugin."""
         click.echo("My custom plugin initialized!")
-    
+
     def cleanup(self):
         """Cleanup when the plugin is unloaded."""
         pass
@@ -176,7 +176,7 @@ from xsoar_cli.utilities import load_config
 @load_config
 def my_command(ctx: click.Context):
     # Access XSOAR client
-    xsoar_client = ctx.obj["server_envs"]["dev"]
+    xsoar_client = ctx.obj["server_envs"]["dev"]["xsoar_client"]
     # Use the client...
 ```
 
@@ -284,11 +284,11 @@ class MyPlugin(XSOARPlugin):
         @click.group()
         def myplugin():
             pass
-        
+
         @click.command()
         def case():  # ✅ Namespaced as 'myplugin case'
             click.echo("My case command")
-        
+
         myplugin.add_command(case)
         return myplugin
 ```
@@ -297,7 +297,7 @@ class MyPlugin(XSOARPlugin):
 
 These command names are reserved by the core CLI:
 - `case` - Case/incident management
-- `config` - Configuration management  
+- `config` - Configuration management
 - `graph` - Dependency graphs
 - `manifest` - Manifest operations
 - `pack` - Content pack operations
@@ -330,7 +330,7 @@ def get_command(self) -> click.Command:
         except Exception as e:
             click.echo(f"Error: {e}", err=True)
             raise click.Abort()
-    
+
     return mycommand
 ```
 
@@ -346,7 +346,7 @@ def my_command(ctx: click.Context):
     # Access configuration
     config = ctx.obj
     # Access XSOAR clients
-    dev_client = ctx.obj["server_envs"]["dev"]
+    dev_client = ctx.obj["server_envs"]["dev"]["xsoar_client"]
 ```
 
 ### 5. Logging
@@ -430,4 +430,4 @@ except ImportError:
 
 You can share plugins by simply sharing the Python file. Users can place it in their plugins directory and it will be automatically discovered.
 
-For more complex plugins, consider packaging them as proper Python packages that install the plugin file automatically.
+For more complex plugins, consider packaging them as proper Python packages that install the plugin file automatically.

{xsoar_cli-1.0.3 → xsoar_cli-1.0.5}/src/xsoar_cli/utilities.py

@@ -1,3 +1,4 @@
+import contextlib
 import json
 from collections.abc import Callable
 from functools import update_wrapper
@@ -7,6 +8,16 @@ import click
 from xsoar_client.xsoar_client import Client
 
 
+def parse_string_to_dict(input_string: str | None, delimiter: str) -> dict:
+    if not input_string:
+        return {}
+    # Parse a string into a python dictionary
+    pairs = [pair.split("=", 1) for pair in input_string.split(delimiter)]
+    # Filter pairs that have exactly 2 parts (key and value) after splitting by "="
+    valid_pairs = [pair for pair in pairs if len(pair) == 2]
+    return {key.strip(): value.strip() for key, value in valid_pairs}
+
+
 def get_config_file_template_contents() -> dict:
     return {
         "default_environment": "dev",
@@ -66,7 +77,11 @@ def load_config(f: Callable) -> Callable:
             ctx.exit(1)
         config = get_config_file_contents(config_file_path)
         parse_config(config, ctx)
-        if "environment" in ctx.params and ctx.params["environment"] not in ctx.obj["server_envs"]:
+        if (
+            "environment" in ctx.params
+            and ctx.params["environment"] not in ctx.obj["server_envs"]
+            and ctx.params["environment"] is not None
+        ):
             click.echo(f"Invalid environment: {ctx.params['environment']}")
             click.echo(f"Available environments as defined in config file are: {list(ctx.obj['server_envs'])}")
             ctx.exit(1)
@@ -75,6 +90,29 @@ def load_config(f: Callable) -> Callable:
     return update_wrapper(wrapper, f)
 
 
+def fail_if_no_artifacts_provider(f: Callable) -> Callable:
+    """
+    This function is only to be used as a decorator for various xsoar-cli subcommands, and only AFTER the load_config decorator has been called.
+    The intention is to fail gracefully if any subcommand is executed which requires an artifacts provider."
+    """
+
+    @click.pass_context
+    def wrapper(ctx: click.Context, *args, **kwargs) -> Callable:  # noqa: ANN002, ANN003
+        if "environment" in ctx.params:  # noqa: SIM108
+            key = ctx.params["environment"]
+        else:
+            key = ctx.obj["default_environment"]
+
+        with contextlib.suppress(KeyError):
+            location = ctx.obj["server_envs"][key].get("artifacts_location", None)
+            if not location:
+                click.echo("Command requires artifacts repository, but no artifacts_location defined in config.")
+                ctx.exit(1)
+        return ctx.invoke(f, *args, **kwargs)
+
+    return update_wrapper(wrapper, f)
+
+
 def parse_config(config: dict, ctx: click.Context) -> None:
     # Set the two XSOAR client objects in Click Context for use in later functions
     ctx.obj = {}
@@ -83,7 +121,8 @@ def parse_config(config: dict, ctx: click.Context) -> None:
     ctx.obj["default_new_case_type"] = config["default_new_case_type"]
     ctx.obj["server_envs"] = {}
     for key in config["server_config"]:
-        ctx.obj["server_envs"][key] = Client(
+        ctx.obj["server_envs"][key] = {}
+        ctx.obj["server_envs"][key]["xsoar_client"] = Client(
             api_token=config["server_config"][key]["api_token"],
             server_url=config["server_config"][key]["base_url"],
             verify_ssl=config["server_config"][key]["verify_ssl"],
@@ -93,3 +132,4 @@ def parse_config(config: dict, ctx: click.Context) -> None:
             artifacts_location=config["server_config"][key].get("artifacts_location", None),
             s3_bucket_name=config["server_config"][key].get("s3_bucket_name", None),
         )
+        ctx.obj["server_envs"][key]["artifacts_location"] = config["server_config"][key].get("artifacts_location", None)