feat(demisto-pack): zip file support

Author: german.ramos.garcia

demisto-pack/MistLang/Integrations/MistLang/MistLang.py

@@ -3,6 +3,9 @@
 from mist.action_run import execute_from_text
 from mist.lang.config import config
 import asyncio, json
+from zipfile import ZipFile
+from io import BytesIO
+import os
 
 # Disable insecure warnings
 requests.packages.urllib3.disable_warnings()  # pylint: disable=no-member
@@ -25,8 +28,14 @@ def baseintegration_dummy(self, url, params) -> Tuple[str, dict]:
         req = requests.get(url)
         if req.status_code == 200:
             try:
-                #TODO: option to download zip files with multple files programs and then download
-                mist_content = req.content.decode("utf-8", "ignore")
+                if url.endswith(".zip"):
+                    zipfile = ZipFile(BytesIO(req.content))
+                    zipfile.extractall("./mist_tmp_zip")
+                    os.chdir('./mist_tmp_zip')
+                    with open("main.mist") as f:
+                        mist_content = f.read()
+                else:
+                    mist_content = req.content.decode("utf-8", "ignore")
                 output = asyncio.run(execute_from_text(mist_content, fn_params=params))
                 result = {"url": url, "raw_output": output}
                 try:
@@ -38,6 +47,10 @@ def baseintegration_dummy(self, url, params) -> Tuple[str, dict]:
                 return "ok", result
             except Exception as e:
                 return "mist file error", {"url": url, "output": str(e)}
+            finally:
+                if os.path.abspath(os.path.curdir).endswith("mist_tmp_zip"):
+                    os.chdir('..')
+                    os.system("rm -rf ./mist_tmp_zip")
         return "network error", {"url": url, "output": str(req.status_code)}
 
 

demisto-pack/MistLang/Integrations/MistLang/MistLang_test.py

@@ -93,3 +93,26 @@ def test_baseintegration_dummy_with_params():
     assert response[1]["raw_output"] == '{"output_message1": "bar", "output_message2": "foo"}\n'
     assert response[1]["output_message1"] == "bar"
     assert response[1]["output_message2"] == "foo"
+
+def test_baseintegration_dummy_zip():
+    """Tests helloworld-say-hello command function.
+
+    Checks the output of the command function with the expected output.
+
+    No mock is needed here because the say_hello_command does not call
+    any external API.
+    """
+    from MistLang import Client, baseintegration_dummy_command
+
+    client = Client(base_url='some_mock_url', verify=False)
+    args = {
+        'url': 'https://raw.githubusercontent.com/BBVA/mist/master/examples/multi_file.zip',
+    }
+    response = baseintegration_dummy_command(client, **args)
+
+    assert response[0] == "ok"
+    assert response[1]["raw_output"] == '{"message": "hello"}\n'
+    assert response[1]["message"] == "hello"
+    
+
+