All files / src/utils/components/feature-access UpgradeDialogProvider.jsx

33.33% Statements 6/18
7.69% Branches 1/13
14.28% Functions 1/7
33.33% Lines 6/18

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71        7x   7x                                                                                                       7x 168x 168x 168x                  
import React, { createContext, useCallback, useContext, useMemo, useState } from "react";
import UpgradeDialog from "@/utils/components/feature-access/UpgradeDialog";
import useFeatureFlag from "@/utils/hooks/useFeatureFlag";
 
const UpgradeDialogContext = createContext(null);
 
export const UpgradeDialogProvider = ({ userType, children }) => {
    const { currentPlan, planName } = useFeatureFlag(userType);
    const [open, setOpen] = useState(false);
    const [dialogState, setDialogState] = useState({
        title: "This feature is locked",
        featureKey: null,
        featureLabel: null,
        featureMeta: null,
        minPlan: null,
    });
 
    const openUpgradeDialog = useCallback((options = {}) => {
        setDialogState({
            title: options.title ?? "This feature is locked",
            featureKey: options.featureKey ?? null,
            featureLabel: options.featureLabel ?? null,
            featureMeta: options.featureMeta ?? null,
            minPlan: options.minPlan ?? null,
        });
        setOpen(true);
    }, []);
 
    const closeUpgradeDialog = useCallback(() => {
        setOpen(false);
    }, []);
 
    const value = useMemo(
        () => ({
            openUpgradeDialog,
            closeUpgradeDialog,
        }),
        [openUpgradeDialog, closeUpgradeDialog]
    );
 
    return (
        <UpgradeDialogContext.Provider value={value}>
            {children}
            <UpgradeDialog
                open={open}
                onClose={closeUpgradeDialog}
                title={dialogState.title}
                featureKey={dialogState.featureKey}
                featureLabel={dialogState.featureLabel}
                featureMeta={dialogState.featureMeta}
                minPlan={dialogState.minPlan}
                currentPlan={currentPlan}
                planName={planName}
            />
        </UpgradeDialogContext.Provider>
    );
};
 
export const useUpgradeDialog = () => {
    const context = useContext(UpgradeDialogContext);
    Eif (!context) {
        return {
            openUpgradeDialog: () => {},
            closeUpgradeDialog: () => {},
        };
    }
    return context;
};
 
export default UpgradeDialogProvider;