injection points

js_modules/dagster-ui/packages/ui-core/src/app/AppTopNav/AppTopNavLinks.tsx

@@ -1,6 +1,7 @@
 import {Box} from '@dagster-io/ui-components';
 import {ReactNode} from 'react';
 import {useHistory} from 'react-router-dom';
+import {FeatureFlag} from 'src/app/FeatureFlags.oss';
 
 import {TopNavLink} from './AppTopNav';
 import {
@@ -11,7 +12,7 @@ import {
   locationPathMatcher,
 } from './activePathMatchers';
 import {DeploymentStatusIcon} from '../../nav/DeploymentStatusIcon';
-import {FeatureFlag, featureEnabled} from '../Flags';
+import {featureEnabled} from '../Flags';
 import {ShortcutHandler} from '../ShortcutHandler';
 
 export type AppNavLinkType = {

js_modules/dagster-ui/packages/ui-core/src/app/FeatureFlags.oss.tsx

@@ -0,0 +1,7 @@
+export enum FeatureFlag {
+  flagDebugConsoleLogging = 'flagDebugConsoleLogging',
+  flagDisableWebsockets = 'flagDisableWebsockets',
+  flagSidebarResources = 'flagSidebarResources',
+  flagDisableAutoLoadDefaults = 'flagDisableAutoLoadDefaults',
+  flagSettingsPage = 'flagSettingsPage',
+}

js_modules/dagster-ui/packages/ui-core/src/app/Flags.tsx

@@ -1,42 +1,33 @@
 import memoize from 'lodash/memoize';
 import {useMemo} from 'react';
+import {FeatureFlag} from 'src/app/FeatureFlags.oss';
 
 import {getJSONForKey} from '../hooks/useStateWithStorage';
 
 export const DAGSTER_FLAGS_KEY = 'DAGSTER_FLAGS';
 
-// Use const because we need to extend this in cloud. https://blog.logrocket.com/extend-enums-typescript/
-export const FeatureFlag = {
-  flagDebugConsoleLogging: 'flagDebugConsoleLogging' as const,
-  flagDisableWebsockets: 'flagDisableWebsockets' as const,
-  flagSidebarResources: 'flagSidebarResources' as const,
-  flagDisableAutoLoadDefaults: 'flagDisableAutoLoadDefaults' as const,
-  flagSettingsPage: 'flagSettingsPage' as const,
-};
-export type FeatureFlagType = keyof typeof FeatureFlag;
-
-export const getFeatureFlags: () => FeatureFlagType[] = memoize(
+export const getFeatureFlags: () => FeatureFlag[] = memoize(
   () => getJSONForKey(DAGSTER_FLAGS_KEY) || [],
 );
 
-export const featureEnabled = memoize((flag: FeatureFlagType) => getFeatureFlags().includes(flag));
+export const featureEnabled = memoize((flag: FeatureFlag) => getFeatureFlags().includes(flag));
 
 type FlagMap = {
-  readonly [F in FeatureFlagType]: boolean;
+  readonly [F in FeatureFlag]: boolean;
 };
 
 export const useFeatureFlags = () => {
   return useMemo(() => {
     const flagSet = new Set(getFeatureFlags());
     const all: Record<string, boolean> = {};
     for (const flag in FeatureFlag) {
-      all[flag] = flagSet.has(flag as FeatureFlagType);
+      all[flag] = flagSet.has(flag as FeatureFlag);
     }
     return all as FlagMap;
   }, []);
 };
 
-export const setFeatureFlags = (flags: FeatureFlagType[]) => {
+export const setFeatureFlags = (flags: FeatureFlag[]) => {
   if (!(flags instanceof Array)) {
     throw new Error('flags must be an array');
   }

js_modules/dagster-ui/packages/ui-core/src/app/UserSettingsButton.tsx

@@ -1,10 +1,10 @@
 import {Icon} from '@dagster-io/ui-components';
 import {useState} from 'react';
+import {getVisibleFeatureFlagRows} from 'src/app/getVisibleFeatureFlagRows.oss';
 
 import {useFeatureFlags} from './Flags';
 import {TopNavButton} from './TopNavButton';
 import {UserSettingsDialog} from './UserSettingsDialog/UserSettingsDialog';
-import {getVisibleFeatureFlagRows} from './getVisibleFeatureFlagRows';
 
 export const UserSettingsButton = () => {
   const {flagSettingsPage} = useFeatureFlags();

js_modules/dagster-ui/packages/ui-core/src/app/UserSettingsDialog/UserSettingsDialog.tsx

@@ -11,13 +11,14 @@ import {
   Tooltip,
 } from '@dagster-io/ui-components';
 import * as React from 'react';
+import {FeatureFlag} from 'src/app/FeatureFlags.oss';
 import {UserPreferences} from 'src/app/UserSettingsDialog/UserPreferences.oss';
 
 import {CodeLinkProtocolSelect} from '../../code-links/CodeLinkProtocol';
-import {FeatureFlagType, getFeatureFlags, setFeatureFlags} from '../Flags';
+import {getFeatureFlags, setFeatureFlags} from '../Flags';
 
 type OnCloseFn = (event: React.SyntheticEvent<HTMLElement>) => void;
-type VisibleFlag = {key: string; label?: React.ReactNode; flagType: FeatureFlagType};
+type VisibleFlag = {key: string; label?: React.ReactNode; flagType: FeatureFlag};
 
 interface DialogProps {
   isOpen: boolean;
@@ -40,15 +41,15 @@ export const UserSettingsDialog = ({isOpen, onClose, visibleFlags}: DialogProps)
 
 interface DialogContentProps {
   onClose: OnCloseFn;
-  visibleFlags: {key: string; label?: React.ReactNode; flagType: FeatureFlagType}[];
+  visibleFlags: {key: string; label?: React.ReactNode; flagType: FeatureFlag}[];
 }
 
 /**
  * Separate the content from the `Dialog` so that we don't prepare its state before
  * we want to render it.
  */
 const UserSettingsDialogContent = ({onClose, visibleFlags}: DialogContentProps) => {
-  const [flags, setFlags] = React.useState<FeatureFlagType[]>(() => getFeatureFlags());
+  const [flags, setFlags] = React.useState<FeatureFlag[]>(() => getFeatureFlags());
   const [reloading, setReloading] = React.useState(false);
 
   const initialFlagState = React.useRef(JSON.stringify([...getFeatureFlags().sort()]));
@@ -57,7 +58,7 @@ const UserSettingsDialogContent = ({onClose, visibleFlags}: DialogContentProps)
     setFeatureFlags(flags);
   });
 
-  const toggleFlag = (flag: FeatureFlagType) => {
+  const toggleFlag = (flag: FeatureFlag) => {
     setFlags(flags.includes(flag) ? flags.filter((f) => f !== flag) : [...flags, flag]);
   };
 

js_modules/dagster-ui/packages/ui-core/src/app/Util.tsx

@@ -1,8 +1,9 @@
 import {cache} from 'idb-lru-cache';
 import memoize from 'lodash/memoize';
 import LRU from 'lru-cache';
+import {FeatureFlag} from 'src/app/FeatureFlags.oss';
 
-import {FeatureFlag, featureEnabled} from './Flags';
+import {featureEnabled} from './Flags';
 import {timeByParts} from './timeByParts';
 
 function twoDigit(v: number) {

js_modules/dagster-ui/packages/ui-core/src/app/getVisibleFeatureFlagRows.tsx → js_modules/dagster-ui/packages/ui-core/src/app/getVisibleFeatureFlagRows.oss.tsx

@@ -1,4 +1,4 @@
-import {FeatureFlag} from './Flags';
+import {FeatureFlag} from 'src/app/FeatureFlags.oss';
 
 /**
  * Open-source feature flags to be displayed in Dagster UI "User settings"

js_modules/dagster-ui/packages/ui-core/src/settings/SettingsLeftPane.tsx

@@ -1,9 +1,9 @@
 import {Box, Icon} from '@dagster-io/ui-components';
 import {useState} from 'react';
 import {useLocation} from 'react-router-dom';
+import {getVisibleFeatureFlagRows} from 'src/app/getVisibleFeatureFlagRows.oss';
 
 import {UserSettingsDialog} from '../app/UserSettingsDialog/UserSettingsDialog';
-import {getVisibleFeatureFlagRows} from '../app/getVisibleFeatureFlagRows';
 import {SideNavItem, SideNavItemConfig} from '../ui/SideNavItem';
 
 const items: SideNavItemConfig[] = [