Authorized.tsx 1.44 KB
Newer Older
1
import { ConnectProps, ConnectState, Route, UserModelState } from '@/models/connect';
ι™ˆεΈ…'s avatar
ι™ˆεΈ… committed
2

3 4 5
import Authorized from '@/utils/Authorized';
import React from 'react';
import Redirect from 'umi/redirect';
6 7
import { connect } from 'dva';
import pathToRegexp from 'path-to-regexp';
8

何乐's avatar
何乐 committed
9
interface AuthComponentProps extends ConnectProps {
10 11 12
  user: UserModelState;
}

13
const getRouteAuthority = (path: string, routeData: Route[]) => {
14
  let authorities: string[] | string | undefined;
何乐's avatar
何乐 committed
15 16 17
  routeData.forEach(route => {
    // match prefix
    if (pathToRegexp(`${route.path}(.*)`).test(path)) {
18 19 20 21
      // exact match
      if (route.path === path) {
        authorities = route.authority || authorities;
      }
何乐's avatar
何乐 committed
22 23 24 25 26 27 28 29 30
      // get children authority recursively
      if (route.routes) {
        authorities = getRouteAuthority(path, route.routes) || authorities;
      }
    }
  });
  return authorities;
};

ι™ˆεΈ…'s avatar
ι™ˆεΈ… committed
31 32 33 34 35
const AuthComponent: React.FC<AuthComponentProps> = ({
  children,
  route = {
    routes: [],
  },
36 37 38
  location = {
    pathname: '',
  },
ι™ˆεΈ…'s avatar
ι™ˆεΈ… committed
39 40
  user,
}) => {
41
  const { currentUser } = user;
ι™ˆεΈ…'s avatar
ι™ˆεΈ… committed
42
  const { routes = [] } = route;
43 44 45
  const isLogin = currentUser && currentUser.name;
  return (
    <Authorized
46
      authority={getRouteAuthority(location.pathname, routes) || ''}
47 48 49 50 51 52 53
      noMatch={isLogin ? <Redirect to="/exception/403" /> : <Redirect to="/user/login" />}
    >
      {children}
    </Authorized>
  );
};

ι™ˆεΈ…'s avatar
ι™ˆεΈ… committed
54
export default connect(({ user }: ConnectState) => ({
55 56
  user,
}))(AuthComponent);