frost-mqtt-client/app/src/preload/index.js

import { contextBridge, ipcRenderer } from 'electron'
import { electronAPI } from '@electron-toolkit/preload'

// Custom APIs for renderer
const api = {
  // 服务器管理
  getServers: () => ipcRenderer.invoke('mqtt:get-servers'),
  addServer: (config) => ipcRenderer.invoke('mqtt:add-server', config),
  updateServer: (id, config) => ipcRenderer.invoke('mqtt:update-server', id, config),
  deleteServer: (id) => ipcRenderer.invoke('mqtt:delete-server', id),

  // 连接管理
  connect: (id) => ipcRenderer.invoke('mqtt:connect', id),
  disconnect: (id) => ipcRenderer.invoke('mqtt:disconnect', id),

  // 主题管理
  addTopic: (serverId, topic, qos) => ipcRenderer.invoke('mqtt:add-topic', serverId, topic, qos),
  removeTopic: (serverId, topic) => ipcRenderer.invoke('mqtt:remove-topic', serverId, topic),
  subscribe: (serverId, topic) => ipcRenderer.invoke('mqtt:subscribe', serverId, topic),
  unsubscribe: (serverId, topic) => ipcRenderer.invoke('mqtt:unsubscribe', serverId, topic),
  subscribeAll: (serverId) => ipcRenderer.invoke('mqtt:subscribe-all', serverId),
  updateTopic: (serverId, topicId, topic, qos) =>
    ipcRenderer.invoke('mqtt:update-topic', serverId, topicId, topic, qos),

  // 消息发布
  publish: (serverId, topic, payload, opts) =>
    ipcRenderer.invoke('mqtt:publish', serverId, topic, payload, opts),

  // 消息管理
  getMessages: (serverId) => ipcRenderer.invoke('mqtt:get-messages', serverId),
  getPublishHistory: (serverId) => ipcRenderer.invoke('mqtt:get-publish-history', serverId),
  clearMessages: (serverId) => ipcRenderer.invoke('mqtt:clear-messages', serverId),
  clearPublishHistory: (serverId) => ipcRenderer.invoke('mqtt:clear-publish-history', serverId),

  // 事件监听
  onStatusChange: (callback) => {
    const handler = (_, data) => callback(data)
    ipcRenderer.on('mqtt:status-change', handler)
    return () => ipcRenderer.removeListener('mqtt:status-change', handler)
  },
  onMessage: (callback) => {
    const handler = (_, data) => callback(data)
    ipcRenderer.on('mqtt:message', handler)
    return () => ipcRenderer.removeListener('mqtt:message', handler)
  },
  onSubscriptionChange: (callback) => {
    const handler = (_, data) => callback(data)
    ipcRenderer.on('mqtt:subscription-change', handler)
    return () => ipcRenderer.removeListener('mqtt:subscription-change', handler)
  },
  onSubscriptionError: (callback) => {
    const handler = (_, data) => callback(data)
    ipcRenderer.on('mqtt:subscription-error', handler)
    return () => ipcRenderer.removeListener('mqtt:subscription-error', handler)
  },
  onPublishError: (callback) => {
    const handler = (_, data) => callback(data)
    ipcRenderer.on('mqtt:publish-error', handler)
    return () => ipcRenderer.removeListener('mqtt:publish-error', handler)
  }
}

// Use `contextBridge` APIs to expose Electron APIs to
// renderer only if context isolation is enabled, otherwise
// just add to the DOM global.
if (process.contextIsolated) {
  try {
    contextBridge.exposeInMainWorld('electron', electronAPI)
    contextBridge.exposeInMainWorld('api', api)
  } catch (error) {
    console.error(error)
  }
} else {
  window.electron = electronAPI
  window.api = api
}