Protect serverless endpoint with workflow endpoint

twentyhq · Dec 17, 2024 · dc1527e · dc1527e
1 parent d0f7270
commit dc1527e
Showing 1 changed file with 8 additions and 9 deletions.
diff --git a/...ty-server/src/engine/metadata-modules/serverless-function/serverless-function.resolver.ts b/...ty-server/src/engine/metadata-modules/serverless-function/serverless-function.resolver.ts
@@ -35,17 +35,16 @@ export class ServerlessFunctionResolver {
   ) {}
 
   async checkFeatureFlag(workspaceId: string) {
-    const isFunctionSettingsEnabled =
-      await this.featureFlagRepository.findOneBy({
-        workspaceId,
-        key: FeatureFlagKey.IsFunctionSettingsEnabled,
-        value: true,
-      });
+    const isWorkflowEnabled = await this.featureFlagRepository.findOneBy({
+      workspaceId,
+      key: FeatureFlagKey.IsWorkflowEnabled,
+      value: true,
+    });
 
-    if (!isFunctionSettingsEnabled) {
+    if (!isWorkflowEnabled) {
       throw new ServerlessFunctionException(
-        `IS_FUNCTION_SETTINGS_ENABLED feature flag is not set to true for this workspace`,
-        ServerlessFunctionExceptionCode.SERVERLESS_FUNCTION_NOT_FOUND,
+        `IS_WORKFLOW_ENABLED feature flag is not set to true for this workspace`,
+        ServerlessFunctionExceptionCode.FEATURE_FLAG_INVALID,
       );
     }
   }