feat: support channels shortcut in wrapper

Author: bardonadam

README.md

@@ -40,6 +40,7 @@ response = activitysmith.notifications.send(
     {
         "title": "Build Failed",
         "message": "CI pipeline failed on main branch",
+        "channels": ["devs", "ops"],  # Optional
     }
 )
 
@@ -59,7 +60,8 @@ start = activitysmith.live_activities.start(
             "current_step": 1,
             "type": "segmented_progress",
             "color": "yellow",
-        }
+        },
+        "channels": ["devs", "ops"],  # Optional
     }
 )
 

activitysmith/client.py

@@ -10,12 +10,36 @@
 from activitysmith_openapi.api.push_notifications_api import PushNotificationsApi
 
 
+def _normalize_channels_target(request: Any) -> Any:
+    if not isinstance(request, dict):
+        return request
+
+    if "target" in request or "channels" not in request:
+        return request
+
+    normalized = dict(request)
+    raw_channels = normalized.pop("channels")
+
+    channels: list[str] = []
+    if isinstance(raw_channels, str):
+        channels = [item.strip() for item in raw_channels.split(",") if item.strip() != ""]
+    elif isinstance(raw_channels, (list, tuple)):
+        channels = [item.strip() for item in raw_channels if isinstance(item, str) and item.strip() != ""]
+
+    if channels:
+        normalized["target"] = {"channels": channels}
+
+    return normalized
+
+
 class NotificationsResource:
     def __init__(self, api: PushNotificationsApi) -> None:
         self._api = api
 
     def send(self, request: Any):
-        return self._api.send_push_notification(push_notification_request=request)
+        return self._api.send_push_notification(
+            push_notification_request=_normalize_channels_target(request)
+        )
 
     # Backward-compatible alias.
     def send_push_notification(self, push_notification_request: Any):
@@ -27,7 +51,9 @@ def __init__(self, api: LiveActivitiesApi) -> None:
         self._api = api
 
     def start(self, request: Any):
-        return self._api.start_live_activity(live_activity_start_request=request)
+        return self._api.start_live_activity(
+            live_activity_start_request=_normalize_channels_target(request)
+        )
 
     def update(self, request: Any):
         return self._api.update_live_activity(live_activity_update_request=request)

tests/test_resources.py

@@ -46,6 +46,22 @@ def test_notifications_short_and_legacy_alias(monkeypatch):
     ]
 
 
+def test_notifications_map_channels_to_target(monkeypatch):
+    monkeypatch.setattr(client_module, "PushNotificationsApi", FakePushNotificationsApi)
+    monkeypatch.setattr(client_module, "LiveActivitiesApi", FakeLiveActivitiesApi)
+
+    client = ActivitySmith(api_key="x")
+    payload = {"title": "Build Failed", "channels": ["devs", "ops"]}
+
+    client.notifications.send(payload)
+    client.notifications.send_push_notification({"title": "Build Failed", "channels": "devs,ops"})
+
+    assert client.notifications._api.calls == [
+        {"push_notification_request": {"title": "Build Failed", "target": {"channels": ["devs", "ops"]}}},
+        {"push_notification_request": {"title": "Build Failed", "target": {"channels": ["devs", "ops"]}}},
+    ]
+
+
 def test_live_activities_short_and_legacy_aliases(monkeypatch):
     monkeypatch.setattr(client_module, "PushNotificationsApi", FakePushNotificationsApi)
     monkeypatch.setattr(client_module, "LiveActivitiesApi", FakeLiveActivitiesApi)
@@ -84,3 +100,36 @@ def test_live_activities_short_and_legacy_aliases(monkeypatch):
         ("update", {"live_activity_update_request": update_payload}),
         ("end", {"live_activity_end_request": end_payload}),
     ]
+
+
+def test_live_activities_start_maps_channels_to_target(monkeypatch):
+    monkeypatch.setattr(client_module, "PushNotificationsApi", FakePushNotificationsApi)
+    monkeypatch.setattr(client_module, "LiveActivitiesApi", FakeLiveActivitiesApi)
+
+    client = ActivitySmith(api_key="x")
+    payload = {
+        "content_state": {
+            "title": "Deploy",
+            "number_of_steps": 4,
+            "current_step": 1,
+            "type": "segmented_progress",
+        },
+        "channels": ["devs", "ops"],
+    }
+
+    client.live_activities.start(payload)
+    client.live_activities.start_live_activity(payload)
+
+    expected = {
+        "content_state": {
+            "title": "Deploy",
+            "number_of_steps": 4,
+            "current_step": 1,
+            "type": "segmented_progress",
+        },
+        "target": {"channels": ["devs", "ops"]},
+    }
+    assert client.live_activities._api.calls == [
+        ("start", {"live_activity_start_request": expected}),
+        ("start", {"live_activity_start_request": expected}),
+    ]