specify action

Author: snopoke

taskbadger/cli.py

@@ -1,10 +1,12 @@
 import subprocess
-from typing import Optional
+from typing import Optional, List, Tuple
 
 import typer
 from rich import print
 
 import taskbadger as tb
+from taskbadger import Action
+from taskbadger import integrations
 from taskbadger.config import get_config, write_config
 from taskbadger.exceptions import ConfigurationError
 
@@ -21,10 +23,21 @@ def _configure_api(ctx):
 
 
 @app.command(context_settings={"allow_extra_args": True, "ignore_unknown_options": True})
-def monitor(ctx: typer.Context, name: str):
+def monitor(
+        ctx: typer.Context,
+        name: str,
+        action_def: Tuple[str, integrations.Integrations, str] = typer.Option(
+            (None, None, None), "--action", "-a", metavar="<trigger integration config>", show_default=False,
+            help="Action definition e.g. \"success,error email to:me@email.com\""
+        )
+):
     """Monitor a command"""
     _configure_api(ctx)
-    task = tb.Task.create(name)
+    action = None
+    if all(action_def):
+        trigger, integration, config = action_def
+        action = Action(trigger, integrations.from_config(integration, config))
+    task = tb.Task.create(name, actions=[action] if action else None)
     try:
         result = subprocess.run(ctx.args, env={"TASKBADGER_TASK_ID": task.id}, shell=True)
     except Exception as e:

taskbadger/integrations.py

@@ -1,9 +1,21 @@
 import dataclasses
-from typing import Any, Dict
+from enum import Enum
+from typing import Any, Dict, List, Generator, Tuple
 
 from taskbadger.internal.models import ActionRequest, ActionRequestConfig
 
 
+class Integrations(str, Enum):
+    email = "email"
+
+
+def from_config(integration: Integrations, config: str):
+    if integration == Integrations.email:
+        split_ = [tuple(item.split(':', 1)) for item in config.split(',')]
+        kwargs = dict(split_)
+        return EmailIntegration(**kwargs)
+
+
 class Integration:
     name: str
 

tests/test_monitor.py

@@ -30,7 +30,19 @@ def test_monitor_error():
     _test_monitor(["not-a-command"], 127)
 
 
-def _test_monitor(command, return_code):
+@mock.patch.dict(os.environ, {
+    "TASKBADGER_ORG": "org",
+    "TASKBADGER_PROJECT": "project",
+    "TASKBADGER_TOKEN": "token",
+}, clear=True)
+def test_monitor_action():
+    _test_monitor(
+        ["echo", "test"], 0, ["-a", "success,error", "email", "to:me@test.com"],
+        action={'trigger': 'success,error', 'integration': 'email', 'config': {'to': 'me@test.com'}}
+    )
+
+
+def _test_monitor(command, return_code, args=None, action=None):
     with (
         mock.patch("taskbadger.sdk.task_create.sync_detailed") as create,
         mock.patch("taskbadger.sdk.task_partial_update.sync_detailed") as update
@@ -39,15 +51,21 @@ def _test_monitor(command, return_code):
         create.return_value = Response(HTTPStatus.OK, b"", {}, task)
 
         update.return_value = Response(HTTPStatus.OK, b"", {}, task)
-        result = runner.invoke(app, ["monitor", "task_name", "--"] + command)
+        args = args or []
+        result = runner.invoke(app, ["monitor", "task_name"] + args + ["--"] + command)
         assert result.exit_code == return_code
 
         settings = _get_settings()
+        request = TaskRequest(name="task_name", status=StatusEnum.PENDING)
+        if action:
+            request.additional_properties = {
+                'actions': [action]
+            }
         create.assert_called_with(
             client=settings.client,
             organization_slug="org",
             project_slug="project",
-            json_body=TaskRequest(name="task_name", status=StatusEnum.PENDING)
+            json_body=request
         )
 
         if return_code == 0: