feat: add Gemini skill integration and multi-user AI setup
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
"""
|
||||
Local AI agent integration setup.
|
||||
Installs the shared ssh.py/encryption.py backend, Claude /ssh command,
|
||||
Codex skill package, platform-specific wrappers, and SSH key material.
|
||||
Codex/Gemini skill packages, platform-specific wrappers, and SSH key material.
|
||||
"""
|
||||
|
||||
import os
|
||||
@@ -12,8 +12,6 @@ import sys
|
||||
|
||||
from core.logger import log
|
||||
|
||||
SHARED_DIR = os.path.expanduser("~/.server-connections")
|
||||
|
||||
# PyInstaller: bundled data is in sys._MEIPASS; otherwise use project dir
|
||||
if getattr(sys, "frozen", False) and hasattr(sys, "_MEIPASS"):
|
||||
_BASE_DIR = sys._MEIPASS
|
||||
@@ -23,25 +21,19 @@ else:
|
||||
SSH_SCRIPT_SRC = os.path.join(_BASE_DIR, "tools", "ssh.py")
|
||||
ENCRYPTION_SRC = os.path.join(_BASE_DIR, "core", "encryption.py")
|
||||
CLAUDE_SKILL_SRC = os.path.join(_BASE_DIR, "tools", "skill-ssh.md")
|
||||
|
||||
CLAUDE_SKILL_DST_DIR = os.path.expanduser("~/.claude/commands")
|
||||
CLAUDE_SKILL_DST = os.path.join(CLAUDE_SKILL_DST_DIR, "ssh.md")
|
||||
SSH_KEY_PATH = os.path.expanduser("~/.ssh/id_ed25519")
|
||||
GLOBAL_CLAUDE_MD = os.path.expanduser("~/.claude/CLAUDE.md")
|
||||
GEMINI_CONTRACT_SRC = os.path.join(_BASE_DIR, "GEMINI.md")
|
||||
|
||||
CODEX_SKILL_SRC_DIR = os.path.join(_BASE_DIR, ".codex", "skills", "server-manager")
|
||||
CODEX_SKILL_DST_ROOT = os.path.expanduser("~/.codex/skills")
|
||||
CODEX_SKILL_DST_DIR = os.path.join(CODEX_SKILL_DST_ROOT, "server-manager")
|
||||
CODEX_SKILL_ENTRY = os.path.join(CODEX_SKILL_DST_DIR, "SKILL.md")
|
||||
CODEX_WRAPPER_SRC_SH = os.path.join(CODEX_SKILL_SRC_DIR, "scripts", "codex-ssh-wrapper.sh")
|
||||
CODEX_WRAPPER_SRC_CMD = os.path.join(CODEX_SKILL_SRC_DIR, "scripts", "codex-ssh-wrapper.cmd")
|
||||
CODEX_WRAPPER_DST = os.path.join(
|
||||
SHARED_DIR,
|
||||
"codex-ssh.cmd" if sys.platform == "win32" else "codex-ssh",
|
||||
)
|
||||
GEMINI_SKILL_SRC_DIR = os.path.join(_BASE_DIR, ".gemini", "skills", "server-manager")
|
||||
GEMINI_WRAPPER_SRC_SH = os.path.join(GEMINI_SKILL_SRC_DIR, "scripts", "gemini-ssh-wrapper.sh")
|
||||
GEMINI_WRAPPER_SRC_CMD = os.path.join(GEMINI_SKILL_SRC_DIR, "scripts", "gemini-ssh-wrapper.cmd")
|
||||
|
||||
_BLOCK_START = "<!-- server-manager:start -->"
|
||||
_BLOCK_END = "<!-- server-manager:end -->"
|
||||
_GEMINI_BLOCK_START = "<!-- server-manager-gemini:start -->"
|
||||
_GEMINI_BLOCK_END = "<!-- server-manager-gemini:end -->"
|
||||
|
||||
GLOBAL_CLAUDE_MD_BLOCK = f"""{_BLOCK_START}
|
||||
## Серверы — ТОЛЬКО через /ssh
|
||||
@@ -80,6 +72,110 @@ python ~/.server-connections/ssh.py --status # online/offline
|
||||
{_BLOCK_END}
|
||||
"""
|
||||
|
||||
GLOBAL_GEMINI_MD_BLOCK = f"""{_GEMINI_BLOCK_START}
|
||||
## ServerManager — use the installed skill
|
||||
|
||||
When a user asks about a server managed by ServerManager, use the installed `server-manager` skill first.
|
||||
|
||||
Preferred discovery commands:
|
||||
|
||||
```bash
|
||||
$HOME/.server-connections/gemini-ssh --list
|
||||
$HOME/.server-connections/gemini-ssh --info ALIAS
|
||||
$HOME/.server-connections/gemini-ssh --status
|
||||
```
|
||||
|
||||
Rules:
|
||||
|
||||
- Never read `~/.server-connections/servers.json`, `settings.json`, or `encryption.py` directly.
|
||||
- Never use `--list-full`.
|
||||
- Never use raw `ssh`, `scp`, `redis-cli`, `psql`, `mysql`, `mc`, or cloud CLIs unless the user explicitly asks to bypass ServerManager.
|
||||
- Choose commands strictly by the endpoint type reported by `--list`.
|
||||
- Use exactly one connection attempt per action and stop on timeout/failure.
|
||||
{_GEMINI_BLOCK_END}
|
||||
"""
|
||||
|
||||
|
||||
def _target_home() -> str:
|
||||
override = os.environ.get("SERVER_MANAGER_TARGET_HOME", "").strip()
|
||||
if override:
|
||||
return os.path.abspath(os.path.expanduser(override))
|
||||
return os.path.expanduser("~")
|
||||
|
||||
|
||||
def _shared_dir() -> str:
|
||||
return os.path.join(_target_home(), ".server-connections")
|
||||
|
||||
|
||||
def _gemini_dir() -> str:
|
||||
return os.path.join(_target_home(), ".gemini")
|
||||
|
||||
|
||||
def _claude_skill_dst_dir() -> str:
|
||||
return os.path.join(_target_home(), ".claude", "commands")
|
||||
|
||||
|
||||
def _claude_skill_dst() -> str:
|
||||
return os.path.join(_claude_skill_dst_dir(), "ssh.md")
|
||||
|
||||
|
||||
def _ssh_key_path() -> str:
|
||||
return os.path.join(_target_home(), ".ssh", "id_ed25519")
|
||||
|
||||
|
||||
def _global_claude_md() -> str:
|
||||
return os.path.join(_target_home(), ".claude", "CLAUDE.md")
|
||||
|
||||
|
||||
def _global_gemini_md() -> str:
|
||||
return os.path.join(_gemini_dir(), "GEMINI.md")
|
||||
|
||||
|
||||
def _codex_skill_dst_root() -> str:
|
||||
return os.path.join(_target_home(), ".codex", "skills")
|
||||
|
||||
|
||||
def _codex_skill_dst_dir() -> str:
|
||||
return os.path.join(_codex_skill_dst_root(), "server-manager")
|
||||
|
||||
|
||||
def _codex_skill_entry() -> str:
|
||||
return os.path.join(_codex_skill_dst_dir(), "SKILL.md")
|
||||
|
||||
|
||||
def _codex_wrapper_dst() -> str:
|
||||
return os.path.join(
|
||||
_shared_dir(),
|
||||
"codex-ssh.cmd" if sys.platform == "win32" else "codex-ssh",
|
||||
)
|
||||
|
||||
|
||||
def _gemini_skill_dst_root() -> str:
|
||||
return os.path.join(_gemini_dir(), "skills")
|
||||
|
||||
|
||||
def _gemini_skill_dst_dir() -> str:
|
||||
return os.path.join(_gemini_skill_dst_root(), "server-manager")
|
||||
|
||||
|
||||
def _gemini_skill_entry() -> str:
|
||||
return os.path.join(_gemini_skill_dst_dir(), "SKILL.md")
|
||||
|
||||
|
||||
def _agents_skill_dst_root() -> str:
|
||||
return os.path.join(_target_home(), ".agents", "skills")
|
||||
|
||||
|
||||
def _agents_skill_dst_dir() -> str:
|
||||
return os.path.join(_agents_skill_dst_root(), "server-manager")
|
||||
|
||||
|
||||
def _gemini_wrapper_dst() -> str:
|
||||
return os.path.join(
|
||||
_shared_dir(),
|
||||
"gemini-ssh.cmd" if sys.platform == "win32" else "gemini-ssh",
|
||||
)
|
||||
|
||||
|
||||
def _ensure_executable(path: str):
|
||||
if sys.platform == "win32" or not os.path.exists(path):
|
||||
@@ -102,27 +198,96 @@ def _copy_tree(src: str, dst: str) -> str:
|
||||
return dst
|
||||
|
||||
|
||||
def _install_wrapper(src: str, dst: str) -> str:
|
||||
return _copy_file(src, dst, executable=(sys.platform != "win32"))
|
||||
|
||||
|
||||
def _skill_script_names() -> list[str]:
|
||||
if sys.platform == "win32":
|
||||
return [
|
||||
os.path.join("scripts", "server-manager-doctor.cmd"),
|
||||
os.path.join("scripts", "codex-ssh-wrapper.cmd"),
|
||||
os.path.join("scripts", "server-manager-gemini-doctor.cmd"),
|
||||
os.path.join("scripts", "gemini-ssh-wrapper.cmd"),
|
||||
]
|
||||
return [
|
||||
os.path.join("scripts", "server-manager-doctor.sh"),
|
||||
os.path.join("scripts", "codex-ssh-wrapper.sh"),
|
||||
os.path.join("scripts", "server-manager-gemini-doctor.sh"),
|
||||
os.path.join("scripts", "gemini-ssh-wrapper.sh"),
|
||||
]
|
||||
|
||||
|
||||
def _ensure_skill_scripts(skill_dir: str):
|
||||
for rel_path in _skill_script_names():
|
||||
_ensure_executable(os.path.join(skill_dir, rel_path))
|
||||
|
||||
|
||||
def _iter_all_user_homes() -> list[str]:
|
||||
homes: list[str] = []
|
||||
|
||||
def add(path: str):
|
||||
expanded = os.path.abspath(os.path.expanduser(path))
|
||||
if os.path.isdir(expanded) and expanded not in homes:
|
||||
homes.append(expanded)
|
||||
|
||||
add(_target_home())
|
||||
|
||||
if sys.platform == "win32":
|
||||
users_root = os.path.join(os.environ.get("SystemDrive", "C:"), "Users")
|
||||
skip = {"public", "default", "default user", "all users"}
|
||||
if os.path.isdir(users_root):
|
||||
for name in sorted(os.listdir(users_root)):
|
||||
if name.lower() in skip:
|
||||
continue
|
||||
add(os.path.join(users_root, name))
|
||||
elif sys.platform == "darwin":
|
||||
add("/var/root")
|
||||
users_root = "/Users"
|
||||
if os.path.isdir(users_root):
|
||||
for name in sorted(os.listdir(users_root)):
|
||||
if name.startswith("."):
|
||||
continue
|
||||
add(os.path.join(users_root, name))
|
||||
else:
|
||||
add("/root")
|
||||
users_root = "/home"
|
||||
if os.path.isdir(users_root):
|
||||
for name in sorted(os.listdir(users_root)):
|
||||
if name.startswith("."):
|
||||
continue
|
||||
add(os.path.join(users_root, name))
|
||||
|
||||
return homes
|
||||
|
||||
|
||||
def check_status() -> dict:
|
||||
"""Check what's installed and what's missing."""
|
||||
shared_dir = _shared_dir()
|
||||
ssh_key_path = _ssh_key_path()
|
||||
return {
|
||||
"shared_dir": os.path.exists(SHARED_DIR),
|
||||
"servers_json": os.path.exists(os.path.join(SHARED_DIR, "servers.json")),
|
||||
"ssh_script": os.path.exists(os.path.join(SHARED_DIR, "ssh.py")),
|
||||
"encryption": os.path.exists(os.path.join(SHARED_DIR, "encryption.py")),
|
||||
"claude_skill_installed": os.path.exists(CLAUDE_SKILL_DST),
|
||||
"codex_skill_installed": os.path.exists(CODEX_SKILL_ENTRY),
|
||||
"codex_wrapper_installed": os.path.exists(CODEX_WRAPPER_DST),
|
||||
"ssh_key_exists": os.path.exists(SSH_KEY_PATH),
|
||||
"ssh_key_pub": os.path.exists(SSH_KEY_PATH + ".pub"),
|
||||
"target_home": _target_home(),
|
||||
"shared_dir": os.path.exists(shared_dir),
|
||||
"servers_json": os.path.exists(os.path.join(shared_dir, "servers.json")),
|
||||
"ssh_script": os.path.exists(os.path.join(shared_dir, "ssh.py")),
|
||||
"encryption": os.path.exists(os.path.join(shared_dir, "encryption.py")),
|
||||
"claude_skill_installed": os.path.exists(_claude_skill_dst()),
|
||||
"codex_skill_installed": os.path.exists(_codex_skill_entry()),
|
||||
"codex_wrapper_installed": os.path.exists(_codex_wrapper_dst()),
|
||||
"gemini_skill_installed": os.path.exists(_gemini_skill_entry()),
|
||||
"gemini_wrapper_installed": os.path.exists(_gemini_wrapper_dst()),
|
||||
"ssh_key_exists": os.path.exists(ssh_key_path),
|
||||
"ssh_key_pub": os.path.exists(ssh_key_path + ".pub"),
|
||||
}
|
||||
|
||||
|
||||
def install_ssh_script() -> str:
|
||||
"""Copy ssh.py and encryption.py to shared dir."""
|
||||
os.makedirs(SHARED_DIR, exist_ok=True)
|
||||
shared_dir = _shared_dir()
|
||||
os.makedirs(shared_dir, exist_ok=True)
|
||||
results = []
|
||||
|
||||
dst = os.path.join(SHARED_DIR, "ssh.py")
|
||||
dst = os.path.join(shared_dir, "ssh.py")
|
||||
if os.path.exists(SSH_SCRIPT_SRC):
|
||||
_copy_file(SSH_SCRIPT_SRC, dst, executable=True)
|
||||
log.info(f"ssh.py installed: {dst}")
|
||||
@@ -132,7 +297,7 @@ def install_ssh_script() -> str:
|
||||
else:
|
||||
results.append("ERROR: ssh.py source not found")
|
||||
|
||||
enc_dst = os.path.join(SHARED_DIR, "encryption.py")
|
||||
enc_dst = os.path.join(shared_dir, "encryption.py")
|
||||
if os.path.exists(ENCRYPTION_SRC):
|
||||
_copy_file(ENCRYPTION_SRC, enc_dst)
|
||||
log.info(f"encryption.py installed: {enc_dst}")
|
||||
@@ -147,54 +312,102 @@ def install_ssh_script() -> str:
|
||||
|
||||
def install_claude_skill() -> str:
|
||||
"""Install /ssh skill for Claude Code."""
|
||||
os.makedirs(CLAUDE_SKILL_DST_DIR, exist_ok=True)
|
||||
claude_skill_dst_dir = _claude_skill_dst_dir()
|
||||
claude_skill_dst = _claude_skill_dst()
|
||||
os.makedirs(claude_skill_dst_dir, exist_ok=True)
|
||||
if os.path.exists(CLAUDE_SKILL_SRC):
|
||||
_copy_file(CLAUDE_SKILL_SRC, CLAUDE_SKILL_DST)
|
||||
log.info(f"Claude skill installed: {CLAUDE_SKILL_DST}")
|
||||
return f"Claude skill installed: {CLAUDE_SKILL_DST}"
|
||||
if os.path.exists(CLAUDE_SKILL_DST):
|
||||
return f"Claude skill already exists: {CLAUDE_SKILL_DST}"
|
||||
_copy_file(CLAUDE_SKILL_SRC, claude_skill_dst)
|
||||
log.info(f"Claude skill installed: {claude_skill_dst}")
|
||||
return f"Claude skill installed: {claude_skill_dst}"
|
||||
if os.path.exists(claude_skill_dst):
|
||||
return f"Claude skill already exists: {claude_skill_dst}"
|
||||
|
||||
skill_content = _generate_skill_content()
|
||||
with open(CLAUDE_SKILL_DST, "w", encoding="utf-8") as f:
|
||||
with open(claude_skill_dst, "w", encoding="utf-8") as f:
|
||||
f.write(skill_content)
|
||||
log.info(f"Claude skill generated: {CLAUDE_SKILL_DST}")
|
||||
return f"Claude skill generated: {CLAUDE_SKILL_DST}"
|
||||
log.info(f"Claude skill generated: {claude_skill_dst}")
|
||||
return f"Claude skill generated: {claude_skill_dst}"
|
||||
|
||||
|
||||
def install_codex_skill() -> str:
|
||||
"""Install ServerManager skill package for Codex and the local wrapper."""
|
||||
results = []
|
||||
codex_skill_dst_dir = _codex_skill_dst_dir()
|
||||
codex_skill_entry = _codex_skill_entry()
|
||||
codex_wrapper_dst = _codex_wrapper_dst()
|
||||
|
||||
if os.path.isdir(CODEX_SKILL_SRC_DIR):
|
||||
_copy_tree(CODEX_SKILL_SRC_DIR, CODEX_SKILL_DST_DIR)
|
||||
for rel_path in [
|
||||
os.path.join("scripts", "server-manager-doctor.sh"),
|
||||
os.path.join("scripts", "server-manager-doctor.cmd"),
|
||||
os.path.join("scripts", "codex-ssh-wrapper.sh"),
|
||||
os.path.join("scripts", "codex-ssh-wrapper.cmd"),
|
||||
]:
|
||||
_ensure_executable(os.path.join(CODEX_SKILL_DST_DIR, rel_path))
|
||||
log.info(f"Codex skill installed: {CODEX_SKILL_DST_DIR}")
|
||||
results.append(f"Codex skill installed: {CODEX_SKILL_DST_DIR}")
|
||||
elif os.path.exists(CODEX_SKILL_ENTRY):
|
||||
results.append(f"Codex skill already exists: {CODEX_SKILL_DST_DIR}")
|
||||
_copy_tree(CODEX_SKILL_SRC_DIR, codex_skill_dst_dir)
|
||||
_ensure_skill_scripts(codex_skill_dst_dir)
|
||||
log.info(f"Codex skill installed: {codex_skill_dst_dir}")
|
||||
results.append(f"Codex skill installed: {codex_skill_dst_dir}")
|
||||
elif os.path.exists(codex_skill_entry):
|
||||
results.append(f"Codex skill already exists: {codex_skill_dst_dir}")
|
||||
else:
|
||||
results.append("ERROR: Codex skill source not found")
|
||||
|
||||
wrapper_src = CODEX_WRAPPER_SRC_CMD if sys.platform == "win32" else CODEX_WRAPPER_SRC_SH
|
||||
if os.path.exists(wrapper_src):
|
||||
_copy_file(wrapper_src, CODEX_WRAPPER_DST, executable=(sys.platform != "win32"))
|
||||
log.info(f"Codex wrapper installed: {CODEX_WRAPPER_DST}")
|
||||
results.append(f"Codex wrapper installed: {CODEX_WRAPPER_DST}")
|
||||
elif os.path.exists(CODEX_WRAPPER_DST):
|
||||
results.append(f"Codex wrapper already exists: {CODEX_WRAPPER_DST}")
|
||||
_copy_file(wrapper_src, codex_wrapper_dst, executable=(sys.platform != "win32"))
|
||||
log.info(f"Codex wrapper installed: {codex_wrapper_dst}")
|
||||
results.append(f"Codex wrapper installed: {codex_wrapper_dst}")
|
||||
elif os.path.exists(codex_wrapper_dst):
|
||||
results.append(f"Codex wrapper already exists: {codex_wrapper_dst}")
|
||||
else:
|
||||
results.append("ERROR: Codex wrapper source not found")
|
||||
|
||||
return "\n".join(results)
|
||||
|
||||
|
||||
def install_gemini_skill() -> str:
|
||||
"""Install ServerManager skill package for Gemini."""
|
||||
results = []
|
||||
gemini_skill_dst_dir = _gemini_skill_dst_dir()
|
||||
gemini_skill_entry = _gemini_skill_entry()
|
||||
agents_skill_dst_dir = _agents_skill_dst_dir()
|
||||
gemini_wrapper_dst = _gemini_wrapper_dst()
|
||||
install_generic_mirror = os.environ.get(
|
||||
"SERVER_MANAGER_INSTALL_GENERIC_SKILL_MIRROR", ""
|
||||
).strip() == "1"
|
||||
|
||||
if os.path.isdir(GEMINI_SKILL_SRC_DIR):
|
||||
_copy_tree(GEMINI_SKILL_SRC_DIR, gemini_skill_dst_dir)
|
||||
_ensure_skill_scripts(gemini_skill_dst_dir)
|
||||
log.info(f"Gemini skill installed: {gemini_skill_dst_dir}")
|
||||
results.append(f"Gemini skill installed: {gemini_skill_dst_dir}")
|
||||
|
||||
if install_generic_mirror:
|
||||
_copy_tree(GEMINI_SKILL_SRC_DIR, agents_skill_dst_dir)
|
||||
_ensure_skill_scripts(agents_skill_dst_dir)
|
||||
log.info(f"Generic agents skill mirror installed: {agents_skill_dst_dir}")
|
||||
results.append(f"Generic agents skill mirror installed: {agents_skill_dst_dir}")
|
||||
elif os.path.exists(agents_skill_dst_dir):
|
||||
shutil.rmtree(agents_skill_dst_dir, ignore_errors=True)
|
||||
log.info(f"Removed generic agents skill mirror to avoid Gemini conflicts: {agents_skill_dst_dir}")
|
||||
results.append(
|
||||
f"Removed generic agents skill mirror to avoid Gemini conflicts: {agents_skill_dst_dir}"
|
||||
)
|
||||
elif os.path.exists(gemini_skill_entry):
|
||||
results.append(f"Gemini skill already exists: {gemini_skill_dst_dir}")
|
||||
else:
|
||||
results.append("ERROR: Gemini skill source not found")
|
||||
|
||||
wrapper_src = GEMINI_WRAPPER_SRC_CMD if sys.platform == "win32" else GEMINI_WRAPPER_SRC_SH
|
||||
if not os.path.exists(wrapper_src):
|
||||
wrapper_src = CODEX_WRAPPER_SRC_CMD if sys.platform == "win32" else CODEX_WRAPPER_SRC_SH
|
||||
|
||||
if os.path.exists(wrapper_src):
|
||||
_install_wrapper(wrapper_src, gemini_wrapper_dst)
|
||||
log.info(f"Gemini wrapper installed: {gemini_wrapper_dst}")
|
||||
results.append(f"Gemini wrapper installed: {gemini_wrapper_dst}")
|
||||
elif os.path.exists(gemini_wrapper_dst):
|
||||
results.append(f"Gemini wrapper already exists: {gemini_wrapper_dst}")
|
||||
else:
|
||||
results.append("ERROR: Gemini wrapper source not found")
|
||||
|
||||
return "\n".join(results)
|
||||
|
||||
|
||||
def install_skill() -> str:
|
||||
"""Backward-compatible alias for the Claude /ssh skill installer."""
|
||||
return install_claude_skill()
|
||||
@@ -202,19 +415,20 @@ def install_skill() -> str:
|
||||
|
||||
def generate_ssh_key() -> str:
|
||||
"""Generate ed25519 SSH key if not exists."""
|
||||
if os.path.exists(SSH_KEY_PATH):
|
||||
return f"Key already exists: {SSH_KEY_PATH}"
|
||||
ssh_key_path = _ssh_key_path()
|
||||
if os.path.exists(ssh_key_path):
|
||||
return f"Key already exists: {ssh_key_path}"
|
||||
|
||||
os.makedirs(os.path.dirname(SSH_KEY_PATH), exist_ok=True)
|
||||
os.makedirs(os.path.dirname(ssh_key_path), exist_ok=True)
|
||||
|
||||
try:
|
||||
subprocess.run(
|
||||
["ssh-keygen", "-t", "ed25519", "-f", SSH_KEY_PATH,
|
||||
["ssh-keygen", "-t", "ed25519", "-f", ssh_key_path,
|
||||
"-N", "", "-C", "server-manager"],
|
||||
check=True, capture_output=True, timeout=15
|
||||
)
|
||||
log.info(f"SSH key generated: {SSH_KEY_PATH}")
|
||||
return f"Key generated: {SSH_KEY_PATH}"
|
||||
log.info(f"SSH key generated: {ssh_key_path}")
|
||||
return f"Key generated: {ssh_key_path}"
|
||||
except FileNotFoundError:
|
||||
hint = "enable OpenSSH optional feature" if sys.platform == "win32" else "install openssh-client"
|
||||
msg = f"ERROR: ssh-keygen not found — {hint}"
|
||||
@@ -228,11 +442,12 @@ def generate_ssh_key() -> str:
|
||||
|
||||
def install_global_claude_md() -> str:
|
||||
"""Add/update server manager section in global ~/.claude/CLAUDE.md."""
|
||||
os.makedirs(os.path.dirname(GLOBAL_CLAUDE_MD), exist_ok=True)
|
||||
global_claude_md = _global_claude_md()
|
||||
os.makedirs(os.path.dirname(global_claude_md), exist_ok=True)
|
||||
|
||||
existing = ""
|
||||
if os.path.exists(GLOBAL_CLAUDE_MD):
|
||||
with open(GLOBAL_CLAUDE_MD, encoding="utf-8") as f:
|
||||
if os.path.exists(global_claude_md):
|
||||
with open(global_claude_md, encoding="utf-8") as f:
|
||||
existing = f.read()
|
||||
|
||||
pattern = re.compile(
|
||||
@@ -242,41 +457,96 @@ def install_global_claude_md() -> str:
|
||||
|
||||
if pattern.search(existing):
|
||||
updated = pattern.sub(GLOBAL_CLAUDE_MD_BLOCK.strip(), existing)
|
||||
with open(GLOBAL_CLAUDE_MD, "w", encoding="utf-8") as f:
|
||||
with open(global_claude_md, "w", encoding="utf-8") as f:
|
||||
f.write(updated)
|
||||
log.info(f"Global CLAUDE.md block updated: {GLOBAL_CLAUDE_MD}")
|
||||
return f"Global CLAUDE.md block updated: {GLOBAL_CLAUDE_MD}"
|
||||
log.info(f"Global CLAUDE.md block updated: {global_claude_md}")
|
||||
return f"Global CLAUDE.md block updated: {global_claude_md}"
|
||||
|
||||
with open(GLOBAL_CLAUDE_MD, "a", encoding="utf-8") as f:
|
||||
with open(global_claude_md, "a", encoding="utf-8") as f:
|
||||
if existing and not existing.endswith("\n"):
|
||||
f.write("\n")
|
||||
f.write("\n" + GLOBAL_CLAUDE_MD_BLOCK)
|
||||
log.info(f"Global CLAUDE.md block added: {GLOBAL_CLAUDE_MD}")
|
||||
return f"Global CLAUDE.md block added: {GLOBAL_CLAUDE_MD}"
|
||||
log.info(f"Global CLAUDE.md block added: {global_claude_md}")
|
||||
return f"Global CLAUDE.md block added: {global_claude_md}"
|
||||
|
||||
|
||||
def install_global_gemini_md() -> str:
|
||||
"""Add/update server manager section in global ~/.gemini/GEMINI.md."""
|
||||
global_gemini_md = _global_gemini_md()
|
||||
os.makedirs(os.path.dirname(global_gemini_md), exist_ok=True)
|
||||
|
||||
existing = ""
|
||||
if os.path.exists(global_gemini_md):
|
||||
with open(global_gemini_md, encoding="utf-8") as f:
|
||||
existing = f.read()
|
||||
|
||||
pattern = re.compile(
|
||||
re.escape(_GEMINI_BLOCK_START) + r".*?" + re.escape(_GEMINI_BLOCK_END),
|
||||
re.DOTALL
|
||||
)
|
||||
|
||||
if pattern.search(existing):
|
||||
updated = pattern.sub(GLOBAL_GEMINI_MD_BLOCK.strip(), existing)
|
||||
with open(global_gemini_md, "w", encoding="utf-8") as f:
|
||||
f.write(updated)
|
||||
log.info(f"Global GEMINI.md block updated: {global_gemini_md}")
|
||||
return f"Global GEMINI.md block updated: {global_gemini_md}"
|
||||
|
||||
with open(global_gemini_md, "a", encoding="utf-8") as f:
|
||||
if existing and not existing.endswith("\n"):
|
||||
f.write("\n")
|
||||
f.write("\n" + GLOBAL_GEMINI_MD_BLOCK)
|
||||
log.info(f"Global GEMINI.md block added: {global_gemini_md}")
|
||||
return f"Global GEMINI.md block added: {global_gemini_md}"
|
||||
|
||||
|
||||
def install_all() -> list[str]:
|
||||
"""Full setup — install everything for Claude Code and Codex."""
|
||||
results = []
|
||||
|
||||
steps = [
|
||||
"""Full setup — install everything for Claude Code, Codex, and Gemini."""
|
||||
all_users = os.environ.get("SERVER_MANAGER_INSTALL_ALL_USERS", "").strip() == "1"
|
||||
base_steps = [
|
||||
("ssh_script", install_ssh_script),
|
||||
("claude_skill", install_claude_skill),
|
||||
("codex_skill", install_codex_skill),
|
||||
("ssh_key", generate_ssh_key),
|
||||
("gemini_skill", install_gemini_skill),
|
||||
("global_claude_md", install_global_claude_md),
|
||||
("global_gemini_md", install_global_gemini_md),
|
||||
]
|
||||
|
||||
for name, func in steps:
|
||||
try:
|
||||
log.info(f"install_all: running {name}")
|
||||
result = func()
|
||||
results.append(result)
|
||||
except Exception as e:
|
||||
msg = f"ERROR ({name}): {e}"
|
||||
log.error(msg)
|
||||
results.append(msg)
|
||||
if not all_users:
|
||||
steps = base_steps[:3] + [("ssh_key", generate_ssh_key)] + base_steps[3:]
|
||||
results = []
|
||||
for name, func in steps:
|
||||
try:
|
||||
log.info(f"install_all: running {name}")
|
||||
result = func()
|
||||
results.append(result)
|
||||
except Exception as e:
|
||||
msg = f"ERROR ({name}): {e}"
|
||||
log.error(msg)
|
||||
results.append(msg)
|
||||
return results
|
||||
|
||||
results = []
|
||||
original_target = os.environ.get("SERVER_MANAGER_TARGET_HOME")
|
||||
for home in _iter_all_user_homes():
|
||||
os.environ["SERVER_MANAGER_TARGET_HOME"] = home
|
||||
results.append(f"[target_home] {home}")
|
||||
for name, func in base_steps:
|
||||
try:
|
||||
log.info(f"install_all(all_users): running {name} for {home}")
|
||||
result = func()
|
||||
results.append(result)
|
||||
except Exception as e:
|
||||
msg = f"ERROR ({name}, {home}): {e}"
|
||||
log.error(msg)
|
||||
results.append(msg)
|
||||
|
||||
if original_target is None:
|
||||
os.environ.pop("SERVER_MANAGER_TARGET_HOME", None)
|
||||
else:
|
||||
os.environ["SERVER_MANAGER_TARGET_HOME"] = original_target
|
||||
|
||||
results.append("INFO: SSH key generation skipped in SERVER_MANAGER_INSTALL_ALL_USERS=1 mode")
|
||||
return results
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user