feat: IDF 连接模式 — 火焰按钮 Build+Flash+Monitor 一键烧录 (调 idf.py, 设置可配路径+自动探测, MCP 开放 esp_build_flash 工具)

Author: Martlet-Tech

src-tauri/src/esp_cmd.rs

@@ -7,7 +7,9 @@ mod window_helper;
 mod encoding_utils;
 mod multi_string;
 mod net_cmd;
+mod esp_cmd;
 
+use esp_cmd::EspHandle;
 use state::{NetState, SerialState};
 use receive_buffer::ReceiveBuffer;
 use mcp_server::McpServerHandle;
@@ -80,6 +82,7 @@ pub fn run() {
         .plugin(tauri_plugin_window_state::Builder::default().build())
         .manage(serial)
         .manage(net)
+        .manage(EspHandle::new())
         .manage(ReceiveBuffer::new())
         .manage(McpServerHandle::new())
         .manage(state::LocaleState::default())
@@ -99,6 +102,11 @@ pub fn run() {
             net_cmd::net_open,
             net_cmd::net_close,
             net_cmd::set_conn_mode,
+            esp_cmd::set_esp_config,
+            esp_cmd::esp_build_flash_start,
+            esp_cmd::esp_flash_cancel,
+            esp_cmd::esp_reset,
+            esp_cmd::detect_esp_paths,
             multi_string::open_multi_string_window,
             multi_string::load_multi_strings,
             multi_string::save_multi_strings,

src-tauri/src/lib.rs

@@ -33,6 +33,7 @@ impl McpServerHandle {
 struct AppState {
     buffer: ReceiveBuffer,
     serial: SerialState,
+    esp: crate::esp_cmd::EspHandle,
     app_handle: tauri::AppHandle,
 }
 
@@ -41,6 +42,7 @@ impl Clone for AppState {
         Self {
             buffer: self.buffer.clone(),
             serial: self.serial.clone(),
+            esp: self.esp.clone(),
             app_handle: self.app_handle.clone(),
         }
     }
@@ -180,6 +182,30 @@ async fn handle_tools_list(id: Value) -> (StatusCode, Json<Value>) {
                     "type": "object",
                     "properties": {}
                 }
+            },
+            {
+                "name": "esp_build_flash",
+                "description": "对 ESP-IDF 项目执行 build + flash(等同于主界面的火焰按钮),完成后芯片自动复位运行",
+                "inputSchema": {
+                    "type": "object",
+                    "properties": {
+                        "project_dir": {
+                            "type": "string",
+                            "description": "ESP-IDF 项目根目录(含 CMakeLists.txt)"
+                        },
+                        "port": {
+                            "type": "string",
+                            "description": "串口, 缺省用当前选中的串口",
+                            "default": ""
+                        },
+                        "baud": {
+                            "type": "integer",
+                            "description": "烧录波特率, 缺省用设置里的值",
+                            "default": 0
+                        }
+                    },
+                    "required": ["project_dir"]
+                }
             }
         ]
     }));
@@ -256,6 +282,82 @@ async fn handle_tools_call(state: &AppState, body: &Value, id: Value) -> (Status
             }));
             (StatusCode::OK, Json(resp))
         }
+        "esp_build_flash" => {
+            let project_dir = args.get("project_dir")
+                .and_then(|v| v.as_str())
+                .unwrap_or("")
+                .to_string();
+            let port = args.get("port")
+                .and_then(|v| v.as_str())
+                .unwrap_or("")
+                .to_string();
+            let baud = args.get("baud").and_then(|v| v.as_u64()).map(|v| v as u32);
+
+            if project_dir.is_empty() {
+                let resp = jsonrpc_error(-32602, "project_dir cannot be empty");
+                return (StatusCode::BAD_REQUEST, Json(resp));
+            }
+
+            let cfg = {
+                let c = state.esp.config.lock().unwrap_or_else(|e| e.into_inner());
+                crate::esp_cmd::EspConfig {
+                    idf_path: c.idf_path.clone(),
+                    python_path: c.python_path.clone(),
+                    baud: baud.unwrap_or(c.baud),
+                }
+            };
+            if cfg.idf_path.trim().is_empty() {
+                let resp = jsonrpc_error(-32603, "ESP-IDF 路径未配置, 请先在系统设置里填写");
+                return (StatusCode::INTERNAL_SERVER_ERROR, Json(resp));
+            }
+
+            {
+                let mut busy = state.esp.busy.lock().unwrap_or_else(|e| e.into_inner());
+                if *busy {
+                    let resp = jsonrpc_error(-32603, "已有烧录任务在运行");
+                    return (StatusCode::INTERNAL_SERVER_ERROR, Json(resp));
+                }
+                *busy = true;
+            }
+
+            let esp = state.esp.clone();
+            let app_handle = state.app_handle.clone();
+            let serial_name = state
+                .serial
+                .port_name
+                .lock()
+                .unwrap_or_else(|e| e.into_inner())
+                .clone()
+                .unwrap_or_default();
+            let port = if port.is_empty() { serial_name } else { port };
+            let result = tokio::task::spawn_blocking(move || {
+                let r = crate::esp_cmd::run_worker(&esp, &app_handle, &cfg, &project_dir, &port);
+                let mut busy = esp.busy.lock().unwrap_or_else(|e| e.into_inner());
+                *busy = false;
+                r
+            })
+            .await;
+
+            match result {
+                Ok(Ok(())) => {
+                    let resp = jsonrpc_success(id, json!({
+                        "content": [{
+                            "type": "text",
+                            "text": "build + flash 成功, 芯片已复位运行"
+                        }]
+                    }));
+                    (StatusCode::OK, Json(resp))
+                }
+                Ok(Err(e)) => {
+                    let resp = jsonrpc_error(-32603, &format!("烧录失败: {}", e));
+                    (StatusCode::INTERNAL_SERVER_ERROR, Json(resp))
+                }
+                Err(e) => {
+                    let resp = jsonrpc_error(-32603, &format!("任务异常: {}", e));
+                    (StatusCode::INTERNAL_SERVER_ERROR, Json(resp))
+                }
+            }
+        }
         _ => {
             let resp = jsonrpc_error(-32602, &format!("Tool not found: {}", tool_name));
             (StatusCode::NOT_FOUND, Json(resp))
@@ -296,6 +398,7 @@ pub async fn mcp_start(
     handle: tauri::State<'_, McpServerHandle>,
     buffer: tauri::State<'_, ReceiveBuffer>,
     serial: tauri::State<'_, SerialState>,
+    esp: tauri::State<'_, crate::esp_cmd::EspHandle>,
     port: u16,
 ) -> Result<(), String> {
     let mut running = handle.running.lock().await;
@@ -307,6 +410,7 @@ pub async fn mcp_start(
     let state = AppState {
         buffer: (*buffer).clone(),
         serial: (*serial).clone(),
+        esp: (*esp).clone(),
         app_handle: app_handle.clone(),
     };
 

src-tauri/src/mcp_server.rs

@@ -1,6 +1,6 @@
 use crate::state::{
-    ConnState, NetIo, NetState, MODE_SERIAL, MODE_TCP_CLIENT, MODE_TCP_SERVER, MODE_UDP_CLIENT,
-    MODE_UDP_SERVER,
+    ConnState, NetIo, NetState, MODE_IDF, MODE_SERIAL, MODE_TCP_CLIENT, MODE_TCP_SERVER,
+    MODE_UDP_CLIENT, MODE_UDP_SERVER,
 };
 use std::io::{Read, Write};
 use std::net::{SocketAddr, TcpListener, TcpStream, ToSocketAddrs, UdpSocket};
@@ -33,7 +33,7 @@ pub async fn set_conn_mode(
     mode: u8,
 ) -> Result<(), String> {
     let state = state.inner();
-    if !(MODE_SERIAL..=MODE_UDP_SERVER).contains(&mode) {
+    if !(MODE_SERIAL..=MODE_IDF).contains(&mode) {
         return Err("Unknown connection mode".into());
     }
     state.mode.store(mode, Ordering::SeqCst);

src-tauri/src/net_cmd.rs

@@ -8,6 +8,7 @@ pub const MODE_TCP_CLIENT: u8 = 1;
 pub const MODE_TCP_SERVER: u8 = 2;
 pub const MODE_UDP_CLIENT: u8 = 3;
 pub const MODE_UDP_SERVER: u8 = 4;
+pub const MODE_IDF: u8 = 5;
 
 pub enum NetIo {
     Stream(std::net::TcpStream),

src-tauri/src/state.rs

@@ -742,6 +742,31 @@ select:focus, input:focus, textarea:focus {
   opacity: 0.7;
 }
 
+/* ===== ESP build/flash dialog ===== */
+.esp-log {
+  height: 320px;
+  overflow-y: auto;
+  background: var(--bg);
+  color: var(--fg);
+  border: 1px solid var(--border);
+  border-radius: 4px;
+  padding: 8px 10px;
+  font-family: Consolas, 'Courier New', monospace;
+  font-size: 12px;
+  line-height: 1.5;
+  white-space: pre-wrap;
+  word-break: break-all;
+  user-select: text;
+  -webkit-user-select: text;
+}
+.esp-log .esp-stage {
+  color: var(--accent);
+  font-weight: 600;
+}
+.esp-log .esp-err {
+  color: var(--danger);
+}
+
 /* ===== Status Bar ===== */
 #status-bar {
   display: flex;

src/css/style.css

@@ -51,6 +51,7 @@
     "tcpServer": "TCP Server",
     "udpClient": "UDP Client",
     "udpServer": "UDP Server",
+    "idf": "IDF",
     "remoteIp": "Remote IP",
     "remotePort": "Remote port",
     "localPort": "Local port"
@@ -123,7 +124,21 @@
     "reconnectInterval": "Interval:",
     "mcp": "MCP Service:",
     "enable": "Enable",
-    "port": "Port:"
+    "port": "Port:",
+    "espIdf": "ESP-IDF path:",
+    "espPython": "Python path (optional):",
+    "espBaud": "Flash baud rate:",
+    "espDetect": "Auto detect"
+  },
+  "esp": {
+    "buildTitle": "ESP-IDF Build + Flash",
+    "flashMonitor": "Flash+Monitor",
+    "needConfig": "Configure ESP-IDF path in Settings first",
+    "needPort": "Select a serial port first",
+    "running": "Flashing...",
+    "cancelling": "Cancelling...",
+    "done": "Done, chip reset & running",
+    "failed": "Failed: {stage}"
   },
   "serialSettings": {
     "title": "Serial Port Settings",

src/i18n/en.json

@@ -51,6 +51,7 @@
     "tcpServer": "TCP Server",
     "udpClient": "UDP Client",
     "udpServer": "UDP Server",
+    "idf": "IDF",
     "remoteIp": "对方IP",
     "remotePort": "对方端口",
     "localPort": "本地端口"
@@ -123,7 +124,21 @@
     "reconnectInterval": "间隔:",
     "mcp": "MCP 服务:",
     "enable": "启用",
-    "port": "端口:"
+    "port": "端口:",
+    "espIdf": "ESP-IDF 路径:",
+    "espPython": "Python 路径(可空):",
+    "espBaud": "烧录波特率:",
+    "espDetect": "自动探测"
+  },
+  "esp": {
+    "buildTitle": "ESP-IDF Build + Flash",
+    "flashMonitor": "Flash+Monitor",
+    "needConfig": "请先在系统设置里配置 ESP-IDF 路径",
+    "needPort": "请先选择串口",
+    "running": "烧录中...",
+    "cancelling": "正在取消...",
+    "done": "完成, 芯片已复位运行",
+    "failed": "失败: {stage}"
   },
   "serialSettings": {
     "title": "串口详细设置",

src/i18n/zh-CN.json

@@ -260,6 +260,22 @@
           <span class="label" style="margin-left:12px" data-i18n="settings.port">端口:</span>
           <input type="number" id="setting-mcp-port" value="9876" min="1024" max="65535" class="small-input" style="width:80px" />
         </div>
+        <div class="setting-row">
+          <label data-i18n="settings.espIdf">ESP-IDF 路径:</label>
+          <input type="text" id="setting-esp-idf" class="file-path-input" list="esp-idf-datalist" spellcheck="false" placeholder="D:\...\esp-idf" />
+          <datalist id="esp-idf-datalist"></datalist>
+          <button id="btn-esp-detect" class="small-btn" data-i18n="settings.espDetect">自动探测</button>
+        </div>
+        <div class="setting-row">
+          <label data-i18n="settings.espPython">Python 路径(可空):</label>
+          <input type="text" id="setting-esp-python" class="file-path-input" list="esp-python-datalist" spellcheck="false" placeholder="...python_env\...\python.exe" />
+          <datalist id="esp-python-datalist"></datalist>
+        </div>
+        <div class="setting-row">
+          <label data-i18n="settings.espBaud">烧录波特率:</label>
+          <input type="number" id="setting-esp-baud" class="small-input" style="width:90px" value="921600" min="0" max="3000000" />
+          <span class="label" style="margin:0 4px" data-i18n="settings.ms">bps</span>
+        </div>
       </div>
       <div class="dialog-footer">
         <button id="setting-apply" class="small-btn primary" data-i18n="common.apply">应用</button>
@@ -336,6 +352,23 @@
     </div>
   </div>
 
+  <div id="esp-flash-overlay" class="overlay hidden">
+    <div id="esp-flash-dialog" class="dialog" style="width:640px;">
+      <div class="dialog-titlebar">
+        <span data-i18n="esp.buildTitle">ESP-IDF Build + Flash</span>
+        <button class="dialog-close title-btn" id="btn-esp-dialog-close"><i data-lucide="x"></i></button>
+      </div>
+      <div class="dialog-body" style="padding:12px;">
+        <pre id="esp-log" class="esp-log"></pre>
+      </div>
+      <div class="dialog-footer">
+        <span id="esp-flash-status" class="label">—</span>
+        <div style="flex:1"></div>
+        <button id="btn-esp-cancel" class="small-btn" data-i18n="common.cancel">取消</button>
+      </div>
+    </div>
+  </div>
+
   <script type="module" src="js/utils.js"></script>
   <script type="module" src="js/titlebar.js"></script>
   <script type="module" src="js/menu.js"></script>

src/index.html

@@ -81,6 +81,14 @@ document.addEventListener('DOMContentLoaded', async () => {
   document.addEventListener('clear-receive', clearReceive);
   listen('clear-receive', clearReceive);
 
+  document.addEventListener('open-monitor', async () => {
+    const s = await getSettings();
+    setTerminalMode(true);
+    applyTerminalUI(true);
+    createTerminal(s);
+    await patchSettings({ mode: 'terminal' });
+  });
+
   document.addEventListener('hex-display-change', (e) => {
     setHexDisplay(e.detail.on);
   });