AvatarDropdown.jsx

import React, { useCallback, useState, useMemo, useEffect } from 'react';
import { LogoutOutlined, LockOutlined } from '@ant-design/icons';
import { Menu, Spin, Button, Form, Modal, message, Avatar, Select } from 'antd';
import { history, useModel } from '@umijs/max';
import { stringify } from 'querystring';
import HeaderDropdown from '../HeaderDropdown';
import styles from './index.less';
import { fakeAccountLoginOut, changePwd } from '@/services/login';
import InitForm from '@/components/InitForm';
import { doFetch } from '@/utils/doFetch';
import { useRequest } from 'ahooks';
import AES from 'crypto-js/aes';
import ECB from 'crypto-js/mode-ecb';
import Pkcs7 from 'crypto-js/pad-pkcs7';
import Utf8 from 'crypto-js/enc-utf8';
import moment from 'moment';
import io from 'socket.io-client';
const { NODE_ENV } = process.env;
const websocketUrl = NODE_ENV == "development" ? "ws://192.168.40.110:9933" : `${window.location.protocol == "http:" ? "ws" : "wss"}://${window.location.host}`;
const socket = io(websocketUrl, {
  reconnectionDelay: 4000,
  transports: ['websocket'],
  autoConnect: true,
});
/**
 * 退出登录，并且将当前的 url 保存
 */
const loginOut = async () => {
  const { logoutRedirectUrl } = await fakeAccountLoginOut();
  if (window.location.pathname !== '/user/login') {
    localStorage.clear();
    window.location.replace(logoutRedirectUrl);
  }
};

const AvatarDropdown = ({ menu }) => {
  const getUserApp = useRequest(async () => {
    let res = await doFetch({ url: '/auth/sysUser/userApp/selection', params: {} });
    return res?.data || {};
  });
  const { initialState, setInitialState } = useModel('@@initialState');
  const { allWebsocket, changeWebsocket } = useModel('useGlobal');
  const [visible, cv] = useState(false),
    [formRef] = Form.useForm(),
    { run, loading } = useRequest(doFetch, {
      manual: true,
      formatResult: (res) => res,
      onSuccess: (result, params) => {
        if (result.code == '0000') {
          cv(false);
          message.success('密码修改成功,请重新登录', 2, () => {
            setInitialState((s) => ({
              ...s,
              currentUser: undefined,
              newMenu: undefined,
            }));
            loginOut();
          });
        }
      },
    });


  useEffect(() => {
    changeWebsocket((s) => ({ ...s, commonWebsocket: socket }));
    socket.on('iotData', (data) => {
      changeWebsocket((s) => ({ ...s, commonWebmsg: data }));
    });
  }, []);
  const fields = useMemo(() => {
    return {
      password: {
        value: null,
        type: 'password',
        title: '密码',
        name: ['password'],
        required: true,
      },
      newPassword: {
        value: null,
        type: 'password',
        title: '新密码',
        name: ['newPassword'],
        required: true,
      },
      confirmPassword: {
        value: null,
        type: 'password',
        title: '确认密码',
        name: ['confirmPassword'],
        required: true,
        checkConfirm: (rule, value) => {
          if (value && value !== formRef.getFieldValue('newPassword')) {
            return Promise.reject('2次密码不一致！');
          } else {
            return Promise.resolve();
          }
        },
      },
    };
  }, []);

  const onMenuClick = (event) => {
    const { key } = event;
    if (key === 'logout') {
      loginOut();
      return;
    } else if (key == 'changepwd') {
      cv(true);
      return;
    }
  };

  const loadings = (
    <HeaderDropdown overlay={menuHeaderDropdown}>
      <React.Fragment>
        <Button onClick={onMenuClick}>
          <Spin
            size="small"
            style={{
              marginLeft: 8,
              marginRight: 8,
            }}
          />
          退出
        </Button>
      </React.Fragment>
    </HeaderDropdown>
  );

  if (!initialState) {
    return loadings;
  }

  const { currentUser } = initialState;
  if (!currentUser || !currentUser.username) {
    return loadings;
  }
  // console.log(
  //   getUserApp?.data?.dataList?.filter((it) => it?.appId == getUserApp?.data?.appId)[0]?.value,
  // );

  let saveData = (values) => {
    let newfields = JSON.parse(JSON.stringify(values));
    delete newfields.confirmPassword;

    let timestamp = moment().valueOf().toString() + 'acb';
    let newtimestamp = AES.encrypt(timestamp, Utf8.parse('NANGAODEAESKEY--'), {
      mode: ECB,
      padding: Pkcs7,
    }).toString();
    let newPassword = AES.encrypt(values.newPassword, Utf8.parse(timestamp), {
      mode: ECB,
      padding: Pkcs7,
    }).toString(),
      password = AES.encrypt(values.password, Utf8.parse(timestamp), {
        mode: ECB,
        padding: Pkcs7,
      }).toString();

    let postData = {
      password,
      newPassword,
      encryptKey: newtimestamp,
    };
    run({ url: '/ngic-auth/sysUser/changePassword', params: { ...postData } });
  };
  const menuHeaderDropdown = (
    <Menu className={styles.menu} selectedKeys={[]} onClick={onMenuClick}>
      <Menu.Item key="logout">
        <LogoutOutlined />
        退出登录
      </Menu.Item>
    </Menu>
  );

  const handleChange = (value) => {
    if (value) {
      window.location.href = value;
    }
  };
  return (
    <div
      style={{
        display: 'flex',
        flexDirection: 'row',
        alignItems: 'center',
      }}
    >
      <Modal
        title="修改密码"
        open={visible}
        onCancel={() => {
          cv(false);
        }}
        destroyOnClose={true}
        maskClosable={false}
        footer={false}
      >
        <InitForm
          formRef={formRef}
          fields={fields}
          col={{ span: 24 }}
          onChange={(changedValues, allValues) => { }}
          submitData={(values, fn) => {
            saveData(values, fn);
          }}
          submitting={loading || !visible}
        />
      </Modal>
      <Select
        style={{ width: 150, marginRight: 8 }}
        value={
          getUserApp?.data?.dataList?.filter((it) => it?.appId == getUserApp?.data?.appId)[0]?.value
        }
        options={getUserApp?.data?.dataList}
        onChange={handleChange}
      />
      <HeaderDropdown overlay={menuHeaderDropdown}>
        <span className={`${styles.action} ${styles.account}`}>
          <Avatar style={{ backgroundColor: '#1890ff' }}>
            {currentUser.fullName.substr(0, 1)}
          </Avatar>
          <span className={`${styles.name} anticon`} style={{ margin: '0 12px' }}>
            {currentUser.fullName}
          </span>
        </span>
      </HeaderDropdown>
    </div>
  );
};

export default AvatarDropdown;