mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-05-03 02:11:48 +00:00
feat: pluggable platform adapter registry + IRC reference implementation
Adds a platform adapter plugin interface so anyone can create new gateway
platforms (IRC, Viber, Line, etc.) as drop-in plugins without modifying
core gateway code.
## Platform Registry (gateway/platform_registry.py)
- PlatformEntry dataclass: name, label, adapter_factory, check_fn,
validate_config, required_env, install_hint, source
- PlatformRegistry singleton with register/unregister/create_adapter
- _create_adapter() in gateway/run.py checks registry first, falls
through to existing if/elif chain for built-in platforms
## Dynamic Platform Enum (gateway/config.py)
- Platform._missing_() accepts unknown string values, creating cached
pseudo-members so Platform('irc') is Platform('irc') holds true
- GatewayConfig.from_dict() now parses plugin platform names from
config.yaml without rejecting them
- get_connected_platforms() delegates to registry for unknown platforms
## Plugin Registration (hermes_cli/plugins.py)
- PluginContext.register_platform() for plugin authors
- Mirrors the existing register_tool() / register_hook() pattern
## IRC Reference Plugin (plugins/platforms/irc/)
- Full async IRC adapter using stdlib asyncio (zero external deps)
- Connects via TLS, handles PING/PONG, nick collision, NickServ auth
- Channel messages require addressing (nick: msg), DMs always dispatch
- Markdown stripping for IRC-clean output, message splitting for
512-byte line limit
- Config via config.yaml extra dict or IRC_* env vars
## Tests (55 new tests)
- Platform enum dynamic members (identity stability, case normalization)
- PlatformRegistry (register, unregister, create, validation, factory)
- GatewayConfig integration (from_dict parsing, get_connected_platforms)
- IRC adapter (init, send, protocol parsing, markdown, requirements)
No existing platform adapters were migrated — the if/elif chain is
untouched. This is Phase 1: prove the interface with a real plugin.
This commit is contained in:
parent
4bede272cf
commit
e1f4de4dd3
10 changed files with 1469 additions and 5 deletions
380
tests/gateway/test_irc_adapter.py
Normal file
380
tests/gateway/test_irc_adapter.py
Normal file
|
|
@ -0,0 +1,380 @@
|
|||
"""Tests for the IRC platform adapter plugin."""
|
||||
|
||||
import asyncio
|
||||
import os
|
||||
import sys
|
||||
import pytest
|
||||
from pathlib import Path
|
||||
from unittest.mock import AsyncMock, MagicMock, patch
|
||||
|
||||
# Ensure the plugins directory is on sys.path for direct import
|
||||
_REPO_ROOT = Path(__file__).resolve().parents[2]
|
||||
_IRC_PLUGIN_DIR = _REPO_ROOT / "plugins" / "platforms" / "irc"
|
||||
if str(_IRC_PLUGIN_DIR) not in sys.path:
|
||||
sys.path.insert(0, str(_IRC_PLUGIN_DIR))
|
||||
|
||||
|
||||
# ── IRC protocol helpers ─────────────────────────────────────────────────
|
||||
|
||||
from adapter import _parse_irc_message, _extract_nick
|
||||
|
||||
|
||||
class TestIRCProtocolHelpers:
|
||||
|
||||
def test_parse_simple_command(self):
|
||||
msg = _parse_irc_message("PING :server.example.com")
|
||||
assert msg["command"] == "PING"
|
||||
assert msg["params"] == ["server.example.com"]
|
||||
assert msg["prefix"] == ""
|
||||
|
||||
def test_parse_prefixed_message(self):
|
||||
msg = _parse_irc_message(":nick!user@host PRIVMSG #channel :Hello world")
|
||||
assert msg["prefix"] == "nick!user@host"
|
||||
assert msg["command"] == "PRIVMSG"
|
||||
assert msg["params"] == ["#channel", "Hello world"]
|
||||
|
||||
def test_parse_numeric_reply(self):
|
||||
msg = _parse_irc_message(":server 001 hermes-bot :Welcome to IRC")
|
||||
assert msg["prefix"] == "server"
|
||||
assert msg["command"] == "001"
|
||||
assert msg["params"] == ["hermes-bot", "Welcome to IRC"]
|
||||
|
||||
def test_parse_nick_collision(self):
|
||||
msg = _parse_irc_message(":server 433 * hermes-bot :Nickname is already in use")
|
||||
assert msg["command"] == "433"
|
||||
|
||||
def test_extract_nick_full_prefix(self):
|
||||
assert _extract_nick("nick!user@host") == "nick"
|
||||
|
||||
def test_extract_nick_bare(self):
|
||||
assert _extract_nick("server.example.com") == "server.example.com"
|
||||
|
||||
|
||||
# ── IRC Adapter ──────────────────────────────────────────────────────────
|
||||
|
||||
from adapter import IRCAdapter, check_requirements, validate_config
|
||||
|
||||
|
||||
class TestIRCAdapterInit:
|
||||
|
||||
def test_init_from_env(self, monkeypatch):
|
||||
monkeypatch.setenv("IRC_SERVER", "irc.test.net")
|
||||
monkeypatch.setenv("IRC_PORT", "6667")
|
||||
monkeypatch.setenv("IRC_NICKNAME", "testbot")
|
||||
monkeypatch.setenv("IRC_CHANNEL", "#test")
|
||||
monkeypatch.setenv("IRC_USE_TLS", "false")
|
||||
|
||||
from gateway.config import PlatformConfig
|
||||
cfg = PlatformConfig(enabled=True)
|
||||
adapter = IRCAdapter(cfg)
|
||||
|
||||
assert adapter.server == "irc.test.net"
|
||||
assert adapter.port == 6667
|
||||
assert adapter.nickname == "testbot"
|
||||
assert adapter.channel == "#test"
|
||||
assert adapter.use_tls is False
|
||||
|
||||
def test_init_from_config_extra(self, monkeypatch):
|
||||
# Clear any env vars
|
||||
for key in ("IRC_SERVER", "IRC_PORT", "IRC_NICKNAME", "IRC_CHANNEL", "IRC_USE_TLS"):
|
||||
monkeypatch.delenv(key, raising=False)
|
||||
|
||||
from gateway.config import PlatformConfig
|
||||
cfg = PlatformConfig(
|
||||
enabled=True,
|
||||
extra={
|
||||
"server": "irc.libera.chat",
|
||||
"port": 6697,
|
||||
"nickname": "hermes",
|
||||
"channel": "#hermes-dev",
|
||||
"use_tls": True,
|
||||
},
|
||||
)
|
||||
adapter = IRCAdapter(cfg)
|
||||
|
||||
assert adapter.server == "irc.libera.chat"
|
||||
assert adapter.port == 6697
|
||||
assert adapter.nickname == "hermes"
|
||||
assert adapter.channel == "#hermes-dev"
|
||||
assert adapter.use_tls is True
|
||||
|
||||
def test_env_overrides_config(self, monkeypatch):
|
||||
monkeypatch.setenv("IRC_SERVER", "env-server.net")
|
||||
|
||||
from gateway.config import PlatformConfig
|
||||
cfg = PlatformConfig(
|
||||
enabled=True,
|
||||
extra={"server": "config-server.net", "channel": "#ch"},
|
||||
)
|
||||
adapter = IRCAdapter(cfg)
|
||||
assert adapter.server == "env-server.net"
|
||||
|
||||
|
||||
class TestIRCAdapterSend:
|
||||
|
||||
@pytest.fixture
|
||||
def adapter(self, monkeypatch):
|
||||
for key in ("IRC_SERVER", "IRC_PORT", "IRC_NICKNAME", "IRC_CHANNEL", "IRC_USE_TLS"):
|
||||
monkeypatch.delenv(key, raising=False)
|
||||
from gateway.config import PlatformConfig
|
||||
cfg = PlatformConfig(
|
||||
enabled=True,
|
||||
extra={
|
||||
"server": "localhost",
|
||||
"port": 6667,
|
||||
"nickname": "testbot",
|
||||
"channel": "#test",
|
||||
"use_tls": False,
|
||||
},
|
||||
)
|
||||
return IRCAdapter(cfg)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_send_not_connected(self, adapter):
|
||||
result = await adapter.send("#test", "hello")
|
||||
assert result.success is False
|
||||
assert "Not connected" in result.error
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_send_success(self, adapter):
|
||||
writer = MagicMock()
|
||||
writer.is_closing = MagicMock(return_value=False)
|
||||
writer.write = MagicMock()
|
||||
writer.drain = AsyncMock()
|
||||
adapter._writer = writer
|
||||
|
||||
result = await adapter.send("#test", "hello world")
|
||||
assert result.success is True
|
||||
assert result.message_id is not None
|
||||
# Verify PRIVMSG was sent
|
||||
writer.write.assert_called()
|
||||
sent_data = writer.write.call_args[0][0]
|
||||
assert b"PRIVMSG #test :hello world" in sent_data
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_send_splits_long_messages(self, adapter):
|
||||
writer = MagicMock()
|
||||
writer.is_closing = MagicMock(return_value=False)
|
||||
writer.write = MagicMock()
|
||||
writer.drain = AsyncMock()
|
||||
adapter._writer = writer
|
||||
|
||||
long_msg = "x" * 1000
|
||||
result = await adapter.send("#test", long_msg)
|
||||
assert result.success is True
|
||||
# Should have been split into multiple PRIVMSG calls
|
||||
assert writer.write.call_count > 1
|
||||
|
||||
|
||||
class TestIRCAdapterMessageParsing:
|
||||
|
||||
@pytest.fixture
|
||||
def adapter(self, monkeypatch):
|
||||
for key in ("IRC_SERVER", "IRC_PORT", "IRC_NICKNAME", "IRC_CHANNEL", "IRC_USE_TLS"):
|
||||
monkeypatch.delenv(key, raising=False)
|
||||
from gateway.config import PlatformConfig
|
||||
cfg = PlatformConfig(
|
||||
enabled=True,
|
||||
extra={
|
||||
"server": "localhost",
|
||||
"port": 6667,
|
||||
"nickname": "hermes",
|
||||
"channel": "#test",
|
||||
"use_tls": False,
|
||||
},
|
||||
)
|
||||
a = IRCAdapter(cfg)
|
||||
a._current_nick = "hermes"
|
||||
a._registered = True
|
||||
return a
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_handle_ping(self, adapter):
|
||||
writer = MagicMock()
|
||||
writer.is_closing = MagicMock(return_value=False)
|
||||
writer.write = MagicMock()
|
||||
writer.drain = AsyncMock()
|
||||
adapter._writer = writer
|
||||
|
||||
await adapter._handle_line("PING :test-server")
|
||||
sent = writer.write.call_args[0][0]
|
||||
assert b"PONG :test-server" in sent
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_handle_welcome(self, adapter):
|
||||
adapter._registered = False
|
||||
adapter._registration_event = asyncio.Event()
|
||||
|
||||
await adapter._handle_line(":server 001 hermes :Welcome to IRC")
|
||||
assert adapter._registered is True
|
||||
assert adapter._registration_event.is_set()
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_handle_nick_collision(self, adapter):
|
||||
writer = MagicMock()
|
||||
writer.is_closing = MagicMock(return_value=False)
|
||||
writer.write = MagicMock()
|
||||
writer.drain = AsyncMock()
|
||||
adapter._writer = writer
|
||||
|
||||
await adapter._handle_line(":server 433 * hermes :Nickname in use")
|
||||
assert adapter._current_nick == "hermes_"
|
||||
sent = writer.write.call_args[0][0]
|
||||
assert b"NICK hermes_" in sent
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_handle_addressed_channel_message(self, adapter):
|
||||
"""Messages addressed to the bot (nick: msg) should be dispatched."""
|
||||
handler = AsyncMock(return_value="response")
|
||||
adapter._message_handler = handler
|
||||
|
||||
# Mock handle_message to capture the event
|
||||
dispatched = []
|
||||
original_dispatch = adapter._dispatch_message
|
||||
|
||||
async def capture_dispatch(**kwargs):
|
||||
dispatched.append(kwargs)
|
||||
|
||||
adapter._dispatch_message = capture_dispatch
|
||||
|
||||
await adapter._handle_line(":user!u@host PRIVMSG #test :hermes: hello there")
|
||||
assert len(dispatched) == 1
|
||||
assert dispatched[0]["text"] == "hello there"
|
||||
assert dispatched[0]["chat_id"] == "#test"
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_ignores_unaddressed_channel_message(self, adapter):
|
||||
dispatched = []
|
||||
|
||||
async def capture_dispatch(**kwargs):
|
||||
dispatched.append(kwargs)
|
||||
|
||||
adapter._dispatch_message = capture_dispatch
|
||||
adapter._message_handler = AsyncMock()
|
||||
|
||||
await adapter._handle_line(":user!u@host PRIVMSG #test :just talking")
|
||||
assert len(dispatched) == 0
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_handle_dm(self, adapter):
|
||||
"""DMs (target == bot nick) should always be dispatched."""
|
||||
dispatched = []
|
||||
|
||||
async def capture_dispatch(**kwargs):
|
||||
dispatched.append(kwargs)
|
||||
|
||||
adapter._dispatch_message = capture_dispatch
|
||||
adapter._message_handler = AsyncMock()
|
||||
|
||||
await adapter._handle_line(":user!u@host PRIVMSG hermes :private message")
|
||||
assert len(dispatched) == 1
|
||||
assert dispatched[0]["text"] == "private message"
|
||||
assert dispatched[0]["chat_type"] == "dm"
|
||||
assert dispatched[0]["chat_id"] == "user"
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_ignores_own_messages(self, adapter):
|
||||
dispatched = []
|
||||
|
||||
async def capture_dispatch(**kwargs):
|
||||
dispatched.append(kwargs)
|
||||
|
||||
adapter._dispatch_message = capture_dispatch
|
||||
adapter._message_handler = AsyncMock()
|
||||
|
||||
await adapter._handle_line(":hermes!bot@host PRIVMSG #test :my own msg")
|
||||
assert len(dispatched) == 0
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_ctcp_action_converted(self, adapter):
|
||||
"""CTCP ACTION (/me) should be converted to text."""
|
||||
dispatched = []
|
||||
|
||||
async def capture_dispatch(**kwargs):
|
||||
dispatched.append(kwargs)
|
||||
|
||||
adapter._dispatch_message = capture_dispatch
|
||||
adapter._message_handler = AsyncMock()
|
||||
|
||||
await adapter._handle_line(":user!u@host PRIVMSG hermes :\x01ACTION waves\x01")
|
||||
assert len(dispatched) == 1
|
||||
assert dispatched[0]["text"] == "* user waves"
|
||||
|
||||
|
||||
class TestIRCAdapterMarkdown:
|
||||
|
||||
def test_strip_bold(self):
|
||||
assert IRCAdapter._strip_markdown("**bold**") == "bold"
|
||||
|
||||
def test_strip_italic(self):
|
||||
assert IRCAdapter._strip_markdown("*italic*") == "italic"
|
||||
|
||||
def test_strip_code(self):
|
||||
assert IRCAdapter._strip_markdown("`code`") == "code"
|
||||
|
||||
def test_strip_link(self):
|
||||
result = IRCAdapter._strip_markdown("[click here](https://example.com)")
|
||||
assert result == "click here (https://example.com)"
|
||||
|
||||
def test_strip_image(self):
|
||||
result = IRCAdapter._strip_markdown("")
|
||||
assert result == "https://example.com/img.png"
|
||||
|
||||
|
||||
# ── Requirements / validation ────────────────────────────────────────────
|
||||
|
||||
|
||||
class TestIRCRequirements:
|
||||
|
||||
def test_check_requirements_with_env(self, monkeypatch):
|
||||
monkeypatch.setenv("IRC_SERVER", "irc.test.net")
|
||||
monkeypatch.setenv("IRC_CHANNEL", "#test")
|
||||
assert check_requirements() is True
|
||||
|
||||
def test_check_requirements_missing_server(self, monkeypatch):
|
||||
monkeypatch.delenv("IRC_SERVER", raising=False)
|
||||
monkeypatch.setenv("IRC_CHANNEL", "#test")
|
||||
assert check_requirements() is False
|
||||
|
||||
def test_check_requirements_missing_channel(self, monkeypatch):
|
||||
monkeypatch.setenv("IRC_SERVER", "irc.test.net")
|
||||
monkeypatch.delenv("IRC_CHANNEL", raising=False)
|
||||
assert check_requirements() is False
|
||||
|
||||
def test_validate_config_from_extra(self, monkeypatch):
|
||||
for key in ("IRC_SERVER", "IRC_CHANNEL"):
|
||||
monkeypatch.delenv(key, raising=False)
|
||||
from gateway.config import PlatformConfig
|
||||
cfg = PlatformConfig(extra={"server": "irc.test.net", "channel": "#test"})
|
||||
assert validate_config(cfg) is True
|
||||
|
||||
def test_validate_config_missing(self, monkeypatch):
|
||||
for key in ("IRC_SERVER", "IRC_CHANNEL"):
|
||||
monkeypatch.delenv(key, raising=False)
|
||||
from gateway.config import PlatformConfig
|
||||
cfg = PlatformConfig(extra={})
|
||||
assert validate_config(cfg) is False
|
||||
|
||||
|
||||
# ── Plugin registration ──────────────────────────────────────────────────
|
||||
|
||||
|
||||
class TestIRCPluginRegistration:
|
||||
"""Test the register() entry point."""
|
||||
|
||||
def test_register_adds_to_registry(self, monkeypatch):
|
||||
monkeypatch.setenv("IRC_SERVER", "irc.test.net")
|
||||
monkeypatch.setenv("IRC_CHANNEL", "#test")
|
||||
|
||||
from gateway.platform_registry import platform_registry
|
||||
|
||||
# Clean up if already registered
|
||||
platform_registry.unregister("irc")
|
||||
|
||||
from adapter import register
|
||||
|
||||
ctx = MagicMock()
|
||||
register(ctx)
|
||||
ctx.register_platform.assert_called_once()
|
||||
call_kwargs = ctx.register_platform.call_args
|
||||
assert call_kwargs[1]["name"] == "irc" or call_kwargs[0][0] == "irc" if call_kwargs[0] else call_kwargs[1]["name"] == "irc"
|
||||
Loading…
Add table
Add a link
Reference in a new issue