prismic-example-3.tsx

const EditorAlert = ({ block, children }) => {
  const { isStaff } = useContext(AuthContext);

  if (!isStaff) {
    return null;
  }

  if (block) {
    return <div className="editor-alert">{children}</div>;
  }

  return <span className="editor-alert">{children}</span>;
};

export default EditorAlert;