Alessandro Magionami
56 lines
1.7 KiB
TypeScript
56 lines
1.7 KiB
TypeScript
import { db } from '@/db/knex'
|
|
import { getFeatureFlags } from '@/modules/shared/helpers/envHelper'
|
|
import { defineRequestDataloaders } from '@/modules/shared/helpers/graphqlHelper'
|
|
import {
|
|
getWorkspaceDomainsFactory,
|
|
getWorkspacesFactory
|
|
} from '@/modules/workspaces/repositories/workspaces'
|
|
import {
|
|
WorkspaceDomain,
|
|
WorkspaceWithOptionalRole
|
|
} from '@/modules/workspacesCore/domain/types'
|
|
import { keyBy } from 'lodash'
|
|
|
|
const { FF_WORKSPACES_MODULE_ENABLED } = getFeatureFlags()
|
|
|
|
declare module '@/modules/core/loaders' {
|
|
interface AllRequestDataLoaders
|
|
extends Partial<ReturnType<typeof dataLoadersDefinition>> {}
|
|
}
|
|
|
|
const dataLoadersDefinition = defineRequestDataloaders(({ ctx, createLoader }) => {
|
|
const getWorkspaces = getWorkspacesFactory({ db })
|
|
const getWorkspaceDomains = getWorkspaceDomainsFactory({ db })
|
|
|
|
return {
|
|
workspaces: {
|
|
/**
|
|
* Get workspace, with the active user's role attached
|
|
*/
|
|
getWorkspace: createLoader<string, WorkspaceWithOptionalRole | null>(
|
|
async (ids) => {
|
|
const results = keyBy(
|
|
await getWorkspaces({ workspaceIds: ids.slice(), userId: ctx.userId }),
|
|
(w) => w.id
|
|
)
|
|
return ids.map((id) => results[id] || null)
|
|
}
|
|
)
|
|
},
|
|
workspaceDomains: {
|
|
/**
|
|
* Get workspace, with the active user's role attached
|
|
*/
|
|
getWorkspaceDomains: createLoader<string, WorkspaceDomain | null>(async (ids) => {
|
|
const results = keyBy(
|
|
await getWorkspaceDomains({ workspaceIds: ids.slice() }),
|
|
(w) => w.id
|
|
)
|
|
return ids.map((id) => results[id] || null)
|
|
})
|
|
}
|
|
}
|
|
})
|
|
|
|
export default FF_WORKSPACES_MODULE_ENABLED ? dataLoadersDefinition : undefined
|