somew1nd
c1a4b49a21
- AgentApiNotifier: Sort candidate process sessions by creation time descending to prioritize the latest active IDE session over stale PIDs. Implement automatic retry across candidate address/token pairs and persist working credentials to shared environment files (~/.gemini/jetski/.jetski_env and ~/.artemis/.artemis_env). - DesktopNotifier: Enable native OS desktop toast notifications by default on macOS, Linux, and Windows unless explicitly disabled or running in CI. Ensure MCP installer sets ARTEMIS_DESKTOP_NOTIFY=true in generated IDE configurations. - ScriptNotifier: Add universal command hook adapter (ARTEMIS_NOTIFY_CMD / MCP_NOTIFY_COMMAND) for custom IDE and CLI integration. - Fault Isolation: Wrap CompositeNotifier and global notify entrypoint in strict try-except blocks to guarantee zero-crash execution even if any individual notifier fails. - Tests & Docs: Add unit tests for session recovery, fault tolerance, script hook execution, and document multi-platform notification architecture in README.
105 lines
4.1 KiB
Python
105 lines
4.1 KiB
Python
# Copyright 2026 Google LLC
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
"""Desktop Toast / OS notification adapter."""
|
|
|
|
import logging
|
|
import os
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
from typing import Any
|
|
|
|
from mcp_server.notifiers.base import BaseNotifier
|
|
|
|
logger = logging.getLogger("mcp_server.notifiers.desktop")
|
|
|
|
|
|
class DesktopNotifier(BaseNotifier):
|
|
"""Notifier that displays native desktop notifications across macOS, Linux, and Windows."""
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
return "desktop"
|
|
|
|
def is_available(self) -> bool:
|
|
try:
|
|
val = os.getenv("ARTEMIS_DESKTOP_NOTIFY", "").lower()
|
|
if val in ("0", "false", "no", "off"):
|
|
return False
|
|
if os.getenv("CI", "").lower() in ("1", "true", "yes") and val not in ("1", "true", "yes"):
|
|
return False
|
|
if sys.platform == "linux" and not shutil.which("notify-send"):
|
|
return False
|
|
if sys.platform == "darwin" and not shutil.which("osascript"):
|
|
return False
|
|
return True
|
|
except Exception:
|
|
return False
|
|
|
|
def notify(
|
|
self,
|
|
conversation_id: str,
|
|
message: str,
|
|
title: str | None = None,
|
|
event_type: str = "completed",
|
|
payload: dict[str, Any] | None = None,
|
|
) -> bool:
|
|
if not self.is_available():
|
|
return False
|
|
|
|
header = title or f"☕ Artemis Task {event_type.capitalize()}"
|
|
clean_body = message.split("\n\n")[0][:120]
|
|
|
|
try:
|
|
if sys.platform == "linux":
|
|
if shutil.which("notify-send"):
|
|
subprocess.run(
|
|
["notify-send", header, clean_body],
|
|
stdout=subprocess.DEVNULL,
|
|
stderr=subprocess.DEVNULL,
|
|
timeout=3,
|
|
)
|
|
return True
|
|
elif sys.platform == "darwin":
|
|
if shutil.which("osascript"):
|
|
script = f'display notification "{clean_body}" with title "{header}"'
|
|
subprocess.run(
|
|
["osascript", "-e", script],
|
|
stdout=subprocess.DEVNULL,
|
|
stderr=subprocess.DEVNULL,
|
|
timeout=3,
|
|
)
|
|
return True
|
|
elif sys.platform == "win32":
|
|
ps_cmd = (
|
|
f"[Windows.UI.Notifications.ToastNotificationManager, Windows.UI.Notifications, ContentType = WindowsRuntime] > $null; "
|
|
f"$template = [Windows.UI.Notifications.ToastNotificationManager]::GetTemplateContent([Windows.UI.Notifications.ToastTemplateType]::ToastText02); "
|
|
f'$textNodes = $template.GetElementsByTagName("text"); '
|
|
f'$textNodes.Item(0).AppendChild($template.CreateTextNode("{header}")) > $null; '
|
|
f'$textNodes.Item(1).AppendChild($template.CreateTextNode("{clean_body}")) > $null; '
|
|
f"$toast = [Windows.UI.Notifications.ToastNotification]::new($template); "
|
|
f'[Windows.UI.Notifications.ToastNotificationManager]::CreateToastNotifier("Artemis").Show($toast);'
|
|
)
|
|
subprocess.run(
|
|
["powershell", "-NoProfile", "-Command", ps_cmd],
|
|
stdout=subprocess.DEVNULL,
|
|
stderr=subprocess.DEVNULL,
|
|
timeout=5,
|
|
)
|
|
return True
|
|
except Exception as e:
|
|
logger.debug(f"Desktop notification failed: {e}")
|
|
return False
|