service.pyā¢46.4 kB
"""Unified service management tool for voice mode services."""
import asyncio
import json
import logging
import os
import platform
import subprocess
import time
from pathlib import Path
from typing import Literal, Optional, Dict, Any, Union
import psutil
from voice_mode.server import mcp
from voice_mode.config import WHISPER_PORT, KOKORO_PORT, LIVEKIT_PORT, SERVICE_AUTO_ENABLE
from voice_mode.utils.services.common import find_process_by_port, check_service_status
from voice_mode.utils.services.whisper_helpers import find_whisper_server, find_whisper_model
from voice_mode.utils.services.kokoro_helpers import find_kokoro_fastapi, has_gpu_support
logger = logging.getLogger("voicemode")
def load_service_file_version(service_name: str, file_type: str) -> Optional[str]:
"""Load version information for a service file."""
versions_file = Path(__file__).parent.parent / "data" / "versions.json"
if not versions_file.exists():
return None
try:
with open(versions_file) as f:
versions = json.load(f)
if file_type == "plist":
filename = f"com.voicemode.{service_name}.plist"
else: # systemd
filename = f"voicemode-{service_name}.service"
return versions.get("service_files", {}).get(filename)
except Exception as e:
logger.error(f"Error loading versions: {e}")
return None
def get_service_config_vars(service_name: str) -> Dict[str, Any]:
"""Get configuration variables for service templates."""
voicemode_dir = os.path.expanduser(os.environ.get("VOICEMODE_BASE_DIR", "~/.voicemode"))
if service_name == "whisper":
whisper_bin = find_whisper_server()
model_file = find_whisper_model()
working_dir = Path(whisper_bin).parent if whisper_bin else voicemode_dir
return {
"WHISPER_BIN": str(whisper_bin) if whisper_bin else "",
"WHISPER_PORT": str(WHISPER_PORT),
"MODEL_FILE": str(model_file) if model_file else "",
"WORKING_DIR": str(working_dir),
"LOG_DIR": os.path.join(voicemode_dir, "logs", "whisper"),
}
elif service_name == "kokoro":
kokoro_dir = find_kokoro_fastapi()
if not kokoro_dir:
kokoro_dir = os.path.join(voicemode_dir, "services", "kokoro")
# Find start script
start_script = None
if platform.system() == "Darwin":
start_script = Path(kokoro_dir) / "start-gpu_mac.sh"
else:
# On Linux, prefer GPU script if GPU is available, otherwise use CPU script
if has_gpu_support():
# Try GPU scripts first
possible_scripts = [
Path(kokoro_dir) / "start-gpu.sh",
Path(kokoro_dir) / "start-cpu.sh" # last resort
]
else:
# No GPU, prefer CPU script
possible_scripts = [
Path(kokoro_dir) / "start-cpu.sh",
Path(kokoro_dir) / "start-gpu.sh" # might work with CPU fallback
]
for script in possible_scripts:
if script.exists():
start_script = script
break
return {
"KOKORO_DIR": str(kokoro_dir),
"KOKORO_PORT": str(KOKORO_PORT),
"START_SCRIPT": str(start_script) if start_script and start_script.exists() else "",
"LOG_DIR": os.path.join(voicemode_dir, "logs", "kokoro"),
}
elif service_name == "livekit":
livekit_bin = "/opt/homebrew/bin/livekit-server" if platform.system() == "Darwin" else "/usr/local/bin/livekit-server"
config_file = os.path.join(voicemode_dir, "config", "livekit.yaml")
return {
"LIVEKIT_BIN": livekit_bin,
"LIVEKIT_PORT": str(LIVEKIT_PORT),
"CONFIG_FILE": config_file,
"LOG_DIR": os.path.join(voicemode_dir, "logs", "livekit"),
"WORKING_DIR": voicemode_dir,
}
else: # frontend
# Get frontend directory
from voice_mode.tools.livekit.frontend import find_frontend_dir
frontend_dir = find_frontend_dir()
if not frontend_dir:
raise RuntimeError("Frontend directory not found")
# Find node binary
node_bin = "/usr/local/bin/node"
possible_node_paths = [
"/usr/local/bin/node",
"/opt/homebrew/bin/node",
"/usr/bin/node",
]
for node_path in possible_node_paths:
if Path(node_path).exists():
node_bin = node_path
break
# Import config to get HOST and PORT values
from voice_mode.config import FRONTEND_HOST, FRONTEND_PORT
return {
"FRONTEND_DIR": str(frontend_dir),
"NODE_BIN": node_bin,
"NODE_PATH": str(Path(node_bin).parent),
"PORT": str(FRONTEND_PORT),
"HOST": FRONTEND_HOST,
"LOG_DIR": os.path.join(voicemode_dir, "logs", "frontend"),
"WORKING_DIR": str(frontend_dir),
}
def get_installed_service_version(service_name: str) -> Optional[str]:
"""Get the version of an installed service file."""
system = platform.system()
if system == "Darwin":
file_path = Path.home() / "Library" / "LaunchAgents" / f"com.voicemode.{service_name}.plist"
else:
file_path = Path.home() / ".config" / "systemd" / "user" / f"voicemode-{service_name}.service"
if not file_path.exists():
return None
try:
content = file_path.read_text()
# Extract version from comment
for line in content.split('\n'):
if 'v' in line and ('<!--' in line or '#' in line):
# Extract version number
import re
match = re.search(r'v(\d+\.\d+\.\d+)', line)
if match:
return match.group(1)
except Exception as e:
logger.debug(f"Could not read version from {file_path}: {e}")
return None
def load_service_template(service_name: str) -> str:
"""Load service file template from templates."""
system = platform.system()
templates_dir = Path(__file__).parent.parent / "templates"
if system == "Darwin":
template_path = templates_dir / "launchd" / f"com.voicemode.{service_name}.plist"
else:
template_path = templates_dir / "systemd" / f"voicemode-{service_name}.service"
if not template_path.exists():
raise FileNotFoundError(f"Service template not found: {template_path}")
return template_path.read_text()
async def status_service(service_name: str) -> str:
"""Get status of a service."""
if service_name == "whisper":
port = WHISPER_PORT
elif service_name == "kokoro":
port = KOKORO_PORT
elif service_name == "livekit":
port = LIVEKIT_PORT
else: # frontend
port = 3000
status, proc = check_service_status(port)
if status == "not_available":
return f"ā {service_name.capitalize()} is not available"
elif status == "forwarded":
return f"""š {service_name.capitalize()} is available via port forwarding
Port: {port} (forwarded)
Local process: Not running
Remote: Accessible"""
try:
with proc.oneshot():
cpu_percent = proc.cpu_percent(interval=0.1)
memory_info = proc.memory_info()
memory_mb = memory_info.rss / 1024 / 1024
create_time = proc.create_time()
cmdline = proc.cmdline()
# Calculate uptime
uptime_seconds = time.time() - create_time
hours = int(uptime_seconds // 3600)
minutes = int((uptime_seconds % 3600) // 60)
seconds = int(uptime_seconds % 60)
if hours > 0:
uptime_str = f"{hours}h {minutes}m {seconds}s"
elif minutes > 0:
uptime_str = f"{minutes}m {seconds}s"
else:
uptime_str = f"{seconds}s"
# Service-specific info
extra_info_parts = []
if service_name == "whisper":
# Get model info
model = "unknown"
model_name = None
for i, arg in enumerate(cmdline):
if arg == "--model" and i + 1 < len(cmdline):
model = Path(cmdline[i + 1]).name
# Extract model name from filename (e.g., ggml-large-v3-turbo.bin -> large-v3-turbo)
if model.startswith("ggml-") and model.endswith(".bin"):
model_name = model[5:-4]
break
extra_info_parts.append(f"Model: {model}")
# Get version and capability info
try:
from voice_mode.utils.services.whisper_version import get_whisper_version_info, check_coreml_model_exists
version_info = get_whisper_version_info()
if version_info.get("version"):
extra_info_parts.append(f"Version: {version_info['version']}")
elif version_info.get("commit"):
extra_info_parts.append(f"Commit: {version_info['commit']}")
# Show Core ML status on Apple Silicon
if platform.machine() == "arm64" and platform.system() == "Darwin":
if version_info.get("coreml_supported"):
# Check if the current model has Core ML
if model_name and check_coreml_model_exists(model_name):
extra_info_parts.append("Core ML: ā Enabled & Active")
else:
extra_info_parts.append("Core ML: ā Supported (model not converted)")
else:
extra_info_parts.append("Core ML: ā Not compiled in")
# Show GPU support
gpu_support = []
if version_info.get("metal_supported"):
gpu_support.append("Metal")
if version_info.get("cuda_supported"):
gpu_support.append("CUDA")
if gpu_support:
extra_info_parts.append(f"GPU: {', '.join(gpu_support)}")
except:
pass
else: # kokoro
# Try to get version info
try:
from voice_mode.utils.services.version_info import get_kokoro_version
version_info = get_kokoro_version()
if version_info.get("api_version"):
extra_info_parts.append(f"API Version: {version_info['api_version']}")
elif version_info.get("version"):
extra_info_parts.append(f"Version: {version_info['version']}")
except:
pass
# Check service file version
installed_version = get_installed_service_version(service_name)
template_version = load_service_file_version(service_name, "plist" if platform.system() == "Darwin" else "service")
if installed_version and template_version:
if installed_version != template_version:
extra_info_parts.append(f"Service files: v{installed_version} (v{template_version} available)")
extra_info_parts.append("š” Run 'service {0} update-service-files' to update".format(service_name))
else:
extra_info_parts.append(f"Service files: v{installed_version} (latest)")
extra_info = ""
if extra_info_parts:
extra_info = "\n " + "\n ".join(extra_info_parts)
return f"""ā
{service_name.capitalize()} is running locally
PID: {proc.pid}
Port: {port}
CPU: {cpu_percent:.1f}%
Memory: {memory_mb:.1f} MB
Uptime: {uptime_str}{extra_info}"""
except Exception as e:
logger.error(f"Error getting process info: {e}")
return f"{service_name.capitalize()} is running (PID: {proc.pid}) but could not get details"
async def start_service(service_name: str) -> str:
"""Start a service."""
# Check if already running
if service_name == "whisper":
port = WHISPER_PORT
elif service_name == "kokoro":
port = KOKORO_PORT
elif service_name == "livekit":
port = LIVEKIT_PORT
else: # frontend
port = 3000
if find_process_by_port(port):
return f"{service_name.capitalize()} is already running on port {port}"
system = platform.system()
# Check if managed by service manager
if system == "Darwin":
plist_path = Path.home() / "Library" / "LaunchAgents" / f"com.voicemode.{service_name}.plist"
if plist_path.exists():
# Use launchctl load
result = subprocess.run(
["launchctl", "load", str(plist_path)],
capture_output=True,
text=True
)
if result.returncode == 0:
# Wait for service to start
for i in range(10):
if find_process_by_port(port):
return f"ā
{service_name.capitalize()} started"
await asyncio.sleep(0.5)
return f"ā ļø {service_name.capitalize()} loaded but not yet listening on port {port}"
else:
error = result.stderr or result.stdout
if "already loaded" in error.lower():
# Service is loaded but maybe not running - try to start it
# This can happen if the service crashed
subprocess.run(["launchctl", "kickstart", "-k", f"gui/{os.getuid()}/com.voicemode.{service_name}"], capture_output=True)
await asyncio.sleep(2)
if find_process_by_port(port):
return f"ā
{service_name.capitalize()} restarted"
return f"ā ļø {service_name.capitalize()} is loaded but failed to start"
return f"ā Failed to start {service_name}: {error}"
elif system == "Linux":
service_file = Path.home() / ".config" / "systemd" / "user" / f"voicemode-{service_name}.service"
if service_file.exists():
# Use systemctl start
result = subprocess.run(
["systemctl", "--user", "start", f"voicemode-{service_name}.service"],
capture_output=True,
text=True
)
if result.returncode == 0:
# Wait for service to start
for i in range(10):
if find_process_by_port(port):
return f"ā
{service_name.capitalize()} started"
await asyncio.sleep(0.5)
return f"ā ļø {service_name.capitalize()} started but not yet listening on port {port}"
else:
error = result.stderr or result.stdout
return f"ā Failed to start {service_name}: {error}"
# Fallback to direct process start
if service_name == "whisper":
# Find whisper-server binary
whisper_bin = find_whisper_server()
if not whisper_bin:
return "ā whisper-server not found. Please run whisper_install first."
# Find model
model_file = find_whisper_model()
if not model_file:
return "ā No Whisper model found. Please run download_model first."
# Start whisper-server
cmd = [str(whisper_bin), "--host", "0.0.0.0", "--port", str(port), "--model", str(model_file)]
elif service_name == "kokoro":
# Find kokoro installation
kokoro_dir = find_kokoro_fastapi()
if not kokoro_dir:
return "ā kokoro-fastapi not found. Please run kokoro_install first."
# Use appropriate start script
if platform.system() == "Darwin":
start_script = Path(kokoro_dir) / "start-gpu_mac.sh"
else:
# On Linux, prefer GPU script if GPU is available, otherwise use CPU script
if has_gpu_support():
# Try GPU scripts first
possible_scripts = [
Path(kokoro_dir) / "start-gpu.sh",
Path(kokoro_dir) / "start-cpu.sh" # last resort
]
else:
# No GPU, prefer CPU script
possible_scripts = [
Path(kokoro_dir) / "start-cpu.sh",
Path(kokoro_dir) / "start-gpu.sh" # might work with CPU fallback
]
start_script = None
for script in possible_scripts:
if script.exists():
start_script = script
break
if not start_script.exists():
return f"ā Start script not found: {start_script}"
cmd = [str(start_script)]
else: # livekit
# Find LiveKit binary
livekit_bin = "/opt/homebrew/bin/livekit-server" if platform.system() == "Darwin" else "/usr/local/bin/livekit-server"
if not Path(livekit_bin).exists():
return "ā LiveKit server not found. Please run 'voice-mode livekit install' first."
# Create config directory if it doesn't exist
config_dir = Path.home() / ".voicemode" / "config"
config_dir.mkdir(parents=True, exist_ok=True)
# Run in dev mode
cmd = [livekit_bin, "--dev"]
try:
# Start the process
process = subprocess.Popen(
cmd,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
cwd=Path(kokoro_dir) if service_name == "kokoro" else None
)
# Wait a moment to check if it started
await asyncio.sleep(2)
if process.poll() is not None:
# Process exited
stderr = process.stderr.read().decode() if process.stderr else ""
return f"ā {service_name.capitalize()} failed to start: {stderr}"
# Verify it's listening
if find_process_by_port(port):
return f"ā
{service_name.capitalize()} started successfully (PID: {process.pid})"
else:
return f"ā ļø {service_name.capitalize()} process started but not listening on port {port} yet"
except Exception as e:
logger.error(f"Error starting {service_name}: {e}")
return f"ā Error starting {service_name}: {str(e)}"
async def stop_service(service_name: str) -> str:
"""Stop a service."""
if service_name == "whisper":
port = WHISPER_PORT
elif service_name == "kokoro":
port = KOKORO_PORT
elif service_name == "livekit":
port = LIVEKIT_PORT
else: # frontend
port = 3000
system = platform.system()
# Check if managed by service manager
if system == "Darwin":
plist_path = Path.home() / "Library" / "LaunchAgents" / f"com.voicemode.{service_name}.plist"
if plist_path.exists():
# Use launchctl unload (without -w to preserve enable state)
result = subprocess.run(
["launchctl", "unload", str(plist_path)],
capture_output=True,
text=True
)
if result.returncode == 0:
return f"ā
{service_name.capitalize()} stopped"
else:
error = result.stderr or result.stdout
# If service is not loaded, that's ok - it's already stopped
if "could not find specified service" in error.lower():
return f"{service_name.capitalize()} is not running"
return f"ā Failed to stop {service_name}: {error}"
elif system == "Linux":
service_file = Path.home() / ".config" / "systemd" / "user" / f"voicemode-{service_name}.service"
if service_file.exists():
# Use systemctl stop
result = subprocess.run(
["systemctl", "--user", "stop", f"voicemode-{service_name}.service"],
capture_output=True,
text=True
)
if result.returncode == 0:
return f"ā
{service_name.capitalize()} stopped"
else:
error = result.stderr or result.stdout
return f"ā Failed to stop {service_name}: {error}"
# Fallback to process termination
proc = find_process_by_port(port)
if not proc:
return f"{service_name.capitalize()} is not running"
try:
pid = proc.pid
proc.terminate()
# Wait for graceful shutdown
try:
proc.wait(timeout=5)
except psutil.TimeoutExpired:
# Force kill if needed
proc.kill()
proc.wait(timeout=5)
return f"ā
{service_name.capitalize()} stopped (was PID: {pid})"
except Exception as e:
logger.error(f"Error stopping {service_name}: {e}")
return f"ā Error stopping {service_name}: {str(e)}"
async def restart_service(service_name: str) -> str:
"""Restart a service."""
stop_result = await stop_service(service_name)
# Brief pause between stop and start
await asyncio.sleep(1)
start_result = await start_service(service_name)
return f"Restart {service_name}:\n{stop_result}\n{start_result}"
async def enable_service(service_name: str) -> str:
"""Enable a service to start at boot/login."""
system = platform.system()
# Check version and offer update if needed
installed_version = get_installed_service_version(service_name)
current_version = load_service_file_version(service_name, "plist" if system == "Darwin" else "service")
if installed_version and current_version and installed_version != current_version:
logger.info(f"Service file update available: {installed_version} -> {current_version}")
try:
if system == "Darwin":
# Load template
template = load_service_template(service_name)
plist_path = Path.home() / "Library" / "LaunchAgents" / f"com.voicemode.{service_name}.plist"
logs_dir = Path.home() / ".voicemode" / "logs"
# Prepare substitutions
if service_name == "whisper":
whisper_bin = find_whisper_server()
if not whisper_bin:
return "ā whisper-server not found. Please run whisper_install first."
model_file = find_whisper_model()
if not model_file:
return "ā No Whisper model found. Please run download_model first."
# Find the start script
# whisper_bin is at ~/.voicemode/services/whisper/build/bin/whisper-server
# start script is at ~/.voicemode/services/whisper/bin/start-whisper-server.sh
install_dir = Path(whisper_bin).parent.parent.parent # Go up from build/bin/whisper-server to whisper dir
start_script_path = install_dir / "bin" / "start-whisper-server.sh"
if not start_script_path.exists():
return f"ā Start script not found: {start_script_path}"
content = template.format(
START_SCRIPT_PATH=str(start_script_path),
LOG_DIR=str(logs_dir),
INSTALL_DIR=str(install_dir)
)
elif service_name == "kokoro":
kokoro_dir = find_kokoro_fastapi()
if not kokoro_dir:
return "ā kokoro-fastapi not found. Please run kokoro_install first."
start_script = Path(kokoro_dir) / "start-gpu_mac.sh"
if not start_script.exists():
return f"ā Start script not found: {start_script}"
content = template.format(
START_SCRIPT=start_script,
KOKORO_DIR=kokoro_dir,
KOKORO_PORT=KOKORO_PORT,
LOG_DIR=logs_dir
)
elif service_name == "livekit":
# Use livekit binary path and config
livekit_bin = "/opt/homebrew/bin/livekit-server" if platform.system() == "Darwin" else "/usr/local/bin/livekit-server"
config_file = Path.home() / ".voicemode" / "config" / "livekit.yaml"
if not Path(livekit_bin).exists():
return "ā LiveKit server not found. Please run 'voice-mode livekit install' first."
content = template.format(
LIVEKIT_BIN=livekit_bin,
LIVEKIT_PORT=LIVEKIT_PORT,
CONFIG_FILE=config_file,
LOG_DIR=logs_dir,
WORKING_DIR=Path.home() / ".voicemode"
)
else: # frontend
# Get service config vars which includes all needed variables
config_vars = get_service_config_vars(service_name)
if "FRONTEND_DIR" not in config_vars:
return "ā Frontend directory not found"
# Add log directory to config vars
config_vars["LOG_DIR"] = str(logs_dir)
# Format template with all config vars
content = template.format(**config_vars)
# Create directories
plist_path.parent.mkdir(parents=True, exist_ok=True)
logs_dir.mkdir(parents=True, exist_ok=True)
# Write plist
plist_path.write_text(content)
# Load with launchctl
result = subprocess.run(
["launchctl", "load", "-w", str(plist_path)],
capture_output=True,
text=True
)
if result.returncode == 0:
return f"ā
{service_name.capitalize()} service enabled. It will start automatically at login.\nPlist: {plist_path}"
else:
error = result.stderr or result.stdout
return f"ā Failed to enable {service_name} service: {error}"
else: # Linux
# Load template
template = load_service_template(service_name)
service_path = Path.home() / ".config" / "systemd" / "user" / f"voicemode-{service_name}.service"
# Prepare substitutions
if service_name == "whisper":
whisper_bin = find_whisper_server()
if not whisper_bin:
return "ā whisper-server not found. Please run whisper_install first."
model_file = find_whisper_model()
if not model_file:
return "ā No Whisper model found. Please run download_model first."
content = template.format(
WHISPER_BIN=whisper_bin,
WHISPER_PORT=WHISPER_PORT,
MODEL_FILE=model_file,
WORKING_DIR=Path(whisper_bin).parent
)
elif service_name == "kokoro":
kokoro_dir = find_kokoro_fastapi()
if not kokoro_dir:
return "ā kokoro-fastapi not found. Please run kokoro_install first."
# On Linux, prefer GPU script if GPU is available, otherwise use CPU script
if has_gpu_support():
# Try GPU scripts first
possible_scripts = [
Path(kokoro_dir) / "start-gpu.sh",
Path(kokoro_dir) / "start-cpu.sh" # last resort
]
else:
# No GPU, prefer CPU script
possible_scripts = [
Path(kokoro_dir) / "start-cpu.sh",
Path(kokoro_dir) / "start-gpu.sh" # might work with CPU fallback
]
start_script = None
for script in possible_scripts:
if script.exists():
start_script = script
break
if not start_script:
gpu_status = "GPU detected" if has_gpu_support() else "No GPU detected"
return f"ā No start script found in {kokoro_dir}. {gpu_status}. Expected one of: start.sh, start-gpu.sh, or start-cpu.sh"
content = template.format(
START_SCRIPT=start_script,
KOKORO_DIR=kokoro_dir,
KOKORO_PORT=KOKORO_PORT
)
elif service_name == "livekit":
# Use livekit binary path and config
livekit_bin = "/usr/local/bin/livekit-server"
config_file = Path.home() / ".voicemode" / "config" / "livekit.yaml"
if not Path(livekit_bin).exists():
return "ā LiveKit server not found. Please run 'voice-mode livekit install' first."
content = template.format(
LIVEKIT_BIN=livekit_bin,
LIVEKIT_PORT=LIVEKIT_PORT,
CONFIG_FILE=config_file,
LOG_DIR=Path.home() / ".voicemode" / "logs",
WORKING_DIR=Path.home() / ".voicemode"
)
else: # frontend
# Get service config vars which includes all needed variables
config_vars = get_service_config_vars(service_name)
if "FRONTEND_DIR" not in config_vars:
return "ā Frontend directory not found"
# Format template with all config vars
content = template.format(**config_vars)
# Write service file
service_path.parent.mkdir(parents=True, exist_ok=True)
service_path.write_text(content)
# Reload and enable
subprocess.run(["systemctl", "--user", "daemon-reload"], check=True)
result = subprocess.run(
["systemctl", "--user", "enable", f"voicemode-{service_name}.service"],
capture_output=True,
text=True
)
if result.returncode == 0:
# Also start it now
subprocess.run(["systemctl", "--user", "start", f"voicemode-{service_name}.service"], check=True)
return f"ā
{service_name.capitalize()} service enabled and started.\nService: {service_path}"
else:
error = result.stderr or result.stdout
return f"ā Failed to enable {service_name} service: {error}"
except Exception as e:
logger.error(f"Error enabling {service_name} service: {e}")
return f"ā Error enabling {service_name} service: {str(e)}"
async def disable_service(service_name: str) -> str:
"""Disable a service from starting at boot/login."""
system = platform.system()
try:
if system == "Darwin":
plist_path = Path.home() / "Library" / "LaunchAgents" / f"com.voicemode.{service_name}.plist"
if not plist_path.exists():
return f"{service_name.capitalize()} service is not installed"
# Unload with launchctl
result = subprocess.run(
["launchctl", "unload", "-w", str(plist_path)],
capture_output=True,
text=True
)
if result.returncode == 0:
# Remove the plist file
plist_path.unlink()
return f"ā
{service_name.capitalize()} service disabled and removed"
else:
error = result.stderr or result.stdout
# If already unloaded, just remove the file
if "Could not find specified service" in error:
plist_path.unlink()
return f"ā
{service_name.capitalize()} service was already disabled, plist removed"
return f"ā Failed to disable {service_name} service: {error}"
else: # Linux
service_name_full = f"voicemode-{service_name}.service"
# Stop and disable
subprocess.run(["systemctl", "--user", "stop", service_name_full], check=True)
result = subprocess.run(
["systemctl", "--user", "disable", service_name_full],
capture_output=True,
text=True
)
if result.returncode == 0:
# Remove service file
service_path = Path.home() / ".config" / "systemd" / "user" / service_name_full
if service_path.exists():
service_path.unlink()
# Reload systemd
subprocess.run(["systemctl", "--user", "daemon-reload"], check=True)
return f"ā
{service_name.capitalize()} service disabled and removed"
else:
error = result.stderr or result.stdout
return f"ā Failed to disable {service_name} service: {error}"
except Exception as e:
logger.error(f"Error disabling {service_name} service: {e}")
return f"ā Error disabling {service_name} service: {str(e)}"
async def update_service_files(service_name: str) -> str:
"""Update service files to the latest version from templates."""
system = platform.system()
# Get template versions
template_versions = load_service_file_version(service_name, "plist" if system == "Darwin" else "service")
if not template_versions:
return "ā Could not load template version information"
# Get installed versions
installed_version = get_installed_service_version(service_name)
if installed_version == template_versions:
return f"ā
Service files are already up to date (version {installed_version})"
try:
# Load template
template_content = load_service_template(service_name)
if system == "Darwin":
# Update launchd plist
plist_path = Path.home() / "Library" / "LaunchAgents" / f"com.voicemode.{service_name}.plist"
# Check if service is running
was_running = find_process_by_port(WHISPER_PORT if service_name == "whisper" else KOKORO_PORT) is not None
if was_running:
# Unload the service first
subprocess.run(["launchctl", "unload", str(plist_path)], capture_output=True)
await asyncio.sleep(1)
# Backup existing file
if plist_path.exists():
backup_path = plist_path.with_suffix(f".backup.{installed_version or 'unknown'}")
plist_path.rename(backup_path)
# Write new plist with current configuration
config_vars = get_service_config_vars(service_name)
final_content = template_content
for key, value in config_vars.items():
final_content = final_content.replace(f"{{{key}}}", str(value))
plist_path.write_text(final_content)
# Also update wrapper script if it exists
wrapper_name = f"start-{service_name}-with-health-check.sh"
wrapper_template = Path(__file__).parent.parent / "templates" / "launchd" / wrapper_name
if wrapper_template.exists():
wrapper_dest = Path(config_vars.get('KOKORO_DIR', config_vars.get('WORKING_DIR', ''))) / wrapper_name
if wrapper_dest.parent.exists():
wrapper_content = wrapper_template.read_text()
for key, value in config_vars.items():
wrapper_content = wrapper_content.replace(f"{{{key}}}", str(value))
wrapper_dest.write_text(wrapper_content)
wrapper_dest.chmod(0o755)
if was_running:
# Reload the service
subprocess.run(["launchctl", "load", str(plist_path)], capture_output=True)
await asyncio.sleep(2)
return f"ā
Updated {service_name} service files from version {installed_version or 'unknown'} to {template_versions}"
else: # Linux
# Update systemd service
service_path = Path.home() / ".config" / "systemd" / "user" / f"voicemode-{service_name}.service"
# Backup existing file
if service_path.exists():
backup_path = service_path.with_suffix(f".backup.{installed_version or 'unknown'}")
service_path.rename(backup_path)
# Write new service file with current configuration
config_vars = get_service_config_vars(service_name)
final_content = template_content
for key, value in config_vars.items():
final_content = final_content.replace(f"{{{key}}}", str(value))
service_path.parent.mkdir(parents=True, exist_ok=True)
service_path.write_text(final_content)
# Reload systemd daemon
subprocess.run(["systemctl", "--user", "daemon-reload"], capture_output=True)
return f"ā
Updated {service_name} service files from version {installed_version or 'unknown'} to {template_versions}"
except Exception as e:
logger.error(f"Error updating service files: {e}")
return f"ā Failed to update service files: {str(e)}"
async def view_logs(service_name: str, lines: Optional[int] = None) -> str:
"""View service logs."""
system = platform.system()
lines = lines or 50
try:
if system == "Darwin":
# Use log show command
cmd = [
"log", "show",
"--predicate", f'process == "{service_name}-server" OR process == "kokoro-fastapi"',
"--last", f"{lines}",
"--style", "compact"
]
result = subprocess.run(cmd, capture_output=True, text=True)
if result.returncode == 0 and result.stdout.strip():
return f"=== Last {lines} log entries for {service_name} ===\n{result.stdout}"
else:
# Fallback to log files
log_dir = Path.home() / ".voicemode" / "logs"
out_log = log_dir / f"{service_name}.out.log"
err_log = log_dir / f"{service_name}.err.log"
logs = []
if out_log.exists():
with open(out_log) as f:
stdout_lines = f.readlines()[-lines:]
if stdout_lines:
logs.append(f"=== stdout (last {len(stdout_lines)} lines) ===")
logs.extend(stdout_lines)
if err_log.exists():
with open(err_log) as f:
stderr_lines = f.readlines()[-lines:]
if stderr_lines:
if logs:
logs.append("")
logs.append(f"=== stderr (last {len(stderr_lines)} lines) ===")
logs.extend(stderr_lines)
if logs:
return "".join(logs).rstrip()
else:
return f"No logs found for {service_name}"
else: # Linux
# Use journalctl
cmd = [
"journalctl", "--user",
"-u", f"voicemode-{service_name}.service",
"-n", str(lines),
"--no-pager"
]
result = subprocess.run(cmd, capture_output=True, text=True)
if result.returncode == 0:
return f"=== Last {lines} journal entries for {service_name} ===\n{result.stdout}"
else:
return f"Failed to retrieve logs: {result.stderr}"
except Exception as e:
logger.error(f"Error viewing logs for {service_name}: {e}")
return f"ā Error viewing logs: {str(e)}"
@mcp.tool()
async def service(
service_name: Literal["whisper", "kokoro", "livekit", "frontend"],
action: Literal["status", "start", "stop", "restart", "enable", "disable", "logs", "update-service-files"] = "status",
lines: Optional[Union[int, str]] = None
) -> str:
"""Unified service management tool for voice mode services.
Manage Whisper (STT) and Kokoro (TTS) services with a single tool.
Args:
service_name: The service to manage ("whisper", "kokoro", or "livekit")
action: The action to perform (default: "status")
- status: Show if service is running and resource usage
- start: Start the service
- stop: Stop the service
- restart: Stop and start the service
- enable: Configure service to start at boot/login
- disable: Remove service from boot/login
- logs: View recent service logs
- update-service-files: Update systemd/launchd service files to latest version
lines: Number of log lines to show (only for logs action, default: 50)
Returns:
Status message indicating the result of the action
Examples:
service("whisper", "status") # Check if Whisper is running
service("kokoro", "start") # Start Kokoro service
service("whisper", "logs", 100) # View last 100 lines of Whisper logs
"""
# Convert lines to integer if provided as string
if lines is not None and isinstance(lines, str):
try:
lines = int(lines)
except ValueError:
logger.warning(f"Invalid lines value '{lines}', using default 50")
lines = 50
# Route to appropriate handler
if action == "status":
return await status_service(service_name)
elif action == "start":
return await start_service(service_name)
elif action == "stop":
return await stop_service(service_name)
elif action == "restart":
return await restart_service(service_name)
elif action == "enable":
return await enable_service(service_name)
elif action == "disable":
return await disable_service(service_name)
elif action == "logs":
return await view_logs(service_name, lines)
elif action == "update-service-files":
return await update_service_files(service_name)
else:
return f"ā Unknown action: {action}"
async def install_service(service_name: str) -> Dict[str, Any]:
"""Install service files for a service."""
try:
system = platform.system()
config_vars = get_service_config_vars(service_name)
# Load template
template_content = load_service_template(service_name)
# Replace placeholders
for key, value in config_vars.items():
template_content = template_content.replace(f"{{{key}}}", str(value))
if system == "Darwin":
# Install launchd plist
plist_path = Path.home() / "Library" / "LaunchAgents" / f"com.voicemode.{service_name}.plist"
plist_path.parent.mkdir(parents=True, exist_ok=True)
plist_path.write_text(template_content)
return {"success": True, "service_file": str(plist_path)}
else:
# Install systemd service
service_path = Path.home() / ".config" / "systemd" / "user" / f"voicemode-{service_name}.service"
service_path.parent.mkdir(parents=True, exist_ok=True)
service_path.write_text(template_content)
# Reload systemd
subprocess.run(["systemctl", "--user", "daemon-reload"], capture_output=True)
return {"success": True, "service_file": str(service_path)}
except Exception as e:
logger.error(f"Error installing service {service_name}: {e}")
return {"success": False, "error": str(e)}
async def uninstall_service(service_name: str) -> Dict[str, Any]:
"""Remove service files for a service."""
try:
system = platform.system()
if system == "Darwin":
plist_path = Path.home() / "Library" / "LaunchAgents" / f"com.voicemode.{service_name}.plist"
if plist_path.exists():
plist_path.unlink()
return {"success": True, "message": f"Removed {plist_path}"}
else:
return {"success": True, "message": "Service file not found"}
else:
service_path = Path.home() / ".config" / "systemd" / "user" / f"voicemode-{service_name}.service"
if service_path.exists():
service_path.unlink()
# Reload systemd
subprocess.run(["systemctl", "--user", "daemon-reload"], capture_output=True)
return {"success": True, "message": f"Removed {service_path}"}
else:
return {"success": True, "message": "Service file not found"}
except Exception as e:
logger.error(f"Error uninstalling service {service_name}: {e}")
return {"success": False, "error": str(e)}