Create workflow setup command (#8406)

Steps to enable workflows:
- enable feature flags
- run metadata sync
- run this command

If the feature flag is not true, the command will fail. Which will be
useful to prevent seeding a wrong workspace.

Including a little fix for send email aciton error
This commit is contained in:
Thomas Trompette
2024-11-08 12:40:11 +01:00
committed by GitHub
parent d8831512d4
commit 813c57fba6
4 changed files with 216 additions and 10 deletions

View File

@ -21,16 +21,16 @@ export class CodeWorkflowAction implements WorkflowAction {
async execute(
workflowStepInput: WorkflowCodeStepInput,
): Promise<WorkflowActionResult> {
const { workspaceId } = this.scopedWorkspaceContextFactory.create();
if (!workspaceId) {
throw new WorkflowStepExecutorException(
'Scoped workspace not found',
WorkflowStepExecutorExceptionCode.SCOPED_WORKSPACE_NOT_FOUND,
);
}
try {
const { workspaceId } = this.scopedWorkspaceContextFactory.create();
if (!workspaceId) {
throw new WorkflowStepExecutorException(
'Scoped workspace not found',
WorkflowStepExecutorExceptionCode.SCOPED_WORKSPACE_NOT_FOUND,
);
}
const result =
await this.serverlessFunctionService.executeOneServerlessFunction(
workflowStepInput.serverlessFunctionId,

View File

@ -0,0 +1,185 @@
import { Logger } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { Command } from 'nest-commander';
import { Repository } from 'typeorm';
import { v4 } from 'uuid';
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';
@Command({
name: 'workflow:seed:views',
description: 'Seed workflow views for workspace.',
})
export class SeedWorkflowViewsCommand extends ActiveWorkspacesCommandRunner {
protected readonly logger: Logger;
constructor(
@InjectRepository(Workspace, 'core')
protected readonly workspaceRepository: Repository<Workspace>,
@InjectRepository(ObjectMetadataEntity, 'metadata')
private readonly objectMetadataRepository: Repository<ObjectMetadataEntity>,
@InjectRepository(FieldMetadataEntity, 'metadata')
private readonly fieldMetadataRepository: Repository<FieldMetadataEntity>,
private readonly twentyORMGlobalManager: TwentyORMGlobalManager,
) {
super(workspaceRepository);
this.logger = new Logger(this.constructor.name);
}
async executeActiveWorkspacesCommand(
_passedParam: string[],
_options: ActiveWorkspacesCommandOptions,
_workspaceIds: string[],
): Promise<void> {
const { dryRun } = _options;
for (const workspaceId of _workspaceIds) {
await this.execute(workspaceId, dryRun);
}
}
private async execute(workspaceId: string, dryRun = false): Promise<void> {
this.logger.log(`Seeding workflow views for workspace: ${workspaceId}`);
const workflowViewId = await this.seedView(
workspaceId,
'workflow',
'All Workflows',
);
await this.seedView(
workspaceId,
'workflowVersion',
'All Workflow Versions',
);
await this.seedView(workspaceId, 'workflowRun', 'All Workflow Runs');
const favoriteRepository =
await this.twentyORMGlobalManager.getRepositoryForWorkspace(
workspaceId,
'favorite',
);
const existingFavorites = await favoriteRepository.find({
where: {
viewId: workflowViewId,
},
});
if (existingFavorites.length > 0) {
this.logger.log(
`Favorite already exists for view: ${existingFavorites[0].id}`,
);
return;
}
if (dryRun) {
this.logger.log(`Dry run: Creating favorite for view: ${workflowViewId}`);
return;
}
await favoriteRepository.insert({
viewId: workflowViewId,
position: 5,
});
}
private async seedView(
workspaceId: string,
nameSingular: string,
viewName: string,
dryRun = false,
): Promise<string> {
const objectMetadata = (
await this.objectMetadataRepository.find({
where: { workspaceId, nameSingular },
})
)?.[0];
if (!objectMetadata) {
throw new Error(`Object metadata not found: ${nameSingular}`);
}
const fieldMetadataName = (
await this.fieldMetadataRepository.find({
where: {
workspaceId,
objectMetadataId: objectMetadata.id,
name: 'name',
},
})
)?.[0];
if (!fieldMetadataName) {
throw new Error(
`Field metadata not found for ${objectMetadata.id}: name`,
);
}
const viewRepository =
await this.twentyORMGlobalManager.getRepositoryForWorkspace(
workspaceId,
'view',
);
const viewFieldRepository =
await this.twentyORMGlobalManager.getRepositoryForWorkspace(
workspaceId,
'viewField',
);
const viewId = v4();
const existingViews = await viewRepository.find({
where: {
objectMetadataId: objectMetadata.id,
name: viewName,
},
});
if (existingViews.length > 0) {
this.logger.log(`View already exists: ${existingViews[0].id}`);
return existingViews[0].id;
}
if (dryRun) {
this.logger.log(`Dry run: Creating view: ${viewName}`);
return viewId;
}
await viewRepository.insert({
id: viewId,
name: viewName,
objectMetadataId: objectMetadata.id,
type: 'table',
key: 'INDEX',
position: 0,
icon: 'IconSettingsAutomation',
kanbanFieldMetadataId: '',
});
await viewFieldRepository.insert({
fieldMetadataId: fieldMetadataName.id,
position: 0,
isVisible: true,
size: 210,
viewId: viewId,
});
return viewId;
}
}

View File

@ -0,0 +1,20 @@
import { Module } from '@nestjs/common';
import { TypeOrmModule } from '@nestjs/typeorm';
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 { SeedWorkflowViewsCommand } from 'src/modules/workflow/common/commands/seed-workflow-views.command';
@Module({
imports: [
TypeOrmModule.forFeature([Workspace], 'core'),
TypeOrmModule.forFeature(
[ObjectMetadataEntity, FieldMetadataEntity],
'metadata',
),
],
providers: [SeedWorkflowViewsCommand],
exports: [SeedWorkflowViewsCommand],
})
export class WorkflowCommandModule {}

View File

@ -1,10 +1,11 @@
import { Module } from '@nestjs/common';
import { WorkflowCommandModule } from 'src/modules/workflow/common/commands/workflow-command.module';
import { WorkflowQueryHookModule } from 'src/modules/workflow/common/query-hooks/workflow-query-hook.module';
import { WorkflowCommonWorkspaceService } from 'src/modules/workflow/common/workspace-services/workflow-common.workspace-service';
@Module({
imports: [WorkflowQueryHookModule],
imports: [WorkflowQueryHookModule, WorkflowCommandModule],
providers: [WorkflowCommonWorkspaceService],
exports: [WorkflowCommonWorkspaceService],
})