bf39fd7b67
v1.2.0: - GUI localization (EN/RU/ZH) with language switcher and persistent selection - About dialog (ⓘ) with app info, features, quick start guide - core/i18n.py — internationalization module with t() function - All GUI components translated via t() keys v1.3.0: - TOTP/2FA tab — Google Authenticator compatible codes with live 30s countdown, one-click copy, per-server secret management - core/totp.py — TOTP module (pyotp, RFC 6238) - core/logger.py — rotating file logger (5MB, 3 backups) - Stronger Fernet encryption key with automatic migration from old key - Thread-safe server store with locks, atomic writes, auto-restore on corruption - Parallel status checks via ThreadPoolExecutor (up to 10 concurrent) - SSH client: explicit channel cleanup, Unix key permissions - Server dialog: port validation (1-65535), TOTP secret field - Language change preserves active tab and server selection - pyotp dependency added Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
135 lines
4.4 KiB
Python
135 lines
4.4 KiB
Python
"""
|
|
Claude Code integration setup.
|
|
Installs ssh.py, encryption.py, /ssh skill, SSH key — everything needed
|
|
for Claude Code to manage servers via the shared servers.json.
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import shutil
|
|
|
|
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
|
|
else:
|
|
_BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
SSH_SCRIPT_SRC = os.path.join(_BASE_DIR, "tools", "ssh.py")
|
|
ENCRYPTION_SRC = os.path.join(_BASE_DIR, "core", "encryption.py")
|
|
SKILL_SRC = os.path.join(_BASE_DIR, "tools", "skill-ssh.md")
|
|
|
|
SKILL_DST_DIR = os.path.expanduser("~/.claude/commands")
|
|
SKILL_DST = os.path.join(SKILL_DST_DIR, "ssh.md")
|
|
SSH_KEY_PATH = os.path.expanduser("~/.ssh/id_ed25519")
|
|
|
|
|
|
def check_status() -> dict:
|
|
"""Check what's installed and what's missing."""
|
|
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")),
|
|
"skill_installed": os.path.exists(SKILL_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)
|
|
results = []
|
|
|
|
# Copy ssh.py
|
|
dst = os.path.join(SHARED_DIR, "ssh.py")
|
|
if os.path.exists(SSH_SCRIPT_SRC):
|
|
shutil.copy2(SSH_SCRIPT_SRC, dst)
|
|
results.append(f"ssh.py installed: {dst}")
|
|
elif os.path.exists(dst):
|
|
results.append(f"ssh.py already exists: {dst}")
|
|
else:
|
|
results.append("ERROR: ssh.py source not found")
|
|
|
|
# Copy encryption.py
|
|
enc_dst = os.path.join(SHARED_DIR, "encryption.py")
|
|
if os.path.exists(ENCRYPTION_SRC):
|
|
shutil.copy2(ENCRYPTION_SRC, enc_dst)
|
|
results.append(f"encryption.py installed: {enc_dst}")
|
|
elif os.path.exists(enc_dst):
|
|
results.append(f"encryption.py already exists: {enc_dst}")
|
|
else:
|
|
results.append("ERROR: encryption.py source not found")
|
|
|
|
return "\n".join(results)
|
|
|
|
|
|
def install_skill() -> str:
|
|
"""Install /ssh skill for Claude Code."""
|
|
os.makedirs(SKILL_DST_DIR, exist_ok=True)
|
|
if os.path.exists(SKILL_SRC):
|
|
shutil.copy2(SKILL_SRC, SKILL_DST)
|
|
return f"Skill installed: {SKILL_DST}"
|
|
# Fallback: check existing
|
|
if os.path.exists(SKILL_DST):
|
|
return f"Skill already exists: {SKILL_DST}"
|
|
# Generate minimal skill
|
|
skill_content = _generate_skill_content()
|
|
with open(SKILL_DST, "w", encoding="utf-8") as f:
|
|
f.write(skill_content)
|
|
return f"Skill generated: {SKILL_DST}"
|
|
|
|
|
|
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}"
|
|
|
|
os.makedirs(os.path.dirname(SSH_KEY_PATH), exist_ok=True)
|
|
|
|
import paramiko
|
|
key = paramiko.Ed25519Key.generate()
|
|
key.write_private_key_file(SSH_KEY_PATH)
|
|
|
|
pub_key = f"ssh-ed25519 {key.get_base64()} server-manager"
|
|
with open(SSH_KEY_PATH + ".pub", "w") as f:
|
|
f.write(pub_key + "\n")
|
|
|
|
return f"Key generated: {SSH_KEY_PATH}"
|
|
|
|
|
|
def install_all() -> list[str]:
|
|
"""Full setup — install everything."""
|
|
results = []
|
|
results.append(install_ssh_script())
|
|
results.append(install_skill())
|
|
results.append(generate_ssh_key())
|
|
return results
|
|
|
|
|
|
def _generate_skill_content() -> str:
|
|
"""Generate /ssh skill markdown."""
|
|
ssh_py_path = os.path.join(SHARED_DIR, "ssh.py").replace("\\", "/")
|
|
return f"""SSH skill for Claude Code.
|
|
|
|
RULES:
|
|
- NEVER read servers.json directly
|
|
- ONLY use ssh.py for all server operations
|
|
- Maximum 1 connection attempt per command
|
|
- If connection fails — report error, do NOT retry automatically
|
|
|
|
Usage:
|
|
python "{ssh_py_path}" ALIAS "command"
|
|
python "{ssh_py_path}" ALIAS --no-sudo "command"
|
|
python "{ssh_py_path}" ALIAS --upload LOCAL REMOTE
|
|
python "{ssh_py_path}" ALIAS --download REMOTE LOCAL
|
|
python "{ssh_py_path}" ALIAS --install-key
|
|
python "{ssh_py_path}" ALIAS --ping
|
|
python "{ssh_py_path}" --list
|
|
python "{ssh_py_path}" --status
|
|
|
|
The user's arguments: $ARGUMENTS
|
|
"""
|