@antv/dumi-theme-antv/dist/model/AIChat.js

AntV website theme based on dumi2.

function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
function _iterableToArrayLimit(r, l) { var t = null == r ? null : "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (null != t) { var e, n, i, u, a = [], f = !0, o = !1; try { if (i = (t = t.call(r)).next, 0 === l) { if (Object(t) !== t) return; f = !1; } else for (; !(f = (e = i.call(t)).done) && (a.push(e.value), a.length !== l); f = !0); } catch (r) { o = !0, n = r; } finally { try { if (!f && null != t.return && (u = t.return(), Object(u) !== u)) return; } finally { if (o) throw n; } } return a; } }
function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
function _regeneratorRuntime() { "use strict"; /*! regenerator-runtime -- Copyright (c) 2014-present, Facebook, Inc. -- license (MIT): https://github.com/facebook/regenerator/blob/main/LICENSE */ _regeneratorRuntime = function _regeneratorRuntime() { return e; }; var t, e = {}, r = Object.prototype, n = r.hasOwnProperty, o = Object.defineProperty || function (t, e, r) { t[e] = r.value; }, i = "function" == typeof Symbol ? Symbol : {}, a = i.iterator || "@@iterator", c = i.asyncIterator || "@@asyncIterator", u = i.toStringTag || "@@toStringTag"; function define(t, e, r) { return Object.defineProperty(t, e, { value: r, enumerable: !0, configurable: !0, writable: !0 }), t[e]; } try { define({}, ""); } catch (t) { define = function define(t, e, r) { return t[e] = r; }; } function wrap(t, e, r, n) { var i = e && e.prototype instanceof Generator ? e : Generator, a = Object.create(i.prototype), c = new Context(n || []); return o(a, "_invoke", { value: makeInvokeMethod(t, r, c) }), a; } function tryCatch(t, e, r) { try { return { type: "normal", arg: t.call(e, r) }; } catch (t) { return { type: "throw", arg: t }; } } e.wrap = wrap; var h = "suspendedStart", l = "suspendedYield", f = "executing", s = "completed", y = {}; function Generator() {} function GeneratorFunction() {} function GeneratorFunctionPrototype() {} var p = {}; define(p, a, function () { return this; }); var d = Object.getPrototypeOf, v = d && d(d(values([]))); v && v !== r && n.call(v, a) && (p = v); var g = GeneratorFunctionPrototype.prototype = Generator.prototype = Object.create(p); function defineIteratorMethods(t) { ["next", "throw", "return"].forEach(function (e) { define(t, e, function (t) { return this._invoke(e, t); }); }); } function AsyncIterator(t, e) { function invoke(r, o, i, a) { var c = tryCatch(t[r], t, o); if ("throw" !== c.type) { var u = c.arg, h = u.value; return h && "object" == _typeof(h) && n.call(h, "__await") ? e.resolve(h.__await).then(function (t) { invoke("next", t, i, a); }, function (t) { invoke("throw", t, i, a); }) : e.resolve(h).then(function (t) { u.value = t, i(u); }, function (t) { return invoke("throw", t, i, a); }); } a(c.arg); } var r; o(this, "_invoke", { value: function value(t, n) { function callInvokeWithMethodAndArg() { return new e(function (e, r) { invoke(t, n, e, r); }); } return r = r ? r.then(callInvokeWithMethodAndArg, callInvokeWithMethodAndArg) : callInvokeWithMethodAndArg(); } }); } function makeInvokeMethod(e, r, n) { var o = h; return function (i, a) { if (o === f) throw new Error("Generator is already running"); if (o === s) { if ("throw" === i) throw a; return { value: t, done: !0 }; } for (n.method = i, n.arg = a;;) { var c = n.delegate; if (c) { var u = maybeInvokeDelegate(c, n); if (u) { if (u === y) continue; return u; } } if ("next" === n.method) n.sent = n._sent = n.arg;else if ("throw" === n.method) { if (o === h) throw o = s, n.arg; n.dispatchException(n.arg); } else "return" === n.method && n.abrupt("return", n.arg); o = f; var p = tryCatch(e, r, n); if ("normal" === p.type) { if (o = n.done ? s : l, p.arg === y) continue; return { value: p.arg, done: n.done }; } "throw" === p.type && (o = s, n.method = "throw", n.arg = p.arg); } }; } function maybeInvokeDelegate(e, r) { var n = r.method, o = e.iterator[n]; if (o === t) return r.delegate = null, "throw" === n && e.iterator.return && (r.method = "return", r.arg = t, maybeInvokeDelegate(e, r), "throw" === r.method) || "return" !== n && (r.method = "throw", r.arg = new TypeError("The iterator does not provide a '" + n + "' method")), y; var i = tryCatch(o, e.iterator, r.arg); if ("throw" === i.type) return r.method = "throw", r.arg = i.arg, r.delegate = null, y; var a = i.arg; return a ? a.done ? (r[e.resultName] = a.value, r.next = e.nextLoc, "return" !== r.method && (r.method = "next", r.arg = t), r.delegate = null, y) : a : (r.method = "throw", r.arg = new TypeError("iterator result is not an object"), r.delegate = null, y); } function pushTryEntry(t) { var e = { tryLoc: t[0] }; 1 in t && (e.catchLoc = t[1]), 2 in t && (e.finallyLoc = t[2], e.afterLoc = t[3]), this.tryEntries.push(e); } function resetTryEntry(t) { var e = t.completion || {}; e.type = "normal", delete e.arg, t.completion = e; } function Context(t) { this.tryEntries = [{ tryLoc: "root" }], t.forEach(pushTryEntry, this), this.reset(!0); } function values(e) { if (e || "" === e) { var r = e[a]; if (r) return r.call(e); if ("function" == typeof e.next) return e; if (!isNaN(e.length)) { var o = -1, i = function next() { for (; ++o < e.length;) if (n.call(e, o)) return next.value = e[o], next.done = !1, next; return next.value = t, next.done = !0, next; }; return i.next = i; } } throw new TypeError(_typeof(e) + " is not iterable"); } return GeneratorFunction.prototype = GeneratorFunctionPrototype, o(g, "constructor", { value: GeneratorFunctionPrototype, configurable: !0 }), o(GeneratorFunctionPrototype, "constructor", { value: GeneratorFunction, configurable: !0 }), GeneratorFunction.displayName = define(GeneratorFunctionPrototype, u, "GeneratorFunction"), e.isGeneratorFunction = function (t) { var e = "function" == typeof t && t.constructor; return !!e && (e === GeneratorFunction || "GeneratorFunction" === (e.displayName || e.name)); }, e.mark = function (t) { return Object.setPrototypeOf ? Object.setPrototypeOf(t, GeneratorFunctionPrototype) : (t.__proto__ = GeneratorFunctionPrototype, define(t, u, "GeneratorFunction")), t.prototype = Object.create(g), t; }, e.awrap = function (t) { return { __await: t }; }, defineIteratorMethods(AsyncIterator.prototype), define(AsyncIterator.prototype, c, function () { return this; }), e.AsyncIterator = AsyncIterator, e.async = function (t, r, n, o, i) { void 0 === i && (i = Promise); var a = new AsyncIterator(wrap(t, r, n, o), i); return e.isGeneratorFunction(r) ? a : a.next().then(function (t) { return t.done ? t.value : a.next(); }); }, defineIteratorMethods(g), define(g, u, "Generator"), define(g, a, function () { return this; }), define(g, "toString", function () { return "[object Generator]"; }), e.keys = function (t) { var e = Object(t), r = []; for (var n in e) r.push(n); return r.reverse(), function next() { for (; r.length;) { var t = r.pop(); if (t in e) return next.value = t, next.done = !1, next; } return next.done = !0, next; }; }, e.values = values, Context.prototype = { constructor: Context, reset: function reset(e) { if (this.prev = 0, this.next = 0, this.sent = this._sent = t, this.done = !1, this.delegate = null, this.method = "next", this.arg = t, this.tryEntries.forEach(resetTryEntry), !e) for (var r in this) "t" === r.charAt(0) && n.call(this, r) && !isNaN(+r.slice(1)) && (this[r] = t); }, stop: function stop() { this.done = !0; var t = this.tryEntries[0].completion; if ("throw" === t.type) throw t.arg; return this.rval; }, dispatchException: function dispatchException(e) { if (this.done) throw e; var r = this; function handle(n, o) { return a.type = "throw", a.arg = e, r.next = n, o && (r.method = "next", r.arg = t), !!o; } for (var o = this.tryEntries.length - 1; o >= 0; --o) { var i = this.tryEntries[o], a = i.completion; if ("root" === i.tryLoc) return handle("end"); if (i.tryLoc <= this.prev) { var c = n.call(i, "catchLoc"), u = n.call(i, "finallyLoc"); if (c && u) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } else if (c) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); } else { if (!u) throw new Error("try statement without catch or finally"); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } } } }, abrupt: function abrupt(t, e) { for (var r = this.tryEntries.length - 1; r >= 0; --r) { var o = this.tryEntries[r]; if (o.tryLoc <= this.prev && n.call(o, "finallyLoc") && this.prev < o.finallyLoc) { var i = o; break; } } i && ("break" === t || "continue" === t) && i.tryLoc <= e && e <= i.finallyLoc && (i = null); var a = i ? i.completion : {}; return a.type = t, a.arg = e, i ? (this.method = "next", this.next = i.finallyLoc, y) : this.complete(a); }, complete: function complete(t, e) { if ("throw" === t.type) throw t.arg; return "break" === t.type || "continue" === t.type ? this.next = t.arg : "return" === t.type ? (this.rval = this.arg = t.arg, this.method = "return", this.next = "end") : "normal" === t.type && e && (this.next = e), y; }, finish: function finish(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.finallyLoc === t) return this.complete(r.completion, r.afterLoc), resetTryEntry(r), y; } }, catch: function _catch(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.tryLoc === t) { var n = r.completion; if ("throw" === n.type) { var o = n.arg; resetTryEntry(r); } return o; } } throw new Error("illegal catch attempt"); }, delegateYield: function delegateYield(e, r, n) { return this.delegate = { iterator: values(e), resultName: r, nextLoc: n }, "next" === this.method && (this.arg = t), y; } }, e; }
function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
import { proxy, subscribe, snapshot } from 'valtio';
import { derive, subscribeKey } from 'valtio/utils';
import localforage from 'localforage';
import FingerprintJS from '@fingerprintjs/fingerprintjs';
import { history } from "dumi";
import { message } from "antd";
import { AIMode } from "../components/AI/constant";
import { trackEvent } from "../utils/analytics";

// --- 配置 ---
// 定义需要持久化的 state key
var PERSIST_KEYS = ['anonymousUserId', 'sessions'];
var STORAGE_KEY = 'ai-chat-state-v1';

// --- 初始状态 ---
var initialState = {
  isInitialized: false,
  anonymousUserId: null,
  sessions: [],
  activeSessionId: null,
  tempMessage: null,
  codeBlock: null,
  lib: null,
  mode: AIMode.implement,
  errorMsg: null
};

// --- valtio Store 创建 ---
export var AIChatStore = proxy(initialState);

// --- 持久化订阅 ---
subscribe(AIChatStore, function () {
  // 确保只在初始化后才进行持久化
  if (!AIChatStore.isInitialized) return;
  var stateSnapshot = snapshot(AIChatStore);

  // console.log('stateSnapshot', stateSnapshot);

  var stateToPersist = {};
  PERSIST_KEYS.forEach(function (key) {
    stateToPersist[key] = stateSnapshot[key];
  });

  // 异步保存，不阻塞UI
  localforage.setItem(STORAGE_KEY, stateToPersist);
});
export var clearEmptySession = function clearEmptySession() {
  AIChatStore.sessions = AIChatStore.sessions.filter(function (s) {
    return s.messages.length > 0;
  });
  if (AIChatStore.sessions.every(function (s) {
    return s.id !== AIChatStore.activeSessionId;
  }) && AIChatStore.sessions.length > 0) {
    AIChatStore.activeSessionId = AIChatStore.sessions[0].id;
  }
};

// --- 初始化逻辑 ---
export var initializeAIChat = /*#__PURE__*/function () {
  var _ref = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee() {
    var persistedState, fp, result, newSession, _AIChatStore$sessions;
    return _regeneratorRuntime().wrap(function _callee$(_context) {
      while (1) switch (_context.prev = _context.next) {
        case 0:
          _context.next = 2;
          return localforage.getItem(STORAGE_KEY);
        case 2:
          persistedState = _context.sent;
          // 2. 合并加载的数据到 store
          if (persistedState) {
            PERSIST_KEYS.forEach(function (key) {
              if (persistedState[key] !== undefined) {
                // @ts-ignore - Directly assigning to the proxy state
                AIChatStore[key] = persistedState[key];
              }
            });
          }

          // 3. 处理初始化边缘情况
          if (AIChatStore.anonymousUserId) {
            _context.next = 18;
            break;
          }
          _context.prev = 5;
          _context.next = 8;
          return FingerprintJS.load();
        case 8:
          fp = _context.sent;
          _context.next = 11;
          return fp.get();
        case 11:
          result = _context.sent;
          // result.visitorId 是一个基于浏览器指纹生成的哈希值
          // 例如：'d8b759e6a5b2f1c3d9e8a6f0b7c5d4a3'
          AIChatStore.anonymousUserId = result.visitorId;
          // console.log('FingerprintJS ID generated:', result.visitorId);
          _context.next = 18;
          break;
        case 15:
          _context.prev = 15;
          _context.t0 = _context["catch"](5);
          // console.error('FingerprintJS failed, falling back to simple ID:', error);
          // 如果指纹生成失败（例如被浏览器插件阻止），回退到一个简单方案
          // 如果需要兼容旧浏览器，可以换成 `import { v4 as uuidv4 } from 'uuid';` 和 `uuidv4()`
          AIChatStore.anonymousUserId = crypto.randomUUID();
        case 18:
          if (AIChatStore.sessions.length === 0) {
            newSession = {
              id: crypto.randomUUID(),
              title: 'New Conversation',
              createdAt: Date.now(),
              messages: []
            };
            AIChatStore.sessions.push(newSession);
            AIChatStore.activeSessionId = newSession.id;
          }
          if (!AIChatStore.activeSessionId || !AIChatStore.sessions.find(function (s) {
            return s.id === AIChatStore.activeSessionId;
          })) {
            AIChatStore.activeSessionId = ((_AIChatStore$sessions = AIChatStore.sessions[0]) === null || _AIChatStore$sessions === void 0 ? void 0 : _AIChatStore$sessions.id) || null;
          }
          clearEmptySession();

          // 4. 标记初始化完成，这将触发持久化订阅
          AIChatStore.isInitialized = true;
          // console.log('Valtio store initialized:', snapshot(AIChatStore));
        case 22:
        case "end":
          return _context.stop();
      }
    }, _callee, null, [[5, 15]]);
  }));
  return function initializeAIChat() {
    return _ref.apply(this, arguments);
  };
}();

// --- 计算属性 (使用 derive) ---
export var derivedState = derive({
  // 获取当前激活的会话对象
  activeSession: function activeSession(get) {
    var _get = get(AIChatStore),
      sessions = _get.sessions,
      activeSessionId = _get.activeSessionId;
    if (!activeSessionId) return null;
    return sessions.find(function (s) {
      return s.id === activeSessionId;
    }) || null;
  }
});

// --- 删除会话 ---
export var handleDeleteSession = function handleDeleteSession(sessionId) {
  var sessionIndex = AIChatStore.sessions.findIndex(function (s) {
    return s.id === sessionId;
  });
  if (sessionIndex === -1) return;

  // 从数组中移除
  AIChatStore.sessions.splice(sessionIndex, 1);

  // 如果删除的是当前激活的会话，需要重置 activeSessionId
  if (AIChatStore.activeSessionId === sessionId) {
    var _AIChatStore$sessions2;
    AIChatStore.activeSessionId = ((_AIChatStore$sessions2 = AIChatStore.sessions[0]) === null || _AIChatStore$sessions2 === void 0 ? void 0 : _AIChatStore$sessions2.id) || null;
  }
};

// --- 重命名会话 ---
export var handleRenameSession = function handleRenameSession(sessionId, newTitle) {
  var session = AIChatStore.sessions.find(function (s) {
    return s.id === sessionId;
  });
  if (session) {
    session.title = newTitle;
  }
};

// --- 置顶会话 ---
export var handlePinSession = function handlePinSession(sessionId) {
  var sessionIndex = AIChatStore.sessions.findIndex(function (s) {
    return s.id === sessionId;
  });
  if (sessionIndex <= 0) return; // 如果找不到或者已经在第一位，则不操作

  // 找到会话，从原位置删除，然后添加到数组头部
  var _AIChatStore$sessions3 = AIChatStore.sessions.splice(sessionIndex, 1),
    _AIChatStore$sessions4 = _slicedToArray(_AIChatStore$sessions3, 1),
    sessionToPin = _AIChatStore$sessions4[0];
  AIChatStore.sessions.unshift(sessionToPin);

  // （可选）置顶后自动激活该会话
  AIChatStore.activeSessionId = sessionId;
};
subscribeKey(AIChatStore, 'activeSessionId', function () {
  AIChatStore.codeBlock = null;
});
export var createPureNewSession = function createPureNewSession(title) {
  var newConversationName = 'New Conversation';
  var existNewSession = AIChatStore.sessions.find(function (s) {
    return s.title === newConversationName && s.messages.length === 0;
  });
  if (existNewSession) {
    AIChatStore.activeSessionId = existNewSession.id;
    return;
  }
  var newSessionId = crypto.randomUUID();
  AIChatStore.sessions.unshift({
    id: newSessionId,
    title: title !== null && title !== void 0 ? title : newConversationName,
    createdAt: Date.now(),
    messages: []
  });
  AIChatStore.activeSessionId = newSessionId;
};
export var createNewSession = function createNewSession(config) {
  // 1. 创建一个新的会话
  createPureNewSession(config.promptText);

  // 2. 创建临时消息并存入 store
  AIChatStore.tempMessage = {
    id: crypto.randomUUID(),
    role: 'user',
    content: config.promptText,
    createdAt: Date.now(),
    mode: config.mode,
    lib: config.lib,
    context: config.context
  };
  if (config.jump) {
    var _config$lang;
    history.push("/".concat((_config$lang = config.lang) !== null && _config$lang !== void 0 ? _config$lang : 'zh', "/ai-playground"));
  }
  // 埋点
  if ((typeof window === "undefined" ? "undefined" : _typeof(window)) === 'object') {
    trackEvent('start_ai_chat', {
      entry_point: config.entry_point,
      mode: AIChatStore.mode,
      lib: AIChatStore.lib,
      page_title: document.title,
      location: location.href
    });
  }
};

/**
 * 清空所有本地AI对话记录和相关状态。用户注销账户或手动请求清除数据时调用。
 *
 * 该函数会执行以下操作：
 * 1. 从 localforage 中删除持久化的状态数据。
 * 2. 将内存中的 AIChatStore 重置为初始状态。
 *
 */
export var clearAllChatData = /*#__PURE__*/function () {
  var _ref2 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2() {
    var initialKeys, _i, _initialKeys, key;
    return _regeneratorRuntime().wrap(function _callee2$(_context2) {
      while (1) switch (_context2.prev = _context2.next) {
        case 0:
          _context2.prev = 0;
          _context2.next = 3;
          return localforage.removeItem(STORAGE_KEY);
        case 3:
          console.log("[AIChat] Persistent state with key \"".concat(STORAGE_KEY, "\" has been removed."));

          // 步骤 2: 将内存中的 valtio store 重置为初始状态。
          // 这会立即更新UI，让所有对话记录从界面上消失。
          // 注意：我们不能直接做 AIChatStore = initialState，
          // 因为 proxy 对象是不可替换的。我们必须逐个属性地重置。
          initialKeys = Object.keys(initialState);
          for (_i = 0, _initialKeys = initialKeys; _i < _initialKeys.length; _i++) {
            key = _initialKeys[_i];
            // @ts-ignore
            AIChatStore[key] = initialState[key];
          }
          console.log('[AIChat] In-memory state has been reset to initial values.');
          _context2.next = 13;
          break;
        case 9:
          _context2.prev = 9;
          _context2.t0 = _context2["catch"](0);
          console.error('[AIChat] Failed to clear all chat data:', _context2.t0);
          message.error('清空对话记录失败，请刷新页面后重试。');
        case 13:
        case "end":
          return _context2.stop();
      }
    }, _callee2, null, [[0, 9]]);
  }));
  return function clearAllChatData() {
    return _ref2.apply(this, arguments);
  };
}();
export function deleteMessage(msgId) {
  derivedState.activeSession.messages = derivedState.activeSession.messages.filter(function (m) {
    return m.id !== msgId;
  });
}
export function branchMessage(index) {
  var messages = derivedState.activeSession.messages.slice(0, index + 1);
  createPureNewSession(messages[0].content);
  AIChatStore.sessions[0].messages = messages;
}