hotfix for lab (#9981)

Issue:
When attempting to toggle a public feature flag that didn't exist in the
workspace's feature flags array, the LabService threw a
FeatureFlagException with code FEATURE_FLAG_NOT_FOUND. This prevented
users from enabling public feature flags for the first time in their
workspace.
Fix:
Modified the LabService to handle non-existent public feature flags by
creating them instead of throwing an error. When the flag doesn't exist,
the service now saves a new feature flag record with the provided key
and value, associates it with the workspace, and returns the newly
created flag.

packages/twenty-front/src/generated/graphql.tsx

@@ -800,7 +800,7 @@ export type Mutation = {
   skipSyncEmailOnboardingStep: OnboardingStepSuccess;
   track: Analytics;
   updateBillingSubscription: BillingUpdateOutput;
-  updateLabPublicFeatureFlag: Scalars['Boolean'];
+  updateLabPublicFeatureFlag: FeatureFlag;
   updateOneField: Field;
   updateOneObject: Object;
   updateOneServerlessFunction: ServerlessFunction;
@@ -2225,7 +2225,7 @@ export type UpdateLabPublicFeatureFlagMutationVariables = Exact<{
 }>;
 
 
-export type UpdateLabPublicFeatureFlagMutation = { __typename?: 'Mutation', updateLabPublicFeatureFlag: boolean };
+export type UpdateLabPublicFeatureFlagMutation = { __typename?: 'Mutation', updateLabPublicFeatureFlag: { __typename?: 'FeatureFlag', id: any, key: FeatureFlagKey, value: boolean } };
 
 export type CreateOidcIdentityProviderMutationVariables = Exact<{
   input: SetupOidcSsoInput;
@@ -3884,7 +3884,11 @@ export type GetEnvironmentVariablesGroupedLazyQueryHookResult = ReturnType<typeo
 export type GetEnvironmentVariablesGroupedQueryResult = Apollo.QueryResult<GetEnvironmentVariablesGroupedQuery, GetEnvironmentVariablesGroupedQueryVariables>;
 export const UpdateLabPublicFeatureFlagDocument = gql`
     mutation UpdateLabPublicFeatureFlag($input: UpdateLabPublicFeatureFlagInput!) {
-  updateLabPublicFeatureFlag(input: $input)
+  updateLabPublicFeatureFlag(input: $input) {
+    id
+    key
+    value
+  }
 }
     `;
 export type UpdateLabPublicFeatureFlagMutationFn = Apollo.MutationFunction<UpdateLabPublicFeatureFlagMutation, UpdateLabPublicFeatureFlagMutationVariables>;

packages/twenty-front/src/modules/settings/lab/graphql/mutations/updateLabPublicFeatureFlag.ts

@@ -4,6 +4,10 @@ export const UPDATE_LAB_PUBLIC_FEATURE_FLAG = gql`
   mutation UpdateLabPublicFeatureFlag(
     $input: UpdateLabPublicFeatureFlagInput!
   ) {
-    updateLabPublicFeatureFlag(input: $input)
+    updateLabPublicFeatureFlag(input: $input) {
+      id
+      key
+      value
+    }
   }
 `;

packages/twenty-front/src/modules/settings/lab/hooks/useLabPublicFeatureFlags.ts

@@ -15,7 +15,29 @@ export const useLabPublicFeatureFlags = () => {
   );
   const labPublicFeatureFlags = useRecoilValue(labPublicFeatureFlagsState);
 
-  const [updateLabPublicFeatureFlag] = useUpdateLabPublicFeatureFlagMutation();
+  const [updateLabPublicFeatureFlag] = useUpdateLabPublicFeatureFlagMutation({
+    onCompleted: (data) => {
+      if (isDefined(currentWorkspace)) {
+        const updatedFlag = data.updateLabPublicFeatureFlag;
+
+        setCurrentWorkspace({
+          ...currentWorkspace,
+          featureFlags: [
+            ...(currentWorkspace.featureFlags?.filter(
+              (flag) => flag.key !== updatedFlag.key,
+            ) ?? []),
+            {
+              ...updatedFlag,
+              workspaceId: currentWorkspace.id,
+            },
+          ],
+        });
+      }
+    },
+    onError: (error) => {
+      setError(error.message);
+    },
+  });
 
   const handleLabPublicFeatureFlagUpdate = async (
     publicFeatureFlag: FeatureFlagKey,
@@ -35,20 +57,8 @@ export const useLabPublicFeatureFlags = () => {
           value,
         },
       },
-      onError: (error) => {
-        setError(error.message);
-      },
     });
 
-    if (isDefined(response.data)) {
-      setCurrentWorkspace({
-        ...currentWorkspace,
-        featureFlags: currentWorkspace.featureFlags?.map((flag) =>
-          flag.key === publicFeatureFlag ? { ...flag, value } : flag,
-        ),
-      });
-    }
-
     return !!response.data;
   };