@apulis/group-selector/lib/index.js

change user group at navbar

"use strict";

function _typeof(obj) { "@babel/helpers - typeof"; if (typeof Symbol === "function" && typeof Symbol.iterator === "symbol") { _typeof = function _typeof(obj) { return typeof obj; }; } else { _typeof = function _typeof(obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }; } return _typeof(obj); }

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

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

var _querystring = require("querystring");

var _service = require("./service");

var _provider = require("@apulis/provider");

var _antd = require("antd");

var _icons = require("@ant-design/icons");

require("./index.less");

var _zh_CN = _interopRequireDefault(require("antd/lib/locale/zh_CN"));

var _en_US = _interopRequireDefault(require("antd/lib/locale/en_US"));

var _ru_RU = _interopRequireDefault(require("antd/lib/locale/ru_RU"));

var _LicenseTip = _interopRequireDefault(require("./components/LicenseTip"));

var _constants = require("@apulis/constants");

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

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 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); }); }; }

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(arr, i) { var _i = arr && (typeof Symbol !== "undefined" && arr[Symbol.iterator] || arr["@@iterator"]); if (_i == null) return; var _arr = []; var _n = true; var _d = false; var _s, _e; try { for (_i = _i.call(arr); !(_n = (_s = _i.next()).done); _n = true) { _arr.push(_s.value); if (i && _arr.length === i) break; } } catch (err) { _d = true; _e = err; } finally { try { if (!_n && _i["return"] != null) _i["return"](); } finally { if (_d) throw _e; } } return _arr; }

function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }

var Option = _antd.Select.Option;

var setLanguage = function setLanguage(locale) {
  localStorage.setItem('umi_locale', locale || 'en-US');
};

var getLanguage = function getLanguage() {
  var map = {
    'zh-CN': _zh_CN.default,
    'ru-RU': _ru_RU.default,
    'en-US': _en_US.default
  };

  if (!localStorage.getItem('umi_locale')) {
    setLanguage('en-US');
    return _en_US.default;
  }

  if (localStorage.getItem('umi_locale')) {
    var lang = localStorage.getItem('umi_locale') || 'en-US';
    return map[lang];
  }

  return _en_US.default;
};

var AGroupSelector = function AGroupSelector(_ref) {
  var _currentUser$userGrou, _currentUser$organiza, _currentUser$organiza2, _currentUser$organiza3, _currentUser$organiza4, _currentUser$userGrou2;

  var onChange = _ref.onChange,
      path = _ref.path,
      isShowLicenseTip = _ref.isShowLicenseTip,
      disabled = _ref.disabled;
  var intl = (0, _provider.useIntl)();

  var _useState = (0, _react.useState)({}),
      _useState2 = _slicedToArray(_useState, 2),
      currentUser = _useState2[0],
      setCurrentUser = _useState2[1];

  var _useState3 = (0, _react.useState)(3),
      _useState4 = _slicedToArray(_useState3, 2),
      currentGroup = _useState4[0],
      setCurrentGroup = _useState4[1];

  var _useState5 = (0, _react.useState)(100),
      _useState6 = _slicedToArray(_useState5, 2),
      expire = _useState6[0],
      setExpire = _useState6[1];

  var fetchLicenseInfo = /*#__PURE__*/function () {
    var _ref2 = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee() {
      var res, remain, queryString;
      return regeneratorRuntime.wrap(function _callee$(_context) {
        while (1) {
          switch (_context.prev = _context.next) {
            case 0:
              _context.next = 2;
              return (0, _service.getLicenseInfo)();

            case 2:
              res = _context.sent;

              if ((res === null || res === void 0 ? void 0 : res.code) === 0) {
                remain = Math.ceil(res.data.remain / (3600 * 24));
                setExpire(remain);
              } else if ((res === null || res === void 0 ? void 0 : res.code) === 530000001) {
                queryString = (0, _querystring.stringify)({
                  redirect: window.location.href
                });
                window.location.href = "".concat(window.location.origin, "/").concat(_constants.USER_LOGIN_URL, "?").concat(queryString, "&license=missing");
              } else {
                _antd.message.info(res.msg);
              }

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

    return function fetchLicenseInfo() {
      return _ref2.apply(this, arguments);
    };
  }();

  var fetchCurrent = /*#__PURE__*/function () {
    var _ref3 = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee2() {
      var _yield$getCurrentUser, code, data;

      return regeneratorRuntime.wrap(function _callee2$(_context2) {
        while (1) {
          switch (_context2.prev = _context2.next) {
            case 0:
              _context2.next = 2;
              return (0, _service.getCurrentUser)();

            case 2:
              _yield$getCurrentUser = _context2.sent;
              code = _yield$getCurrentUser.code;
              data = _yield$getCurrentUser.data;

              if (code === 0) {
                setCurrentUser(data);
              }

            case 6:
            case "end":
              return _context2.stop();
          }
        }
      }, _callee2);
    }));

    return function fetchCurrent() {
      return _ref3.apply(this, arguments);
    };
  }();

  var fetchRefreshToken = /*#__PURE__*/function () {
    var _ref4 = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee3() {
      var _yield$refreshToken, code, msg, data;

      return regeneratorRuntime.wrap(function _callee3$(_context3) {
        while (1) {
          switch (_context3.prev = _context3.next) {
            case 0:
              _context3.next = 2;
              return (0, _service.refreshToken)();

            case 2:
              _yield$refreshToken = _context3.sent;
              code = _yield$refreshToken.code;
              msg = _yield$refreshToken.msg;
              data = _yield$refreshToken.data;

              if (code === 0) {
                path ? window.location.href = "".concat(window.location.origin).concat(path) : window.location.reload();
                localStorage.token = data.token;
              } else {
                _antd.message.info(msg);
              }

            case 7:
            case "end":
              return _context3.stop();
          }
        }
      }, _callee3);
    }));

    return function fetchRefreshToken() {
      return _ref4.apply(this, arguments);
    };
  }();

  var submitGroup = /*#__PURE__*/function () {
    var _ref5 = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee4(id) {
      var _yield$changeGroupApi, code, msg;

      return regeneratorRuntime.wrap(function _callee4$(_context4) {
        while (1) {
          switch (_context4.prev = _context4.next) {
            case 0:
              _context4.next = 2;
              return (0, _service.changeGroup)(id);

            case 2:
              _yield$changeGroupApi = _context4.sent;
              code = _yield$changeGroupApi.code;
              msg = _yield$changeGroupApi.msg;

              if (code === 0) {
                fetchRefreshToken();
              } else {
                _antd.message.info(msg);
              }

            case 6:
            case "end":
              return _context4.stop();
          }
        }
      }, _callee4);
    }));

    return function submitGroup(_x) {
      return _ref5.apply(this, arguments);
    };
  }();

  var changeGroup = function changeGroup(v) {
    submitGroup(v);
    onChange === null || onChange === void 0 ? void 0 : onChange(v);
  };

  (0, _react.useEffect)(function () {
    fetchCurrent();
  }, []);
  (0, _react.useEffect)(function () {
    setCurrentGroup((currentUser === null || currentUser === void 0 ? void 0 : currentUser.userGroupId) || 3);

    if (isShowLicenseTip) {
      fetchLicenseInfo();
    } // 手动在iam编辑了用户组信息后，token失效，需要重新获取token


    if ((currentUser === null || currentUser === void 0 ? void 0 : currentUser.userGroupId) && (currentUser === null || currentUser === void 0 ? void 0 : currentUser.userGroups)) {
      var groupIds = currentUser === null || currentUser === void 0 ? void 0 : currentUser.userGroups.map(function (g) {
        return g.id;
      });
      var curGroupId = currentUser === null || currentUser === void 0 ? void 0 : currentUser.userGroupId;

      if (!groupIds.includes(curGroupId)) {
        console.log('token is not valid');
        localStorage.removeItem('token');
        fetchCurrent();
      }
    }
  }, [currentUser]);
  return /*#__PURE__*/_react.default.createElement("div", {
    style: {
      display: 'flex',
      alignItems: 'center'
    }
  }, isShowLicenseTip && expire <= 30 && /*#__PURE__*/_react.default.createElement(_LicenseTip.default, {
    expire: expire
  }), (currentUser === null || currentUser === void 0 ? void 0 : (_currentUser$userGrou = currentUser.userGroup) === null || _currentUser$userGrou === void 0 ? void 0 : _currentUser$userGrou.type) !== 1 && /*#__PURE__*/_react.default.createElement("div", {
    className: "nav-item"
  }, [3, 4].includes(currentUser === null || currentUser === void 0 ? void 0 : (_currentUser$organiza = currentUser.organization) === null || _currentUser$organiza === void 0 ? void 0 : _currentUser$organiza.status) && /*#__PURE__*/_react.default.createElement(_antd.Popover, {
    placement: "bottom",
    content: intl.getMessage('groupSelector.warnTip', '该组织被冻结，无法使用任何资源')
  }, /*#__PURE__*/_react.default.createElement(_icons.WarningFilled, {
    className: "warn-icon"
  })), /*#__PURE__*/_react.default.createElement(_icons.TeamOutlined, {
    className: "org-icon"
  }), /*#__PURE__*/_react.default.createElement(_antd.Popover, {
    placement: "bottom",
    content: /*#__PURE__*/_react.default.createElement(_react.default.Fragment, null, /*#__PURE__*/_react.default.createElement("div", null, intl.getMessage('groupSelector.organizationAccount', '组织账号'), ":", ' ', currentUser === null || currentUser === void 0 ? void 0 : (_currentUser$organiza2 = currentUser.organization) === null || _currentUser$organiza2 === void 0 ? void 0 : _currentUser$organiza2.account), /*#__PURE__*/_react.default.createElement("div", null, intl.getMessage('groupSelector.organizationName', '组织名称'), ":", ' ', currentUser === null || currentUser === void 0 ? void 0 : (_currentUser$organiza3 = currentUser.organization) === null || _currentUser$organiza3 === void 0 ? void 0 : _currentUser$organiza3.nickName))
  }, /*#__PURE__*/_react.default.createElement("span", {
    className: "ellipsis"
  }, currentUser === null || currentUser === void 0 ? void 0 : (_currentUser$organiza4 = currentUser.organization) === null || _currentUser$organiza4 === void 0 ? void 0 : _currentUser$organiza4.nickName))), /*#__PURE__*/_react.default.createElement(_antd.Select, {
    style: {
      width: '200px'
    },
    value: currentGroup,
    onChange: function onChange(v) {
      changeGroup(v);
    },
    disabled: disabled
  }, currentUser === null || currentUser === void 0 ? void 0 : (_currentUser$userGrou2 = currentUser.userGroups) === null || _currentUser$userGrou2 === void 0 ? void 0 : _currentUser$userGrou2.map(function (g) {
    return /*#__PURE__*/_react.default.createElement(Option, {
      key: g.id,
      value: g.id
    }, g.account);
  })));
};

var ProviderWarp = function ProviderWarp(props) {
  return /*#__PURE__*/_react.default.createElement(_antd.ConfigProvider, {
    locale: getLanguage()
  }, /*#__PURE__*/_react.default.createElement(_provider.ConfigProviderWrap, null, /*#__PURE__*/_react.default.createElement(AGroupSelector, props)));
};

var _default = ProviderWarp;
exports.default = _default;