Weishen/Hydro
2
0 Fork 0
mirror of https://gitcode.com/gh_mirrors/hy/Hydro.git synced 2026-10-20 01:38:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

core: return current udoc on login success (#794)

Browse Source
This commit is contained in:
panda authored and GitHub committed 2024-05-01 23:33:42 +08:00
1 parent 87adfd288c
commit c68b2b346b
1 file changed
+4 -1
Download Patch File Download Diff File Expand all files Collapse all files
packages/hydrooj/src/handler/user.ts
+4 -1
View File
@@ -120,6 +120,7 @@ class UserLoginHandler extends Handler {
await udoc.checkPassword(password); await udoc.checkPassword(password);
await user.setById(udoc._id, { loginat: new Date(), loginip: this.request.ip }); await user.setById(udoc._id, { loginat: new Date(), loginip: this.request.ip });
if (!udoc.hasPriv(PRIV.PRIV_USER_PROFILE)) throw new BlacklistedError(uname, udoc.banReason); if (!udoc.hasPriv(PRIV.PRIV_USER_PROFILE)) throw new BlacklistedError(uname, udoc.banReason);
this.context.HydroContext.user = udoc;
this.session.viewLang = ''; this.session.viewLang = '';
this.session.uid = udoc._id; this.session.uid = udoc._id;
this.session.sudo = null; this.session.sudo = null;
@@ -224,7 +225,8 @@ class UserLogoutHandler extends Handler {
this.response.template = 'user_logout.html'; this.response.template = 'user_logout.html';
} }
async post() { async post(domainId: string) {
this.context.HydroContext.user = await user.getById(domainId, 0);
this.session.uid = 0; this.session.uid = 0;
this.session.sudo = null; this.session.sudo = null;
this.session.sudoUid = null; this.session.sudoUid = null;
@@ -322,6 +324,7 @@ class UserRegisterWithCodeHandler extends Handler {
if (this.session.viewLang) $set.viewLang = this.session.viewLang; if (this.session.viewLang) $set.viewLang = this.session.viewLang;
if (Object.keys($set).length) await user.setById(uid, $set); if (Object.keys($set).length) await user.setById(uid, $set);
if (tdoc.oauth) await oauth.set(tdoc.oauth[1], uid); if (tdoc.oauth) await oauth.set(tdoc.oauth[1], uid);
this.context.HydroContext.user = await user.getById(domainId, uid);
this.session.viewLang = ''; this.session.viewLang = '';
this.session.uid = uid; this.session.uid = uid;
this.session.sudoUid = null; this.session.sudoUid = null;