Merge pull request chigwell#282 from vsem-azamat/feat/pin-message-notify

feat(messages): add optional notify to pin_message

Author: chigwell

telegram_mcp/tools/messages.py

@@ -1795,15 +1795,23 @@ async def delete_messages_bulk(
 @with_account(readonly=False)
 @validate_id("chat_id")
 async def pin_message(
-    chat_id: Union[int, str], message_id: int, account: Optional[str] = None
+    chat_id: Union[int, str],
+    message_id: int,
+    notify: bool = False,
+    account: Optional[str] = None,
 ) -> str:
     """
     Pin a message in a chat.
+
+    Args:
+        chat_id: The ID or username of the chat.
+        message_id: The ID of the message to pin.
+        notify: Notify chat members about the pin. Default False pins silently.
     """
     try:
         cl = get_client(account)
         entity = await resolve_entity(chat_id, cl)
-        await cl.pin_message(entity, message_id)
+        await cl.pin_message(entity, message_id, notify=notify)
         return f"Message {message_id} pinned in chat {chat_id}."
     except Exception as e:
         return log_and_format_error("pin_message", e, chat_id=chat_id, message_id=message_id)

tests/test_pin_notify.py

@@ -0,0 +1,47 @@
+import pytest
+
+from telegram_mcp.tools import messages
+
+
+class _PinClient:
+    def __init__(self):
+        self.calls = []
+
+    async def pin_message(self, entity, message_id, **kwargs):
+        self.calls.append((entity, message_id, kwargs))
+
+
+@pytest.fixture
+def client(monkeypatch):
+    cl = _PinClient()
+
+    async def fake_resolve(chat_id, client=None):
+        return "entity"
+
+    monkeypatch.setattr(messages, "get_client", lambda account=None: cl)
+    monkeypatch.setattr(messages, "resolve_entity", fake_resolve)
+    return cl
+
+
+@pytest.mark.asyncio
+async def test_pin_message_is_silent_by_default(client):
+    result = await messages.pin_message(chat_id=1, message_id=2)
+
+    assert result == "Message 2 pinned in chat 1."
+    assert client.calls == [("entity", 2, {"notify": False})]
+
+
+@pytest.mark.asyncio
+async def test_pin_message_notifies_when_requested(client):
+    result = await messages.pin_message(chat_id=1, message_id=2, notify=True)
+
+    assert result == "Message 2 pinned in chat 1."
+    assert client.calls == [("entity", 2, {"notify": True})]
+
+
+def test_pin_message_schema_exposes_optional_notify():
+    tool = messages.mcp._tool_manager.get_tool("pin_message")
+    notify = tool.parameters["properties"]["notify"]
+
+    assert notify["default"] is False
+    assert "notify" not in tool.parameters.get("required", [])