@stssocialst-stp/fe-payroll 0.3.5 → 0.3.7

package/dist/index.js

@@ -762,7 +762,12 @@ var messages_en = {
 	"payroll.bistp.status.pending": "Sent to Bank",
 	"payroll.bistp.status.skippedNib": "No NIB",
 	"payroll.bistp.status.sendFailed": "Send Failed",
+	"payroll.bistp.status.enviados": "Sent to BISTP",
 	"payroll.bistp.progress": "BISTP Progress",
+	"payroll.async.generating": "Generating beneficiaries in background...",
+	"payroll.async.createdSoFar": "{benefitCount} created so far",
+	"payroll.async.note": "This process may take a few minutes. You can close this page - the process continues in the background.",
+	"payroll.async.timedOut": "The background generation process took too long. Please refresh this page to check the current status.",
 	"payroll.route.beneficiariesManagement": "Beneficiary Management",
 	"payroll.beneficiaries.management.title": "Beneficiary Management",
 	"payroll.beneficiaries.management.noPermission": "You do not have permission to manage beneficiaries.",
@@ -3211,6 +3216,99 @@ var mapDispatchToProps$b = function mapDispatchToProps(dispatch) {
 };
 var PaymentApproveForPaymentSummary = reactRedux.connect(mapStateToProps$b, mapDispatchToProps$b)(PaymentApproveForPaymentDialog);
 
+var POLL_INTERVAL_MS = 5000;
+var TIMEOUT_MS = 600000;
+var COUNT_QUERY = function COUNT_QUERY(payrollUuid) {
+  return "\n  {\n    benefitConsumptionByPayroll(payrollUuid: \"".concat(payrollUuid, "\", first: 0, isDeleted: false) {\n      totalCount\n    }\n  }\n");
+};
+function useAsyncPayrollProgress(_ref) {
+  var payrollUuid = _ref.payrollUuid,
+    payrollStatus = _ref.payrollStatus;
+  var dispatch = reactRedux.useDispatch();
+  var _useState = React.useState(0),
+    _useState2 = _slicedToArray__default["default"](_useState, 2),
+    benefitCount = _useState2[0],
+    setBenefitCount = _useState2[1];
+  var _useState3 = React.useState(false),
+    _useState4 = _slicedToArray__default["default"](_useState3, 2),
+    isGenerating = _useState4[0],
+    setIsGenerating = _useState4[1];
+  var _useState5 = React.useState(false),
+    _useState6 = _slicedToArray__default["default"](_useState5, 2),
+    timedOut = _useState6[0],
+    setTimedOut = _useState6[1];
+  var pollRef = React.useRef(null);
+  var startTimeRef = React.useRef(null);
+  var prevCountRef = React.useRef(0);
+  var stableCountRounds = React.useRef(0);
+  var stopPolling = React.useCallback(function () {
+    if (pollRef.current) {
+      clearInterval(pollRef.current);
+      pollRef.current = null;
+    }
+  }, []);
+  var fetchCount = React.useCallback(function () {
+    if (!payrollUuid) return;
+    dispatch(feCore.graphql(COUNT_QUERY(payrollUuid), 'POLL_BENEFIT_COUNT')).then(function (response) {
+      var _response$payload$dat, _response$payload;
+      var count = (_response$payload$dat = response === null || response === void 0 || (_response$payload = response.payload) === null || _response$payload === void 0 || (_response$payload = _response$payload.data) === null || _response$payload === void 0 || (_response$payload = _response$payload.benefitConsumptionByPayroll) === null || _response$payload === void 0 ? void 0 : _response$payload.totalCount) !== null && _response$payload$dat !== void 0 ? _response$payload$dat : 0;
+      if (count === 0) {
+        setBenefitCount(count);
+        setIsGenerating(true);
+        stableCountRounds.current = 0;
+        prevCountRef.current = count;
+        return;
+      }
+      setBenefitCount(count);
+      if (count === prevCountRef.current) {
+        stableCountRounds.current += 1;
+      } else {
+        stableCountRounds.current = 0;
+      }
+      prevCountRef.current = count;
+      if (stableCountRounds.current >= 2) {
+        setIsGenerating(false);
+        stopPolling();
+      }
+    })["catch"](function () {
+      setBenefitCount(function (c) {
+        return Math.max(c, 0);
+      });
+    });
+  }, [payrollUuid, dispatch, stopPolling]);
+  React.useEffect(function () {
+    var shouldPoll = payrollUuid && payrollStatus === 'PENDING_APPROVAL';
+    if (shouldPoll && !pollRef.current) {
+      setTimedOut(false);
+      startTimeRef.current = Date.now();
+      stableCountRounds.current = 0;
+      prevCountRef.current = 0;
+      fetchCount();
+      pollRef.current = setInterval(function () {
+        if (Date.now() - startTimeRef.current >= TIMEOUT_MS) {
+          setIsGenerating(false);
+          setTimedOut(true);
+          stopPolling();
+          return;
+        }
+        fetchCount();
+      }, POLL_INTERVAL_MS);
+    }
+    if (!shouldPoll && pollRef.current) {
+      setIsGenerating(false);
+      stopPolling();
+    }
+    return function () {
+      return stopPolling();
+    };
+  }, [payrollUuid, payrollStatus, fetchCount, stopPolling]);
+  return {
+    isGenerating: isGenerating,
+    benefitCount: benefitCount,
+    timedOut: timedOut
+  };
+}
+
 var useStyles$b = styles$7.makeStyles(function (theme) {
   return {
     paper: theme.paper.paper,
@@ -3257,12 +3355,64 @@ function PayrollTab(_ref) {
   var modulesManager = feCore.useModulesManager();
   var _useTranslations = feCore.useTranslations(MODULE_NAME, modulesManager),
     formatMessage = _useTranslations.formatMessage;
+  var _useAsyncPayrollProgr = useAsyncPayrollProgress({
+      payrollUuid: payrollUuid,
+      payrollStatus: payroll === null || payroll === void 0 ? void 0 : payroll.status
+    }),
+    isGenerating = _useAsyncPayrollProgr.isGenerating,
+    benefitCount = _useAsyncPayrollProgr.benefitCount,
+    timedOut = _useAsyncPayrollProgr.timedOut;
+  var _useState3 = React.useState(false),
+    _useState4 = _slicedToArray__default["default"](_useState3, 2),
+    generationComplete = _useState4[0],
+    setGenerationComplete = _useState4[1];
+  var _useState5 = React.useState(0),
+    _useState6 = _slicedToArray__default["default"](_useState5, 2),
+    refreshBenefitsKey = _useState6[0],
+    setRefreshBenefitsKey = _useState6[1];
+  React.useEffect(function () {
+    if (!isGenerating && benefitCount > 0 && !generationComplete) {
+      setGenerationComplete(true);
+      setRefreshBenefitsKey(function (k) {
+        return k + 1;
+      });
+    }
+  }, [isGenerating, benefitCount, generationComplete]);
   var downloadPayrollData = function downloadPayrollData(payrollUuid, payrollName) {
     downloadPayroll(payrollUuid, payrollName);
   };
   return /*#__PURE__*/React__default["default"].createElement(core.Paper, {
     className: classes.paper
+  }, (isGenerating || timedOut) && payrollUuid && !isPayrollFromFailedInvoices && /*#__PURE__*/React__default["default"].createElement(core.Grid, {
+    container: true,
+    spacing: 2,
+    style: {
+      padding: '16px'
+    }
   }, /*#__PURE__*/React__default["default"].createElement(core.Grid, {
+    item: true,
+    xs: 12
+  }, /*#__PURE__*/React__default["default"].createElement(core.Typography, {
+    variant: "body1"
+  }, timedOut ? formatMessage('payroll.async.timedOut') : formatMessage('payroll.async.generating')), !timedOut && /*#__PURE__*/React__default["default"].createElement(core.LinearProgress, {
+    variant: "indeterminate",
+    style: {
+      marginTop: '8px'
+    }
+  }), /*#__PURE__*/React__default["default"].createElement(core.Typography, {
+    variant: "caption",
+    style: {
+      display: 'block',
+      marginTop: '4px'
+    }
+  }, !timedOut && formatMessage('payroll.async.createdSoFar', {
+    benefitCount: benefitCount
+  })), /*#__PURE__*/React__default["default"].createElement(core.Typography, {
+    variant: "caption",
+    style: {
+      display: 'block'
+    }
+  }, !timedOut && formatMessage('payroll.async.note')))), /*#__PURE__*/React__default["default"].createElement(core.Grid, {
     container: true,
     className: "".concat(classes.tableTitle, " ").concat(classes.tabs)
   }, /*#__PURE__*/React__default["default"].createElement("div", {
@@ -3313,7 +3463,8 @@ function PayrollTab(_ref) {
     payrollUuid: payrollUuid,
     isInTask: isInTask,
     isPayrollFromFailedInvoices: isPayrollFromFailedInvoices,
-    payroll: payroll
+    payroll: payroll,
+    refreshBenefitsKey: refreshBenefitsKey
   }));
 }
 
@@ -5706,13 +5857,15 @@ function BenefitConsumptionsTabPanel(_ref2) {
   var value = _ref2.value,
     rights = _ref2.rights,
     payrollUuid = _ref2.payrollUuid,
-    isPayrollFromFailedInvoices = _ref2.isPayrollFromFailedInvoices;
+    isPayrollFromFailedInvoices = _ref2.isPayrollFromFailedInvoices,
+    refreshBenefitsKey = _ref2.refreshBenefitsKey;
   return /*#__PURE__*/React__default["default"].createElement(feCore.PublishedComponent, {
     pubRef: "policyHolder.TabPanel",
     module: "payroll",
     index: BENEFIT_CONSUMPTION_LIST_TAB_VALUE,
     value: value
   }, rights.includes(RIGHT_PAYROLL_SEARCH) && payrollUuid && /*#__PURE__*/React__default["default"].createElement(BenefitConsumptionSearcher$1, {
+    key: refreshBenefitsKey,
     rights: rights,
     payrollUuid: payrollUuid,
     isPayrollFromFailedInvoices: isPayrollFromFailedInvoices
@@ -6380,11 +6533,93 @@ function useBeneficiarios() {
     _useState6 = _slicedToArray__default["default"](_useState5, 2),
     error = _useState6[0],
     setError = _useState6[1];
-  function request(_x) {
+  function poll(_x) {
+    return _poll.apply(this, arguments);
+  }
+  function _poll() {
+    _poll = _asyncToGenerator__default["default"](/*#__PURE__*/_regeneratorRuntime__default["default"].mark(function _callee(clientMutationId) {
+      var INTERVAL_MS, TIMEOUT_MS, started, _result$data, _node$jsonExt$output, _node$jsonExt, resp, result, node;
+      return _regeneratorRuntime__default["default"].wrap(function (_context) {
+        while (1) switch (_context.prev = _context.next) {
+          case 0:
+            INTERVAL_MS = 5000;
+            TIMEOUT_MS = 20 * 60 * 1000; // 20 minutos
+            started = Date.now();
+          case 1:
+            if (!(Date.now() - started < TIMEOUT_MS)) {
+              _context.next = 8;
+              break;
+            }
+            _context.next = 2;
+            return new Promise(function (resolve) {
+              return setTimeout(resolve, INTERVAL_MS);
+            });
+          case 2:
+            _context.next = 3;
+            return fetch("".concat(feCore.baseApiUrl, "/graphql"), {
+              method: 'POST',
+              headers: buildHeaders(),
+              credentials: 'same-origin',
+              body: JSON.stringify({
+                query: "\n            query {\n              mutationLogs(clientMutationId: \"".concat(clientMutationId, "\") {\n                edges { node { status error jsonExt clientMutationLabel } }\n              }\n            }\n          ")
+              })
+            });
+          case 3:
+            resp = _context.sent;
+            _context.next = 4;
+            return resp.json();
+          case 4:
+            result = _context.sent;
+            node = result === null || result === void 0 || (_result$data = result.data) === null || _result$data === void 0 || (_result$data = _result$data.mutationLogs) === null || _result$data === void 0 || (_result$data = _result$data.edges) === null || _result$data === void 0 || (_result$data = _result$data[0]) === null || _result$data === void 0 ? void 0 : _result$data.node;
+            if (node) {
+              _context.next = 5;
+              break;
+            }
+            return _context.abrupt("continue", 1);
+          case 5:
+            // Update UI with label if needed
+            setOutput("A processar: ".concat(node.clientMutationLabel, "..."));
+            if (!(node.status === 2)) {
+              _context.next = 6;
+              break;
+            }
+            return _context.abrupt("return", {
+              success: true,
+              output: (_node$jsonExt$output = (_node$jsonExt = node.jsonExt) === null || _node$jsonExt === void 0 ? void 0 : _node$jsonExt.output) !== null && _node$jsonExt$output !== void 0 ? _node$jsonExt$output : null,
+              error: null
+            });
+          case 6:
+            if (!(node.status === 1)) {
+              _context.next = 7;
+              break;
+            }
+            return _context.abrupt("return", {
+              success: false,
+              output: null,
+              error: node.error
+            });
+          case 7:
+            _context.next = 1;
+            break;
+          case 8:
+            return _context.abrupt("return", {
+              success: false,
+              output: null,
+              error: 'Timeout: operação excedeu 20 minutos'
+            });
+          case 9:
+          case "end":
+            return _context.stop();
+        }
+      }, _callee);
+    }));
+    return _poll.apply(this, arguments);
+  }
+  function request(_x2) {
     return _request.apply(this, arguments);
   }
   function _request() {
-    _request = _asyncToGenerator__default["default"](/*#__PURE__*/_regeneratorRuntime__default["default"].mark(function _callee(path) {
+    _request = _asyncToGenerator__default["default"](/*#__PURE__*/_regeneratorRuntime__default["default"].mark(function _callee2(path) {
       var _ref,
         _ref$multipart,
         multipart,
@@ -6396,15 +6631,15 @@ function useBeneficiarios() {
         contentType,
         _data,
         message,
-        _args = arguments,
+        _args2 = arguments,
         _t;
-      return _regeneratorRuntime__default["default"].wrap(function (_context) {
-        while (1) switch (_context.prev = _context.next) {
+      return _regeneratorRuntime__default["default"].wrap(function (_context2) {
+        while (1) switch (_context2.prev = _context2.next) {
           case 0:
-            _ref = _args.length > 1 && _args[1] !== undefined ? _args[1] : {}, _ref$multipart = _ref.multipart, multipart = _ref$multipart === void 0 ? false : _ref$multipart, _ref$body = _ref.body, body = _ref$body === void 0 ? null : _ref$body;
+            _ref = _args2.length > 1 && _args2[1] !== undefined ? _args2[1] : {}, _ref$multipart = _ref.multipart, multipart = _ref$multipart === void 0 ? false : _ref$multipart, _ref$body = _ref.body, body = _ref$body === void 0 ? null : _ref$body;
             setLoading(true);
             setError(null);
-            _context.prev = 1;
+            _context2.prev = 1;
             init = {
               method: 'POST',
               credentials: 'same-origin'
@@ -6415,54 +6650,50 @@ function useBeneficiarios() {
               init.headers = buildHeaders();
               if (body) init.body = JSON.stringify(body);
             }
-            _context.next = 2;
+            _context2.next = 2;
             return fetch("".concat(BASE).concat(path), init);
           case 2:
-            response = _context.sent;
+            response = _context2.sent;
             data = null;
             contentType = response.headers.get('content-type') || '';
             if (!contentType.includes('application/json')) {
-              _context.next = 4;
+              _context2.next = 4;
               break;
             }
-            _context.next = 3;
+            _context2.next = 3;
             return response.json();
           case 3:
-            data = _context.sent;
+            data = _context2.sent;
           case 4:
             if (response.ok) {
-              _context.next = 5;
+              _context2.next = 5;
               break;
             }
             message = ((_data = data) === null || _data === void 0 ? void 0 : _data.message) || "HTTP ".concat(response.status);
             setError(message);
-            return _context.abrupt("return", {
+            return _context2.abrupt("return", {
               ok: false,
               message: message,
               data: data
             });
           case 5:
-            return _context.abrupt("return", {
+            return _context2.abrupt("return", {
               ok: true,
               data: data
             });
           case 6:
-            _context.prev = 6;
-            _t = _context["catch"](1);
+            _context2.prev = 6;
+            _t = _context2["catch"](1);
             setError(_t.message);
-            return _context.abrupt("return", {
+            return _context2.abrupt("return", {
               ok: false,
               message: _t.message
             });
           case 7:
-            _context.prev = 7;
-            setLoading(false);
-            return _context.finish(7);
-          case 8:
           case "end":
-            return _context.stop();
+            return _context2.stop();
         }
-      }, _callee, null, [[1, 6, 7, 8]]);
+      }, _callee2, null, [[1, 6]]);
     }));
     return _request.apply(this, arguments);
   }
@@ -6470,50 +6701,53 @@ function useBeneficiarios() {
     return _backup.apply(this, arguments);
   }
   function _backup() {
-    _backup = _asyncToGenerator__default["default"](/*#__PURE__*/_regeneratorRuntime__default["default"].mark(function _callee2() {
+    _backup = _asyncToGenerator__default["default"](/*#__PURE__*/_regeneratorRuntime__default["default"].mark(function _callee3() {
       var payrollId,
         _yield$request,
         ok,
         data,
-        blob,
-        a,
-        _args2 = arguments;
-      return _regeneratorRuntime__default["default"].wrap(function (_context2) {
-        while (1) switch (_context2.prev = _context2.next) {
+        pollResult,
+        _args3 = arguments;
+      return _regeneratorRuntime__default["default"].wrap(function (_context3) {
+        while (1) switch (_context3.prev = _context3.next) {
           case 0:
-            payrollId = _args2.length > 0 && _args2[0] !== undefined ? _args2[0] : null;
-            _context2.next = 1;
+            payrollId = _args3.length > 0 && _args3[0] !== undefined ? _args3[0] : null;
+            _context3.next = 1;
             return request('/backup/', {
               body: payrollId ? {
                 payroll_id: payrollId
               } : {}
             });
           case 1:
-            _yield$request = _context2.sent;
+            _yield$request = _context3.sent;
             ok = _yield$request.ok;
             data = _yield$request.data;
-            if (ok) {
-              blob = new Blob([JSON.stringify(data, null, 2)], {
-                type: 'application/json'
-              });
-              a = document.createElement('a');
-              a.href = URL.createObjectURL(blob);
-              a.download = "backup_beneficiarios_".concat(new Date().toISOString().slice(0, 19).replace(/:/g, '-'), ".json");
-              document.body.appendChild(a);
-              a.click();
-              document.body.removeChild(a);
-              URL.revokeObjectURL(a.href);
-              setOutput("Backup conclu\xEDdo: ".concat(JSON.stringify(data.counts)));
+            if (!ok) {
+              _context3.next = 3;
+              break;
             }
-            return _context2.abrupt("return", {
+            setOutput('Backup iniciado. A aguardar conclusão...');
+            _context3.next = 2;
+            return poll(data.clientMutationId);
+          case 2:
+            pollResult = _context3.sent;
+            if (pollResult.success) {
+              setOutput("Backup conclu\xEDdo: ".concat(pollResult.output));
+              // Handle download trigger if file info is in jsonExt
+            } else {
+              setError(pollResult.error);
+            }
+          case 3:
+            setLoading(false);
+            return _context3.abrupt("return", {
               ok: ok,
               data: data
             });
-          case 2:
+          case 4:
           case "end":
-            return _context2.stop();
+            return _context3.stop();
         }
-      }, _callee2);
+      }, _callee3);
     }));
     return _backup.apply(this, arguments);
   }
@@ -6521,7 +6755,7 @@ function useBeneficiarios() {
     return _limpar.apply(this, arguments);
   }
   function _limpar() {
-    _limpar = _asyncToGenerator__default["default"](/*#__PURE__*/_regeneratorRuntime__default["default"].mark(function _callee3() {
+    _limpar = _asyncToGenerator__default["default"](/*#__PURE__*/_regeneratorRuntime__default["default"].mark(function _callee4() {
       var _ref2,
         _ref2$dryRun,
         dryRun,
@@ -6530,12 +6764,13 @@ function useBeneficiarios() {
         _yield$request2,
         ok,
         data,
-        _args3 = arguments;
-      return _regeneratorRuntime__default["default"].wrap(function (_context3) {
-        while (1) switch (_context3.prev = _context3.next) {
+        pollResult,
+        _args4 = arguments;
+      return _regeneratorRuntime__default["default"].wrap(function (_context4) {
+        while (1) switch (_context4.prev = _context4.next) {
           case 0:
-            _ref2 = _args3.length > 0 && _args3[0] !== undefined ? _args3[0] : {}, _ref2$dryRun = _ref2.dryRun, dryRun = _ref2$dryRun === void 0 ? true : _ref2$dryRun, _ref2$forceSkipFinanc = _ref2.forceSkipFinancial, forceSkipFinancial = _ref2$forceSkipFinanc === void 0 ? false : _ref2$forceSkipFinanc;
-            _context3.next = 1;
+            _ref2 = _args4.length > 0 && _args4[0] !== undefined ? _args4[0] : {}, _ref2$dryRun = _ref2.dryRun, dryRun = _ref2$dryRun === void 0 ? true : _ref2$dryRun, _ref2$forceSkipFinanc = _ref2.forceSkipFinancial, forceSkipFinancial = _ref2$forceSkipFinanc === void 0 ? false : _ref2$forceSkipFinanc;
+            _context4.next = 1;
             return request('/limpar/', {
               body: {
                 username: 'admin',
@@ -6544,27 +6779,42 @@ function useBeneficiarios() {
               }
             });
           case 1:
-            _yield$request2 = _context3.sent;
+            _yield$request2 = _context4.sent;
             ok = _yield$request2.ok;
             data = _yield$request2.data;
-            if (ok) setOutput(data.output);
-            return _context3.abrupt("return", {
+            if (!ok) {
+              _context4.next = 3;
+              break;
+            }
+            setOutput('Limpeza iniciada. A aguardar conclusão...');
+            _context4.next = 2;
+            return poll(data.clientMutationId);
+          case 2:
+            pollResult = _context4.sent;
+            if (pollResult.success) {
+              setOutput(pollResult.output);
+            } else {
+              setError(pollResult.error);
+            }
+          case 3:
+            setLoading(false);
+            return _context4.abrupt("return", {
               ok: ok,
               data: data
             });
-          case 2:
+          case 4:
           case "end":
-            return _context3.stop();
+            return _context4.stop();
         }
-      }, _callee3);
+      }, _callee4);
     }));
     return _limpar.apply(this, arguments);
   }
-  function importar(_x2) {
+  function importar(_x3) {
     return _importar.apply(this, arguments);
   }
   function _importar() {
-    _importar = _asyncToGenerator__default["default"](/*#__PURE__*/_regeneratorRuntime__default["default"].mark(function _callee4(file) {
+    _importar = _asyncToGenerator__default["default"](/*#__PURE__*/_regeneratorRuntime__default["default"].mark(function _callee5(file) {
       var _ref3,
         _ref3$dryRun,
         dryRun,
@@ -6576,11 +6826,12 @@ function useBeneficiarios() {
         _yield$request3,
         ok,
         data,
-        _args4 = arguments;
-      return _regeneratorRuntime__default["default"].wrap(function (_context4) {
-        while (1) switch (_context4.prev = _context4.next) {
+        pollResult,
+        _args5 = arguments;
+      return _regeneratorRuntime__default["default"].wrap(function (_context5) {
+        while (1) switch (_context5.prev = _context5.next) {
           case 0:
-            _ref3 = _args4.length > 1 && _args4[1] !== undefined ? _args4[1] : {}, _ref3$dryRun = _ref3.dryRun, dryRun = _ref3$dryRun === void 0 ? false : _ref3$dryRun, payrollId = _ref3.payrollId, benefitPlanId = _ref3.benefitPlanId, benefitType = _ref3.benefitType, sheet = _ref3.sheet;
+            _ref3 = _args5.length > 1 && _args5[1] !== undefined ? _args5[1] : {}, _ref3$dryRun = _ref3.dryRun, dryRun = _ref3$dryRun === void 0 ? false : _ref3$dryRun, payrollId = _ref3.payrollId, benefitPlanId = _ref3.benefitPlanId, benefitType = _ref3.benefitType, sheet = _ref3.sheet;
             fd = new FormData();
             fd.append('file', file);
             fd.append('username', 'admin');
@@ -6589,33 +6840,48 @@ function useBeneficiarios() {
             if (benefitPlanId) fd.append('benefit_plan_id', benefitPlanId);
             if (benefitType) fd.append('benefit_type', benefitType);
             if (sheet) fd.append('sheet', sheet);
-            _context4.next = 1;
+            _context5.next = 1;
             return request('/importar/', {
               multipart: true,
               body: fd
             });
           case 1:
-            _yield$request3 = _context4.sent;
+            _yield$request3 = _context5.sent;
             ok = _yield$request3.ok;
             data = _yield$request3.data;
-            if (ok) setOutput(data.output);
-            return _context4.abrupt("return", {
+            if (!ok) {
+              _context5.next = 3;
+              break;
+            }
+            setOutput('Importação iniciada. A aguardar conclusão...');
+            _context5.next = 2;
+            return poll(data.clientMutationId);
+          case 2:
+            pollResult = _context5.sent;
+            if (pollResult.success) {
+              setOutput(pollResult.output);
+            } else {
+              setError(pollResult.error);
+            }
+          case 3:
+            setLoading(false);
+            return _context5.abrupt("return", {
               ok: ok,
               data: data
             });
-          case 2:
+          case 4:
           case "end":
-            return _context4.stop();
+            return _context5.stop();
         }
-      }, _callee4);
+      }, _callee5);
     }));
     return _importar.apply(this, arguments);
   }
-  function restore(_x3) {
+  function restore(_x4) {
     return _restore.apply(this, arguments);
   }
   function _restore() {
-    _restore = _asyncToGenerator__default["default"](/*#__PURE__*/_regeneratorRuntime__default["default"].mark(function _callee5(file) {
+    _restore = _asyncToGenerator__default["default"](/*#__PURE__*/_regeneratorRuntime__default["default"].mark(function _callee6(file) {
       var _ref4,
         _ref4$dryRun,
         dryRun,
@@ -6625,35 +6891,51 @@ function useBeneficiarios() {
         _yield$request4,
         ok,
         data,
-        _args5 = arguments;
-      return _regeneratorRuntime__default["default"].wrap(function (_context5) {
-        while (1) switch (_context5.prev = _context5.next) {
+        pollResult,
+        _args6 = arguments;
+      return _regeneratorRuntime__default["default"].wrap(function (_context6) {
+        while (1) switch (_context6.prev = _context6.next) {
           case 0:
-            _ref4 = _args5.length > 1 && _args5[1] !== undefined ? _args5[1] : {}, _ref4$dryRun = _ref4.dryRun, dryRun = _ref4$dryRun === void 0 ? true : _ref4$dryRun, _ref4$skipPhases = _ref4.skipPhases, skipPhases = _ref4$skipPhases === void 0 ? [] : _ref4$skipPhases;
+            _ref4 = _args6.length > 1 && _args6[1] !== undefined ? _args6[1] : {}, _ref4$dryRun = _ref4.dryRun, dryRun = _ref4$dryRun === void 0 ? true : _ref4$dryRun, _ref4$skipPhases = _ref4.skipPhases, skipPhases = _ref4$skipPhases === void 0 ? [] : _ref4$skipPhases;
             fd = new FormData();
             fd.append('file', file);
             fd.append('username', 'admin');
             fd.append('dry_run', String(dryRun));
             if (skipPhases.length) fd.append('skip_phases', skipPhases.join(','));
-            _context5.next = 1;
+            _context6.next = 1;
             return request('/restore/', {
               multipart: true,
               body: fd
             });
           case 1:
-            _yield$request4 = _context5.sent;
+            _yield$request4 = _context6.sent;
             ok = _yield$request4.ok;
             data = _yield$request4.data;
-            if (ok) setOutput(data.output);
-            return _context5.abrupt("return", {
+            if (!ok) {
+              _context6.next = 3;
+              break;
+            }
+            setOutput('Restore iniciado. A aguardar conclusão...');
+            _context6.next = 2;
+            return poll(data.clientMutationId);
+          case 2:
+            pollResult = _context6.sent;
+            if (pollResult.success) {
+              setOutput(pollResult.output);
+            } else {
+              setError(pollResult.error);
+            }
+          case 3:
+            setLoading(false);
+            return _context6.abrupt("return", {
               ok: ok,
               data: data
             });
-          case 2:
+          case 4:
           case "end":
-            return _context5.stop();
+            return _context6.stop();
         }
-      }, _callee5);
+      }, _callee6);
     }));
     return _restore.apply(this, arguments);
   }