cozy-search/dist/components/AssistantProvider.js

UI components about search bar and IA assistant

"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.useAssistant = exports.default = exports.AssistantContext = void 0;

var _regenerator = _interopRequireDefault(require("@babel/runtime/regenerator"));

var _asyncToGenerator2 = _interopRequireDefault(require("@babel/runtime/helpers/asyncToGenerator"));

var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));

var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));

var _set = _interopRequireDefault(require("lodash/set"));

var _react = _interopRequireWildcard(require("react"));

var _reactRouterDom = require("react-router-dom");

var _cozyClient = require("cozy-client");

var _useRealtime3 = _interopRequireDefault(require("cozy-realtime/dist/useRealtime"));

var _helpers = require("./helpers");

var _queries = require("./queries");

function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }

function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }

function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); enumerableOnly && (symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; })), keys.push.apply(keys, symbols); } return keys; }

function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = null != arguments[i] ? arguments[i] : {}; i % 2 ? ownKeys(Object(source), !0).forEach(function (key) { (0, _defineProperty2.default)(target, key, source[key]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)) : ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } return target; }

var AssistantContext = /*#__PURE__*/_react.default.createContext();

exports.AssistantContext = AssistantContext;

var useAssistant = function useAssistant() {
  var context = (0, _react.useContext)(AssistantContext);

  if (!context) {
    throw new Error('useAssistant must be used within a AssistantProvider');
  }

  return context;
};

exports.useAssistant = useAssistant;

var AssistantProvider = function AssistantProvider(_ref) {
  var children = _ref.children;

  var _useParams = (0, _reactRouterDom.useParams)(),
      conversationId = _useParams.conversationId;

  var client = (0, _cozyClient.useClient)();

  var _useState = (0, _react.useState)({
    message: {},
    status: 'idle',
    messagesId: []
  }),
      _useState2 = (0, _slicedToArray2.default)(_useState, 2),
      assistantState = _useState2[0],
      setAssistantState = _useState2[1];

  (0, _useRealtime3.default)(client, (0, _defineProperty2.default)({}, _queries.CHAT_CONVERSATIONS_DOCTYPE, {
    created: function created(res) {
      (0, _helpers.pushMessagesIdInState)(conversationId, res, setAssistantState);
    },
    updated: function updated(res) {
      (0, _helpers.pushMessagesIdInState)(conversationId, res, setAssistantState);
    }
  }), []);
  (0, _useRealtime3.default)(client, (0, _defineProperty2.default)({}, _queries.CHAT_EVENTS_DOCTYPE, {
    created: function created(res) {
      setAssistantState(function (prevState) {
        // to exclude realtime messages if not relevant to the actual conversation
        if (!(0, _helpers.isMessageForThisConversation)(res, prevState.messagesId)) {
          return prevState;
        }

        if (res.object === 'done') {
          if (prevState.status !== 'idle') {
            return _objectSpread(_objectSpread({}, prevState), {}, {
              status: 'idle'
            });
          }
        }

        if (res.object === 'delta') {
          var message = (0, _set.default)(prevState.message, res.position, res.content);
          return _objectSpread(_objectSpread({}, prevState), {}, {
            message: message,
            status: 'writing'
          });
        }

        return prevState;
      });
    }
  }), []);
  var clearAssistant = (0, _react.useCallback)(function () {
    return setAssistantState({
      message: {},
      status: 'pending',
      messagesId: []
    });
  }, []);
  var onAssistantExecute = (0, _react.useCallback)( /*#__PURE__*/function () {
    var _ref3 = (0, _asyncToGenerator2.default)( /*#__PURE__*/_regenerator.default.mark(function _callee(_ref2, callback) {
      var value, conversationId;
      return _regenerator.default.wrap(function _callee$(_context) {
        while (1) {
          switch (_context.prev = _context.next) {
            case 0:
              value = _ref2.value, conversationId = _ref2.conversationId;

              if (value) {
                _context.next = 3;
                break;
              }

              return _context.abrupt("return");

            case 3:
              callback === null || callback === void 0 ? void 0 : callback();
              clearAssistant();
              _context.next = 7;
              return client.stackClient.fetchJSON('POST', "/ai/chat/conversations/".concat(conversationId), {
                q: value
              });

            case 7:
              setAssistantState(function (v) {
                return _objectSpread(_objectSpread({}, v), {}, {
                  status: 'pending'
                });
              });

            case 8:
            case "end":
              return _context.stop();
          }
        }
      }, _callee);
    }));

    return function (_x, _x2) {
      return _ref3.apply(this, arguments);
    };
  }(), [client, clearAssistant]);
  var value = (0, _react.useMemo)(function () {
    return {
      assistantState: assistantState,
      setAssistantState: setAssistantState,
      clearAssistant: clearAssistant,
      onAssistantExecute: onAssistantExecute
    };
  }, [assistantState, clearAssistant, onAssistantExecute]);
  return /*#__PURE__*/_react.default.createElement(AssistantContext.Provider, {
    value: value
  }, children);
};

var _default = /*#__PURE__*/_react.default.memo(AssistantProvider);

exports.default = _default;