-
Notifications
You must be signed in to change notification settings - Fork 2.5k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Set missing labelIdentifier to custom objects (#8750)
## Context Following #8729 This command backfills missing labelIdentifier for custom objects
- Loading branch information
Showing
3 changed files
with
117 additions
and
0 deletions.
There are no files selected for viewing
108 changes: 108 additions & 0 deletions
108
...mands/upgrade-version/0-33/0-33-set-missing-label-identifier-to-custom-objects.command.ts
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,108 @@ | ||
import { InjectRepository } from '@nestjs/typeorm'; | ||
|
||
import chalk from 'chalk'; | ||
import { Command } from 'nest-commander'; | ||
import { IsNull, Repository } from 'typeorm'; | ||
|
||
import { | ||
ActiveWorkspacesCommandOptions, | ||
ActiveWorkspacesCommandRunner, | ||
} from 'src/database/commands/active-workspaces.command'; | ||
import { Workspace } from 'src/engine/core-modules/workspace/workspace.entity'; | ||
import { FieldMetadataEntity } from 'src/engine/metadata-modules/field-metadata/field-metadata.entity'; | ||
import { ObjectMetadataEntity } from 'src/engine/metadata-modules/object-metadata/object-metadata.entity'; | ||
import { TwentyORMGlobalManager } from 'src/engine/twenty-orm/twenty-orm-global.manager'; | ||
import { CUSTOM_OBJECT_STANDARD_FIELD_IDS } from 'src/engine/workspace-manager/workspace-sync-metadata/constants/standard-field-ids'; | ||
|
||
interface SetMissingLabelIdentifierToCustomObjectsCommandOptions | ||
extends ActiveWorkspacesCommandOptions {} | ||
|
||
@Command({ | ||
name: 'upgrade-0.33:set-missing-label-identifier-to-custom-objects', | ||
description: 'Set missing labelIdentifier to custom objects', | ||
}) | ||
export class SetMissingLabelIdentifierToCustomObjectsCommand extends ActiveWorkspacesCommandRunner { | ||
constructor( | ||
@InjectRepository(Workspace, 'core') | ||
protected readonly workspaceRepository: Repository<Workspace>, | ||
@InjectRepository(FieldMetadataEntity, 'metadata') | ||
private readonly fieldMetadataRepository: Repository<FieldMetadataEntity>, | ||
@InjectRepository(ObjectMetadataEntity, 'metadata') | ||
private readonly objectMetadataRepository: Repository<ObjectMetadataEntity>, | ||
private readonly twentyORMGlobalManager: TwentyORMGlobalManager, | ||
) { | ||
super(workspaceRepository); | ||
} | ||
|
||
async executeActiveWorkspacesCommand( | ||
_passedParam: string[], | ||
options: SetMissingLabelIdentifierToCustomObjectsCommandOptions, | ||
workspaceIds: string[], | ||
): Promise<void> { | ||
this.logger.log( | ||
'Running command to set missing labelIdentifier to custom objects', | ||
); | ||
|
||
for (const workspaceId of workspaceIds) { | ||
this.logger.log(`Running command for workspace ${workspaceId}`); | ||
|
||
try { | ||
await this.setMissingLabelIdentifierToCustomObjectsForWorkspace( | ||
workspaceId, | ||
options, | ||
); | ||
} catch (error) { | ||
this.logger.log( | ||
chalk.red( | ||
`Running command on workspace ${workspaceId} failed with error: ${error}, ${error.stack}`, | ||
), | ||
); | ||
continue; | ||
} finally { | ||
this.logger.log( | ||
chalk.green(`Finished running command for workspace ${workspaceId}.`), | ||
); | ||
} | ||
} | ||
|
||
this.logger.log(chalk.green(`Command completed!`)); | ||
} | ||
|
||
private async setMissingLabelIdentifierToCustomObjectsForWorkspace( | ||
workspaceId: string, | ||
options: SetMissingLabelIdentifierToCustomObjectsCommandOptions, | ||
): Promise<void> { | ||
const customObjects = await this.objectMetadataRepository.find({ | ||
where: { | ||
workspaceId, | ||
labelIdentifierFieldMetadataId: IsNull(), | ||
isCustom: true, | ||
}, | ||
}); | ||
|
||
for (const customObject of customObjects) { | ||
const labelIdentifierFieldMetadata = | ||
await this.fieldMetadataRepository.findOne({ | ||
where: { | ||
workspaceId, | ||
objectMetadataId: customObject.id, | ||
standardId: CUSTOM_OBJECT_STANDARD_FIELD_IDS.name, | ||
}, | ||
}); | ||
|
||
if (labelIdentifierFieldMetadata && !options.dryRun) { | ||
await this.objectMetadataRepository.update(customObject.id, { | ||
labelIdentifierFieldMetadataId: labelIdentifierFieldMetadata.id, | ||
}); | ||
} | ||
|
||
if (options.verbose) { | ||
this.logger.log( | ||
chalk.yellow( | ||
`Set labelIdentifierFieldMetadataId for custom object ${customObject.nameSingular}`, | ||
), | ||
); | ||
} | ||
} | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters