platform/server/collaborator/src/platform.ts
2025-03-20 17:21:51 +07:00

127 lines
3.7 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, {
Client,
PersonId,
pickPrimarySocialId,
systemAccountUuid,
TxOperations,
WorkspaceUuid
} from '@hcengineering/core'
import { createClient, getTransactorEndpoint } from '@hcengineering/server-client'
import { Token, generateToken } from '@hcengineering/server-token'
import { getClient as getAccountClient } from '@hcengineering/account-client'
import config from './config'
async function connect (token: string): Promise<Client> {
const endpoint = await getTransactorEndpoint(token)
return await createClient(endpoint, token)
}
async function getTxOperations (client: Client, token: Token, isDerived: boolean = false): Promise<TxOperations> {
let primarySocialString: PersonId
if (token.account === systemAccountUuid) {
primarySocialString = core.account.System
} else {
const rawToken = generateToken(token.account, token.workspace, { service: 'collaborator' })
const accountClient = getAccountClient(config.AccountsUrl, rawToken)
const socialIds = await accountClient.getSocialIds()
primarySocialString = pickPrimarySocialId(socialIds)._id
}
return new TxOperations(client, primarySocialString, isDerived)
}
/**
* @public
*/
export interface ClientFactoryParams {
derived: boolean
}
/**
* @public
*/
export type ClientFactory = (params?: ClientFactoryParams) => Promise<TxOperations>
/**
* @public
*/
export function simpleClientFactory (token: Token): ClientFactory {
return async (params?: ClientFactoryParams) => {
const derived = params?.derived ?? false
const client = await connect(generateToken(token.account, token.workspace, { service: 'collaborator' }))
return await getTxOperations(client, token, derived)
}
}
/**
* @public
*/
export function reusableClientFactory (token: Token, controller: Controller): ClientFactory {
return async (params?: ClientFactoryParams) => {
const derived = params?.derived ?? false
const workspaceClient = await controller.get(token.workspace)
return await getTxOperations(workspaceClient.client, token, derived)
}
}
/**
* @public
*/
export class Controller {
private readonly workspaces: Map<WorkspaceUuid, WorkspaceClient> = new Map<WorkspaceUuid, WorkspaceClient>()
async get (workspace: WorkspaceUuid): Promise<WorkspaceClient> {
let client = this.workspaces.get(workspace)
if (client === undefined) {
client = await WorkspaceClient.create(workspace)
this.workspaces.set(workspace, client)
}
return client
}
async close (): Promise<void> {
for (const workspace of this.workspaces.values()) {
await workspace.close()
}
this.workspaces.clear()
}
}
/**
* @public
*/
export class WorkspaceClient {
private constructor (
readonly workspace: WorkspaceUuid,
readonly client: Client
) {}
static async create (workspace: WorkspaceUuid): Promise<WorkspaceClient> {
const token = generateToken(systemAccountUuid, workspace, { service: 'collaborator' })
const client = await connect(token)
return new WorkspaceClient(workspace, client)
}
async close (): Promise<void> {
await this.client.close()
}
}