|
|
|
import { Dictionary } from 'lodash';
|
|
|
|
import { BUILTIN_ROLES, PRIV } from './builtin';
|
|
|
|
import { Udoc, DomainDoc } from '../interface';
|
|
|
|
import * as db from '../service/db';
|
|
|
|
|
|
|
|
const coll = db.collection('domain');
|
|
|
|
const collUser = db.collection('domain.user');
|
|
|
|
|
|
|
|
export async function add(domainId: string, owner: number, name: string, bulletin: string) {
|
|
|
|
await coll.insertOne({
|
|
|
|
_id: domainId, owner, name, bulletin, roles: {},
|
|
|
|
});
|
|
|
|
return domainId;
|
|
|
|
}
|
|
|
|
|
|
|
|
export function get(domainId: string): Promise<DomainDoc | null> {
|
|
|
|
return coll.findOne({ _id: domainId });
|
|
|
|
}
|
|
|
|
|
|
|
|
export function getMany(query: any, sort: any, page: number, limit: number): Promise<DomainDoc[]> {
|
|
|
|
return coll.find(query).sort(sort).skip((page - 1) * limit).limit(limit)
|
|
|
|
.toArray();
|
|
|
|
}
|
|
|
|
|
|
|
|
export function getMulti(query: any = {}) {
|
|
|
|
return coll.find(query);
|
|
|
|
}
|
|
|
|
|
|
|
|
export function edit(domainId: string, $set: any) {
|
|
|
|
return coll.updateOne({ _id: domainId }, { $set });
|
|
|
|
}
|
|
|
|
|
|
|
|
export async function inc(domainId: string, field: string, n: number): Promise<DomainDoc | null> {
|
|
|
|
const res = await coll.findOneAndUpdate(
|
|
|
|
{ _id: domainId },
|
|
|
|
{ $inc: { [field]: n } },
|
|
|
|
{ returnOriginal: false },
|
|
|
|
);
|
|
|
|
return res.value;
|
|
|
|
}
|
|
|
|
|
|
|
|
export async function getList(domainIds: string[]): Promise<Dictionary<DomainDoc>> {
|
|
|
|
const r = {};
|
|
|
|
// eslint-disable-next-line no-await-in-loop
|
|
|
|
for (const domainId of domainIds) r[domainId] = await get(domainId);
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
|
|
|
|
export function setUserRole(domainId: string, uid: number, role: string) {
|
|
|
|
return collUser.updateOne({ uid, domainId }, { role }, { upsert: true });
|
|
|
|
}
|
|
|
|
|
|
|
|
export async function getRoles(arg: string | DomainDoc) {
|
|
|
|
let ddoc: DomainDoc;
|
|
|
|
if (typeof arg === 'string') ddoc = await get(arg);
|
|
|
|
else ddoc = arg;
|
|
|
|
const roles = [];
|
|
|
|
const r = [];
|
|
|
|
for (const role in ddoc.roles) {
|
|
|
|
roles.push({ _id: role, perm: BigInt(ddoc.roles[role]) });
|
|
|
|
r.push(role);
|
|
|
|
}
|
|
|
|
for (const role in BUILTIN_ROLES) {
|
|
|
|
if (!r.includes(role)) {
|
|
|
|
roles.push({ _id: role, perm: BUILTIN_ROLES[role] });
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return roles;
|
|
|
|
}
|
|
|
|
|
|
|
|
export async function setRoles(domainId: string, roles: Dictionary<bigint>) {
|
|
|
|
const current = await get(domainId);
|
|
|
|
for (const role in roles) {
|
|
|
|
current.roles[role] = roles[role].toString();
|
|
|
|
}
|
|
|
|
return await coll.updateOne({ _id: domainId }, { $set: { roles: current.roles } });
|
|
|
|
}
|
|
|
|
|
|
|
|
export async function addRole(domainId: string, name: string, permission: bigint) {
|
|
|
|
const current = await get(domainId);
|
|
|
|
current.roles[name] = permission.toString();
|
|
|
|
return await coll.updateOne({ _id: domainId }, { $set: { roles: current.roles } });
|
|
|
|
}
|
|
|
|
|
|
|
|
export async function deleteRoles(domainId: string, roles: string[]) {
|
|
|
|
const current = await get(domainId);
|
|
|
|
for (const role of roles) delete current.roles[role];
|
|
|
|
await Promise.all([
|
|
|
|
coll.updateOne({ _id: domainId }, { $set: current }),
|
|
|
|
collUser.updateMany({ domainId, role: { $in: roles } }, { $set: { role: 'default' } }),
|
|
|
|
]);
|
|
|
|
}
|
|
|
|
|
|
|
|
export async function getDomainUser(domainId: string, udoc: Udoc) {
|
|
|
|
let dudoc = await collUser.findOne({ domainId, uid: udoc._id });
|
|
|
|
dudoc = dudoc || {};
|
|
|
|
if (udoc._id === 1) dudoc.role = 'guest';
|
|
|
|
if (udoc.priv & PRIV.PRIV_MANAGE_ALL_DOMAIN) dudoc.role = 'admin';
|
|
|
|
dudoc.role = dudoc.role || 'default';
|
|
|
|
const ddoc = await get(domainId);
|
|
|
|
dudoc.perm = ddoc.roles[dudoc.role]
|
|
|
|
? BigInt(ddoc.roles[dudoc.role])
|
|
|
|
: BUILTIN_ROLES[dudoc.role];
|
|
|
|
return dudoc;
|
|
|
|
}
|
|
|
|
|
|
|
|
export function setMultiUserInDomain(domainId: string, query: any, params: any) {
|
|
|
|
return collUser.updateMany({ domainId, ...query }, { $set: params });
|
|
|
|
}
|
|
|
|
|
|
|
|
export function getMultiInDomain(domainId: string, query: any = {}) {
|
|
|
|
return collUser.find({ domainId, ...query });
|
|
|
|
}
|
|
|
|
|
|
|
|
export function setUserInDomain(domainId: string, uid: number, params: any) {
|
|
|
|
return collUser.updateOne({ domainId, uid }, { $set: params });
|
|
|
|
}
|
|
|
|
|
|
|
|
export async function incUserInDomain(domainId: string, uid: number, field: string, n = 1) {
|
|
|
|
// @ts-ignore
|
|
|
|
const dudoc = await getDomainUser(domainId, { _id: uid });
|
|
|
|
dudoc[field] = dudoc[field] + n || n;
|
|
|
|
await setUserInDomain(domainId, uid, { [field]: dudoc[field] });
|
|
|
|
return dudoc;
|
|
|
|
}
|
|
|
|
|
|
|
|
export async function getDictUserByDomainId(uid: number) {
|
|
|
|
const dudocs = await collUser.find({ uid }).toArray();
|
|
|
|
const ddocs = await coll.find({ owner: uid }).toArray();
|
|
|
|
const dudict = {};
|
|
|
|
for (const dudoc of dudocs) {
|
|
|
|
// eslint-disable-next-line no-await-in-loop
|
|
|
|
dudict[dudoc.domainId] = await get(dudoc.domainId);
|
|
|
|
}
|
|
|
|
for (const ddoc of ddocs) {
|
|
|
|
dudict[ddoc._id] = ddoc;
|
|
|
|
}
|
|
|
|
return dudict;
|
|
|
|
}
|
|
|
|
|
|
|
|
global.Hydro.model.domain = {
|
|
|
|
getRoles,
|
|
|
|
add,
|
|
|
|
inc,
|
|
|
|
get,
|
|
|
|
getMany,
|
|
|
|
edit,
|
|
|
|
getMulti,
|
|
|
|
getList,
|
|
|
|
setRoles,
|
|
|
|
addRole,
|
|
|
|
deleteRoles,
|
|
|
|
setUserRole,
|
|
|
|
getDomainUser,
|
|
|
|
setMultiUserInDomain,
|
|
|
|
setUserInDomain,
|
|
|
|
incUserInDomain,
|
|
|
|
getMultiInDomain,
|
|
|
|
getDictUserByDomainId,
|
|
|
|
};
|