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

core: user_group: draft

Browse Source
This commit is contained in:
undefined committed 2022-01-07 18:20:05 +08:00
1 parent 07c5793fb8
commit f699d67019
5 files changed
+99 -55
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

packages/hydrooj/src/handler/home.ts
+20 -16
View File
@@ -24,7 +24,7 @@ import * as training from '../model/training';
import user from '../model/user';
import * as bus from '../service/bus';
import {
Connection, ConnectionHandler, Handler, param, Route, Types,
Connection, ConnectionHandler, Handler, param, query, Route, Types,
} from '../service/server';
const { geoip, useragent } = global.Hydro.lib;
@@ -291,28 +291,32 @@ class UserChangemailWithCodeHandler extends Handler {
}
class HomeDomainHandler extends Handler {
async get() {
let ddocs: DomainDoc[] = [];
let dudict: Record<string, DomainDoc> = {};
if (!this.user.hasPriv(PRIV.PRIV_VIEW_ALL_DOMAIN)) {
@query('all', Types.Boolean)
async get(domainId: string, all: boolean) {
let res: DomainDoc[] = [];
let dudict: Record<string, any> = {};
if (!all) {
dudict = await domain.getDictUserByDomainId(this.user._id);
const dids = Object.keys(dudict);
ddocs = await domain.getMulti({ _id: { $in: dids } }).toArray();
res = await domain.getMulti({ _id: { $in: dids } }).toArray();
} else {
const _ddocs = await domain.getMulti().toArray();
for (const ddoc of _ddocs) {
// eslint-disable-next-line no-await-in-loop
const pcount = await ProblemModel.count(ddoc._id, {});
if (pcount >= 3 || ddoc.owner === this.user._id) {
dudict[ddoc._id] = ddoc;
ddocs.push(ddoc);
}
}
this.checkPriv(PRIV.PRIV_VIEW_ALL_DOMAIN);
res = await domain.getMulti().toArray();
await Promise.all(res.map(async (ddoc) => {
dudict[ddoc._id] = await user.getById(domainId, this.user._id);
}));
}
const canManage = {};
for (const ddoc of ddocs) {
const ddocs = [];
for (const ddoc of res) {
// eslint-disable-next-line no-await-in-loop
const udoc = (await user.getById(ddoc._id, this.user._id))!;
const dudoc = dudict[ddoc._id];
if (['default', 'guest'].includes(dudoc.role)) {
delete dudict[ddoc._id];
continue;
}
ddocs.push(ddoc);
canManage[ddoc._id] = udoc.hasPerm(PERM.PERM_EDIT_DOMAIN)
|| udoc.hasPriv(PRIV.PRIV_MANAGE_ALL_DOMAIN);
}