Add Handbook.md benchmark tasks

2026-06-24 12:44:34 -07:00
commit 72f9a65917
1800 changed files with 323589 additions and 0 deletions
@@ -0,0 +1,136 @@
+"""Direct message tool handlers."""
+
+from __future__ import annotations
+
+import re
+from typing import Any
+
+from slack_mock.models import SlackChannel, SlackMessage, SlackMessageType, SlackState
+from slack_mock.state import (
+    add_message,
+    generate_mpim_channel_id,
+    generate_timestamp,
+    get_bot_user_id,
+    get_state,
+    get_user,
+    mutate_state,
+)
+from slack_mock.tools.common import (
+    channel_payload,
+    create_channel_object,
+    empty_message,
+    is_direct_conversation,
+    message_payload,
+    model_dump,
+    now_seconds,
+)
+
+
+def open_dm(user_id: str) -> dict[str, Any]:
+    return _open_dm(user_id)
+
+
+def _open_dm(user_id: str) -> dict[str, Any]:
+    state = get_state()
+    user = get_user(user_id)
+    if user is None:
+        return {"ok": False, "error": "user_not_found", "channel": {}}
+    for channel in state.channels.values():
+        if channel.is_im and channel.user == user_id:
+            return {"ok": True, "channel": channel_payload(channel)}
+
+    def _open(state: SlackState) -> SlackChannel:
+        now = now_seconds()
+        channel_id = f"D{str(now)[-6:]}{re.sub(r'[^a-zA-Z0-9]', '', user_id)}"
+        channel = create_channel_object(channel_id, user.name, is_private=True, is_im=True, user_id=user_id)
+        state.channels[channel_id] = channel
+        state.messages[channel_id] = []
+        return channel
+
+    channel = mutate_state(_open)
+    return {"ok": True, "channel": channel_payload(channel)}
+
+
+def _mpim_members(user_ids: list[str]) -> tuple[list[str], list[str]]:
+    unique_user_ids = list(dict.fromkeys(user_ids))
+    bot_user_id = get_bot_user_id()
+    other_user_ids = sorted(user_id for user_id in unique_user_ids if user_id != bot_user_id)
+    return [bot_user_id, *other_user_ids], other_user_ids
+
+
+def _mpim_name(other_user_ids: list[str]) -> str:
+    state = get_state()
+    names = [state.users[user_id].name for user_id in other_user_ids]
+    normalized_names = [re.sub(r"[^a-z0-9_-]+", "-", name.casefold()).strip("-") for name in names]
+    return f"mpdm-{'--'.join(normalized_names)}-1"
+
+
+def open_mpim(user_ids: list[str]) -> dict[str, Any]:
+    return _open_mpim(user_ids)
+
+
+def _open_mpim(user_ids: list[str]) -> dict[str, Any]:
+    state = get_state()
+    member_ids, other_user_ids = _mpim_members(user_ids)
+    if len(other_user_ids) < 2:
+        return {"ok": False, "error": "not_enough_users", "channel": {}}
+    missing_user_ids = [user_id for user_id in other_user_ids if user_id not in state.users]
+    if missing_user_ids:
+        return {"ok": False, "error": "user_not_found", "channel": {}}
+
+    member_set = set(member_ids)
+    channel_name = _mpim_name(other_user_ids)
+    for channel in state.channels.values():
+        if not channel.is_mpim:
+            continue
+        if channel.members and set(channel.members) == member_set:
+            return {"ok": True, "channel": channel_payload(channel)}
+        if channel.members is None and channel.name == channel_name:
+            return {"ok": True, "channel": channel_payload(channel)}
+
+    def _open(state: SlackState) -> SlackChannel:
+        channel_id = generate_mpim_channel_id()
+        channel = create_channel_object(
+            channel_id,
+            channel_name,
+            is_private=True,
+            is_mpim=True,
+            members=member_ids,
+        )
+        state.channels[channel_id] = channel
+        state.messages[channel_id] = []
+        return channel
+
+    channel = mutate_state(_open)
+    return {"ok": True, "channel": channel_payload(channel)}
+
+
+def list_dms(limit: int = 20) -> dict[str, Any]:
+    dms = sorted(
+        (channel for channel in get_state().channels.values() if is_direct_conversation(channel)),
+        key=lambda channel: channel.updated,
+        reverse=True,
+    )
+    return {"ok": True, "channels": model_dump(dms[: limit or 20])}
+
+
+def send_dm(user_id: str, text: str) -> dict[str, Any]:
+    opened = _open_dm(user_id)
+    if not opened.get("ok"):
+        return {"ok": False, "error": opened.get("error"), "channel": "", "ts": "", "message": empty_message()}
+    channel_id = opened["channel"]["id"]
+    ts = generate_timestamp()
+    message = SlackMessage(type=SlackMessageType.MESSAGE, user=get_bot_user_id(), text=text, ts=ts, team="T_MOCK")
+    add_message(channel_id, message)
+    return {"ok": True, "channel": channel_id, "ts": ts, "message": message_payload(message)}
+
+
+def send_mpim(user_ids: list[str], text: str) -> dict[str, Any]:
+    opened = _open_mpim(user_ids)
+    if not opened.get("ok"):
+        return {"ok": False, "error": opened.get("error"), "channel": "", "ts": "", "message": empty_message()}
+    channel_id = opened["channel"]["id"]
+    ts = generate_timestamp()
+    message = SlackMessage(type=SlackMessageType.MESSAGE, user=get_bot_user_id(), text=text, ts=ts, team="T_MOCK")
+    add_message(channel_id, message)
+    return {"ok": True, "channel": channel_id, "ts": ts, "message": message_payload(message)}