From ccc63446c3262db7ef42ef9a3e21eb5fcf79d8e1 Mon Sep 17 00:00:00 2001
From: Fredrik Strand Oseberg <fredrik.no@gmail.com>
Date: Mon, 16 Sep 2024 11:13:18 +0200
Subject: [PATCH] Fix/default strategy permission cherry pick (#8144)

Cherry pick #8113, #8114 and #8112

---------

Co-authored-by: Thomas Heartman <thomas@getunleash.io>
---
 .../feature/StrategyTypes/StrategyVariants.tsx       | 12 ++++++++++--
 .../ProjectDefaultStrategySettings.tsx               | 10 +++++++++-
 .../EditDefaultStrategy.tsx                          |  5 +++--
 .../ProjectDefaultStrategyForm.tsx                   | 10 +++++++++-
 .../ProjectEnvironmentDefaultStrategy.tsx            | 10 ++++++++--
 .../features/project-environments/environments.ts    |  3 ++-
 6 files changed, 41 insertions(+), 9 deletions(-)

diff --git a/frontend/src/component/feature/StrategyTypes/StrategyVariants.tsx b/frontend/src/component/feature/StrategyTypes/StrategyVariants.tsx
index da2a70c75fc8..33cd56e7dfad 100644
--- a/frontend/src/component/feature/StrategyTypes/StrategyVariants.tsx
+++ b/frontend/src/component/feature/StrategyTypes/StrategyVariants.tsx
@@ -27,7 +27,15 @@ export const StrategyVariants: FC<{
     projectId: string;
     environment: string;
     editable?: boolean;
-}> = ({ strategy, setStrategy, projectId, environment, editable }) => {
+    permission?: string | string[];
+}> = ({
+    strategy,
+    setStrategy,
+    projectId,
+    environment,
+    editable,
+    permission = UPDATE_FEATURE_ENVIRONMENT_VARIANTS,
+}) => {
     const { trackEvent } = usePlausibleTracker();
     const [variantsEdit, setVariantsEdit] = useState<IFeatureVariantEdit[]>([]);
     const theme = useTheme();
@@ -153,7 +161,7 @@ export const StrategyVariants: FC<{
             <PermissionButton
                 onClick={addVariant}
                 variant='outlined'
-                permission={UPDATE_FEATURE_ENVIRONMENT_VARIANTS}
+                permission={permission}
                 projectId={projectId}
                 environmentId={environment}
                 data-testid='ADD_STRATEGY_VARIANT_BUTTON'
diff --git a/frontend/src/component/project/Project/ProjectSettings/ProjectDefaultStrategySettings/ProjectDefaultStrategySettings.tsx b/frontend/src/component/project/Project/ProjectSettings/ProjectDefaultStrategySettings/ProjectDefaultStrategySettings.tsx
index 0874164aa493..ae11b1f93ad4 100644
--- a/frontend/src/component/project/Project/ProjectSettings/ProjectDefaultStrategySettings/ProjectDefaultStrategySettings.tsx
+++ b/frontend/src/component/project/Project/ProjectSettings/ProjectDefaultStrategySettings/ProjectDefaultStrategySettings.tsx
@@ -6,6 +6,7 @@ import { PageContent } from 'component/common/PageContent/PageContent';
 import { PageHeader } from 'component/common/PageHeader/PageHeader';
 import {
     PROJECT_DEFAULT_STRATEGY_READ,
+    PROJECT_DEFAULT_STRATEGY_WRITE,
     UPDATE_PROJECT,
 } from 'component/providers/AccessProvider/permissions';
 import { Alert, styled } from '@mui/material';
@@ -29,7 +30,14 @@ export const ProjectDefaultStrategySettings = () => {
     usePageTitle(`Project default strategy configuration – ${projectName}`);
 
     if (
-        !hasAccess([UPDATE_PROJECT, PROJECT_DEFAULT_STRATEGY_READ], projectId)
+        !hasAccess(
+            [
+                UPDATE_PROJECT,
+                PROJECT_DEFAULT_STRATEGY_READ,
+                PROJECT_DEFAULT_STRATEGY_WRITE,
+            ],
+            projectId,
+        )
     ) {
         return (
             <PageContent
diff --git a/frontend/src/component/project/Project/ProjectSettings/ProjectDefaultStrategySettings/ProjectEnvironment/ProjectEnvironmentDefaultStrategy/EditDefaultStrategy.tsx b/frontend/src/component/project/Project/ProjectSettings/ProjectDefaultStrategySettings/ProjectEnvironment/ProjectEnvironmentDefaultStrategy/EditDefaultStrategy.tsx
index a305175207ee..34eaad10b8c6 100644
--- a/frontend/src/component/project/Project/ProjectSettings/ProjectDefaultStrategySettings/ProjectEnvironment/ProjectEnvironmentDefaultStrategy/EditDefaultStrategy.tsx
+++ b/frontend/src/component/project/Project/ProjectSettings/ProjectDefaultStrategySettings/ProjectEnvironment/ProjectEnvironmentDefaultStrategy/EditDefaultStrategy.tsx
@@ -6,7 +6,7 @@ import { useStrategy } from 'hooks/api/getters/useStrategy/useStrategy';
 import { useEffect, useState } from 'react';
 import { formatUnknownError } from 'utils/formatUnknownError';
 import FormTemplate from 'component/common/FormTemplate/FormTemplate';
-import { UPDATE_FEATURE_STRATEGY } from 'component/providers/AccessProvider/permissions';
+import { PROJECT_DEFAULT_STRATEGY_WRITE } from 'component/providers/AccessProvider/permissions';
 import type { IStrategy } from 'interfaces/strategy';
 import { useRequiredQueryParam } from 'hooks/useRequiredQueryParam';
 import type { ISegment } from 'interfaces/segment';
@@ -19,6 +19,7 @@ import { usePlausibleTracker } from 'hooks/usePlausibleTracker';
 import { ProjectDefaultStrategyForm } from './ProjectDefaultStrategyForm';
 import type { CreateFeatureStrategySchema } from 'openapi';
 import useProjectOverview from 'hooks/api/getters/useProjectOverview/useProjectOverview';
+import { UPDATE_PROJECT } from '@server/types/permissions';
 
 export const useDefaultStrategy = (
     projectId: string,
@@ -154,7 +155,7 @@ const EditDefaultStrategy = () => {
                 environmentId={environmentId}
                 onSubmit={onSubmit}
                 loading={loading}
-                permission={UPDATE_FEATURE_STRATEGY}
+                permission={[PROJECT_DEFAULT_STRATEGY_WRITE, UPDATE_PROJECT]}
                 errors={errors}
                 isChangeRequest={false}
             />
diff --git a/frontend/src/component/project/Project/ProjectSettings/ProjectDefaultStrategySettings/ProjectEnvironment/ProjectEnvironmentDefaultStrategy/ProjectDefaultStrategyForm.tsx b/frontend/src/component/project/Project/ProjectSettings/ProjectDefaultStrategySettings/ProjectEnvironment/ProjectEnvironmentDefaultStrategy/ProjectDefaultStrategyForm.tsx
index dda1c80342b8..7ad83f8dd3d3 100644
--- a/frontend/src/component/project/Project/ProjectSettings/ProjectDefaultStrategySettings/ProjectEnvironment/ProjectEnvironmentDefaultStrategy/ProjectDefaultStrategyForm.tsx
+++ b/frontend/src/component/project/Project/ProjectSettings/ProjectDefaultStrategySettings/ProjectEnvironment/ProjectEnvironmentDefaultStrategy/ProjectDefaultStrategyForm.tsx
@@ -21,11 +21,15 @@ import { FeatureStrategyConstraints } from 'component/feature/FeatureStrategy/Fe
 import { FeatureStrategyType } from 'component/feature/FeatureStrategy/FeatureStrategyType/FeatureStrategyType';
 import { FeatureStrategyTitle } from 'component/feature/FeatureStrategy/FeatureStrategyForm/FeatureStrategyTitle/FeatureStrategyTitle';
 import { StrategyVariants } from 'component/feature/StrategyTypes/StrategyVariants';
+import {
+    PROJECT_DEFAULT_STRATEGY_WRITE,
+    UPDATE_PROJECT,
+} from '@server/types/permissions';
 
 interface IProjectDefaultStrategyFormProps {
     projectId: string;
     environmentId: string;
-    permission: string;
+    permission: string | string[];
     onSubmit: () => void;
     onCancel?: () => void;
     loading: boolean;
@@ -186,6 +190,10 @@ export const ProjectDefaultStrategyForm = ({
                         setStrategy={setStrategy}
                         environment={environmentId}
                         projectId={projectId}
+                        permission={[
+                            PROJECT_DEFAULT_STRATEGY_WRITE,
+                            UPDATE_PROJECT,
+                        ]}
                     />
                 }
             />
diff --git a/frontend/src/component/project/Project/ProjectSettings/ProjectDefaultStrategySettings/ProjectEnvironment/ProjectEnvironmentDefaultStrategy/ProjectEnvironmentDefaultStrategy.tsx b/frontend/src/component/project/Project/ProjectSettings/ProjectDefaultStrategySettings/ProjectEnvironment/ProjectEnvironmentDefaultStrategy/ProjectEnvironmentDefaultStrategy.tsx
index e7786bf2b5f0..a340a3742dff 100644
--- a/frontend/src/component/project/Project/ProjectSettings/ProjectDefaultStrategySettings/ProjectEnvironment/ProjectEnvironmentDefaultStrategy/ProjectEnvironmentDefaultStrategy.tsx
+++ b/frontend/src/component/project/Project/ProjectSettings/ProjectDefaultStrategySettings/ProjectEnvironment/ProjectEnvironmentDefaultStrategy/ProjectEnvironmentDefaultStrategy.tsx
@@ -1,13 +1,16 @@
 import { useRequiredPathParam } from 'hooks/useRequiredPathParam';
 import { StrategyItemContainer } from 'component/common/StrategyItemContainer/StrategyItemContainer';
 import PermissionIconButton from 'component/common/PermissionIconButton/PermissionIconButton';
-import { UPDATE_FEATURE_STRATEGY } from 'component/providers/AccessProvider/permissions';
 import { Link } from 'react-router-dom';
 import Edit from '@mui/icons-material/Edit';
 import { StrategyExecution } from 'component/feature/FeatureView/FeatureOverview/FeatureOverviewEnvironments/FeatureOverviewEnvironment/EnvironmentAccordionBody/StrategyDraggableItem/StrategyItem/StrategyExecution/StrategyExecution';
 import type { ProjectEnvironmentType } from 'interfaces/environments';
 import { useMemo } from 'react';
 import type { CreateFeatureStrategySchema } from 'openapi';
+import {
+    PROJECT_DEFAULT_STRATEGY_WRITE,
+    UPDATE_PROJECT,
+} from '@server/types/permissions';
 
 interface ProjectEnvironmentDefaultStrategyProps {
     environment: ProjectEnvironmentType;
@@ -59,7 +62,10 @@ const ProjectEnvironmentDefaultStrategy = ({
                 actions={
                     <>
                         <PermissionIconButton
-                            permission={UPDATE_FEATURE_STRATEGY}
+                            permission={[
+                                PROJECT_DEFAULT_STRATEGY_WRITE,
+                                UPDATE_PROJECT,
+                            ]}
                             environmentId={environmentId}
                             projectId={projectId}
                             component={Link}
diff --git a/src/lib/features/project-environments/environments.ts b/src/lib/features/project-environments/environments.ts
index a80010edce45..0ccc75656d7c 100644
--- a/src/lib/features/project-environments/environments.ts
+++ b/src/lib/features/project-environments/environments.ts
@@ -5,6 +5,7 @@ import {
     type IUnleashServices,
     serializeDates,
     UPDATE_PROJECT,
+    PROJECT_DEFAULT_STRATEGY_WRITE,
 } from '../../types';
 import type { Logger } from '../../logger';
 import type EnvironmentService from './environment-service';
@@ -105,7 +106,7 @@ export default class EnvironmentsController extends Controller {
             method: 'post',
             path: `${PREFIX}/:environment/default-strategy`,
             handler: this.updateDefaultStrategyForProjectEnvironment,
-            permission: UPDATE_PROJECT,
+            permission: [UPDATE_PROJECT, PROJECT_DEFAULT_STRATEGY_WRITE],
             middleware: [
                 openApiService.validPath({
                     tags: ['Projects'],