eoss-ui 0.6.38 → 0.6.40

package/src/utils/util.js

@@ -26,6 +26,7 @@ const $darkColorCount = 4;
 const publicKey = 'abcdefghijklmnopqrstuvwxyz12345678';
 let loginMsg;
 let initAuth;
+let $_http;
 
 const JSONbigToString = JSONbig({ storeAsString: true });
 import { MessageBox, Loading } from 'eoss-element';
@@ -89,290 +90,293 @@ const ajax = function ({
   } else {
     method = method.toLowerCase();
   }
-  let baseURL = host === true ? getStorage('host') : host === false ? '' : host;
-  let http = axios.create({
-    baseURL: baseURL,
-    timeout: 99000,
-    transformResponse: [
-      function (data) {
-        try {
-          return JSONbigToString.parse(data);
-        } catch (err) {
-          return data;
+  let http = $_http;
+  if (!http) {
+    let baseURL = host === true ? getStorage('host') : host === false ? '' : host;
+    http = axios.create({
+      baseURL: baseURL,
+      timeout: 99000,
+      transformResponse: [
+        function (data) {
+          try {
+            return JSONbigToString.parse(data);
+          } catch (err) {
+            return data;
+          }
         }
+      ]
+    });
+    // 请求拦截
+    http.interceptors.request.use(
+      (config) => {
+        config.headers.common = getStorage();
+        // 根据请求的信息(请求方式,url,请求get/post数据),产生map的key
+        // let requestKey = getRequestKey(config);
+        // // 判断请求是否重复
+        // if (pendingRequest.has(requestKey)) {
+        //   // 取消上次请求
+        //   let cancel = pendingRequest.get(requestKey);
+        //   cancel();
+        //   // 删除请求信息
+        //   pendingRequest.delete(requestKey);
+        // }
+        // // 把请求信息,添加请求到map当中
+        // // 生成取消方法
+        // config.cancelToken =
+        //   config.cancelToken ||
+        //   new axios.CancelToken(cancel => {
+        //     // 把取消方法添加到map
+        //     if (!pendingRequest.has(requestKey)) {
+        //       pendingRequest.set(requestKey, cancel);
+        //     }
+        //   });
+        return config;
+      },
+      (err) => {
+        return err;
       }
-    ]
-  });
-  // 请求拦截
-  http.interceptors.request.use(
-    (config) => {
-      config.headers.common = getStorage();
-      // 根据请求的信息(请求方式,url,请求get/post数据),产生map的key
-      // let requestKey = getRequestKey(config);
-      // // 判断请求是否重复
-      // if (pendingRequest.has(requestKey)) {
-      //   // 取消上次请求
-      //   let cancel = pendingRequest.get(requestKey);
-      //   cancel();
-      //   // 删除请求信息
-      //   pendingRequest.delete(requestKey);
-      // }
-      // // 把请求信息,添加请求到map当中
-      // // 生成取消方法
-      // config.cancelToken =
-      //   config.cancelToken ||
-      //   new axios.CancelToken(cancel => {
-      //     // 把取消方法添加到map
-      //     if (!pendingRequest.has(requestKey)) {
-      //       pendingRequest.set(requestKey, cancel);
-      //     }
-      //   });
-      return config;
-    },
-    (err) => {
-      return err;
-    }
-  );
-  // 响应拦截
-  http.interceptors.response.use(
-    (response) => {
-      // delPendingRequest(response.config);
-      loading && loading.close();
-      if (
-        response.headers.authorization &&
-        response.headers.authorization !== getStorage('token')
-      ) {
+    );
+    // 响应拦截
+    http.interceptors.response.use(
+      (response) => {
+        // delPendingRequest(response.config);
+        loading && loading.close();
         if (
-          localStorage.getItem('storage') === 'localStorage' ||
-          localStorage.getItem('token') ||
-          localStorage.getItem('Authorization')
+          response.headers.authorization &&
+          response.headers.authorization !== getStorage('token')
         ) {
-          localStorage.setItem('token', response.headers.authorization);
-          localStorage.setItem('Authorization', response.headers.authorization);
-        } else {
-          sessionStorage.setItem('token', response.headers.authorization);
-          sessionStorage.setItem(
-            'Authorization',
-            response.headers.authorization
-          );
+          if (
+            localStorage.getItem('storage') === 'localStorage' ||
+            localStorage.getItem('token') ||
+            localStorage.getItem('Authorization')
+          ) {
+            localStorage.setItem('token', response.headers.authorization);
+            localStorage.setItem('Authorization', response.headers.authorization);
+          } else {
+            sessionStorage.setItem('token', response.headers.authorization);
+            sessionStorage.setItem(
+              'Authorization',
+              response.headers.authorization
+            );
+          }
         }
-      }
-      if (
-        (response.headers.ssId ||
-          response.headers.Ssid ||
-          response.headers.ssid) &&
-        (response.headers.ssId ||
-          response.headers.SsId ||
-          response.headers.ssid) !== getStorage('ssId')
-      ) {
         if (
-          localStorage.getItem('storage') === 'localStorage' ||
-          localStorage.getItem('ssId')
+          (response.headers.ssId ||
+            response.headers.Ssid ||
+            response.headers.ssid) &&
+          (response.headers.ssId ||
+            response.headers.SsId ||
+            response.headers.ssid) !== getStorage('ssId')
         ) {
-          localStorage.setItem(
-            'ssId',
-            response.headers.ssId ||
+          if (
+            localStorage.getItem('storage') === 'localStorage' ||
+            localStorage.getItem('ssId')
+          ) {
+            localStorage.setItem(
+              'ssId',
+              response.headers.ssId ||
               response.headers.SsId ||
               response.headers.ssid
-          );
-        } else {
-          sessionStorage.setItem(
-            'ssId',
-            response.headers.ssId ||
+            );
+          } else {
+            sessionStorage.setItem(
+              'ssId',
+              response.headers.ssId ||
               response.headers.SsId ||
               response.headers.ssid
-          );
-        }
-      }
-      if (response.status === 200) {
-        if (
-          response.data.rCode === 64 ||
-          response.data.rCode === 65 ||
-          response.data.rCode === 67 ||
-          response.data.rCode === 69
-        ) {
-          removeStorage([
-            'Authorization',
-            'token',
-            'ssId',
-            'userId',
-            'userName',
-            'deviceUnique',
-            'menus',
-            'useCaseCodes',
-            'mainConfig',
-            'jump',
-            'showLaunch'
-          ]);
-          let href = '';
-          try {
-            href = win.top.location.href;
-          } catch (error) {
-            href = win.location.href;
+            );
           }
+        }
+        if (response.status === 200) {
           if (
-            href.indexOf('#/login') === -1 &&
-            href.indexOf('/login.html') === -1
+            response.data.rCode === 64 ||
+            response.data.rCode === 65 ||
+            response.data.rCode === 67 ||
+            response.data.rCode === 69
           ) {
-            let remind = sessionStorage.getItem('remind');
-            if (!remind) {
-              clearTimeout(loginMsg);
-              loginMsg = setTimeout(() => {
-                sessionStorage.setItem('remind', 1);
-                MessageBox.confirm(
-                  response.data.rCode === 69
-                    ? '该账号在其他地方已登陆!'
-                    : '登录已过期，请重新登录!',
-                  '提示',
-                  response.data.rCode === 69
-                    ? {
+            removeStorage([
+              'Authorization',
+              'token',
+              'ssId',
+              'userId',
+              'userName',
+              'deviceUnique',
+              'menus',
+              'useCaseCodes',
+              'mainConfig',
+              'jump',
+              'showLaunch'
+            ]);
+            let href = '';
+            try {
+              href = win.top.location.href;
+            } catch (error) {
+              href = win.location.href;
+            }
+            if (
+              href.indexOf('#/login') === -1 &&
+              href.indexOf('/login.html') === -1
+            ) {
+              let remind = sessionStorage.getItem('remind');
+              if (!remind) {
+                clearTimeout(loginMsg);
+                loginMsg = setTimeout(() => {
+                  sessionStorage.setItem('remind', 1);
+                  MessageBox.confirm(
+                    response.data.rCode === 69
+                      ? '该账号在其他地方已登陆!'
+                      : '登录已过期，请重新登录!',
+                    '提示',
+                    response.data.rCode === 69
+                      ? {
                         confirmButtonText: '确定',
                         closeOnClickModal: false,
                         type: 'warning'
                       }
-                    : {
+                      : {
                         confirmButtonText: '确定',
                         cancelButtonText: '取消',
                         closeOnClickModal: false,
                         type: 'warning'
                       }
-                )
-                  .then(() => {
-                    try {
-                      const loginPage =
-                        getStorage('login') || getStorage('loginPage');
-                      if (loginPage) {
-                        let src;
-                        if (!startWith(loginPage, ['http', '/'], true)) {
-                          let pathname = win.top.location.pathname;
-                          if (pathname !== '/') {
-                            pathname = pathname.split('/');
-                            pathname.splice(pathname.length - 1);
-                            pathname = pathname.join('/');
-                            src = pathname + '/' + loginPage.replace('./', '');
+                  )
+                    .then(() => {
+                      try {
+                        const loginPage =
+                          getStorage('login') || getStorage('loginPage');
+                        if (loginPage) {
+                          let src;
+                          if (!startWith(loginPage, ['http', '/'], true)) {
+                            let pathname = win.top.location.pathname;
+                            if (pathname !== '/') {
+                              pathname = pathname.split('/');
+                              pathname.splice(pathname.length - 1);
+                              pathname = pathname.join('/');
+                              src = pathname + '/' + loginPage.replace('./', '');
+                            } else {
+                              src = pathname + loginPage.replace('./', '');
+                            }
                           } else {
-                            src = pathname + loginPage.replace('./', '');
+                            src = loginPage;
                           }
+                          win.top.location.href = src;
+                        } else if (
+                          win.top.location.href.indexOf('main.html') > -1
+                        ) {
+                          win.top.location.href = './login.html';
                         } else {
-                          src = loginPage;
-                        }
-                        win.top.location.href = src;
-                      } else if (
-                        win.top.location.href.indexOf('main.html') > -1
-                      ) {
-                        win.top.location.href = './login.html';
-                      } else {
-                        const hash = win.top.location.hash;
-                        if (hash) {
-                          const len = win.top.location.href.indexOf(hash);
-                          win.top.location.href =
-                            win.location.href.slice(0, len) + '#/login';
-                        } else {
-                          win.top.location.href = '/login.html';
+                          const hash = win.top.location.hash;
+                          if (hash) {
+                            const len = win.top.location.href.indexOf(hash);
+                            win.top.location.href =
+                              win.location.href.slice(0, len) + '#/login';
+                          } else {
+                            win.top.location.href = '/login.html';
+                          }
                         }
+                      } catch (error) {
+                        win.postMessage({ type: 1 }, '*');
                       }
-                    } catch (error) {
-                      win.postMessage({ type: 1 }, '*');
+                    })
+                    .catch((e) => {
+                      sessionStorage.removeItem('remind');
+                    });
+                }, 2000);
+              }
+            }
+          } else if (response.data.rCode === 61) {
+            clearTimeout(initAuth);
+            initAuth = setTimeout(() => {
+              MessageBox({
+                title: '请进行二次鉴权',
+                closeOnClickModal: false,
+                closeOnPressEscape: false,
+                showConfirmButton: false,
+                dangerouslyUseHTMLString: true,
+                scroll: false,
+                customClass: 'es-retrial-auth-msg',
+                render: (h, msgBox) => {
+                  return h('es-retrial-auth', {
+                    ref: 'auth',
+                    props: {
+                      reload:
+                        response.data.results.reload ||
+                        response.data.results.refresh,
+                      group: response.data.results.retrialAuthGroupIds,
+                      type: response.data.results.retrialAuthType,
+                      msgBox: msgBox
                     }
-                  })
-                  .catch((e) => {
-                    sessionStorage.removeItem('remind');
                   });
-              }, 2000);
-            }
+                },
+                callback: (res, obj) => {
+                  obj.$children.forEach((item) => {
+                    item.clearTimeouts && item.clearTimeouts();
+                    return;
+                  });
+                }
+              });
+            }, 1000);
           }
-        } else if (response.data.rCode === 61) {
-          clearTimeout(initAuth);
-          initAuth = setTimeout(() => {
-            MessageBox({
-              title: '请进行二次鉴权',
-              closeOnClickModal: false,
-              closeOnPressEscape: false,
-              showConfirmButton: false,
-              dangerouslyUseHTMLString: true,
-              scroll: false,
-              customClass: 'es-retrial-auth-msg',
-              render: (h, msgBox) => {
-                return h('es-retrial-auth', {
-                  ref: 'auth',
-                  props: {
-                    reload:
-                      response.data.results.reload ||
-                      response.data.results.refresh,
-                    group: response.data.results.retrialAuthGroupIds,
-                    type: response.data.results.retrialAuthType,
-                    msgBox: msgBox
-                  }
-                });
-              },
-              callback: (res, obj) => {
-                obj.$children.forEach((item) => {
-                  item.clearTimeouts && item.clearTimeouts();
-                  return;
-                });
-              }
-            });
-          }, 1000);
+          // eslint-disable-next-line no-undef
+          return Promise.resolve(JSON.parse(JSON.stringify(response.data)));
         }
-        // eslint-disable-next-line no-undef
-        return Promise.resolve(JSON.parse(JSON.stringify(response.data)));
-      }
-    },
-    (error) => {
-      loading && loading.close();
-      // if (!axios.isCancel(error)) {
-      //   delPendingRequest(error.config || {});
-      // }
-      if (error && error.response) {
-        if (error.response.status) {
-          switch (error.response.status) {
-            case 400:
-              error.message = '错误请求';
-              break;
-            case 401:
-              error.message = '未授权，请重新登录';
-              break;
-            case 403:
-              error.message = '服务器错误';
-              break;
-            case 404:
-              error.message = '网络请求不存在';
-              break;
-            case 405:
-              error.message = '请求方法错误';
-              break;
-            case 408:
-              error.message = '请求超时';
-              break;
-            case 500:
-              error.message = '服务器错误';
-              break;
-            case 501:
-              error.message = '网络未实现';
-              break;
-            case 502:
-              error.message = '网络错误';
-              break;
-            case 503:
-              error.message = '服务不可用';
-              break;
-            case 504:
-              error.message = '网络超时';
-              break;
-            case 505:
-              error.message = 'http版本不支持该请求';
-              break;
-            default:
-              error.message = `连接错误${error.response.status}`;
+      },
+      (error) => {
+        loading && loading.close();
+        // if (!axios.isCancel(error)) {
+        //   delPendingRequest(error.config || {});
+        // }
+        if (error && error.response) {
+          if (error.response.status) {
+            switch (error.response.status) {
+              case 400:
+                error.message = '错误请求';
+                break;
+              case 401:
+                error.message = '未授权，请重新登录';
+                break;
+              case 403:
+                error.message = '服务器错误';
+                break;
+              case 404:
+                error.message = '网络请求不存在';
+                break;
+              case 405:
+                error.message = '请求方法错误';
+                break;
+              case 408:
+                error.message = '请求超时';
+                break;
+              case 500:
+                error.message = '服务器错误';
+                break;
+              case 501:
+                error.message = '网络未实现';
+                break;
+              case 502:
+                error.message = '网络错误';
+                break;
+              case 503:
+                error.message = '服务不可用';
+                break;
+              case 504:
+                error.message = '网络超时';
+                break;
+              case 505:
+                error.message = 'http版本不支持该请求';
+                break;
+              default:
+                error.message = `连接错误${error.response.status}`;
+            }
+          } else {
+            error.message = '无法请求,请联系管理员';
           }
-        } else {
-          error.message = '无法请求,请联系管理员';
         }
+        // eslint-disable-next-line no-undef
+        return Promise.reject(error);
       }
-      // eslint-disable-next-line no-undef
-      return Promise.reject(error);
-    }
-  );
+    );
+  }
   let userId = getStorage('userId');
   if (
     !Object.prototype.hasOwnProperty.call(data, 'userId') &&
@@ -481,7 +485,177 @@ const ajax = function ({
     ...configs
   });
 };
+/**
+ * ajaxStream
+ * @desc:ajax流式请求
+ * @author huangbo
+ * @date 2022年5月7日
+ * @param {Object} [controller] - controller
+ * @param {String} [method] - 请求类型
+ * @param {String} [url] - 请求地址
+ * @param {String} [host] - host
+ * @param {Object} [data] - 请求参数
+ * @param {Object} [params] - 请求参数
+ **/
+const ajaxStream = function ({ controller, method = 'post', host, url, data, params, progress, success, fail, final, ...configs }) {
+  try {
+    // 如果有正在进行的请求，取消它
+    if (controller) {
+      controller.abort();
+    }
+    controller = new AbortController();
+    const baseURL = host || getStorage('baseHost') || '';
+    let userId = getStorage('userId');
+    if (
+      data && !Object.prototype.hasOwnProperty.call(data, 'userId') &&
+      userId
+    ) {
+      data.userId = userId;
+    }
+    if (
+      params && !Object.prototype.hasOwnProperty.call(params, 'userId') &&
+      userId
+    ) {
+      params.userId = getStorage('userId');
+    }
+    let info = {
+      time: 0,
+      conversationId: '',
+      content: '',
+      thinkContent: '',
+      primeval: '',
+      chunk: ''
+    };
+    let partialData = '';
+    let lastProcessedIndex = 0;
+    let done = false;
+    let finishThink = false;
+    ajax({
+      timeout: 990000,
+      format: false,
+      method,
+      url: baseURL + url,
+      signal: controller.signal,
+      responseType: 'text',
+      data,
+      params,
+      ...configs,
+      onDownloadProgress: progressEvent => {
+        // 获取从上次处理后新增的数据
+        const currentResponse = (progressEvent.event || progressEvent).currentTarget.responseText;
+        const chunk = currentResponse.slice(lastProcessedIndex);
+        if (chunk === 'data:') {
+          return;
+        }
+
+        // 更新偏移量
+        lastProcessedIndex = currentResponse.length;
+
+        // 假设数据是以 'data:' 开头的 SSE 格式
+        partialData += chunk;
+        const lines = partialData.split('\n');
+        let remainingData = ''; // 用于保存未完整处理的部分数据
+        let primeval = '';
+        let chunks = '';
+        lines.forEach(line => {
+          if (line === '') {
+            return;
+          }
+          if (line.startsWith('data:')) {
+            const data = line.replace('data:', '').trim();
+            if (data) {
+              try {
+                // 假设数据是 JSON 格式
+                const parsedData = JSON.parse(data);
+                info.time = parsedData.time;
+                info.conversationId = parsedData.conversationId;
+                const thinkRegex = /<think>([\s\S]*?)<\/think>([\s\S]*)?/;
+                const match = parsedData.message.match(thinkRegex);
+                if (match) {
+                  const thinkContent = match[1] ? match[1].trim() : ''; // <think> 标签内的数据
+                  const content = match[2] ? match[2].trim() : ''; // <think> 标签后的数据
+                  // 更新 AI 消息内容
+                  info.thinkContent += thinkContent;
+                  info.content += content;
+                } else {
+                  const startRegex = /<think>([\s\S]*)/;
+                  const endRegex = /([\s\S]*?)<\/think>([\s\S]*)?/;
+                  const start = parsedData.message.match(startRegex);
+                  const end = parsedData.message.match(endRegex);
+                  if (start) {
+                    const thinkContent = start[1] ? start[1].trim() : ''; // <think> 标签内的数据
+                    info.thinkContent += thinkContent;
+                  } else {
+                    if (end) {
+                      finishThink = true;
+                      const thinkContent = end[1] ? end[1].trim() : ''; // <think> 标签后的数据
+                      const content = end[2] ? end[2].trim() : '';
+                      // 更新 AI 消息内容
+                      info.thinkContent += thinkContent;
+                      info.content += content;
+                      chunks += content;
+                    } else if (finishThink) {
+                      info.content += parsedData.message;
+                      chunks += parsedData.message;
+                    } else {
+                      info.thinkContent += parsedData.message;
+                    }
+                  }
+                }
+                primeval += parsedData.message;
+                done = parsedData.done;
+              } catch (parseError) {
+                console.error('解析数据出错:', parseError);
+              }
+            }
+          } else {
+            // 保存未处理完的部分数据
+            remainingData += line + '\n';
+          }
+        });
+        info.primeval = primeval;
+        info.chunk = chunks;
 
+        // 将未完整处理的数据保留到 partialData 中
+        partialData = remainingData;
+        progress && progress({ message: info, controller, done });
+      }
+    }).then(() => {
+      // 处理请求完成后的信息
+      const originalText = info.content;
+      const regex = /<think>([\s\S]*?)<\/think>([\s\S]*)/;
+      const match = originalText.match(regex);
+      if (match) {
+        info.thinkContent = match ? match[1].trim() : '';
+        info.content = match ? match[2].trim() : '';
+      }
+      success && success({ message: info, controller });
+    })
+      .catch(() => {
+        // 处理请求完成后的信息
+        const originalText = info.content || '';
+        const regex = /<think>([\s\S]*?)<\/think>([\s\S]*)/;
+        const match = originalText.match(regex);
+        if (match) {
+          info.thinkContent = match ? match[1].trim() : '';
+          info.content = match ? match[2].trim() : '';
+        }
+        fail && fail({ status: 1, message: info, controller });
+      });
+  } catch (error) {
+    console.error('聊天请求出错:', error);
+    let errorMessage = '抱歉，发生了一个错误，请稍后再试。';
+    if (error.code === 'ERR_NETWORK') {
+      errorMessage = '网络连接错误，请检查服务器是否正在运行或网络连接是否正常。';
+    } else if (error.response) {
+      errorMessage = `服务器错误 (${error.response.status}): ${error.response.data || '未知错误'
+        }`;
+    }
+    fail && fail({ status: 0, message: { role: 'assistant', content: errorMessage }, controller });
+  } finally {
+    final && final({ status: -1, message: null, controller });
+  }
+};
 /**
  * arrUnique
  * @desc:数组元素对象去重
@@ -552,7 +726,7 @@ const browser = function () {
  * @author huangbo
  * @date 2022年5月7日
  **/
-const busEmit = function (vm = this, {method, args}) {
+const busEmit = function (vm = this, { method, args }) {
   const bus = win.$wujie ? win.$wujie.bus : vm.bus || vm.$root.Bus;
   bus && bus.$emit(method, args);
   window.postMessage({ method, query: args }, '*');
@@ -1430,7 +1604,7 @@ const getMainConfig = function (callback) {
         callback(res.results);
       }
     })
-    .catch(() => {});
+    .catch(() => { });
 };
 
 /**
@@ -2219,7 +2393,7 @@ const isLogined = function ({
           removeStorage();
         }
       })
-      .catch(() => {});
+      .catch(() => { });
     return;
   }
   let pathname = '';
@@ -2252,7 +2426,7 @@ const isLogined = function ({
       Object.prototype.hasOwnProperty.call(to.query, 'serverId') &&
       Object.prototype.hasOwnProperty.call(to.query, 'authType')
     ) {
-      let loadingInstance = Loading.service({text: '页面加载中', spinner: 'el-icon-loading', background: 'rgba(0, 0, 0, 0.65)'});
+      let loadingInstance = Loading.service({ text: '页面加载中', spinner: 'el-icon-loading', background: 'rgba(0, 0, 0, 0.65)' });
       ajax({
         method: 'post',
         url: authCenter,
@@ -2372,7 +2546,7 @@ const isLogined = function ({
         })
         .catch((e) => {
           loadingInstance.close();
-});
+        });
     } else if (
       token ||
       to.path === '/' ||
@@ -3244,10 +3418,10 @@ const watermark = function (option) {
   };
   let options = option
     ? extend(
-        true,
-        config,
-        typeof option === 'string' ? { html: option } : option
-      )
+      true,
+      config,
+      typeof option === 'string' ? { html: option } : option
+    )
     : config;
   let wrap = document.getElementById('watermark');
   if (!wrap) {
@@ -3327,6 +3501,7 @@ const watermark = function (option) {
 
 export default {
   ajax,
+  ajaxStream,
   arrUnique,
   average,
   browser,