实现客户端的单点登录

.env.shenranqi

@@ -1,4 +1,4 @@
-VITE_h5_client_url=http://10.9.18.138:18900
+VITE_h5_client_url=https://aiagent.szgas.com.cn
 ###  VITE_h5_client_url=https://bqw-120.ii999.live:20038
 #VITE_h5_client_url=http://127.0.0.1:18901
 VITE_HsAppCode=1

.idea/misc.xml

@@ -1,6 +1,6 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
-  <component name="ProjectRootManager" version="2" languageLevel="JDK_17" default="true" project-jdk-name="17" project-jdk-type="JavaSDK">
+  <component name="ProjectRootManager" version="2" languageLevel="JDK_1_8" default="true" project-jdk-name="17" project-jdk-type="JavaSDK">
     <output url="file://$PROJECT_DIR$/out" />
   </component>
 </project>

electron-builder.yml

@@ -1,4 +1,4 @@
-appId: com.huashiai.dify-market-manager-gui
+appId: com.szgas.dify-market-manager-gui
 compression: maximum
 productName: 深燃智能体平台
 directories:
@@ -12,7 +12,7 @@ files:
 asarUnpack:
   - resources/**
 win:
-  executableName: 深燃智能体平台
+  executableName: 深圳燃气智能体平台
 nsis:
   artifactName: ${name}-${version}-setup.${ext}
   shortcutName: ${productName}
@@ -32,7 +32,7 @@ linux:
   target:
     - AppImage
     - deb
-  maintainer: huashiai.org
+  maintainer: szgas.org
   category: Utility
 appImage:
   artifactName: ${name}-${version}.${ext}

package.json

@@ -1,10 +1,10 @@
 {
-  "name": "dify-market-manager-gui",
+  "name": "SZGAS_AIAgent",
   "version": "1.0.0",
-  "description": "百千万AI智能体共创平台",
+  "description": "深圳燃气 AI智能体",
   "main": "./out/main/index.js",
-  "author": "huashiai.com",
+  "author": "szgas.com",
-  "homepage": "https://huashiai.com",
+  "homepage": "https://szgas.com",
   "scripts": {
     "format": "prettier --write .",
     "lint": "eslint --cache .",
@@ -37,6 +37,24 @@
     "build:icon": "./node_modules/.bin/electron-icon-builder --input=./build/icon.png --output=./build/",
     "make": "electron-forge make"
   },
+  "build": {
+    "protocols": [{
+      "name": "aiAgent Protocol",
+      "schemes": ["aiAgent"]
+    }],
+    "win": {
+      "target": "nsis",
+      "signingHashAlgorithms": ["sha256"]
+    },
+    "mac": {
+      "extendInfo": {
+        "CFBundleURLTypes": [{
+          "CFBundleURLName": "MyApp Protocol",
+          "CFBundleURLSchemes": ["myapp"]
+        }]
+      }
+    }
+  },
   "permissions": [
     "audioCapture",
     "videoCapture"
@@ -51,6 +69,7 @@
     "electron-log": "^5.4.0",
     "electron-store": "^8.0.0",
     "electron-updater": "^6.3.9",
+    "jquery": "^3.7.1",
     "ws": "^8.18.2",
     "xe-utils": "^3.7.4"
   },

src/main/index.js

@@ -1,7 +1,7 @@
 import { app, shell, BrowserWindow, ipcMain, Menu, session, screen, dialog } from 'electron'
 import { electronApp, optimizer } from '@electron-toolkit/utils'
 import Store from 'electron-store'
-import { createWindow, createDrageWindow, unregisterAllShortcuts, getMainWindow } from './window.js'
+import { createWindow, createDrageWindow, unregisterAllShortcuts, getMainWindow,createOAuthWindow } from './window.js'
 import { setupIPC } from './ipc.js'
 import { createTray, destroyTray,clearBrowserCache } from './tray.js'
 import XEUtils from 'xe-utils'
@@ -15,6 +15,8 @@ import { setStoreValue, getStoreValue ,deleteStore} from './store.js'
 import AutoLaunch from 'auto-launch'
 import axios from 'axios'
 
+const PROTOCOL_NAME = 'aiAgent'
+
 import WebSocketClient from './utils/WebSocketClient';
 
 let wsClient=null
@@ -163,6 +165,81 @@ function cleanupCacheDirectories() {
   }
 }
 
+/**
+ * 注册自定义协议为系统默认处理程序
+ */
+function registerDeepLinkProtocol() {
+  try {
+    if (process.platform === 'win32') {
+      if (process.defaultApp) {
+        app.setAsDefaultProtocolClient(PROTOCOL_NAME, process.execPath, [path.resolve(process.argv[1])])
+      } else {
+        app.setAsDefaultProtocolClient(PROTOCOL_NAME)
+      }
+    } else {
+      app.setAsDefaultProtocolClient(PROTOCOL_NAME)
+    }
+    logger.info(`已注册自定义协议: ${PROTOCOL_NAME}`)
+  } catch (error) {
+    logger.error('注册自定义协议失败:', error)
+  }
+}
+
+/**
+ * 从命令行参数提取深链地址与 code 参数
+ * 返回对象 { url, code }；未找到返回 null
+ */
+function extractDeepLinkFromArgs(args = []) {
+  try {
+    const protoPrefix = `${PROTOCOL_NAME.toLowerCase()}://`
+    const found = (args || []).find(
+      (arg) => typeof arg === 'string' && arg.toLowerCase().startsWith(protoPrefix)
+    )
+    if (!found) return null
+
+    let code = null
+    try {
+      const u = new URL(found)
+      const params = new URLSearchParams(u.search || '')
+      code = params.get('code')
+    } catch (e) {
+      const m = found.match(/[?&]code=([^&]+)/i)
+      code = m ? decodeURIComponent(m[1]) : null
+    }
+
+    return { url: found, code }
+  } catch (error) {
+    logger.error('提取深链地址失败:', error)
+    return null
+  }
+}
+
+/**
+ * 处理深链地址
+ */
+function handleDeepLink(payload) {
+  try {
+    const url = typeof payload === 'string' ? payload : payload?.url
+    const code = typeof payload === 'object' ? payload?.code : null
+    logger.info('Deep link URL:', url)
+    if (code) {
+      logger.info('Deep link code:', code)
+    }
+    const mainWindow = BrowserWindow.getAllWindows()[0] || getMainWindow()
+    if (mainWindow) {
+      const h5_client_url=getStoreValue("h5_client_url")+"/pc_client/autologin?code="+code
+
+      logger.info("==================================== mainWindow.loadURL:"+h5_client_url)
+      // 加载存储的 URL
+      mainWindow.loadURL(h5_client_url)
+      mainWindow.show()
+      mainWindow.focus()
+    }
+  } catch (error) {
+    logger.error('处理深链地址失败:', error)
+  }
+}
+
 /**
  * 单应用启动实现
  * 请求单一实例锁，
@@ -181,6 +258,10 @@ if (!gotTheLock) {
   // 监听第二个实例被运行时
   app.on('second-instance', (event, commandLine, workingDirectory) => {
     logger.info('检测到第二个实例启动，激活现有实例')
+    const deeplink = extractDeepLinkFromArgs(commandLine)
+    if (deeplink) {
+      handleDeepLink(deeplink)
+    }
     // 当有第二个实例被运行时，激活之前的实例并将焦点置于其窗口
     const windows = BrowserWindow.getAllWindows()
     if (windows.length > 0) {
@@ -196,6 +277,8 @@ if (!gotTheLock) {
   app.whenReady().then(() => {
     logger.info('应用已准备就绪，开始初始化...');
 
+    registerDeepLinkProtocol()
+
     // 清理缓存目录
     cleanupCacheDirectories()
 
@@ -213,6 +296,8 @@ if (!gotTheLock) {
       optimizer.watchWindowShortcuts(window)
     })
 
+
+
     // 根据存储的状态决定是否创建悬浮窗口
     if (getStoreValue('showDrageWindow')) {
        createDrageWindow()
@@ -247,8 +332,20 @@ if (!gotTheLock) {
     })
 
     logger.info('应用初始化完成');
+
+    const initialDeeplink = extractDeepLinkFromArgs(process.argv)
+    if (initialDeeplink) {
+      handleDeepLink(initialDeeplink)
+    }
   })
 
+    app.on('open-url', (event, url) => {
+      logger.info('Opened with URL:', url)
+      event.preventDefault()
+      // 处理url，例如解析参数并打开相应窗口
+      logger.info('Opened with URL:', url)
+    })
+
   // 修改窗口关闭行为
   app.on('window-all-closed',  async (event) => {
     if (process.platform !== 'darwin') {

src/main/tray.js

@@ -90,7 +90,7 @@ export function createTray() {
   tray = new Tray(icon)
 
   // 设置托盘图标的提示文本
-  tray.setToolTip('Dify Market Manager')
+  tray.setToolTip('深圳燃气 AI智能体')
 
   // 创建右键菜单
   const contextMenu = Menu.buildFromTemplate([

src/main/utils/WebSocketClient.js

@@ -81,7 +81,7 @@ class WebSocketClient {
 
     setTimeout(() => {
       this.lockReconnect = false;
-      this.createConnect(); // 重新获取本地缓存并创建连接
+      // this.createConnect(); // 重新获取本地缓存并创建连接
     }, this.reconnectInterval);
   }
 

src/main/window.js

@@ -13,6 +13,7 @@ import {checkForUpdates} from "./utils/updateUtils"
 import dayjs from 'dayjs'
 
 let mainWindow = null
+let oauthWindow = null
 let difyfullScreenWindow = null
 let drageWindow = null
 let apiConfigWindow = null
@@ -51,6 +52,49 @@ async function checkAndApplyDeviceAccessPrivilege() {
   }
 }
 
+export async function createOAuthWindow(){
+  logger.info('创建授权窗口')
+
+  Menu.setApplicationMenu(null)
+
+  // Create the browser window.
+  oauthWindow = new BrowserWindow({
+    width: 1480,
+    height: 980,
+    minWidth: 1480,
+    minHeight: 980,
+    show: false,
+    media: {
+      audio: true,
+      video: true
+    },
+    icon: icon,
+    webPreferences: {
+      nodeIntegration: false,
+      nodeIntegrationInWorker: true,
+      preload: join(__dirname, '../preload/index.js'),
+      // sandbox: false
+    }
+  })
+
+  // 检查本地存储中是否已有 h5_client_url
+  const existingH5ClientUrl = getStoreValue("h5_client_url")
+  if (!existingH5ClientUrl || existingH5ClientUrl.trim() === '') {
+    // 只有当值不存在或为空时，才设置默认值
+    // setStoreValue("h5_client_url", "http://work.ii999.live:20040")
+    // setStoreValue("h5_client_url", "http://10.102.8.56:18900")
+    // setStoreValue("h5_client_url", "http://68.66.24.160:18900")
+    setStoreValue("h5_client_url", import.meta.env.VITE_h5_client_url)
+  }
+
+  const h5_client_url=getStoreValue("h5_client_url")+"/pc_client/autologin"
+
+  logger.info("==================================== mainWindow.loadURL:"+h5_client_url)
+  // 加载存储的 URL
+  oauthWindow.loadURL(h5_client_url)
+
+
+}
 
 export async function createWindow()  {
   // 如果窗口已经存在，直接显示并返回
@@ -63,6 +107,9 @@ export async function createWindow()  {
 
   Menu.setApplicationMenu(null)
 
+  // createOAuthWindow()
+  // return
+
   // Create the browser window.
   mainWindow = new BrowserWindow({
     width: 1480,
@@ -300,6 +347,9 @@ function registerShortcuts(window=null) {
   app.on('will-quit', () => {
     globalShortcut.unregisterAll() // 注销所有快捷键
   })
+
+
+
 }
 
 export async function createNewWindow(url, access_token, refresh_token,sandbox=false) {

src/preload/index.js

@@ -1,21 +1,16 @@
-import { contextBridge , ipcRenderer} from 'electron'
+import { contextBridge, ipcRenderer } from 'electron'
 import { electronAPI } from '@electron-toolkit/preload'
 
 // Custom APIs for renderer
 const api = {}
 
-// 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)
-
+    ipcRenderer.on('data-from-main', (event, key, value) => {
-    ipcRenderer.on('data-from-main', (event, key,value) => {
+      localStorage.setItem(key, value)
-      localStorage.setItem(key, value);
+    })
-    });
-
   } catch (error) {
     console.error(error)
   }
@@ -23,5 +18,26 @@ if (process.contextIsolated) {
   window.electron = electronAPI
   window.api = api
 
-
+  // ✅ 重点修复：覆盖 require('jquery') 让它返回 jQuery
+  try {
+    // 保存原始 require 避免循环
+    const originalRequire = require;
+    // 重写 require 函数，让 'jquery' 请求返回 jQuery
+    require = (request) => {
+      if (request === 'jquery') {
+        // 确保 jQuery 已加载
+        if (!window.jQuery) {
+          window.jQuery = originalRequire('jquery');
+          window.$ = window.jQuery;
+        }
+        return window.jQuery;
+      }
+      return originalRequire(request);
+    };
+    // 初始化 jQuery（触发上面的逻辑）
+    window.jQuery = require('jquery');
+    window.$ = window.jQuery;
+  } catch (e) {
+    console.error('Failed to load jQuery:', e);
+  }
 }