6179ded862
CustomTkinter desktop app for managing remote servers. Features: SSH terminal, SFTP file transfer, key management, background status monitoring, server CRUD with dark theme. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
117 lines
4.3 KiB
Python
117 lines
4.3 KiB
Python
"""
|
|
Keys tab — SSH key management: view, generate, install.
|
|
"""
|
|
|
|
import os
|
|
import threading
|
|
import customtkinter as ctk
|
|
from core.ssh_client import SSHClientWrapper
|
|
|
|
|
|
class KeysTab(ctk.CTkFrame):
|
|
def __init__(self, master, store):
|
|
super().__init__(master, fg_color="transparent")
|
|
self.store = store
|
|
self._current_alias: str | None = None
|
|
|
|
# Key info
|
|
ctk.CTkLabel(self, text="SSH Key", font=ctk.CTkFont(size=16, weight="bold"), anchor="w").pack(fill="x", padx=15, pady=(15, 5))
|
|
|
|
self.key_path_label = ctk.CTkLabel(self, text="", anchor="w", text_color="#9ca3af")
|
|
self.key_path_label.pack(fill="x", padx=15)
|
|
|
|
self.pub_key_box = ctk.CTkTextbox(self, height=80, font=ctk.CTkFont(family="Consolas", size=11), state="disabled")
|
|
self.pub_key_box.pack(fill="x", padx=15, pady=(5, 10))
|
|
|
|
# Buttons
|
|
btn_frame = ctk.CTkFrame(self, fg_color="transparent")
|
|
btn_frame.pack(fill="x", padx=15, pady=5)
|
|
|
|
self.gen_btn = ctk.CTkButton(btn_frame, text="Generate Key", command=self._generate)
|
|
self.gen_btn.pack(side="left", padx=(0, 10))
|
|
|
|
self.install_btn = ctk.CTkButton(btn_frame, text="Install on Server", fg_color="#22c55e", hover_color="#16a34a", command=self._install)
|
|
self.install_btn.pack(side="left")
|
|
|
|
self.copy_btn = ctk.CTkButton(btn_frame, text="Copy Public Key", fg_color="#6b7280", command=self._copy_key)
|
|
self.copy_btn.pack(side="right")
|
|
|
|
# Status log
|
|
self.status_log = ctk.CTkTextbox(self, height=120, font=ctk.CTkFont(family="Consolas", size=11), state="disabled")
|
|
self.status_log.pack(fill="both", expand=True, padx=15, pady=(10, 15))
|
|
|
|
self._refresh_key_info()
|
|
|
|
def set_server(self, alias: str | None):
|
|
self._current_alias = alias
|
|
|
|
def _refresh_key_info(self):
|
|
key_path = self.store.get_ssh_key_path()
|
|
pub_path = key_path + ".pub"
|
|
self.key_path_label.configure(text=f"Path: {key_path}")
|
|
|
|
self.pub_key_box.configure(state="normal")
|
|
self.pub_key_box.delete("1.0", "end")
|
|
|
|
if os.path.exists(pub_path):
|
|
with open(pub_path, "r") as f:
|
|
pub_key = f.read().strip()
|
|
self.pub_key_box.insert("1.0", pub_key)
|
|
self.gen_btn.configure(state="disabled", text="Key exists")
|
|
else:
|
|
self.pub_key_box.insert("1.0", "No key found. Click 'Generate Key' to create one.")
|
|
self.gen_btn.configure(state="normal", text="Generate Key")
|
|
|
|
self.pub_key_box.configure(state="disabled")
|
|
|
|
def _log(self, text: str):
|
|
self.status_log.configure(state="normal")
|
|
self.status_log.insert("end", text + "\n")
|
|
self.status_log.configure(state="disabled")
|
|
self.status_log.see("end")
|
|
|
|
def _generate(self):
|
|
try:
|
|
key_path = self.store.get_ssh_key_path()
|
|
wrapper = SSHClientWrapper({"alias": "temp", "ip": "0", "user": "root"}, key_path)
|
|
msg = wrapper.generate_key()
|
|
self._log(msg)
|
|
self._refresh_key_info()
|
|
except Exception as e:
|
|
self._log(f"[ERROR] {e}")
|
|
|
|
def _install(self):
|
|
if not self._current_alias:
|
|
self._log("[!] No server selected")
|
|
return
|
|
|
|
server = self.store.get_server(self._current_alias)
|
|
if not server:
|
|
return
|
|
|
|
self.install_btn.configure(state="disabled", text="Installing...")
|
|
|
|
def _do():
|
|
try:
|
|
wrapper = SSHClientWrapper(server, self.store.get_ssh_key_path())
|
|
msg = wrapper.install_key()
|
|
self.after(0, lambda: self._log(f"[{self._current_alias}] {msg}"))
|
|
except Exception as e:
|
|
self.after(0, lambda: self._log(f"[ERROR] {e}"))
|
|
finally:
|
|
self.after(0, lambda: self.install_btn.configure(state="normal", text="Install on Server"))
|
|
|
|
threading.Thread(target=_do, daemon=True).start()
|
|
|
|
def _copy_key(self):
|
|
key_path = self.store.get_ssh_key_path()
|
|
pub_path = key_path + ".pub"
|
|
if os.path.exists(pub_path):
|
|
with open(pub_path, "r") as f:
|
|
pub_key = f.read().strip()
|
|
self.clipboard_clear()
|
|
self.clipboard_append(pub_key)
|
|
self._log("Public key copied to clipboard")
|
|
else:
|
|
self._log("[!] No public key to copy")
|