@difizen/ai-flow/es/components/AIBasic/PromptEditor/plugins/component-picker-block/hooks.js

Scalable, out-of-the-box, agent-oriented flow

function _toConsumableArray(arr) { return _arrayWithoutHoles(arr) || _iterableToArray(arr) || _unsupportedIterableToArray(arr) || _nonIterableSpread(); }
function _nonIterableSpread() { throw new TypeError("Invalid attempt to spread non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
function _iterableToArray(iter) { if (typeof Symbol !== "undefined" && iter[Symbol.iterator] != null || iter["@@iterator"] != null) return Array.from(iter); }
function _arrayWithoutHoles(arr) { if (Array.isArray(arr)) return _arrayLikeToArray(arr); }
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; }
import { AndroidFilled } from '@ant-design/icons';
import { useLexicalComposerContext } from '@lexical/react/LexicalComposerContext';
import { $insertNodes } from 'lexical';
import { useMemo } from 'react';
import { $createCustomTextNode } from "../custom-text/node";
import { INSERT_VARIABLE_VALUE_BLOCK_COMMAND } from "../variable-block/index";
import { PickerBlockMenuOption } from "./menu";
import { VariableMenuItem } from "./variable-option";
import React from 'react';
import { jsx as _jsx } from "react/jsx-runtime";
export var useVariableOptions = function useVariableOptions(variableBlock, queryString) {
  var _useLexicalComposerCo = useLexicalComposerContext(),
    _useLexicalComposerCo2 = _slicedToArray(_useLexicalComposerCo, 1),
    editor = _useLexicalComposerCo2[0];
  var options = useMemo(function () {
    if (!(variableBlock !== null && variableBlock !== void 0 && variableBlock.variables)) {
      return [];
    }
    var baseOptions = variableBlock.variables.map(function (item) {
      return new PickerBlockMenuOption({
        key: item.value,
        group: 'prompt variable',
        render: function render(_ref) {
          var queryString = _ref.queryString,
            isSelected = _ref.isSelected,
            onSelect = _ref.onSelect,
            onSetHighlight = _ref.onSetHighlight;
          return /*#__PURE__*/_jsx(VariableMenuItem, {
            title: item.value,
            icon: /*#__PURE__*/_jsx(AndroidFilled, {
              className: "w-[14px] h-[14px] text-[#2970FF]"
            }),
            queryString: queryString,
            isSelected: isSelected,
            onClick: onSelect,
            onMouseEnter: onSetHighlight
          });
        },
        onSelect: function onSelect() {
          editor.dispatchCommand(INSERT_VARIABLE_VALUE_BLOCK_COMMAND, "{{".concat(item.value, "}}"));
        }
      });
    });
    if (!queryString) {
      return baseOptions;
    }
    var regex = new RegExp(queryString, 'i');
    return baseOptions.filter(function (option) {
      return regex.test(option.key);
    });
  }, [editor, queryString, variableBlock]);
  var addOption = useMemo(function () {
    return new PickerBlockMenuOption({
      key: '添加新变量',
      group: 'prompt variable',
      render: function render(_ref2) {
        var queryString = _ref2.queryString,
          isSelected = _ref2.isSelected,
          onSelect = _ref2.onSelect,
          onSetHighlight = _ref2.onSetHighlight;
        return /*#__PURE__*/_jsx(VariableMenuItem, {
          title: '添加新变量',
          icon: /*#__PURE__*/_jsx(AndroidFilled, {
            className: "mr-2 w-[14px] h-[14px] text-[#2970FF]"
          }),
          queryString: queryString,
          isSelected: isSelected,
          onClick: onSelect,
          onMouseEnter: onSetHighlight
        });
      },
      onSelect: function onSelect() {
        editor.update(function () {
          var prefixNode = $createCustomTextNode('{{');
          var suffixNode = $createCustomTextNode('}}');
          $insertNodes([prefixNode, suffixNode]);
          prefixNode.select();
        });
      }
    });
  }, [editor]);
  return useMemo(function () {
    return variableBlock !== null && variableBlock !== void 0 && variableBlock.show ? [].concat(_toConsumableArray(options), [addOption]) : [];
  }, [options, addOption, variableBlock === null || variableBlock === void 0 ? void 0 : variableBlock.show]);
};
export var useExternalToolOptions = function useExternalToolOptions(externalToolBlockType, queryString) {
  var _useLexicalComposerCo3 = useLexicalComposerContext(),
    _useLexicalComposerCo4 = _slicedToArray(_useLexicalComposerCo3, 1),
    editor = _useLexicalComposerCo4[0];
  var options = useMemo(function () {
    if (!(externalToolBlockType !== null && externalToolBlockType !== void 0 && externalToolBlockType.externalTools)) {
      return [];
    }
    var baseToolOptions = externalToolBlockType.externalTools.map(function (item) {
      return new PickerBlockMenuOption({
        key: item.name,
        group: 'external tool',
        render: function render(_ref3) {
          var queryString = _ref3.queryString,
            isSelected = _ref3.isSelected,
            onSelect = _ref3.onSelect,
            onSetHighlight = _ref3.onSetHighlight;
          return /*#__PURE__*/_jsx(VariableMenuItem, {
            title: item.name,
            icon: /*#__PURE__*/_jsx(AndroidFilled, {
              className: "!w-[14px] !h-[14px]"
              // icon={item.icon}
              // background={item.icon_background}
            }),
            extraElement: /*#__PURE__*/_jsx("div", {
              className: "text-xs text-gray-400",
              children: item.variableName
            }),
            queryString: queryString,
            isSelected: isSelected,
            onClick: onSelect,
            onMouseEnter: onSetHighlight
          });
        },
        onSelect: function onSelect() {
          editor.dispatchCommand(INSERT_VARIABLE_VALUE_BLOCK_COMMAND, "{{".concat(item.variableName, "}}"));
        }
      });
    });
    if (!queryString) {
      return baseToolOptions;
    }
    var regex = new RegExp(queryString, 'i');
    return baseToolOptions.filter(function (option) {
      return regex.test(option.key);
    });
  }, [editor, queryString, externalToolBlockType]);
  var addOption = useMemo(function () {
    return new PickerBlockMenuOption({
      key: '添加工具',
      group: 'external tool',
      render: function render(_ref4) {
        var queryString = _ref4.queryString,
          isSelected = _ref4.isSelected,
          onSelect = _ref4.onSelect,
          onSetHighlight = _ref4.onSetHighlight;
        return /*#__PURE__*/_jsx(VariableMenuItem, {
          title: '添加工具',
          icon: /*#__PURE__*/_jsx(AndroidFilled, {
            className: "mr-2 w-[14px] h-[14px] text-[#444CE7]"
          }),
          extraElement: /*#__PURE__*/_jsx(AndroidFilled, {
            className: "w-3 h-3 text-gray-400"
          }),
          queryString: queryString,
          isSelected: isSelected,
          onClick: onSelect,
          onMouseEnter: onSetHighlight
        });
      },
      onSelect: function onSelect() {
        var _externalToolBlockTyp;
        externalToolBlockType === null || externalToolBlockType === void 0 || (_externalToolBlockTyp = externalToolBlockType.onAddExternalTool) === null || _externalToolBlockTyp === void 0 || _externalToolBlockTyp.call(externalToolBlockType);
      }
    });
  }, [externalToolBlockType]);
  return useMemo(function () {
    return externalToolBlockType !== null && externalToolBlockType !== void 0 && externalToolBlockType.show ? [].concat(_toConsumableArray(options), [addOption]) : [];
  }, [options, addOption, externalToolBlockType === null || externalToolBlockType === void 0 ? void 0 : externalToolBlockType.show]);
};
export var useOptions = function useOptions(variableBlock, externalToolBlockType, queryString) {
  var variableOptions = useVariableOptions(variableBlock, queryString);
  var externalToolOptions = useExternalToolOptions(externalToolBlockType, queryString);
  return useMemo(function () {
    return {
      allFlattenOptions: [].concat(_toConsumableArray(variableOptions), _toConsumableArray(externalToolOptions))
    };
  }, [variableOptions, externalToolOptions]);
};