@electerm/electerm-react 2.7.9 → 2.8.6

package/client/components/terminal/zmodem-client.js

@@ -0,0 +1,385 @@
+/**
+ * Zmodem client handler for web terminal
+ * Handles UI interactions and communicates with server-side zmodem
+ */
+
+import { throttle } from 'lodash-es'
+import { filesize } from 'filesize'
+import { transferTypeMap } from '../../common/constants.js'
+import { getLocalFileInfo } from '../sftp/file-read.js'
+
+const ZMODEM_SAVE_PATH_KEY = 'zmodem-save-path'
+
+/**
+ * ZmodemClient class handles zmodem UI and client-side logic
+ */
+export class ZmodemClient {
+  constructor (terminal) {
+    this.terminal = terminal
+    this.socket = null
+    this.isActive = false
+    this.currentTransfer = null
+    this.savePath = null
+    this.transferStartTime = 0
+  }
+
+  /**
+   * Initialize zmodem client with socket
+   * @param {WebSocket} socket - WebSocket connection
+   */
+  init (socket) {
+    this.socket = socket
+    this.setupMessageHandler()
+  }
+
+  /**
+   * Setup message handler for zmodem events from server
+   */
+  setupMessageHandler () {
+    if (!this.socket) return
+
+    // Use addEventListener to avoid conflicting with attach-addon's onmessage
+    this.messageHandler = (event) => {
+      // Check if it's a JSON message (zmodem control message)
+      if (typeof event.data === 'string') {
+        try {
+          const msg = JSON.parse(event.data)
+          if (msg.action === 'zmodem-event') {
+            this.handleServerEvent(msg)
+          }
+        } catch (e) {
+          // Not JSON, ignore
+        }
+      }
+    }
+    this.socket.addEventListener('message', this.messageHandler)
+  }
+
+  /**
+   * Handle server zmodem events
+   * @param {Object} msg - Message from server
+   */
+  handleServerEvent (msg) {
+    const { event } = msg
+
+    switch (event) {
+      case 'receive-start':
+        this.onReceiveStart()
+        break
+      case 'send-start':
+        this.onSendStart()
+        break
+      case 'file-start':
+        this.onFileStart(msg.name, msg.size)
+        break
+      case 'file-prepared':
+        this.onFilePrepared(msg.name, msg.path, msg.size)
+        break
+      case 'progress':
+        this.onProgress(msg)
+        break
+      case 'file-complete':
+        this.onFileComplete(msg.name, msg.path)
+        break
+      case 'file-skipped':
+        this.onFileSkipped(msg.name, msg.message)
+        break
+      case 'session-end':
+        this.onSessionEnd()
+        break
+    }
+  }
+
+  /**
+   * Send message to server
+   * @param {Object} msg - Message to send
+   */
+  sendToServer (msg) {
+    if (this.socket && this.socket.readyState === WebSocket.OPEN) {
+      this.socket.send(JSON.stringify({
+        action: 'zmodem-event',
+        ...msg
+      }))
+    }
+  }
+
+  /**
+   * Handle receive session start
+   */
+  async onReceiveStart () {
+    this.isActive = true
+    this.writeBanner('RECEIVE')
+
+    // Ask user for save directory
+    const savePath = await this.openSaveFolderSelect()
+    if (savePath) {
+      this.savePath = savePath
+      this.sendToServer({
+        event: 'set-save-path',
+        path: savePath
+      })
+    } else {
+      // User cancelled, end session
+      this.sendToServer({
+        event: 'cancel'
+      })
+      this.onSessionEnd()
+    }
+  }
+
+  /**
+   * Handle send session start
+   */
+  async onSendStart () {
+    this.isActive = true
+    this.writeBanner('SEND')
+
+    // Ask user to select files
+    const files = await this.openFileSelect()
+    if (files && files.length > 0) {
+      this.sendToServer({
+        event: 'send-files',
+        files
+      })
+    } else {
+      // User cancelled, end session
+      this.sendToServer({
+        event: 'cancel'
+      })
+      this.onSessionEnd()
+    }
+  }
+
+  /**
+   * Handle file start event
+   * @param {string} name - File name
+   * @param {number} size - File size
+   */
+  onFileStart (name, size) {
+    this.currentTransfer = {
+      name,
+      size,
+      transferred: 0,
+      type: this.savePath ? transferTypeMap.download : transferTypeMap.upload,
+      path: null
+    }
+    this.transferStartTime = Date.now()
+    this.writeProgress()
+  }
+
+  /**
+   * Handle file prepared event (for receive)
+   * @param {string} name - File name
+   * @param {string} path - File path
+   * @param {number} size - File size
+   */
+  onFilePrepared (name, path, size) {
+    // Store the full path for display
+    if (this.currentTransfer) {
+      this.currentTransfer.path = path
+    }
+  }
+
+  /**
+   * Handle progress update
+   * @param {Object} msg - Progress message
+   */
+  onProgress (msg) {
+    if (!this.currentTransfer) {
+      this.currentTransfer = {
+        name: msg.name,
+        size: msg.size,
+        transferred: 0,
+        type: msg.type,
+        path: msg.path || null
+      }
+      this.transferStartTime = Date.now()
+    }
+
+    this.currentTransfer.transferred = msg.transferred
+    this.currentTransfer.serverSpeed = msg.speed // Use server's speed calculation
+    this.currentTransfer.path = msg.path || this.currentTransfer.path
+    this.writeProgress()
+  }
+
+  /**
+   * Handle file complete
+   * @param {string} name - File name
+   * @param {string} path - File path
+   */
+  onFileComplete (name, path) {
+    if (this.currentTransfer) {
+      this.currentTransfer.transferred = this.currentTransfer.size
+      this.currentTransfer.path = path
+      // Call directly to ensure 100% is displayed immediately
+      this._doWriteProgress(true)
+      // Add newline after completion
+      this.writeToTerminal('\r\n')
+    }
+    this.currentTransfer = null
+  }
+
+  /**
+   * Handle file skipped (already exists on remote side)
+   * @param {string} name - File name
+   * @param {string} message - Skip message
+   */
+  onFileSkipped (name, message) {
+    this.writeToTerminal(`\r\n\x1b[33m\x1b[1mSKIPPED: ${name} - ${message}\x1b[0m\r\n`)
+    this.currentTransfer = null
+  }
+
+  /**
+   * Handle session end
+   */
+  onSessionEnd () {
+    this.isActive = false
+    this.currentTransfer = null
+    this.savePath = null
+    if (this.terminal && this.terminal.term) {
+      this.terminal.term.focus()
+      this.terminal.term.write('\r\n')
+    }
+  }
+
+  /**
+   * Write banner to terminal
+   * @param {string} type - 'RECEIVE' or 'SEND'
+   */
+  writeBanner (type) {
+    const border = '='.repeat(50)
+    this.writeToTerminal(`\r\n${border}\r\n`)
+    this.writeToTerminal('\x1b[33m\x1b[1mRecommend use trzsz instead: https://github.com/trzsz/trzsz\x1b[0m\r\n')
+    this.writeToTerminal(`${border}\r\n\r\n`)
+    this.writeToTerminal(`\x1b[32m\x1b[1mZMODEM::${type}::START\x1b[0m\r\n`)
+  }
+
+  /**
+   * Write progress to terminal (throttled for updates, immediate for completion)
+   * @param {boolean} isComplete - Whether this is the final completion display
+   */
+  writeProgress = throttle((isComplete = false) => {
+    this._doWriteProgress(isComplete)
+  }, 500)
+
+  /**
+   * Internal function to actually write progress
+   * @param {boolean} isComplete - Whether this is the final completion display
+   */
+  _doWriteProgress (isComplete = false) {
+    if (!this.currentTransfer || !this.terminal?.term) return
+
+    const { name, size, transferred, path, serverSpeed } = this.currentTransfer
+    const percent = size > 0 ? Math.floor(transferred * 100 / size) : 100
+
+    // Use server's speed if available, otherwise calculate locally
+    const speed = serverSpeed || 0
+
+    // Use full path if available, otherwise just name
+    const displayName = path || name
+
+    // filesize expects bytes and formats to human readable
+    const formatSize = (bytes) => filesize(bytes)
+
+    // Clear line and write progress
+    const str = `\r\x1b[2K\x1b[32m${displayName}\x1b[0m: ${percent}%, ${formatSize(transferred)}/${formatSize(size)}, ${formatSize(speed)}/s${isComplete ? ' \x1b[32m\x1b[1m[DONE]\x1b[0m' : ''}`
+    this.writeToTerminal(str + '\r')
+  }
+
+  /**
+   * Write text to terminal
+   * @param {string} text - Text to write
+   */
+  writeToTerminal (text) {
+    if (this.terminal && this.terminal.term) {
+      this.terminal.term.write(text)
+    }
+  }
+
+  /**
+   * Open file select dialog
+   * @returns {Promise<Array>} - Selected files
+   */
+  openFileSelect = async () => {
+    const properties = [
+      'openFile',
+      'multiSelections',
+      'showHiddenFiles',
+      'noResolveAliases',
+      'treatPackageAsDirectory',
+      'dontAddToRecent'
+    ]
+    const files = await window.api.openDialog({
+      title: 'Choose some files to send',
+      message: 'Choose some files to send',
+      properties
+    }).catch(() => false)
+    if (!files || !files.length) {
+      return null
+    }
+    const r = []
+    for (const filePath of files) {
+      const stat = await getLocalFileInfo(filePath)
+      r.push({ ...stat, filePath, path: filePath })
+    }
+    return r
+  }
+
+  /**
+   * Open save folder select dialog
+   * @returns {Promise<string>} - Selected folder path
+   */
+  openSaveFolderSelect = async () => {
+    // Try to use last saved path
+    const lastPath = window.localStorage.getItem(ZMODEM_SAVE_PATH_KEY)
+
+    const savePaths = await window.api.openDialog({
+      title: 'Choose a folder to save file(s)',
+      message: 'Choose a folder to save file(s)',
+      defaultPath: lastPath || undefined,
+      properties: [
+        'openDirectory',
+        'showHiddenFiles',
+        'createDirectory',
+        'noResolveAliases',
+        'treatPackageAsDirectory',
+        'dontAddToRecent'
+      ]
+    }).catch(() => false)
+    if (!savePaths || !savePaths.length) {
+      return null
+    }
+    // Save for next time
+    window.localStorage.setItem(ZMODEM_SAVE_PATH_KEY, savePaths[0])
+    return savePaths[0]
+  }
+
+  /**
+   * Cancel ongoing transfer
+   */
+  cancel () {
+    if (!this.isActive) return
+    this.writeToTerminal('\r\n\x1b[33m\x1b[1mZMODEM transfer cancelled by user\x1b[0m\r\n')
+    this.sendToServer({
+      event: 'cancel'
+    })
+    this.onSessionEnd()
+  }
+
+  /**
+   * Clean up resources
+   */
+  destroy () {
+    if (this.socket && this.messageHandler) {
+      this.socket.removeEventListener('message', this.messageHandler)
+      this.messageHandler = null
+    }
+    this.isActive = false
+    this.currentTransfer = null
+    this.savePath = null
+    this.socket = null
+    this.terminal = null
+  }
+}
+
+export default ZmodemClient

package/client/components/terminal-info/data-cols-parser.jsx

@@ -1,6 +1,6 @@
 import { copy } from '../../common/clipboard'
 import filesizeParser from 'filesize-parser'
-import { formatBytes } from '../../common/byte-format'
+import { filesize } from 'filesize'
 
 const valueParserMaps = {
   size: v => v,
@@ -23,7 +23,8 @@ function copyValue (event) {
 export default (data) => {
   return Object.keys(data).map(k => {
     const rd = (txt) => {
-      const r = k === 'mem' ? formatBytes(parseInt(txt, 10)) : txt
+      // txt is in KB, convert to bytes for filesize function
+      const r = k === 'mem' ? filesize(parseInt(txt, 10) * 1024) : txt
       const itemProps = {
         className: 'activity-item pointer',
         'data-content': r,

package/client/components/terminal-info/network.jsx

@@ -5,7 +5,7 @@
 import { Table } from 'antd'
 import { isEmpty } from 'lodash-es'
 import { useEffect, useState } from 'react'
-import { formatBytes } from '../../common/byte-format'
+import { filesize } from 'filesize'
 import copy from 'json-deep-copy'
 import { ApiOutlined } from '@ant-design/icons'
 
@@ -92,7 +92,8 @@ export default function TerminalInfoDisk (props) {
       },
       render: (v) => {
         if (k === 'up' || k === 'down') {
-          return formatBytes(v)
+          console.log('render traffic', k, v)
+          return filesize(v || 0)
         }
         return v
       }