import { instead } from "@lib/api/patcher";
import { lazyDestructure } from "@lib/utils/lazy";
import { OptionalKeys } from "@lib/utils/types";
import { findByProps } from "@metro/utils";
import { AccessibilityInfo, Platform, View } from "react-native";

const { ContextMenu: _ContextMenu } = lazyDestructure(() => findByProps("ContextMenu"));

type ContextMenuItem = {
    label: string,
    iconSource: import("react-native").ImageSourcePropType,
    action: () => unknown,
    [key: string]: any;
};

type ContextMenuProps = {
    triggerOnLongPress: boolean;
    items: ContextMenuItem[] | ContextMenuItem[][];
    align: OptionalKeys<["left", "right", "above", "below", "auto"]>;
    title: string;
    children: React.FC;
    [key: string]: any;
};

export default function ContextMenu(props: ContextMenuProps) {
    const ref = React.useRef(null);

    React.useEffect(() => {
        // hacky discord fix for android where pressing back with context menu open crashes the client
        if (Platform.OS !== "android") return;

        // @ts-ignore
        const ctxMenuReactTag = ref.current?._children?.[0]?._nativeTag;

        return () => {
            if (!ctxMenuReactTag) return;

            const unpatch = instead("setAccessibilityFocus", AccessibilityInfo, ([tag], orig) => {
                if (tag !== ctxMenuReactTag) {
                    return orig.apply(AccessibilityInfo, [tag]);
                } else {
                    unpatch(); // and skip this call
                }
            });
        };
    }, []);

    return <View ref={ref}>
        <_ContextMenu
            {...props}
        />
    </View>;
}