platform/models/training/src/migration.ts
Andrey Sobolev 6b519bfb0e
UBERF-9633: More proper fix (#8248)
Signed-off-by: Andrey Sobolev <haiodo@gmail.com>
2025-03-17 18:17:25 +07:00

98 lines
2.9 KiB
TypeScript

//
// Copyright © 2024 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 { Class, Doc, DOMAIN_SEQUENCE, Sequence, TxOperations, type Ref, type TypedSpace } from '@hcengineering/core'
import {
tryMigrate,
tryUpgrade,
type MigrateOperation,
type MigrationClient,
type MigrationUpgradeClient
} from '@hcengineering/model'
import core from '@hcengineering/model-core'
import training, { trainingId } from '@hcengineering/training'
import { DOMAIN_TRAINING } from './types'
export const trainingOperation: MigrateOperation = {
async migrate (client: MigrationClient, mode): Promise<void> {
await tryMigrate(mode, client, trainingId, [
{
state: 'migrateSequnce',
mode: 'upgrade',
func: async (client: MigrationClient) => {
await client.update(
DOMAIN_TRAINING,
{ _class: 'training:class:Sequence' as Ref<Class<Doc>> },
{ _class: core.class.Sequence }
)
await client.move(DOMAIN_TRAINING, { _class: core.class.Sequence }, DOMAIN_SEQUENCE)
}
}
])
},
async upgrade (state: Map<string, Set<string>>, client: () => Promise<MigrationUpgradeClient>, mode): Promise<void> {
await tryUpgrade(mode, state, client, trainingId, [
{
state: 'create-defaults',
func: async (client) => {
const tx = new TxOperations(client, core.account.System)
await ensureTypedSpace(tx)
await ensureSequence(tx)
}
}
])
}
}
async function ensureTypedSpace (tx: TxOperations): Promise<Ref<TypedSpace>> {
const existing = await tx.findOne(core.class.TypedSpace, {
_id: training.space.Trainings
})
if (existing !== undefined) {
return existing._id
}
return await tx.createDoc(
core.class.TypedSpace,
core.space.Space,
{
name: 'Trainings',
description: 'Space for all trainings',
private: false,
archived: false,
autoJoin: true,
members: [],
type: training.spaceType.Trainings
},
training.space.Trainings
)
}
async function ensureSequence (tx: TxOperations): Promise<Ref<Sequence>> {
const existing = await tx.findOne(core.class.Sequence, {
attachedTo: training.class.Training
})
if (existing !== undefined) {
return existing._id
}
return await tx.createDoc(core.class.Sequence, training.space.Trainings, {
attachedTo: training.class.Training,
sequence: 0
})
}