mirror of
https://github.com/hcengineering/platform.git
synced 2025-04-03 06:21:28 +00:00
648 lines
21 KiB
TypeScript
648 lines
21 KiB
TypeScript
//
|
|
// Copyright © 2023 Hardcore Engineering Inc.
|
|
//
|
|
// Licensed under the Eclipse Public License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License. You may
|
|
// obtain a copy of the License at https://www.eclipse.org/legal/epl-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
//
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
import core, {
|
|
Account,
|
|
AccountRole,
|
|
AttachedDoc,
|
|
Class,
|
|
DOMAIN_MODEL,
|
|
Doc,
|
|
DocumentQuery,
|
|
Domain,
|
|
FindResult,
|
|
LookupData,
|
|
MeasureContext,
|
|
ObjQueryType,
|
|
Position,
|
|
PullArray,
|
|
Ref,
|
|
SearchOptions,
|
|
SearchQuery,
|
|
SearchResult,
|
|
Space,
|
|
Tx,
|
|
TxCUD,
|
|
TxCreateDoc,
|
|
TxProcessor,
|
|
TxRemoveDoc,
|
|
TxUpdateDoc,
|
|
TxWorkspaceEvent,
|
|
WorkspaceEvent,
|
|
clone,
|
|
generateId,
|
|
systemAccountEmail,
|
|
toFindResult,
|
|
type SessionData
|
|
} from '@hcengineering/core'
|
|
import platform, { PlatformError, Severity, Status } from '@hcengineering/platform'
|
|
import {
|
|
BaseMiddleware,
|
|
Middleware,
|
|
ServerFindOptions,
|
|
TxMiddlewareResult,
|
|
type PipelineContext
|
|
} from '@hcengineering/server-core'
|
|
import { isOwner, isSystem } from './utils'
|
|
type SpaceWithMembers = Pick<Space, '_id' | 'members' | 'private' | '_class'>
|
|
|
|
/**
|
|
* @public
|
|
*/
|
|
export class SpaceSecurityMiddleware extends BaseMiddleware implements Middleware {
|
|
private allowedSpaces: Record<Ref<Account>, Ref<Space>[]> = {}
|
|
private readonly spacesMap = new Map<Ref<Space>, SpaceWithMembers>()
|
|
private readonly privateSpaces = new Set<Ref<Space>>()
|
|
private readonly _domainSpaces = new Map<string, Set<Ref<Space>> | Promise<Set<Ref<Space>>>>()
|
|
private readonly publicSpaces = new Set<Ref<Space>>()
|
|
private readonly systemSpaces = new Set<Ref<Space>>()
|
|
|
|
wasInit: Promise<void> | boolean = false
|
|
|
|
private readonly mainSpaces = new Set([
|
|
core.space.Configuration,
|
|
core.space.DerivedTx,
|
|
core.space.Model,
|
|
core.space.Space,
|
|
core.space.Workspace,
|
|
core.space.Tx
|
|
])
|
|
|
|
private constructor (
|
|
private readonly skipFindCheck: boolean,
|
|
context: PipelineContext,
|
|
next?: Middleware
|
|
) {
|
|
super(context, next)
|
|
}
|
|
|
|
static async create (
|
|
skipFindCheck: boolean,
|
|
ctx: MeasureContext,
|
|
context: PipelineContext,
|
|
next: Middleware | undefined
|
|
): Promise<SpaceSecurityMiddleware> {
|
|
return new SpaceSecurityMiddleware(skipFindCheck, context, next)
|
|
}
|
|
|
|
private resyncDomains (): void {
|
|
this.wasInit = false
|
|
}
|
|
|
|
private addMemberSpace (member: Ref<Account>, space: Ref<Space>): void {
|
|
const arr = this.allowedSpaces[member] ?? []
|
|
arr.push(space)
|
|
this.allowedSpaces[member] = arr
|
|
}
|
|
|
|
private addSpace (space: SpaceWithMembers): void {
|
|
this.spacesMap.set(space._id, space)
|
|
if (space.private) {
|
|
this.privateSpaces.add(space._id)
|
|
} else {
|
|
this.publicSpaces.add(space._id)
|
|
}
|
|
for (const member of space.members) {
|
|
this.addMemberSpace(member, space._id)
|
|
}
|
|
}
|
|
|
|
async init (ctx: MeasureContext): Promise<void> {
|
|
if (this.wasInit === true) {
|
|
return
|
|
}
|
|
if (this.wasInit === false) {
|
|
this.wasInit = (async () => {
|
|
await ctx.with('init-space-security', {}, async (ctx) => {
|
|
ctx.contextData = undefined
|
|
const spaces: SpaceWithMembers[] =
|
|
(await this.next?.findAll(
|
|
ctx,
|
|
core.class.Space,
|
|
{},
|
|
{
|
|
projection: {
|
|
private: 1,
|
|
_class: 1,
|
|
_id: 1,
|
|
members: 1
|
|
}
|
|
}
|
|
)) ?? []
|
|
this.spacesMap.clear()
|
|
this.publicSpaces.clear()
|
|
this.systemSpaces.clear()
|
|
for (const space of spaces) {
|
|
if (space._class === core.class.SystemSpace) {
|
|
this.systemSpaces.add(space._id)
|
|
} else {
|
|
this.addSpace(space)
|
|
}
|
|
}
|
|
})
|
|
})()
|
|
}
|
|
if (this.wasInit instanceof Promise) {
|
|
await this.wasInit
|
|
this.wasInit = true
|
|
}
|
|
}
|
|
|
|
private removeMemberSpace (member: Ref<Account>, space: Ref<Space>): void {
|
|
const arr = this.allowedSpaces[member]
|
|
if (arr !== undefined) {
|
|
const index = arr.findIndex((p) => p === space)
|
|
if (index !== -1) {
|
|
arr.splice(index, 1)
|
|
this.allowedSpaces[member] = arr
|
|
}
|
|
}
|
|
}
|
|
|
|
private removeSpace (_id: Ref<Space>): void {
|
|
const space = this.spacesMap.get(_id)
|
|
if (space !== undefined) {
|
|
for (const member of space.members) {
|
|
this.removeMemberSpace(member, space._id)
|
|
}
|
|
}
|
|
this.spacesMap.delete(_id)
|
|
this.privateSpaces.delete(_id)
|
|
this.publicSpaces.delete(_id)
|
|
}
|
|
|
|
private handleCreate (tx: TxCUD<Space>): void {
|
|
const createTx = tx as TxCreateDoc<Space>
|
|
if (!this.context.hierarchy.isDerived(createTx.objectClass, core.class.Space)) return
|
|
if (createTx.objectClass === core.class.SystemSpace) {
|
|
this.systemSpaces.add(createTx.objectId)
|
|
} else {
|
|
const res = TxProcessor.createDoc2Doc<Space>(createTx)
|
|
this.addSpace(res)
|
|
}
|
|
}
|
|
|
|
private pushMembersHandle (
|
|
ctx: MeasureContext,
|
|
addedMembers: Ref<Account> | Position<Ref<Account>>,
|
|
space: Ref<Space>
|
|
): void {
|
|
if (typeof addedMembers === 'object') {
|
|
for (const member of addedMembers.$each) {
|
|
this.addMemberSpace(member, space)
|
|
}
|
|
this.brodcastEvent(ctx, addedMembers.$each, space)
|
|
} else {
|
|
this.addMemberSpace(addedMembers, space)
|
|
this.brodcastEvent(ctx, [addedMembers], space)
|
|
}
|
|
}
|
|
|
|
private pullMembersHandle (
|
|
ctx: MeasureContext,
|
|
removedMembers: Partial<Ref<Account>> | PullArray<Ref<Account>>,
|
|
space: Ref<Space>
|
|
): void {
|
|
if (typeof removedMembers === 'object') {
|
|
const { $in } = removedMembers as PullArray<Ref<Account>>
|
|
if ($in !== undefined) {
|
|
for (const member of $in) {
|
|
this.removeMemberSpace(member, space)
|
|
}
|
|
this.brodcastEvent(ctx, $in, space)
|
|
}
|
|
} else {
|
|
this.removeMemberSpace(removedMembers, space)
|
|
this.brodcastEvent(ctx, [removedMembers], space)
|
|
}
|
|
}
|
|
|
|
private syncMembers (ctx: MeasureContext, members: Ref<Account>[], space: SpaceWithMembers): void {
|
|
const oldMembers = new Set(space.members)
|
|
const newMembers = new Set(members)
|
|
const changed: Ref<Account>[] = []
|
|
for (const old of oldMembers) {
|
|
if (!newMembers.has(old)) {
|
|
this.removeMemberSpace(old, space._id)
|
|
changed.push(old)
|
|
}
|
|
}
|
|
for (const newMem of newMembers) {
|
|
if (!oldMembers.has(newMem)) {
|
|
this.addMemberSpace(newMem, space._id)
|
|
changed.push(newMem)
|
|
}
|
|
}
|
|
if (changed.length > 0) {
|
|
this.brodcastEvent(ctx, changed, space._id)
|
|
}
|
|
}
|
|
|
|
private brodcastEvent (ctx: MeasureContext<SessionData>, users: Ref<Account>[], space?: Ref<Space>): void {
|
|
const targets = this.getTargets(users)
|
|
const tx: TxWorkspaceEvent = {
|
|
_class: core.class.TxWorkspaceEvent,
|
|
_id: generateId(),
|
|
event: WorkspaceEvent.SecurityChange,
|
|
modifiedBy: core.account.System,
|
|
modifiedOn: Date.now(),
|
|
objectSpace: space ?? core.space.DerivedTx,
|
|
space: core.space.DerivedTx,
|
|
params: null
|
|
}
|
|
ctx.contextData.broadcast.txes.push(tx)
|
|
ctx.contextData.broadcast.targets['security' + tx._id] = (it) => {
|
|
// TODO: I'm not sure it is called
|
|
if (it._id === tx._id) {
|
|
return targets
|
|
}
|
|
}
|
|
}
|
|
|
|
private broadcastNonMembers (ctx: MeasureContext, space: SpaceWithMembers): void {
|
|
const users = this.context.modelDb.findAllSync(core.class.Account, { _id: { $nin: space?.members } })
|
|
this.brodcastEvent(
|
|
ctx,
|
|
users.map((p) => p._id),
|
|
space._id
|
|
)
|
|
}
|
|
|
|
private async handleUpdate (ctx: MeasureContext, tx: TxCUD<Space>): Promise<void> {
|
|
await this.init(ctx)
|
|
|
|
const updateDoc = tx as TxUpdateDoc<Space>
|
|
if (!this.context.hierarchy.isDerived(updateDoc.objectClass, core.class.Space)) return
|
|
|
|
const space = this.spacesMap.get(updateDoc.objectId)
|
|
if (space !== undefined) {
|
|
if (updateDoc.operations.private !== undefined) {
|
|
if (updateDoc.operations.private) {
|
|
this.privateSpaces.add(updateDoc.objectId)
|
|
this.publicSpaces.delete(updateDoc.objectId)
|
|
this.broadcastNonMembers(ctx, space)
|
|
} else if (!updateDoc.operations.private) {
|
|
this.privateSpaces.delete(updateDoc.objectId)
|
|
this.publicSpaces.add(updateDoc.objectId)
|
|
this.broadcastNonMembers(ctx, space)
|
|
}
|
|
}
|
|
|
|
if (updateDoc.operations.members !== undefined) {
|
|
this.syncMembers(ctx, updateDoc.operations.members, space)
|
|
}
|
|
if (updateDoc.operations.$push?.members !== undefined) {
|
|
this.pushMembersHandle(ctx, updateDoc.operations.$push.members, space._id)
|
|
}
|
|
|
|
if (updateDoc.operations.$pull?.members !== undefined) {
|
|
this.pullMembersHandle(ctx, updateDoc.operations.$pull.members, space._id)
|
|
}
|
|
const updatedSpace = TxProcessor.updateDoc2Doc(space as any, updateDoc)
|
|
this.spacesMap.set(updateDoc.objectId, updatedSpace)
|
|
}
|
|
}
|
|
|
|
private handleRemove (tx: TxCUD<Space>): void {
|
|
const removeTx = tx as TxRemoveDoc<Space>
|
|
if (!this.context.hierarchy.isDerived(removeTx.objectClass, core.class.Space)) return
|
|
if (removeTx._class !== core.class.TxCreateDoc) return
|
|
this.removeSpace(tx.objectId)
|
|
}
|
|
|
|
private async handleTx (ctx: MeasureContext, tx: TxCUD<Space>): Promise<void> {
|
|
await this.init(ctx)
|
|
if (tx._class === core.class.TxCreateDoc) {
|
|
this.handleCreate(tx)
|
|
} else if (tx._class === core.class.TxUpdateDoc) {
|
|
await this.handleUpdate(ctx, tx)
|
|
} else if (tx._class === core.class.TxRemoveDoc) {
|
|
this.handleRemove(tx)
|
|
}
|
|
}
|
|
|
|
getTargets (accounts: Ref<Account>[]): string[] {
|
|
const users = this.context.modelDb.findAllSync(core.class.Account, { _id: { $in: accounts } })
|
|
const res = users.map((p) => p.email)
|
|
res.push(systemAccountEmail)
|
|
return res
|
|
}
|
|
|
|
private async processTxSpaceDomain (sctx: MeasureContext, actualTx: TxCUD<Doc>): Promise<void> {
|
|
if (actualTx._class === core.class.TxCreateDoc) {
|
|
const ctx = actualTx as TxCreateDoc<Doc>
|
|
const doc = TxProcessor.createDoc2Doc(ctx)
|
|
const domain = this.context.hierarchy.getDomain(ctx.objectClass)
|
|
const key = this.getKey(domain)
|
|
const space = (doc as any)[key]
|
|
if (space === undefined) return
|
|
;(await this.getDomainSpaces(sctx, domain)).add(space)
|
|
} else if (actualTx._class === core.class.TxUpdateDoc) {
|
|
const updTx = actualTx as TxUpdateDoc<Doc>
|
|
const domain = this.context.hierarchy.getDomain(updTx.objectClass)
|
|
const key = this.getKey(domain)
|
|
const space = (updTx.operations as any)[key]
|
|
if (space !== undefined) {
|
|
;(await this.getDomainSpaces(sctx, domain)).add(space)
|
|
}
|
|
}
|
|
}
|
|
|
|
private async processTx (ctx: MeasureContext<SessionData>, tx: Tx): Promise<void> {
|
|
const h = this.context.hierarchy
|
|
if (TxProcessor.isExtendsCUD(tx._class)) {
|
|
const cudTx = tx as TxCUD<Doc>
|
|
const isSpace = h.isDerived(cudTx.objectClass, core.class.Space)
|
|
if (isSpace) {
|
|
const account = ctx.contextData.account
|
|
if (account.role === AccountRole.Guest) {
|
|
throw new PlatformError(new Status(Severity.ERROR, platform.status.Forbidden, {}))
|
|
}
|
|
await this.handleTx(ctx, cudTx as TxCUD<Space>)
|
|
}
|
|
await this.processTxSpaceDomain(ctx, tx as TxCUD<Doc>)
|
|
if (h.isDerived(cudTx.objectClass, core.class.Account) && cudTx._class === core.class.TxUpdateDoc) {
|
|
const cud = cudTx as TxUpdateDoc<Account>
|
|
if (cud.operations.role !== undefined) {
|
|
this.brodcastEvent(ctx, [cud.objectId])
|
|
}
|
|
}
|
|
} else if (tx._class === core.class.TxWorkspaceEvent) {
|
|
const event = tx as TxWorkspaceEvent
|
|
if (event.event === WorkspaceEvent.BulkUpdate) {
|
|
this.resyncDomains()
|
|
}
|
|
}
|
|
}
|
|
|
|
async tx (ctx: MeasureContext<SessionData>, txes: Tx[]): Promise<TxMiddlewareResult> {
|
|
await this.init(ctx)
|
|
const account = ctx.contextData.account
|
|
if (account.role === AccountRole.DocGuest) {
|
|
throw new PlatformError(new Status(Severity.ERROR, platform.status.Forbidden, {}))
|
|
}
|
|
const processed = new Set<Ref<Tx>>()
|
|
ctx.contextData.contextCache.set('processed', processed)
|
|
for (const tx of txes) {
|
|
processed.add(tx._id)
|
|
await this.processTx(ctx, tx)
|
|
}
|
|
return await this.provideTx(ctx, txes)
|
|
}
|
|
|
|
override async handleBroadcast (ctx: MeasureContext<SessionData>): Promise<void> {
|
|
const processed: Set<Ref<Tx>> = ctx.contextData.contextCache.get('processed') ?? new Set<Ref<Tx>>()
|
|
ctx.contextData.contextCache.set('processed', processed)
|
|
for (const txd of ctx.contextData.broadcast.txes) {
|
|
if (!processed.has(txd._id)) {
|
|
await this.processTx(ctx, txd)
|
|
}
|
|
}
|
|
for (const tx of ctx.contextData.broadcast.txes) {
|
|
if (TxProcessor.isExtendsCUD(tx._class)) {
|
|
// TODO: Do we need security check here?
|
|
const cudTx = tx as TxCUD<Doc>
|
|
await this.processTxSpaceDomain(ctx, cudTx)
|
|
} else if (tx._class === core.class.TxWorkspaceEvent) {
|
|
const event = tx as TxWorkspaceEvent
|
|
if (event.event === WorkspaceEvent.BulkUpdate) {
|
|
this.resyncDomains()
|
|
}
|
|
}
|
|
}
|
|
|
|
ctx.contextData.broadcast.targets.spaceSec = (tx) => {
|
|
const space = this.spacesMap.get(tx.objectSpace)
|
|
if (space === undefined) return undefined
|
|
if (this.systemSpaces.has(space._id) || this.mainSpaces.has(space._id)) return undefined
|
|
|
|
return space.members.length === 0 ? undefined : this.getTargets(space?.members)
|
|
}
|
|
|
|
await this.next?.handleBroadcast(ctx)
|
|
}
|
|
|
|
private getAllAllowedSpaces (account: Account, isData: boolean): Ref<Space>[] {
|
|
const userSpaces = this.allowedSpaces[account._id] ?? []
|
|
const res = [...userSpaces, account._id as string as Ref<Space>, ...this.systemSpaces, ...this.mainSpaces]
|
|
return isData ? res : [...res, ...this.publicSpaces]
|
|
}
|
|
|
|
async getDomainSpaces (ctx: MeasureContext, domain: Domain): Promise<Set<Ref<Space>>> {
|
|
let domainSpaces = this._domainSpaces.get(domain)
|
|
if (domainSpaces === undefined) {
|
|
const p = (
|
|
this.next?.groupBy<Ref<Space>, Doc>(ctx, domain, this.getKey(domain)) ?? Promise.resolve(new Map())
|
|
).then((r) => new Set<Ref<Space>>(r.keys()))
|
|
this._domainSpaces.set(domain, p)
|
|
domainSpaces = await p
|
|
this._domainSpaces.set(domain, domainSpaces)
|
|
}
|
|
return domainSpaces instanceof Promise ? await domainSpaces : domainSpaces
|
|
}
|
|
|
|
private async filterByDomain (
|
|
ctx: MeasureContext,
|
|
domain: Domain,
|
|
spaces: Ref<Space>[]
|
|
): Promise<{ result: Set<Ref<Space>>, allDomainSpaces: boolean, domainSpaces: Set<Ref<Space>> }> {
|
|
const domainSpaces = await this.getDomainSpaces(ctx, domain)
|
|
const result = new Set(spaces.filter((p) => domainSpaces.has(p)))
|
|
return {
|
|
result,
|
|
allDomainSpaces: result.size === domainSpaces.size,
|
|
domainSpaces
|
|
}
|
|
}
|
|
|
|
private async mergeQuery<T extends Doc>(
|
|
ctx: MeasureContext,
|
|
account: Account,
|
|
query: ObjQueryType<T['space']>,
|
|
domain: Domain,
|
|
isSpace: boolean
|
|
): Promise<ObjQueryType<T['space']> | undefined> {
|
|
const spaces = await this.filterByDomain(ctx, domain, this.getAllAllowedSpaces(account, !isSpace))
|
|
if (query == null) {
|
|
if (spaces.allDomainSpaces) {
|
|
return undefined
|
|
}
|
|
return { $in: Array.from(spaces.result) }
|
|
}
|
|
if (typeof query === 'string') {
|
|
if (!spaces.result.has(query)) {
|
|
return { $in: [] }
|
|
}
|
|
} else if (query.$in != null) {
|
|
query.$in = query.$in.filter((p) => spaces.result.has(p))
|
|
if (query.$in.length === spaces.domainSpaces.size) {
|
|
// all domain spaces
|
|
delete query.$in
|
|
}
|
|
} else {
|
|
if (spaces.allDomainSpaces) {
|
|
delete query.$in
|
|
} else {
|
|
query.$in = Array.from(spaces.result)
|
|
}
|
|
}
|
|
if (Object.keys(query).length === 0) {
|
|
return undefined
|
|
}
|
|
return query
|
|
}
|
|
|
|
private getKey (domain: string): string {
|
|
return domain === 'tx' ? 'objectSpace' : domain === 'space' ? '_id' : 'space'
|
|
}
|
|
|
|
override async findAll<T extends Doc>(
|
|
ctx: MeasureContext<SessionData>,
|
|
_class: Ref<Class<T>>,
|
|
query: DocumentQuery<T>,
|
|
options?: ServerFindOptions<T>
|
|
): Promise<FindResult<T>> {
|
|
await this.init(ctx)
|
|
|
|
const domain = this.context.hierarchy.getDomain(_class)
|
|
const newQuery = clone(query)
|
|
const account = ctx.contextData.account
|
|
const isSpace = this.context.hierarchy.isDerived(_class, core.class.Space)
|
|
const field = this.getKey(domain)
|
|
|
|
let clientFilterSpaces: Set<Ref<Space>> | undefined
|
|
|
|
if (!isSystem(account, ctx) && account.role !== AccountRole.DocGuest && domain !== DOMAIN_MODEL) {
|
|
if (!isOwner(account, ctx) || !isSpace) {
|
|
if (query[field] !== undefined) {
|
|
const res = await this.mergeQuery(ctx, account, query[field], domain, isSpace)
|
|
if (res === undefined) {
|
|
// eslint-disable-next-line @typescript-eslint/no-dynamic-delete
|
|
delete newQuery[field]
|
|
} else {
|
|
newQuery[field] = res
|
|
if (typeof res === 'object') {
|
|
if (Array.isArray(res.$in) && res.$in.length === 1 && Object.keys(res).length === 1) {
|
|
newQuery[field] = res.$in[0]
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
const spaces = await this.filterByDomain(ctx, domain, this.getAllAllowedSpaces(account, !isSpace))
|
|
if (spaces.allDomainSpaces) {
|
|
// eslint-disable-next-line @typescript-eslint/no-dynamic-delete
|
|
delete newQuery[field]
|
|
} else if (spaces.result.size === 1) {
|
|
newQuery[field] = Array.from(spaces.result)[0]
|
|
if (options !== undefined) {
|
|
options.allowedSpaces = Array.from(spaces.result)
|
|
} else {
|
|
options = { allowedSpaces: Array.from(spaces.result) }
|
|
}
|
|
} else {
|
|
// Check if spaces > 85% of all domain spaces, in this case return all and filter on client.
|
|
if (spaces.result.size / spaces.domainSpaces.size > 0.85 && options?.limit === undefined) {
|
|
clientFilterSpaces = spaces.result
|
|
delete newQuery.space
|
|
} else {
|
|
newQuery[field] = { $in: Array.from(spaces.result) }
|
|
if (options !== undefined) {
|
|
options.allowedSpaces = Array.from(spaces.result)
|
|
} else {
|
|
options = { allowedSpaces: Array.from(spaces.result) }
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
let findResult = await this.provideFindAll(ctx, _class, !this.skipFindCheck ? newQuery : query, options)
|
|
if (clientFilterSpaces !== undefined) {
|
|
const cfs = clientFilterSpaces
|
|
findResult = toFindResult(
|
|
findResult.filter((it) => cfs.has((it as any)[field])),
|
|
findResult.total,
|
|
findResult.lookupMap
|
|
)
|
|
}
|
|
if (!isOwner(account, ctx) && account.role !== AccountRole.DocGuest) {
|
|
if (options?.lookup !== undefined) {
|
|
for (const object of findResult) {
|
|
if (object.$lookup !== undefined) {
|
|
this.filterLookup(ctx, object.$lookup)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return findResult
|
|
}
|
|
|
|
override async searchFulltext (
|
|
ctx: MeasureContext<SessionData>,
|
|
query: SearchQuery,
|
|
options: SearchOptions
|
|
): Promise<SearchResult> {
|
|
await this.init(ctx)
|
|
const newQuery = { ...query }
|
|
const account = ctx.contextData.account
|
|
if (!isSystem(account, ctx)) {
|
|
const allSpaces = this.getAllAllowedSpaces(account, true)
|
|
if (query.classes !== undefined) {
|
|
const res = new Set<Ref<Space>>()
|
|
const passedDomains = new Set<string>()
|
|
for (const _class of query.classes) {
|
|
const domain = this.context.hierarchy.getDomain(_class)
|
|
if (passedDomains.has(domain)) {
|
|
continue
|
|
}
|
|
passedDomains.add(domain)
|
|
const spaces = await this.filterByDomain(ctx, domain, allSpaces)
|
|
for (const space of spaces.result) {
|
|
res.add(space)
|
|
}
|
|
}
|
|
newQuery.spaces = [...res]
|
|
} else {
|
|
newQuery.spaces = allSpaces
|
|
}
|
|
}
|
|
const result = await this.provideSearchFulltext(ctx, newQuery, options)
|
|
return result
|
|
}
|
|
|
|
filterLookup<T extends Doc>(ctx: MeasureContext, lookup: LookupData<T>): void {
|
|
if (Object.keys(lookup).length === 0) return
|
|
const account = ctx.contextData.account
|
|
if (isSystem(account, ctx)) return
|
|
const allowedSpaces = new Set(this.getAllAllowedSpaces(account, true))
|
|
for (const key in lookup) {
|
|
const val = lookup[key]
|
|
if (Array.isArray(val)) {
|
|
const arr: AttachedDoc[] = []
|
|
for (const value of val) {
|
|
if (allowedSpaces.has(value.space)) {
|
|
arr.push(value)
|
|
}
|
|
}
|
|
lookup[key] = arr as any
|
|
} else if (val !== undefined) {
|
|
if (!allowedSpaces.has(val.space)) {
|
|
lookup[key] = undefined
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|