posthog/.storybook/decorators/withFeatureFlags.tsx

import { useFeatureFlags } from '~/mocks/browser'
import type { DecoratorFn } from '@storybook/react'

/** Global story decorator that allows setting feature flags.
 *
 * ```ts
 * export default {
 *   title: 'My story',
 *   component: MyComponent,
 *   parameters: {
 *     featureFlags: ['hogql'], // add flags here
 *   },
 * } as ComponentMeta<typeof MyComponent>
 * ```
 */
export const withFeatureFlags: DecoratorFn = (Story, { parameters }) => {
    if (parameters.featureFlags) {
        useFeatureFlags(parameters.featureFlags)
    }

    return <Story />
}