tachybase_todo/packages/plugin-users/src/middlewares/parseToken.ts
2022-04-09 15:30:43 +08:00

49 lines
1.2 KiB
TypeScript

import { Context, Next } from '@nocobase/actions';
import UsersPlugin from '../server';
export function parseToken(options?: { plugin: UsersPlugin }) {
return async function parseToken(ctx: Context, next: Next) {
const user = await findUserByToken(ctx, options.plugin);
if (user) {
ctx.state.currentUser = user;
setCurrentRole(ctx, user);
}
return next();
};
}
function setCurrentRole(ctx, user) {
const userRoles = user.get('roles');
let userRole;
if (userRoles.length == 1) {
userRole = userRoles[0].get('name');
} else if (userRoles.length > 1) {
const defaultRole = userRoles.findIndex((role) => role.get('rolesUsers').default);
userRole = (defaultRole !== -1 ? userRoles[defaultRole] : userRoles[0]).get('name');
}
if (userRole) {
ctx.state.currentRole = userRole;
}
}
async function findUserByToken(ctx: Context, plugin: UsersPlugin) {
const token = ctx.getBearerToken();
if (!token) {
return null;
}
try {
const { userId } = await plugin.jwtService.decode(token);
return await ctx.db.getRepository('users').findOne({
filter: {
id: userId,
},
appends: ['roles'],
});
} catch (error) {
console.warn(error);
}
}