refactor code, ẩn thông tin ssh

core/ssh_update_flash.py

@@ -0,0 +1,92 @@
+"""
+Luồng SSH cho chế độ "Update Firmware" (thiết bị đã cài sẵn OpenWrt).
+
+Đặc điểm:
+  - Thiết bị đang chạy bình thường, SSH đã mở sẵn với pass đã biết.
+  - Không có bước Telnet / đặt lại mật khẩu.
+  - Kết nối trực tiếp → upload SCP → verify → sync + sysupgrade.
+
+Public API:
+  flash_device_update_ssh(ip, firmware_path, user, password,
+                          backup_password, status_cb)
+"""
+
+import time
+
+import paramiko
+
+from core.ssh_utils import (
+    _create_ssh_client,
+    _upload_firmware,
+    _verify_firmware,
+    _sync_and_sysupgrade,
+)
+
+
+def flash_device_update_ssh(ip, firmware_path, user="root",
+                             password="admin123a", backup_password="",
+                             status_cb=None):
+    """
+    Cập nhật firmware lên thiết bị OpenWrt đang chạy qua SSH / sysupgrade.
+
+    Luồng:
+      1. Kết nối SSH (thử password, nếu lỗi auth thử backup_password)
+      2. Upload firmware qua SCP lên /tmp/
+      3. Verify file tồn tại
+      4. sync + sysupgrade
+
+    Args:
+        ip              : địa chỉ IP thiết bị
+        firmware_path   : đường dẫn file .bin trên máy tính
+        user            : SSH username (mặc định "root")
+        password        : SSH password chính
+        backup_password : SSH password dự phòng nếu password chính sai
+        status_cb       : callback(str) để cập nhật trạng thái lên UI
+
+    Returns:
+        "DONE"     — update thành công
+        "FAIL: …"  — thất bại, kèm lý do
+    """
+
+    # ── STEP 1: Kết nối SSH ─────────────────────────────────────────
+    if status_cb:
+        status_cb("Connecting SSH...")
+
+    client = None
+
+    # Thử password chính trước
+    try:
+        client = _create_ssh_client(ip, user, password)
+    except paramiko.AuthenticationException:
+        # Thử backup_password nếu có
+        if backup_password:
+            try:
+                client = _create_ssh_client(ip, user, backup_password)
+            except paramiko.AuthenticationException:
+                return "FAIL: SSH authentication failed (wrong password)"
+            except paramiko.SSHException as e:
+                return f"FAIL: SSH error — {e}"
+            except Exception as e:
+                return f"FAIL: Cannot connect — {e}"
+        else:
+            return "FAIL: SSH authentication failed"
+    except paramiko.SSHException as e:
+        return f"FAIL: SSH error — {e}"
+    except Exception as e:
+        return f"FAIL: Cannot connect — {e}"
+
+    # ── STEP 2–4: Upload → Verify → sysupgrade ─────────────────────
+    try:
+        remote_path = _upload_firmware(client, firmware_path, status_cb)
+        _verify_firmware(client, remote_path, status_cb)
+        return _sync_and_sysupgrade(client, remote_path, status_cb)
+
+    except RuntimeError as e:
+        return f"FAIL: {e}"
+    except Exception as e:
+        return f"FAIL: {e}"
+    finally:
+        try:
+            client.close()
+        except Exception:
+            pass